Class: Richat::Command
- Inherits:
-
Object
- Object
- Richat::Command
- Defined in:
- lib/richat/command.rb
Constant Summary collapse
- EXIT_CODE =
0
- NEXT_CODE =
1
- PROMPT_CHANGED_CODE =
2
- SYS_CMD_CODE =
3
- SYS_CHAT_CODE =
4
- TOGGLE_CONTEXT_CODE =
5
- RESET_CONTEXT_CODE =
6
Class Attribute Summary collapse
-
.prompt ⇒ Object
readonly
Returns the value of attribute prompt.
-
.prompt_id ⇒ Object
readonly
Returns the value of attribute prompt_id.
Class Method Summary collapse
- .call(user_input, sys_cmd_mode) ⇒ Object
- .handle_choose_prompt(arg) ⇒ Object
- .handle_config ⇒ Object
- .handle_exit ⇒ Object
- .handle_help ⇒ Object
- .handle_prompt ⇒ Object
- .handle_system_command(cmd) ⇒ Object
- .kill_process ⇒ Object
- .load_default_prompt ⇒ Object
- .print_exception(message) ⇒ Object
- .print_info(message) ⇒ Object
- .print_welcome ⇒ Object
- .process_exist? ⇒ Boolean
- .prompt_id_list ⇒ Object
Class Attribute Details
.prompt ⇒ Object (readonly)
Returns the value of attribute prompt.
12 13 14 |
# File 'lib/richat/command.rb', line 12 def prompt @prompt end |
.prompt_id ⇒ Object (readonly)
Returns the value of attribute prompt_id.
12 13 14 |
# File 'lib/richat/command.rb', line 12 def prompt_id @prompt_id end |
Class Method Details
.call(user_input, sys_cmd_mode) ⇒ Object
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
# File 'lib/richat/command.rb', line 14 def call(user_input, sys_cmd_mode) user_input = user_input.strip if sys_cmd_mode return SYS_CHAT_CODE if Config.get("sys_cmd", "deactivate_keywords").include?(user_input) handle_system_command(user_input) else return SYS_CMD_CODE if Config.get("sys_cmd", "activate_keywords").include?(user_input) 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 =~ /^\/prompt\s*/ if user_input == "/prompt" handle_prompt else handle_choose_prompt(user_input.split(" ").last) end elsif user_input == "/context" TOGGLE_CONTEXT_CODE elsif user_input == "/context-reset" RESET_CONTEXT_CODE end end end |
.handle_choose_prompt(arg) ⇒ Object
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 |
# File 'lib/richat/command.rb', line 131 def handle_choose_prompt(arg) id_list = prompt_id_list if id_list.empty? print_exception("Prompt files not found.") return NEXT_CODE end if id_list.include?(arg) current_prompt_id = arg elsif arg =~ /\d/ pt_id = id_list[arg.to_i] if pt_id.nil? print_exception("Prompt not found.") return NEXT_CODE end current_prompt_id = pt_id else print_exception("Prompt not found.") return NEXT_CODE end if current_prompt_id == @prompt_id print_exception("Prompt not changed.") return NEXT_CODE end prompt_content = File.read(File.join(File.(Config.get("prompt", "prompt_dir")), current_prompt_id)) if prompt_content.empty? print_exception("Prompt is empty.") return NEXT_CODE end puts "Changed prompt to #{current_prompt_id}." @prompt_id = current_prompt_id @prompt = prompt_content PROMPT_CHANGED_CODE end |
.handle_config ⇒ Object
88 89 90 91 92 |
# File 'lib/richat/command.rb', line 88 def handle_config puts "\e[32mConfiguration file path is #{File.("~/.richat/config.json")}\e[0m" puts JSON.pretty_generate(Config.get_config) NEXT_CODE end |
.handle_exit ⇒ Object
58 59 60 61 62 |
# File 'lib/richat/command.rb', line 58 def handle_exit kill_process puts "Bye!" EXIT_CODE end |
.handle_help ⇒ Object
94 95 96 97 98 99 100 101 102 103 |
# File 'lib/richat/command.rb', line 94 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" puts "\e[32m/context\e[0m toggle chat context" puts "\e[32m/context-reset\e[0m reset chat context" puts "\e[32m/help\e[0m show help info" NEXT_CODE end |
.handle_prompt ⇒ Object
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# File 'lib/richat/command.rb', line 109 def handle_prompt id_list = prompt_id_list if id_list.empty? print_exception "#{File.(Config.get("prompt", "prompt_dir"))} has no prompt files." return NEXT_CODE end if prompt.nil? || prompt.empty? puts "\e[32mCurrent prompt not set.\e[0m" puts "Use command \e[32m/prompt prompt_index\e[0m or \e[32m/prompt prompt_id\e[0m to set prompt." end id_list.each_with_index do |pt, idx| if pt == prompt_id puts "#{idx}. \e[32m#{pt}\e[0m" else puts "#{idx}. #{pt}" end end NEXT_CODE end |
.handle_system_command(cmd) ⇒ Object
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
# File 'lib/richat/command.rb', line 42 def handle_system_command(cmd) if (match = /^cd\s?(.*)$/.match(cmd)) fp = File.(match[1].empty? ? "~" : match[1]) begin Dir.chdir(fp) rescue puts "cd: no such file or directory: #{fp}" end else @pid = Process.spawn(cmd) Process.wait(@pid) end NEXT_CODE end |
.kill_process ⇒ Object
64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
# File 'lib/richat/command.rb', line 64 def kill_process return if @pid.nil? if Gem.win_platform? system("taskkill /F /PID #{@pid}", err: File::NULL) else return unless process_exist? begin Process.kill("TERM", @pid) rescue nil end end @pid = nil end |
.load_default_prompt ⇒ Object
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 |
# File 'lib/richat/command.rb', line 181 def load_default_prompt @prompt ||= begin @prompt_id = Config.get("prompt", "default") if @prompt_id.nil? || @prompt_id.empty? @prompt_id = nil return end begin File.read(File.join(File.(Config.get("prompt", "prompt_dir")), @prompt_id)) rescue => e print_exception(e.) exit end end end |
.print_exception(message) ⇒ Object
169 170 171 |
# File 'lib/richat/command.rb', line 169 def print_exception() puts "\e[31m#{}\e[0m" end |
.print_info(message) ⇒ Object
173 174 175 |
# File 'lib/richat/command.rb', line 173 def print_info() puts "\e[32m#{}\e[0m" end |
.print_welcome ⇒ Object
177 178 179 |
# File 'lib/richat/command.rb', line 177 def print_welcome puts "Richat is a command-line ChatGPT tool implemented in Ruby that supports highly customizable configuration, press \e[32m/help\e[0m to display help info. If you have any suggestions or questions, please feel free to provide feedback on https://github.com/fzdp/richat." end |
.process_exist? ⇒ Boolean
79 80 81 82 83 84 85 86 |
# File 'lib/richat/command.rb', line 79 def process_exist? begin Process.getpgid(@pid) true rescue false end end |