Module: Roast::Tools::Cmd

Extended by:
Cmd
Included in:
Cmd
Defined in:
lib/roast/tools/cmd.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object

Add this method to be included in other classes



13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/roast/tools/cmd.rb', line 13

def included(base)
  base.class_eval do
    function(
      :cmd,
      'Run a command in the current working directory (e.g. "ls", "rake", "ruby"). ' \
        "You may use this tool to execute tests and verify if they pass.",
      command: { type: "string", description: "The command to run in a bash shell." },
    ) do |params|
      Roast::Tools::Cmd.call(params[:command])
    end
  end
end

Instance Method Details

#call(command) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/roast/tools/cmd.rb', line 27

def call(command)
  Roast::Helpers::Logger.info("🔧 Running command: #{command}\n")

  # Validate the command starts with one of the allowed prefixes
  allowed_prefixes = ["pwd", "find", "ls", "rake", "ruby", "dev"]
  command_prefix = command.split(" ").first

  err = "Error: Command not allowed. Only commands starting with #{allowed_prefixes.join(", ")} are permitted."
  return err unless allowed_prefixes.any? do |prefix|
    command_prefix == prefix
  end

  # Execute the command in the current working directory
  result = ""

  # Use a full shell environment for commands, especially for 'dev'
  if command_prefix == "dev"
    # Use bash -l -c to ensure we get a login shell with all environment variables
    full_command = "bash -l -c '#{command.gsub("'", "\\'")}'"
    IO.popen(full_command, chdir: Dir.pwd) do |io|
      result = io.read
    end
  else
    # For other commands, use the original approach
    IO.popen(command, chdir: Dir.pwd) do |io|
      result = io.read
    end
  end

  exit_status = $CHILD_STATUS.exitstatus

  # Return the command output along with exit status information
  output = "Command: #{command}\n"
  output += "Exit status: #{exit_status}\n"
  output += "Output:\n#{result}"

  output
rescue StandardError => e
  "Error running command: #{e.message}".tap do |error_message|
    Roast::Helpers::Logger.error(error_message + "\n")
    Roast::Helpers::Logger.debug(e.backtrace.join("\n") + "\n") if ENV["DEBUG"]
  end
end