octo 0.0.2

data/.gitignore

@@ -0,0 +1,2 @@
+*.swp
+*.swo

data/bin/octo

@@ -0,0 +1,90 @@
+#!/usr/bin/env ruby
+
+require 'gli'
+require 'octo'
+
+# Install signal handlers
+Signal.trap('SIGINT') do
+  exit
+end
+
+Signal.trap('SIGKILL') do
+  exit
+end
+
+Signal.trap('SIGTERM') do
+  exit
+end
+
+include GLI::App
+
+program_desc 'Run commands on multiple hosts'
+
+flag [:t,:tasklist], :default_value => File.join(ENV['HOME'],'.todolist')
+
+octo = nil
+pre do |global_opts, command, opts, args|
+  octo = Octo.new(opts)
+  true
+end
+
+desc 'Manage profiles'
+long_desc 'Manage profile that will be used to run commands. Each profile consists of a set of servers.'
+command :profile do |profile|
+  profile.desc 'List all profiles'
+  profile.long_desc 'List all the profiles that have been previously configured along with their server.'
+  profile.arg_name '[profile]'
+  profile.command :list do |list|
+    list.action do |global_opts, opts, args|
+      if args.empty?
+        octo.list.each do |profile|
+          puts "#{profile}:"
+          octo.config[profile].each do |server|
+            puts "- #{server}"
+          end
+        end
+      else
+        exit_now!("#{args.first} is not a defined profile") unless octo.config.keys.include? args.first
+
+        puts "#{args.first}:"
+        octo.config[args.first].each do |server|
+          puts "- #{server}"
+        end
+      end
+    end
+  end
+
+  profile.desc 'Add a host to a profile'
+  profile.arg_name 'profile server'
+  profile.long_desc 'Add a host to a profile, the host should be specified using the form user@hostname, eg. nan@nine27.com.'
+  profile.command :add do |add|
+    add.action do |global_opts, opts, args|
+      octo.add args[0], args[1]
+    end
+  end
+
+  profile.desc 'Remove a host from a profile'
+  profile.long_desc 'Remove a host from a profile, the host should be specified using the form user@hostname, eg. nan@nine27.com.'
+  profile.arg_name 'profile server'
+  profile.command :rm do |rm|
+    rm.action do |global_opts, opts, args|
+      octo.rm args[0], args[1]
+    end
+  end
+end
+
+desc 'Run a command'
+long_desc 'Run a command in parallel on all configured hosts in the profile, eg.
+octo run cluster_1 \'tail -f /var/log/httpd/error_log\'.'
+command :run do |cmd|
+  cmd.action do |global_opts, tops, args|
+    octo.run args[0], args[1]
+  end
+end
+
+post do |global_opts, command, opts, args|
+  octo.save
+  true
+end
+
+exit run(ARGV)

data/lib/octo.rb

@@ -0,0 +1,34 @@
+require 'yaml'
+require 'net/ssh/multi'
+require 'term/ansicolor'
+
+class Octo
+  require 'octo/profile'
+  include Octo::Profile
+  include Term::ANSIColor
+
+  def initialize(options = {})
+    @options = options.merge({
+      file: false,
+      multi: false
+    })
+
+    self.load
+  end
+
+  def run(profile, command)
+    Net::SSH::Multi.start do |session|
+      @config[profile].each do |server|
+        session.use server
+      end
+
+      session.exec command do |ch, stream, data|
+        data.lines.each do |line|
+          stream = stream == :stderr ? $stderr : $stdout
+          stream.puts "[#{green(ch.properties[:server].to_s)}] #{line}"
+        end
+      end
+      session.loop
+    end
+  end
+end

data/lib/octo/profile.rb

@@ -0,0 +1,41 @@
+require 'octo'
+
+module Octo::Profile
+  attr_reader :config
+
+  def config_file
+    "#{Dir.home}/.octorc"
+  end
+
+  def list(profile = nil)
+    if profile.nil?
+      @config.keys
+    else
+      @config[profile]
+    end
+  end
+
+  def add(profile, server)
+    @config[profile] = [] if @config[profile].nil?
+    @config[profile] << server
+  end
+
+  def rm(profile, server)
+    @config[profile].delete(server)
+    @config.delete(profile) if @config[profile].empty?
+  end
+
+  def load
+    if File.exists?(config_file)
+      @config = YAML.load_file(config_file)
+    else
+      @config = {}
+    end
+  end
+
+  def save
+    File.open(config_file, 'w') do |f|
+      f.write YAML.dump(@config)
+    end
+  end
+end

data/octo.gemspec

@@ -0,0 +1,16 @@
+Gem::Specification.new do |s|
+  s.name         = 'octo'
+  s.summary      = 'Run commands in parallel on multiple hosts'
+  s.description  = 'A lightweight script that allows you to run commands in parallel on multiple hosts'
+  s.version      = '0.0.2'
+  s.author       = 'Nan Zhong'
+  s.email        = 'nan@nine27.com'
+  s.homepage     = 'https://github.com/nanzhong/octo'
+  s.files        = `git ls-files`.split($\)
+
+  s.executables << 'octo'
+
+  s.add_dependency 'net-ssh-multi'
+  s.add_dependency 'gli'
+  s.add_dependency 'term-ansicolor'
+end

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification
+name: octo
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Nan Zhong
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh-multi
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: gli
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: term-ansicolor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A lightweight script that allows you to run commands in parallel on multiple
+  hosts
+email: nan@nine27.com
+executables:
+- octo
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- bin/octo
+- lib/octo.rb
+- lib/octo/profile.rb
+- octo.gemspec
+homepage: https://github.com/nanzhong/octo
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Run commands in parallel on multiple hosts
+test_files: []