improve prompt and follow directory changes

Author: stakach

README.md

@@ -6,7 +6,7 @@ Need to Google that command you can't quite remember? Use a Large Language Model
 * Just describe what you want to do and the LLM will generate the commands for you.
 * You can be vague, you will be prompted for parameter details (file and folder names, branches, commit messages etc)
 
-<img width="621" alt="example" src="https://user-images.githubusercontent.com/368013/232493882-6bb8b4f8-6988-41f1-9f25-a1685e0c1750.png">
+<img width="623" alt="example" src="https://user-images.githubusercontent.com/368013/232639264-cd136de1-a1cd-4e32-ba39-ee860793d9de.png">
 
 ## Installation
 

src/llm-cli.cr

@@ -18,9 +18,11 @@ rescue error : LLM::CLI::Chat::Error
 end
 shell = LLM::CLI::Shell.new
 prompt = LLM::CLI::Prompt.new({
-  "No user assistance",
+  "No user assistance, command ordering is important",
+  "You are running on #{shell.operating_system}",
   "Commands to be executed on the following shell: #{File.basename shell.selected}",
   "Wrap unknown command parameters in <brackets>",
+  "you might need to change directory before executing subsequent commands",
 })
 
 # grab the users request
@@ -99,6 +101,21 @@ begin
           error: :inherit
         ).wait
         puts "Command failed with exit code: #{status.exit_code}" unless status.success?
+
+        # change directory as required
+        if execute.starts_with?("cd ") || execute.includes?("&& cd ")
+          args = Process.parse_arguments(execute)
+          found = nil
+          args.each_with_index do |cmd, index|
+            if cmd == "cd"
+              found = index + 1
+              break
+            end
+          end
+          if found && (new_dir = args[found]?)
+            Dir.cd(new_dir)
+          end
+        end
       end
     end
   rescue error

src/llm-cli/shell.cr

@@ -22,6 +22,24 @@ class LLM::CLI::Shell
     shell
   end
 
+  {% begin %}
+    def operating_system
+      {% if flag?(:win32) %}
+        "windows"
+      {% elsif flag?(:darwin) %}
+        "macOS"
+      {% elsif flag?(:musl) %}
+        "musl linux"
+      {% elsif flag?(:linux) %}
+        "linux"
+      {% elsif flag?(:bsd) %}
+        "bsd"
+      {% else %}
+        "posix"
+      {% end %}
+    end
+  {% end %}
+
   def get_input(prompt)
     print prompt
     STDIN.gets.to_s.chomp