Create ssh-session-cli

Author: darkofabijan

ssh-session-cli

@@ -0,0 +1,92 @@
+#!/usr/bin/env ruby
+
+require "json"
+
+def bold(string)
+  "\e[33m#{string}\e[0m"
+end
+
+def load_commands(type)
+  setup_commands    = JSON.parse(File.read("/var/run/commands/setup_commands"))
+  job_commands      = JSON.parse(File.read("/var/run/commands/job_commands"))
+  post_job_commands = JSON.parse(File.read("/var/run/commands/post_job_commands"))
+
+  case type
+  when :setup
+    setup_commands
+  when :job
+    job_commands
+  when :post_job
+    post_job_commands
+  when :all
+   setup_commands + job_commands + post_job_commands
+  else
+    raise "Unrecognized command type: #{type}"
+  end
+end
+
+def display_help
+  puts "Semaphore CLI for SSH session."
+  puts ""
+  puts "  #{bold "run:setup"}     - Runs setup commands for this job"
+  puts "  #{bold "run:job"}       - Runs commands specific for this job"
+  puts "  #{bold "run:post_job"}  - Runs post_job commands for this job"
+  puts "  #{bold "run:all"}       - Runs all commands for this job"
+  puts ""
+  puts "  #{bold "show:setup"}    - Show setup commands for this job"
+  puts "  #{bold "show:job"}      - Show commands specific for this job"
+  puts "  #{bold "show:post_job"} - Show post_job commands for this job"
+  puts "  #{bold "show:all"}      - Show all commands for this job"
+  puts ""
+end
+
+def show(type)
+  load_commands(type).each.with_index do |cmd, index|
+    puts "#{bold (index + 1).to_s.rjust(2)}: #{cmd["command_string"]}"
+  end
+end
+
+def run(type)
+  load_commands(type).each do |cmd|
+    puts bold("$ #{cmd["command_string"]}")
+
+    # we need to use login shell if we want 'nvm' and 'rvm' to be available
+    system("bash", "-c", "-l", cmd["command_string"])
+
+    status = $?.exitstatus
+
+    if status != 0
+      puts ""
+      puts bold("Command Failed. Exit Status: #{status}")
+      exit(status)
+    end
+
+    puts ""
+  end
+end
+
+if ARGV.size == 0
+  display_help
+  exit
+end
+
+case ARGV[0]
+when "run:setup"
+  run(:setup)
+when "run:job"
+  run(:job)
+when "run:post_job"
+  run(:post_job)
+when "run:all"
+  run(:all)
+when "show:setup"
+  show(:setup)
+when "show:job"
+  show(:job)
+when "show:post_job"
+  show(:post_job)
+when "show:all"
+  show(:all)
+else
+  abort "[ERROR] Unrecognized command"
+end