broham 0.0.9 → 0.0.10

data/bin/broham-hosts

@@ -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

data/bin/broham-register

@@ -2,13 +2,133 @@
 require 'rubygems'
 require 'broham'
 require 'broham/script'
-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.get_cluster_settings
-Broham.get_command_line_args :role_name
-role_name, broham_args = [Settings[:role_name], Settings[:set]||{}]
+[
+  # ['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{Registering as #{role_name} with #{broham_args.inspect}}
-resp = Broham.register role_name, broham_args
-node = Broham.host role_name
-$stderr.puts 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-sup

@@ -2,12 +2,133 @@
 require 'rubygems'
 require 'broham'
 require 'broham/script'
-require 'json'
 
-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{Retrieving '#{role_name}'}
-resp = Broham.host role_name
-$stderr.puts 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-unregister

@@ -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