lib/koma/cli.rb in koma-0.10.1 vs lib/koma/cli.rb in koma-0.11.0

@@ -32,39 +32,39 @@
       else
         puts JSON.pretty_generate(gathered)
       end
     end
 
-    desc 'run-command <host1,host2,..> <command>', 'run command on a remote machine'
+    desc 'run-command <host1,host2,..> <command1> <command2> ...', 'run command on a remote machine'
     option :yaml, type: :boolean, desc: 'stdout YAML', aliases: :Y
     option :identity_file, type: :string, banner: '<identity_file>', desc: 'identity file', aliases: :i
     option :port, type: :numeric, banner: '<port>', desc: 'port', aliases: :p
-    def run_command(host = nil, command = nil)
+    def run_command(host = nil, *commands)
       if host.nil?
         STDERR.puts 'ERROR: "koma run-command" was called with no arguments'
         STDERR.puts 'Usage: "koma run-command <host1,host2,..> <command>"'
         return
       end
-      if command.nil?
-        command = host
+      if commands.empty?
+        commands = [host]
         begin
           stdin = timeout(1) { $stdin.read }
         rescue Timeout::Error
         end
         if stdin.nil?
-          STDERR.puts 'ERROR: "koma run-command" was called with no arguments'
-          STDERR.puts 'Usage: "koma run-command <host1,host2,..> <command>"'
+          STDERR.puts 'ERROR: "koma run-commands" was called with no arguments'
+          STDERR.puts 'Usage: "koma run-commands <host1,host2,..> <commands>"'
           return
         end
         ret = stdin.split("\n").select { |line| line =~ /^Host ([^\s\*]+)/ }.map do |line|
           line =~ /^Host ([^\s]+)/
           Regexp.last_match[1]
         end
         host = ret.join(',')
       end
       backend = Koma::Backend::Ssh.new(host, options)
       backend.stdin = stdin if stdin
-      gathered = backend.run_command(command)
+      gathered = backend.run_commands(commands)
       if options[:yaml]
         puts YAML.dump(gathered)
       else
         puts JSON.pretty_generate(gathered)
       end