broham 0.0.9 → 0.0.10

data/README.textile

@@ -64,6 +64,13 @@ h4. Commandline Interface
   $ broham-unregister-all chad nfs_server
   Removing nfs_server from chad cluster
   {"timestamp":["20100405093238"],"fqdn":["nfs.infochimps.org"],"client_path":["/home"],"server_path":["/home"],"role":["nfs_server"],"private_ip":["10.123.156.231"],"default_ip":["10.123.156.231"],"public_ip":["104.136.251.50"]}
+
+  # Alternative interface:
+  broham-sup host               # show host
+  broham-sup_yall hosts         # show all hosts matching the given (start-of-string-anchored) regex
+  broham-yo host                # register as given host
+  broham-diss host              # unregister given host
+  broham-fuck_all_yall hosts_re # unregister all hosts matching the given (start-of-string-anchored) regex
 </code></pre>
 
 h4. IRB Usage

data/VERSION

@@ -1 +1 @@
-0.0.9
+0.0.10

data/bin/broham

@@ -3,13 +3,132 @@ require 'rubygems'
 require 'broham'
 require 'broham/script'
 
-Broham.get_cluster_settings
-Broham.get_command_line_args :command, :role_name
-role_name = Settings[:role_name]
+[
+  # ['sup',             '', '(alias for hosts)'],
+  # ['yo',              '', '(alias for register)'],
+  # ['diss',            '', '(alias for unregister)'],
+  # ['fuck_all_yall',   '', '(alias for unregister_like)'],
+  # ['word',            '', '(alias for get)'],
+  #
+  ['hosts',           'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['register',        'role [--extra_param=val...]', 'Registers this machine for the given role; stuff in any leftover params'],
+  ['unregister',      'role',                        'Unregister that role'],
+  ['unregister_like', 'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['get',             'role_pattern attr',           'show attr for all hosts prefixed by role_pattern']
+].each{|cmd| Configliere::COMMANDS << cmd }
 
+class BrohamScript
+  attr_accessor :role_name, :attr
+  def command()   Settings.command ; end
+  def role_name() Settings[:role_name] ; end
+  def attr()      Settings[:attr]      ; end
+
+  def hosts_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Listing hosts like #{role_name}}
+    dump_table Broham.hosts_like(role_name)
+  end
+
+  def register_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register(role_name)
+    dump_table [new_bro]
+  end
+
+  def register_as_next_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register_as_next(role_name)
+    dump_table [new_bro]
+  end
+
+  def unregister_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering #{role_name}}
+    dead_bro = Broham.unregister(role_name)
+    dump_table [dead_bro]
+  end
+
+  def unregister_like_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering all hosts like #{role_name}}
+    dump_table Broham.unregister_like role_name
+  end
+
+  def get_cmd
+    BrohamScript.get_commandline_predicates :role_name, :attr
+    $stderr.puts %Q{Getting #{attr} for hosts like #{role_name}}
+    Broham.hosts_like(role_name).map do |bro|
+      $stdout.puts bro[attr]
+    end
+  end
+
+  def run
+    case Settings.command.to_s
+    when 'hosts', 'sup'
+      hosts_cmd
+    when 'register', 'yo'
+      register_cmd
+    when 'unregister', 'diss'
+      unregister_cmd
+    when 'unregister_like', 'fuck_all_yall'
+      unregister_like_cmd
+    when 'get', 'word'
+      get_cmd
+    when ''
+      Settings.die "Please use one of the commands listed above"
+    else
+      Settings.die "Don't know how to run command #{command}"
+    end
+  end
+
+protected
+
+  # Dump a list of hashes as a formatted, equally-spaced table
+  def dump_table hosts_list
+    attr_lens = {}
+    # find all used fields, and their width
+    hosts_list.each do |host|
+      host.to_hash.each{|k,v| attr_lens[k] = [v.to_s.length, attr_lens[k].to_i].max }
+    end
+    # account for length of attr titles
+    attr_lens.each{|attr,len| attr_lens[attr] = [len, attr.length].max }
+    # take attrs in order, putting role first (and excluding id)
+    attrs = ['role', (attr_lens.keys - ['role', 'id']).sort].flatten
+    $stderr.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%attr}.join("\t")
+    hosts_list.each do |host|
+      $stdout.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%host[attr]}.join("\t")
+    end
+  end
+
+  # Loads defaults
+  def self.get_cluster_settings
+    Settings.read(ENV['HOME']+'/.poolparty/poolparty.yaml')
+    Settings.resolve!
+    Broham.establish_connection
+  end
+
+  # takes a list of attributes and pops each, in turn, from the commandline
+  # (the Settings.rest list)
+  def self.get_commandline_predicates *arg_names
+    arg_names.each do |arg_name|
+      Settings[arg_name] = Settings.rest.shift
+    end
+    check_args! *arg_names
+  end
+
+  #
+  def self.check_args! *arg_names
+    arg = arg_names.last
+    if Settings[arg].blank? then Settings.die "ERROR: Please supply #{arg_names.join(", ")} after the command." end
+  end
 
-case Settings[:command].to_s
-when 'public_ips' then puts Broham.hosts_like(role_name).map{|bro| bro[:public_ip]}.join("\n")
-else raise "Bad command #{Settings[:command]}"
 end
 
+BrohamScript.get_cluster_settings
+BrohamScript.new.run

data/bin/broham-diss

@@ -3,11 +3,132 @@ require 'rubygems'
 require 'broham'
 require 'broham/script'
 
-Broham.get_cluster_settings
-Broham.get_command_line_args :role_name
-role_name = Settings[:role_name]
+[
+  # ['sup',             '', '(alias for hosts)'],
+  # ['yo',              '', '(alias for register)'],
+  # ['diss',            '', '(alias for unregister)'],
+  # ['fuck_all_yall',   '', '(alias for unregister_like)'],
+  # ['word',            '', '(alias for get)'],
+  #
+  ['hosts',           'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['register',        'role [--extra_param=val...]', 'Registers this machine for the given role; stuff in any leftover params'],
+  ['unregister',      'role',                        'Unregister that role'],
+  ['unregister_like', 'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['get',             'role_pattern attr',           'show attr for all hosts prefixed by role_pattern']
+].each{|cmd| Configliere::COMMANDS << cmd }
 
-$stderr.puts %Q{Removing #{role_name}}
-resps = Broham.unregister role_name
-$stderr.puts resps.map{|resp| resp.to_pretty_json }
+class BrohamScript
+  attr_accessor :role_name, :attr
+  def command()   Settings.command ; end
+  def role_name() Settings[:role_name] ; end
+  def attr()      Settings[:attr]      ; end
 
+  def hosts_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Listing hosts like #{role_name}}
+    dump_table Broham.hosts_like(role_name)
+  end
+
+  def register_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register(role_name)
+    dump_table [new_bro]
+  end
+
+  def register_as_next_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register_as_next(role_name)
+    dump_table [new_bro]
+  end
+
+  def unregister_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering #{role_name}}
+    dead_bro = Broham.unregister(role_name)
+    dump_table [dead_bro]
+  end
+
+  def unregister_like_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering all hosts like #{role_name}}
+    dump_table Broham.unregister_like role_name
+  end
+
+  def get_cmd
+    BrohamScript.get_commandline_predicates :role_name, :attr
+    $stderr.puts %Q{Getting #{attr} for hosts like #{role_name}}
+    Broham.hosts_like(role_name).map do |bro|
+      $stdout.puts bro[attr]
+    end
+  end
+
+  def run
+    case Settings.command.to_s
+    when 'hosts', 'sup'
+      hosts_cmd
+    when 'register', 'yo'
+      register_cmd
+    when 'unregister', 'diss'
+      unregister_cmd
+    when 'unregister_like', 'fuck_all_yall'
+      unregister_like_cmd
+    when 'get', 'word'
+      get_cmd
+    when ''
+      Settings.die "Please use one of the commands listed above"
+    else
+      Settings.die "Don't know how to run command #{command}"
+    end
+  end
+
+protected
+
+  # Dump a list of hashes as a formatted, equally-spaced table
+  def dump_table hosts_list
+    attr_lens = {}
+    # find all used fields, and their width
+    hosts_list.each do |host|
+      host.to_hash.each{|k,v| attr_lens[k] = [v.to_s.length, attr_lens[k].to_i].max }
+    end
+    # account for length of attr titles
+    attr_lens.each{|attr,len| attr_lens[attr] = [len, attr.length].max }
+    # take attrs in order, putting role first (and excluding id)
+    attrs = ['role', (attr_lens.keys - ['role', 'id']).sort].flatten
+    $stderr.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%attr}.join("\t")
+    hosts_list.each do |host|
+      $stdout.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%host[attr]}.join("\t")
+    end
+  end
+
+  # Loads defaults
+  def self.get_cluster_settings
+    Settings.read(ENV['HOME']+'/.poolparty/poolparty.yaml')
+    Settings.resolve!
+    Broham.establish_connection
+  end
+
+  # takes a list of attributes and pops each, in turn, from the commandline
+  # (the Settings.rest list)
+  def self.get_commandline_predicates *arg_names
+    arg_names.each do |arg_name|
+      Settings[arg_name] = Settings.rest.shift
+    end
+    check_args! *arg_names
+  end
+
+  #
+  def self.check_args! *arg_names
+    arg = arg_names.last
+    if Settings[arg].blank? then Settings.die "ERROR: Please supply #{arg_names.join(", ")} after the command." end
+  end
+
+end
+
+BrohamScript.get_cluster_settings
+BrohamScript.new.run

data/bin/broham-fuck_all_yall

@@ -3,11 +3,132 @@ require 'rubygems'
 require 'broham'
 require 'broham/script'
 
-Broham.get_cluster_settings
-Broham.get_command_line_args :role_name
-role_name = Settings[:role_name]
+[
+  # ['sup',             '', '(alias for hosts)'],
+  # ['yo',              '', '(alias for register)'],
+  # ['diss',            '', '(alias for unregister)'],
+  # ['fuck_all_yall',   '', '(alias for unregister_like)'],
+  # ['word',            '', '(alias for get)'],
+  #
+  ['hosts',           'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['register',        'role [--extra_param=val...]', 'Registers this machine for the given role; stuff in any leftover params'],
+  ['unregister',      'role',                        'Unregister that role'],
+  ['unregister_like', 'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['get',             'role_pattern attr',           'show attr for all hosts prefixed by role_pattern']
+].each{|cmd| Configliere::COMMANDS << cmd }
 
-$stderr.puts %Q{Removing all like #{role_name}}
-resps = Broham.unregister_like role_name
-$stderr.puts resps.map{|resp| resp.to_pretty_json }
+class BrohamScript
+  attr_accessor :role_name, :attr
+  def command()   Settings.command ; end
+  def role_name() Settings[:role_name] ; end
+  def attr()      Settings[:attr]      ; end
 
+  def hosts_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Listing hosts like #{role_name}}
+    dump_table Broham.hosts_like(role_name)
+  end
+
+  def register_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register(role_name)
+    dump_table [new_bro]
+  end
+
+  def register_as_next_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register_as_next(role_name)
+    dump_table [new_bro]
+  end
+
+  def unregister_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering #{role_name}}
+    dead_bro = Broham.unregister(role_name)
+    dump_table [dead_bro]
+  end
+
+  def unregister_like_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering all hosts like #{role_name}}
+    dump_table Broham.unregister_like role_name
+  end
+
+  def get_cmd
+    BrohamScript.get_commandline_predicates :role_name, :attr
+    $stderr.puts %Q{Getting #{attr} for hosts like #{role_name}}
+    Broham.hosts_like(role_name).map do |bro|
+      $stdout.puts bro[attr]
+    end
+  end
+
+  def run
+    case Settings.command.to_s
+    when 'hosts', 'sup'
+      hosts_cmd
+    when 'register', 'yo'
+      register_cmd
+    when 'unregister', 'diss'
+      unregister_cmd
+    when 'unregister_like', 'fuck_all_yall'
+      unregister_like_cmd
+    when 'get', 'word'
+      get_cmd
+    when ''
+      Settings.die "Please use one of the commands listed above"
+    else
+      Settings.die "Don't know how to run command #{command}"
+    end
+  end
+
+protected
+
+  # Dump a list of hashes as a formatted, equally-spaced table
+  def dump_table hosts_list
+    attr_lens = {}
+    # find all used fields, and their width
+    hosts_list.each do |host|
+      host.to_hash.each{|k,v| attr_lens[k] = [v.to_s.length, attr_lens[k].to_i].max }
+    end
+    # account for length of attr titles
+    attr_lens.each{|attr,len| attr_lens[attr] = [len, attr.length].max }
+    # take attrs in order, putting role first (and excluding id)
+    attrs = ['role', (attr_lens.keys - ['role', 'id']).sort].flatten
+    $stderr.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%attr}.join("\t")
+    hosts_list.each do |host|
+      $stdout.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%host[attr]}.join("\t")
+    end
+  end
+
+  # Loads defaults
+  def self.get_cluster_settings
+    Settings.read(ENV['HOME']+'/.poolparty/poolparty.yaml')
+    Settings.resolve!
+    Broham.establish_connection
+  end
+
+  # takes a list of attributes and pops each, in turn, from the commandline
+  # (the Settings.rest list)
+  def self.get_commandline_predicates *arg_names
+    arg_names.each do |arg_name|
+      Settings[arg_name] = Settings.rest.shift
+    end
+    check_args! *arg_names
+  end
+
+  #
+  def self.check_args! *arg_names
+    arg = arg_names.last
+    if Settings[arg].blank? then Settings.die "ERROR: Please supply #{arg_names.join(", ")} after the command." end
+  end
+
+end
+
+BrohamScript.get_cluster_settings
+BrohamScript.new.run

data/bin/broham-get

@@ -0,0 +1,134 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'broham'
+require 'broham/script'
+
+[
+  # ['sup',             '', '(alias for hosts)'],
+  # ['yo',              '', '(alias for register)'],
+  # ['diss',            '', '(alias for unregister)'],
+  # ['fuck_all_yall',   '', '(alias for unregister_like)'],
+  # ['word',            '', '(alias for get)'],
+  #
+  ['hosts',           'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['register',        'role [--extra_param=val...]', 'Registers this machine for the given role; stuff in any leftover params'],
+  ['unregister',      'role',                        'Unregister that role'],
+  ['unregister_like', 'role_pattern',                'Unregister all hosts prefixed by role_pattern'],
+  ['get',             'role_pattern attr',           'show attr for all hosts prefixed by role_pattern']
+].each{|cmd| Configliere::COMMANDS << cmd }
+
+class BrohamScript
+  attr_accessor :role_name, :attr
+  def command()   Settings.command ; end
+  def role_name() Settings[:role_name] ; end
+  def attr()      Settings[:attr]      ; end
+
+  def hosts_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Listing hosts like #{role_name}}
+    dump_table Broham.hosts_like(role_name)
+  end
+
+  def register_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register(role_name)
+    dump_table [new_bro]
+  end
+
+  def register_as_next_cmd
+    # Settings.define :set, :description => %Q{Any arg prefixed with "--set" will become an extra arg to register: 'broham-register foo --set-path=/path/to/foo' sets :path => '/path/to/foo' as an additional attribute}, :type => Hash
+    broham_args = Settings[:set]||{}
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Registering #{role_name} -- #{broham_args}}
+    new_bro = Broham.register_as_next(role_name)
+    dump_table [new_bro]
+  end
+
+  def unregister_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering #{role_name}}
+    dead_bro = Broham.unregister(role_name)
+    dump_table [dead_bro]
+  end
+
+  def unregister_like_cmd
+    BrohamScript.get_commandline_predicates :role_name
+    $stderr.puts %Q{Unregistering all hosts like #{role_name}}
+    dump_table Broham.unregister_like role_name
+  end
+
+  def get_cmd
+    BrohamScript.get_commandline_predicates :role_name, :attr
+    $stderr.puts %Q{Getting #{attr} for hosts like #{role_name}}
+    Broham.hosts_like(role_name).map do |bro|
+      $stdout.puts bro[attr]
+    end
+  end
+
+  def run
+    case Settings.command.to_s
+    when 'hosts', 'sup'
+      hosts_cmd
+    when 'register', 'yo'
+      register_cmd
+    when 'unregister', 'diss'
+      unregister_cmd
+    when 'unregister_like', 'fuck_all_yall'
+      unregister_like_cmd
+    when 'get', 'word'
+      get_cmd
+    when ''
+      Settings.die "Please use one of the commands listed above"
+    else
+      Settings.die "Don't know how to run command #{command}"
+    end
+  end
+
+protected
+
+  # Dump a list of hashes as a formatted, equally-spaced table
+  def dump_table hosts_list
+    attr_lens = {}
+    # find all used fields, and their width
+    hosts_list.each do |host|
+      host.to_hash.each{|k,v| attr_lens[k] = [v.to_s.length, attr_lens[k].to_i].max }
+    end
+    # account for length of attr titles
+    attr_lens.each{|attr,len| attr_lens[attr] = [len, attr.length].max }
+    # take attrs in order, putting role first (and excluding id)
+    attrs = ['role', (attr_lens.keys - ['role', 'id']).sort].flatten
+    $stderr.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%attr}.join("\t")
+    hosts_list.each do |host|
+      $stdout.puts attrs.map{|attr| "%-#{attr_lens[attr]}s"%host[attr]}.join("\t")
+    end
+  end
+
+  # Loads defaults
+  def self.get_cluster_settings
+    Settings.read(ENV['HOME']+'/.poolparty/poolparty.yaml')
+    Settings.resolve!
+    Broham.establish_connection
+  end
+
+  # takes a list of attributes and pops each, in turn, from the commandline
+  # (the Settings.rest list)
+  def self.get_commandline_predicates *arg_names
+    arg_names.each do |arg_name|
+      Settings[arg_name] = Settings.rest.shift
+    end
+    check_args! *arg_names
+  end
+
+  #
+  def self.check_args! *arg_names
+    arg = arg_names.last
+    if Settings[arg].blank? then Settings.die "ERROR: Please supply #{arg_names.join(", ")} after the command." end
+  end
+
+end
+
+BrohamScript.get_cluster_settings
+BrohamScript.new.run