octo 0.0.4 → 0.0.5

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6f725787f4c6ae31626f1eceed7504ed32eb98f3
+  data.tar.gz: 3e3214b8d920d863483ba69b558ec78a5696dccc
+SHA512:
+  metadata.gz: 95a553226385139f430c0ab7b3cce4b7e4d254a040e30e24dd348e59eaa79f0457167ab3626ea2a34036c93f2a658d9241ff48810076f7bc4026d10efd4c3aae
+  data.tar.gz: bed80746eeecec5bb79b557c27ad18000155e7999d3e04e19438dcabd244413ff3ae73a506fe7eb2981181fc0bf3b14f2d2ee6df928084144cda94b2d8ca28c5

data/bin/octo

@@ -20,6 +20,9 @@ include GLI::App
 
 program_desc 'Run commands on multiple hosts'
 
+desc 'Set the mode tht octo will run in (eg. ssh, mysql)'
+flag [:m, :mode], must_match: /(mysql)|(ssh)/i, default_value: 'ssh', type: String
+
 octo = nil
 pre do |global_opts, command, opts, args|
   octo = Octo.new(opts)
@@ -35,17 +38,19 @@ command :profile do |profile|
   profile.command :list do |list|
     list.action do |global_opts, opts, args|
       if args.empty?
-        octo.list.each do |profile|
+        octo.config[global_opts[:m]].each do |profile, servers|
           puts "#{profile}:"
-          octo.config[profile].each do |server|
+          servers.each do |server|
             puts "- #{server}"
           end
         end
       else
-        exit_now!("#{args.first} is not a defined profile") unless octo.config.keys.include? args.first
+        unless octo.profile_exists? global_opts[:m], args.first
+          exit_now!("#{args.first} is not a defined profile")
+        end
 
         puts "#{args.first}:"
-        octo.config[args.first].each do |server|
+        octo.list(global_opts[:m], args.first).each do |server|
           puts "- #{server}"
         end
       end
@@ -57,7 +62,7 @@ command :profile do |profile|
   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]
+      octo.add global_opts[:m], args[0], args[1]
     end
   end
 
@@ -66,7 +71,11 @@ command :profile do |profile|
   profile.arg_name 'profile server'
   profile.command :rm do |rm|
     rm.action do |global_opts, opts, args|
-      octo.rm args[0], args[1]
+      unless octo.profile_exists? global_opts[:m], args.first
+        exit_now!("#{args.first} is not a defined profile")
+      end
+
+      octo.rm global_opts[:m], args[0], args[1]
     end
   end
 end
@@ -76,7 +85,17 @@ 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]
+    unless octo.profile_exists? global_opts[:m], args.first
+      exit_now!("#{args.first} is not a defined profile")
+    end
+
+    case global_opts[:m]
+    when 'ssh'
+      octo.run_ssh args[0], args[1]
+    when 'mysql'
+      exit_now!("No query specified") if args[1].nil? || args[1].empty?
+      octo.run_mysql args[0], args[1]
+    end
   end
 end
 

data/lib/octo/profile.rb

@@ -7,22 +7,32 @@ module Octo::Profile
     "#{Dir.home}/.octorc"
   end
 
-  def list(profile = nil)
+  def profile_exists?(type, profile)
+    @config[type].keys.include? profile
+  end
+
+  def list(type, profile = nil)
     if profile.nil?
-      @config.keys
+      @config[type].keys
     else
-      @config[profile]
+      @config[type].each do |name, servers|
+        if name == profile
+          return servers
+        end
+      end
+
+      return nil
     end
   end
 
-  def add(profile, server)
-    @config[profile] = [] if @config[profile].nil?
-    @config[profile] << server
+  def add(type, profile, server)
+    @config[type][profile] = [] if @config[type][profile].nil?
+    @config[type][profile] << server
   end
 
-  def rm(profile, server)
-    @config[profile].delete(server)
-    @config.delete(profile) if @config[profile].empty?
+  def rm(type, profile, server)
+    @config[type][profile].delete(server)
+    @config[type].delete(profile) if @config[type][profile].empty?
   end
 
   def load

data/lib/octo.rb

@@ -1,5 +1,7 @@
 require 'yaml'
 require 'net/ssh/multi'
+require 'mysql2'
+require 'terminal-table'
 require 'term/ansicolor'
 
 class Octo
@@ -16,9 +18,9 @@ class Octo
     self.load
   end
 
-  def run(profile, command)
+  def run_ssh(profile, command)
     Net::SSH::Multi.start do |session|
-      @config[profile].each do |server|
+      @config['ssh'][profile].each do |server|
         session.use server
       end
 
@@ -31,4 +33,30 @@ class Octo
       session.loop
     end
   end
+
+  def run_mysql(profile, query)
+    @config['mysql'][profile].each do |server|
+      print "Running query on #{server}... "
+
+      begin
+        server_data = server.match(/(.+):(.+)@(.+)\/(.+)/)
+        client = Mysql2::Client.new(username: server_data[1],
+                                    password: server_data[2],
+                                    host: server_data[3],
+                                    database: server_data[4])
+
+        results = client.query(query)
+        puts "#{results.count} results"
+        table = Terminal::Table.new(headings: results.fields) do |t|
+          results.each(as: :array) do |row|
+            t << row
+          end
+        end
+
+        puts table
+      rescue Exception => e
+        puts "ERROR: #{e.inspect}"
+      end
+    end
+  end
 end

data/octo.gemspec

@@ -2,7 +2,7 @@ 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.4'
+  s.version      = '0.0.5'
   s.author       = 'Nan Zhong'
   s.email        = 'nan@nine27.com'
   s.homepage     = 'https://github.com/nanzhong/octo'
@@ -11,6 +11,8 @@ Gem::Specification.new do |s|
   s.executables << 'octo'
 
   s.add_dependency 'net-ssh-multi'
+  s.add_dependency 'mysql2'
   s.add_dependency 'gli'
   s.add_dependency 'term-ansicolor'
+  s.add_dependency 'terminal-table'
 end

metadata

@@ -1,62 +1,83 @@
 --- !ruby/object:Gem::Specification
 name: octo
 version: !ruby/object:Gem::Version
-  version: 0.0.4
-  prerelease: 
+  version: 0.0.5
 platform: ruby
 authors:
 - Nan Zhong
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-20 00:00:00.000000000 Z
+date: 2013-03-13 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: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    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
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    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
@@ -74,26 +95,25 @@ files:
 - octo.gemspec
 homepage: https://github.com/nanzhong/octo
 licenses: []
+metadata: {}
 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
+rubygems_version: 2.0.0
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: Run commands in parallel on multiple hosts
 test_files: []