aicli 0.3.1 → 0.4.0

data/lib/aicli/helpers/theme.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require 'pastel'
+require 'tty-prompt'
+
+module AiCli
+  module Helpers
+    module Theme
+      module_function
+
+      def accent(text)
+        pastel.public_send(Constants::ACCENT_COLOR, text)
+      end
+
+      def bold(text)
+        pastel.bold(text)
+      end
+
+      def dim(text)
+        pastel.dim(text)
+      end
+
+      def status(config)
+        "#{Constants::PROJECT_NAME}, #{config['PROVIDER']} / #{config['MODEL']}"
+      end
+
+      def show_status(config)
+        puts accent(status(config))
+      end
+
+      def prompt(**options)
+        TTY::Prompt.new(active_color: Constants::ACCENT_COLOR, **options)
+      end
+
+      def pastel
+        @pastel ||= Pastel.new
+      end
+    end
+  end
+end

data/lib/aicli/prompt.rb

@@ -1,102 +1,46 @@
 # frozen_string_literal: true
 
-require 'open3'
 require 'clipboard'
-require 'pastel'
-require 'tty-prompt'
-require 'tty-spinner'
 
 module AiCli
   module Prompt
     module_function
 
     def run(use_prompt: nil, silent_mode: false)
-      init_i18n
-
       config = Helpers::Config.get
-      skip_explanation = silent_mode || config['SILENT_MODE']
-
-      pastel = Pastel.new
-      puts ''
-      puts pastel.cyan(Helpers::Constants::PROJECT_NAME)
-      puts pastel.dim("#{config['PROVIDER']} / #{config['MODEL']}")
-
-      the_prompt = use_prompt.nil? || use_prompt.strip.empty? ? ask_prompt : use_prompt
+      prompt_text = use_prompt.to_s.strip.empty? ? ask_prompt : use_prompt
 
-      spinner = TTY::Spinner.new("[:spinner] #{Helpers::I18n.t('Loading...')}", format: :dots)
-      spinner.auto_spin
-
-      result = Helpers::Completion.get_script_and_info(
-        prompt: the_prompt,
-        config: config
-      )
-
-      spinner.success(Helpers::I18n.t('Your script') + ':')
-      puts ''
-      script = result[:read_script].call(->(chunk) { print chunk })
-      puts ''
-      puts ''
-
-      explanation_text = ''
-      unless skip_explanation
-        spinner = TTY::Spinner.new("[:spinner] #{Helpers::I18n.t('Getting explanation...')}", format: :dots)
-        spinner.auto_spin
-
-        info = result[:read_info].call(->(chunk) { print chunk })
-        if info.nil? || info.empty?
-          explanation = Helpers::Completion.get_explanation(
-            script: script,
-            config: config
-          )
-          spinner.success(Helpers::I18n.t('Explanation') + ':')
-          puts ''
-          explanation_text = explanation[:read_explanation].call(->(chunk) { print chunk })
-          puts ''
-          puts ''
-        else
-          explanation_text = info
-          spinner.success(Helpers::I18n.t('Explanation') + ':')
-        end
-      end
-
-      append_prompt_context(the_prompt, script, explanation_text)
-      run_or_revise_flow(script, config, silent_mode)
+      Helpers::Theme.show_status(config)
+      script = Helpers::Completion.get_script(prompt: prompt_text, config: config)
+      save_context(prompt_text, script, '')
+      present_script_menu(script, config, silent_mode, leading_newline: true)
     end
 
-    def init_i18n
-      config = Helpers::Config.get
-      Helpers::I18n.set_language(config['LANGUAGE'])
-    rescue StandardError
-      Helpers::I18n.set_language('en')
-    end
-
-    def examples
-      [
-        Helpers::I18n.t('delete all log files'),
-        Helpers::I18n.t('list js files'),
-        Helpers::I18n.t('fetch me a random joke'),
-        Helpers::I18n.t('list all commits')
-      ]
-    end
-
-    def ask_prompt(initial = nil)
-      prompt = TTY::Prompt.new(interrupt: :error)
+    def ask_prompt
+      prompt = Helpers::Theme.prompt(interrupt: :error)
       Helpers::Context.seed_prompt_history(prompt)
 
       loop do
-        begin
-          return prompt.ask(Helpers::I18n.t('What would you like me to do?')) do |q|
-            q.required true
-            q.default initial || Helpers::I18n.t('Say hello')
-            q.validate(/.+/, Helpers::I18n.t('Please enter a prompt.'))
-          end
-        rescue TTY::Reader::InputInterrupt
-          print "\r\e[2K"
+        return prompt.ask(Helpers::I18n.t('What would you like me to do?')) do |q|
+          q.required true
+          q.default Helpers::I18n.t('Say hello')
+          q.validate(/.+/, Helpers::I18n.t('Please enter a prompt.'))
         end
+      rescue TTY::Reader::InputInterrupt
+        print "\r\e[2K"
+      end
+    end
+
+    def ask_revision
+      Helpers::Theme.prompt(interrupt: :exit).ask(
+        Helpers::I18n.t('What would you like me to change in this script?')
+      ) do |q|
+        q.required true
+        q.validate(/.+/, Helpers::I18n.t('Please enter a prompt.'))
       end
     end
 
-    def append_prompt_context(user_prompt, script, explanation)
+    def save_context(user_prompt, script, explanation = '')
       assistant = +"```\n#{script}\n```"
       assistant << "\n\n#{explanation}" unless explanation.to_s.strip.empty?
 
@@ -106,43 +50,65 @@ module AiCli
       Helpers::Context.save_messages(messages)
     end
 
-    def ask_revision
-      prompt = TTY::Prompt.new(interrupt: :exit)
-      prompt.ask(Helpers::I18n.t('What would you like me to change in this script?')) do |q|
-        q.required true
-        q.validate(/.+/, Helpers::I18n.t('Please enter a prompt.'))
-      end
+    def append_explanation_to_context(explanation)
+      text = explanation.to_s.strip
+      return if text.empty?
+
+      messages = Helpers::Context.load_messages
+      return if messages.empty?
+
+      last = messages.last
+      return unless last['role'] == 'assistant' && !last['content'].include?(text)
+
+      last['content'] = "#{last['content']}\n\n#{text}"
+      Helpers::Context.save_messages(messages)
+    end
+
+    def display_script(script, leading_newline: false)
+      text = script.to_s.strip
+      return if text.empty?
+
+      puts '' if leading_newline
+      puts Helpers::Theme.bold(text)
     end
 
-    def run_script(script)
-      puts "#{Helpers::I18n.t('Running')}: #{script}"
+    def display_explanation(text)
+      stripped = text.to_s.strip
+      return if stripped.empty?
+
       puts ''
-      system(ENV['SHELL'] || 'bash', '-c', script)
-      Helpers::ShellHistory.record_executed(script)
+      puts Helpers::Theme.dim(stripped)
     end
 
-    def run_or_revise_flow(script, config, silent_mode)
-      prompt = TTY::Prompt.new(interrupt: :exit)
-      empty_script = script.strip.empty?
+    def present_script_menu(script, config, silent_mode, leading_newline: false, show_script: true)
+      display_script(script, leading_newline: leading_newline) if show_script
+      run_action_menu(script, config, silent_mode)
+    end
 
-      choices = []
-      unless empty_script
-        choices << { name: "✅ #{Helpers::I18n.t('Yes')}", value: :yes }
-        choices << { name: "📝 #{Helpers::I18n.t('Edit')}", value: :edit }
-      end
-      choices << { name: "🔁 #{Helpers::I18n.t('Revise')}", value: :revise }
-      choices << { name: "📋 #{Helpers::I18n.t('Copy')}", value: :copy }
-      choices << { name: "❌ #{Helpers::I18n.t('Cancel')}", value: :cancel }
+    def run_action_menu(script, config, silent_mode)
+      empty = script.strip.empty?
+      message = empty ? Helpers::I18n.t('Revise this script?') : Helpers::I18n.t('Run this script?')
 
-      message = empty_script ? Helpers::I18n.t('Revise this script?') : Helpers::I18n.t('Run this script?')
-      answer = prompt.select(message, choices)
+      puts ''
+      answer = Helpers::ScriptActionMenu.select(
+        question: message,
+        config: config,
+        silent_mode: silent_mode,
+        empty_script: empty
+      )
 
       case answer
       when :yes
-        run_script(script)
+        puts ''
+        Helpers::ShellExec.run(script)
       when :edit
-        new_script = prompt.ask(Helpers::I18n.t('you can edit script here'), default: script)
-        run_script(new_script) if new_script && !new_script.empty?
+        edited = Helpers::ScriptEditor.ask(script)
+        present_script_menu(edited, config, silent_mode, show_script: false) if edited && !edited.empty?
+      when :explain
+        text = Helpers::Completion.get_explanation(script: script, config: config)
+        display_explanation(text)
+        append_explanation_to_context(text)
+        present_script_menu(script, config, silent_mode, leading_newline: !text.to_s.strip.empty?)
       when :revise
         revision_flow(script, config, silent_mode)
       when :copy
@@ -154,45 +120,14 @@ module AiCli
       end
     end
 
-    def revision_flow(current_script, config, silent_mode)
+    def revision_flow(script, config, silent_mode)
       revision = ask_revision
-
-      spinner = TTY::Spinner.new("[:spinner] #{Helpers::I18n.t('Loading...')}", format: :dots)
-      spinner.auto_spin
-
-      result = Helpers::Completion.get_revision(
-        prompt: revision,
-        code: current_script,
-        config: config
-      )
-
-      spinner.success(Helpers::I18n.t('Your new script') + ':')
-      puts ''
-      script = result[:read_script].call(->(chunk) { print chunk })
-      puts ''
-      puts ''
-
-      unless silent_mode
-        info_spinner = TTY::Spinner.new("[:spinner] #{Helpers::I18n.t('Getting explanation...')}", format: :dots)
-        info_spinner.auto_spin
-
-        explanation = Helpers::Completion.get_explanation(
-          script: script,
-          config: config
-        )
-
-        info_spinner.success(Helpers::I18n.t('Explanation') + ':')
-        puts ''
-        explanation[:read_explanation].call(->(chunk) { print chunk })
-        puts ''
-        puts ''
-      end
-
-      run_or_revise_flow(script, config, silent_mode)
+      revised = Helpers::Completion.get_revision(prompt: revision, code: script, config: config)
+      present_script_menu(revised, config, silent_mode, leading_newline: true)
     end
 
-    private_class_method :init_i18n, :examples, :ask_prompt, :ask_revision,
-                         :run_script, :run_or_revise_flow, :revision_flow,
-                         :append_prompt_context
+    private_class_method :ask_prompt, :ask_revision, :save_context, :append_explanation_to_context,
+                         :display_script, :display_explanation, :present_script_menu,
+                         :run_action_menu, :revision_flow
   end
 end

data/lib/aicli/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiCli
-  VERSION = '0.3.1'
+  VERSION = '0.4.0'
 end

data/lib/aicli.rb

@@ -2,11 +2,14 @@
 
 require_relative 'aicli/version'
 require_relative 'aicli/helpers/constants'
+require_relative 'aicli/helpers/theme'
+require_relative 'aicli/helpers/script_action_menu'
 require_relative 'aicli/helpers/error'
 require_relative 'aicli/helpers/i18n'
 require_relative 'aicli/helpers/os_detect'
-require_relative 'aicli/helpers/strip_regex_patterns'
 require_relative 'aicli/helpers/shell_history'
+require_relative 'aicli/helpers/shell_exec'
+require_relative 'aicli/helpers/script_editor'
 require_relative 'aicli/helpers/llm'
 require_relative 'aicli/helpers/completion'
 require_relative 'aicli/helpers/config'

data/locales/en.yml

@@ -1,11 +1,7 @@
 ---
-Starting new conversation: Starting new conversation
 Goodbye!: Goodbye!
-send a message ('exit' to quit): send a message ('exit' to quit)
 Please enter a prompt.: Please enter a prompt.
 THINKING...: THINKING...
-Please set your OpenAI API key via `ai config set OPENAI_KEY=<your token>`: Please
-  set your OpenAI API key via `ai config set OPENAI_KEY=<your token>`
 Set config: Set config
 Enter your OpenAI API key: Enter your OpenAI API key
 Enter your Anthropic API key: Enter your Anthropic API key
@@ -14,7 +10,6 @@ Provider: Provider
 Pick a provider: Pick a provider
 OpenAI Key: OpenAI Key
 Anthropic Key: Anthropic Key
-Please enter a key: Please enter a key
 OpenAI API Endpoint: OpenAI API Endpoint
 Enter your OpenAI API Endpoint: Enter your OpenAI API Endpoint
 Silent Mode: Silent Mode
@@ -22,51 +17,30 @@ Enable silent mode?: Enable silent mode?
 Model: Model
 Pick a model.: Pick a model.
 No models found for this provider.: No models found for this provider.
-Enter the model you want to use: Enter the model you want to use
 Language: Language
 Enter the language you want to use: Enter the language you want to use
 What would you like me to do?: What would you like me to do?
-delete all log files: delete all log files
-list js files: list js files
-fetch me a random joke: fetch me a random joke
-list all commits: list all commits
 Say hello: Say hello
-you can edit script here: you can edit script here
-What would you like me to change in this script?: What would you like me to change
-  in this script?
-e.g.: e.g.
-e.g. change the folder name: e.g. change the folder name
-Your script: Your script
-Loading...: Loading...
-Getting explanation...: Getting explanation...
-Explanation: Explanation
+Edit script: Edit script
+Edit with arrow keys, Enter to confirm, Ctrl+C to cancel: Edit with arrow keys, Enter to confirm, Ctrl+C to cancel
+What would you like me to change in this script?: What would you like me to change in this script?
 Run this script?: Run this script?
-Run it?: Run it?
 Revise this script?: Revise this script?
-'Yes': 'Yes'
-Lets go!: Lets go!
-Edit: Edit
-Make some adjustments before running: Make some adjustments before running
+Yes: Yes
+Modify: Modify
+Exit: Exit
+Explain: Explain
 Revise: Revise
-Give feedback via prompt and get a new result: Give feedback via prompt and get a
-  new result
 Copy: Copy
-Copy the generated script to your clipboard: Copy the generated script to your clipboard
 Cancel: Cancel
-Exit the program: Exit the program
 Running: Running
 Copied to clipboard!: Copied to clipboard!
-Your new script: Your new script
 Invalid config property: Invalid config property
+Invalid mode: Invalid mode
 Shell detection failed unexpectedly: Shell detection failed unexpectedly
-Invalid model: Invalid model
 Error: Error
 Missing required parameter: Missing required parameter
-Please open a Bug report with the information above: Please open a Bug report with
-  the information above
-Prompt to run: Prompt to run
+Please open a Bug report with the information above: Please open a Bug report with the information above
 You: You
-Ask for a shell command. Type exit to quit.: Ask for a shell command. Type exit to quit.
-Restored context: Restored context
-messages: messages
 Shell integration installed.: Shell integration installed.
+Run it?: Run it?

data/locales/it.yml

@@ -1,11 +1,7 @@
 ---
-Starting new conversation: Avvia una nuova conversazione
 Goodbye!: Arrivederci!
-send a message ('exit' to quit): invia un messaggio ('exit' per uscire)
 Please enter a prompt.: Per favore inserisci un prompt.
 THINKING...: pensando...
-Please set your OpenAI API key via `ai config set OPENAI_KEY=<your token>`: Per favore
-  imposta la tua chiave API OpenAI tramite `ai config set OPENAI_KEY=<your token>`
 Set config: Imposta la configurazione
 Enter your OpenAI API key: Inserisci la tua chiave API OpenAI
 Enter your Anthropic API key: Inserisci la tua chiave API Anthropic
@@ -14,7 +10,6 @@ Provider: Provider
 Pick a provider: Scegli un provider
 OpenAI Key: Chiave OpenAI
 Anthropic Key: Chiave Anthropic
-Please enter a key: Per favore inserisci una chiave
 OpenAI API Endpoint: Endpoint API OpenAI
 Enter your OpenAI API Endpoint: Inserisci il tuo endpoint API OpenAI
 Silent Mode: Modalità silenziosa
@@ -22,50 +17,30 @@ Enable silent mode?: Abilitare la modalità silenziosa?
 Model: Modello
 Pick a model.: Scegli un modello.
 No models found for this provider.: Nessun modello trovato per questo provider.
-Enter the model you want to use: Inserisci il modello che vuoi utilizzare
 Language: Lingua
 Enter the language you want to use: Inserisci la lingua che vuoi utilizzare
 What would you like me to do?: Cosa vorresti che facessi?
-delete all log files: elimina tutti i file di log
-list js files: elenca i file js
-fetch me a random joke: raccontami una barzelletta
-list all commits: elenca tutti i commit
 Say hello: Saluta
-you can edit script here: puoi modificare lo script qui
-What would you like me to change in this script?: Cosa vorresti che cambiassi in questo
-  script?
-e.g.: es.
-e.g. change the folder name: es. cambia il nome della cartella
-Your script: Il tuo script
-Loading...: Caricamento...
-Getting explanation...: Ottieni spiegazione...
-Explanation: Spiegazione
+Edit script: Modifica lo script
+Edit with arrow keys, Enter to confirm, Ctrl+C to cancel: Modifica con le frecce, Invio per confermare, Ctrl+C per annullare
+What would you like me to change in this script?: Cosa vorresti che cambiassi in questo script?
 Run this script?: Esegui questo script?
-Run it?: Lo eseguo?
 Revise this script?: Rivedi questo script?
-'Yes': Si
-Lets go!: Vai!
-Edit: Modifica
-Make some adjustments before running: Fai alcune modifiche prima di eseguire
-Revise: Revisiona
-Give feedback via prompt and get a new result: Dai un feedback tramite prompt e ottieni
-  un nuovo risultato
+Yes: Sì
+Modify: Modifica
+Exit: Esci
+Explain: Spiega
+Revise: Rivedi
 Copy: Copia
-Copy the generated script to your clipboard: Copia lo script generato negli appunti
 Cancel: Annulla
-Exit the program: Esci dal programma
 Running: Esecuzione
 Copied to clipboard!: Copiato negli appunti!
-Your new script: Il tuo nuovo script
 Invalid config property: Proprietà di configurazione non valida
-Shell detection failed unexpectedly: Rilevamento shell fallito inaspettatamente
-Invalid model: Modello non valido
+Invalid mode: Modalità non valida
+Shell detection failed unexpectedly: Rilevamento shell fallito in modo imprevisto
 Error: Errore
-Please open a Bug report with the information above: Apri un report di bug con le
-  informazioni sopra
-Prompt to run: Prompt da eseguire
+Missing required parameter: Parametro richiesto mancante
+Please open a Bug report with the information above: Apri una segnalazione bug con le informazioni sopra
 You: Tu
-Ask for a shell command. Type exit to quit.: Chiedi un comando shell. Digita exit per uscire.
-Restored context: Contesto ripristinato
-messages: messaggi
 Shell integration installed.: Integrazione shell installata.
+Run it?: Lo eseguo?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aicli
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Antonio Molinari
@@ -37,6 +37,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.8'
+- !ruby/object:Gem::Dependency
+  name: reline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.0
 - !ruby/object:Gem::Dependency
   name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
@@ -122,8 +136,11 @@ files:
 - lib/aicli/helpers/i18n.rb
 - lib/aicli/helpers/llm.rb
 - lib/aicli/helpers/os_detect.rb
+- lib/aicli/helpers/script_action_menu.rb
+- lib/aicli/helpers/script_editor.rb
+- lib/aicli/helpers/shell_exec.rb
 - lib/aicli/helpers/shell_history.rb
-- lib/aicli/helpers/strip_regex_patterns.rb
+- lib/aicli/helpers/theme.rb
 - lib/aicli/prompt.rb
 - lib/aicli/version.rb
 - locales/ar.yml

data/lib/aicli/helpers/strip_regex_patterns.rb

@@ -1,21 +0,0 @@
-# frozen_string_literal: true
-
-module AiCli
-  module Helpers
-    module StripRegexPatterns
-      module_function
-
-      def call(input_string, pattern_list)
-        pattern_list.reduce(input_string) do |current, pattern|
-          next current if pattern.nil?
-
-          if pattern.is_a?(Regexp)
-            current.gsub(pattern, '')
-          else
-            current.gsub(pattern.to_s, '')
-          end
-        end
-      end
-    end
-  end
-end