richat 0.2.5 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a6a6598af5c6e17f582012237108fdb8476d3c8c2b015114a8f2e5b72fcd8e02
-  data.tar.gz: 2103424385ccbc8342beca1be859faeb2c42a5f15ae4188b8253651d18a438d1
+  metadata.gz: 9b76efa6dcbdfe7a9f2184c7823b6c0292a6ce8b35ed44378679c06b0c847503
+  data.tar.gz: cda10863c00442ecd0a5c24c5758251bcfa1bb04b84f623843c2b6df30d1ead5
 SHA512:
-  metadata.gz: e6a3aad954fb88a24cf79e9f334d5912fd0fac875b74c4e585d4fa10626179ce22c0912686b8b7e2ac4642ed77fdab7962b2a63a5fbca8c0df4ebf821e4ecad4
-  data.tar.gz: c4012e6a086e7bddfdcc1e65a488f4248254e4ab3d814319806cd53725f12e8b4f5ac6eff78b1df5203398e34d82825f3082ccc0a96823b891e295a779e70510
+  metadata.gz: 3251caa055e0e5e60b9cac6981e81b52c083eebf06c5c5a4d4e2f115171f4445ed79142694ef514c1f067aed13276f0e03230f85896e8fce425804abb2cbbd16
+  data.tar.gz: '04008b119d99774c298f20f3d631a938221be60951df86e8e1c0eb6630060f3762561cfc62798f6cfa65746465c1b704febf63f1834dda82da248679d5adc6ea'

data/lib/richat/command.rb

@@ -3,19 +3,21 @@ module Richat
     EXIT_CODE = 0
     NEXT_CODE = 1
     PROMPT_CHANGED_CODE = 2
+    SYS_CMD_CODE = 3
+    SYS_CHAT_CODE = 4
 
     class << self
       attr_reader :prompt, :prompt_id
 
-      def call(user_input)
+      def call(user_input, sys_cmd_mode=false)
         user_input = user_input.strip
+        return handle_system_command(user_input, sys_cmd_mode) if user_input.start_with?(*Config.get("sys_cmd", "activate_keywords"))
+        return handle_exit if Config.get("shell", "exit_keywords").include?(user_input)
         return unless user_input.start_with?("/")
         if user_input == "/help"
           handle_help
         elsif user_input == "/config"
           handle_config
-        elsif user_input == "/exit"
-          handle_exit
         elsif user_input =~ /^\/prompt\s*/
           if user_input == "/prompt"
             handle_prompt
@@ -25,6 +27,25 @@ module Richat
         end
       end
 
+      def handle_system_command(user_input, sys_cmd_mode)
+        Config.get("sys_cmd", "activate_keywords").each do |sub_str|
+          if user_input.start_with?(sub_str)
+            cmd = user_input[sub_str.length..-1].strip
+            return SYS_CHAT_CODE if Config.get("sys_cmd", "deactivate_keywords").include?(cmd) && sys_cmd_mode
+            if (match = /^cd\s?(.*)$/.match(cmd))
+              Dir.chdir(File.expand_path(match[1].empty? ? "~" : match[1]))
+            else
+              system(cmd)
+            end
+            if sys_cmd_mode
+              return NEXT_CODE
+            else
+              return SYS_CMD_CODE, sub_str
+            end
+          end
+        end
+      end
+
       def handle_exit
         puts "Bye"
         EXIT_CODE
@@ -37,6 +58,7 @@ module Richat
       end
 
       def handle_help
+        puts "Version #{VERSION}"
         puts "\e[32m/exit\e[0m exit Richat"
         puts "\e[32m/config\e[0m show configuration"
         puts "\e[32m/prompt\e[0m show prompt list"

data/lib/richat/config.rb

@@ -1,3 +1,5 @@
+require 'json'
+
 module Richat
   class Config
     DEFAULT_CONFIG = {
@@ -18,7 +20,12 @@ module Richat
         "save_shell_history" => true,
         "enable_chat_context" => true,
         "show_welcome_info" => true,
-        "shell_history_file" => "~/.richat/history.txt"
+        "shell_history_file" => "~/.richat/history.txt",
+        "exit_keywords" => ["/exit", "q", "quit", "exit"]
+      },
+      "sys_cmd" => {
+        "activate_keywords" => [">", "!"],
+        "deactivate_keywords" => ["q", "quit", "exit"]
       },
       "prompt" => {
         "prompt_dir" => "~/.richat/prompts",

data/lib/richat/shell.rb

@@ -33,17 +33,21 @@ module Richat
       end
 
       Command.print_welcome if Config.get("shell", "show_welcome_info")
+      sys_cmd_mode = false
+      current_cmd_prefix = nil
 
       begin
-        while (user_content = Readline.readline("\e[32m#{Command.prompt_id&.+" "}\e[0m\e[33m>> \e[0m", true))
+        while (user_content = Readline.readline(shell_prompt(sys_cmd_mode, current_cmd_prefix), true))
           if user_content.empty?
             Readline::HISTORY&.pop
             next
           end
 
+          user_content = current_cmd_prefix + user_content if sys_cmd_mode
           File.open(history_path, 'a') { |f| f.puts(user_content) } if enable_full_completion
 
-          if (code = Command.call(user_content))
+          code, _ = Command.call(user_content, sys_cmd_mode)
+          if code
             if code == Command::NEXT_CODE
               next
             elsif code == Command::EXIT_CODE
@@ -51,6 +55,13 @@ module Richat
             elsif code == Command::PROMPT_CHANGED_CODE
               context_messages = [{ role: 'system', content: Command.prompt }]
               next
+            elsif code == Command::SYS_CMD_CODE
+              sys_cmd_mode = true
+              current_cmd_prefix = _
+              next
+            elsif code == Command::SYS_CHAT_CODE
+              sys_cmd_mode = false
+              next
             end
           end
 
@@ -84,5 +95,15 @@ module Richat
         Command.handle_exit
       end
     end
+
+    private
+
+    def shell_prompt(sys_cmd_mode, current_cmd_prefix)
+      if sys_cmd_mode
+        "\e[32m#{Command.prompt_id&.+" "}\e[0m\e[33m>>\e[0m \e[32m#{current_cmd_prefix}\e[0m"
+      else
+        "\e[32m#{Command.prompt_id&.+" "}\e[0m\e[33m>> \e[0m"
+      end
+    end
   end
 end

data/lib/richat/version.rb

@@ -1,3 +1,3 @@
 module Richat
-  VERSION = "0.2.5"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: richat
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.3.0
 platform: ruby
 authors:
 - fzdp
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-05 00:00:00.000000000 Z
+date: 2023-04-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -67,8 +67,9 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.12'
 description: Richat is a command-line ChatGPT tool implemented in Ruby that supports
-  highly customizable configuration. It can save chat logs, performs fuzzy searches
-  on historical inputs, allows for prompt customization and switching at any time
+  highly customizable configuration. It can save chat contents, performs fuzzy searches
+  on historical inputs, allows for prompt switching at any time and can event act
+  as Linux terminal.
 email:
 - fzdp01@gmail.com
 executables: