broham 0.0.9 → 0.0.10

data/bin/broham-unregister_like

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

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

@@ -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/broham.gemspec

@@ -5,14 +5,14 @@
 
 Gem::Specification.new do |s|
   s.name = %q{broham}
-  s.version = "0.0.9"
+  s.version = "0.0.10"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Philip (flip) Kromer"]
-  s.date = %q{2010-04-14}
+  s.date = %q{2010-05-02}
   s.description = %q{Bro! Broham always knows where his bros are, bro. Using broham, a newly-created cloud machine can annouce its availability for a certain role ("nfs_server" or "db_slave-2"), allowing any other interested nodes to discover its public_ip, private_ip, etc. See also: http://j.mp/amongbros}
   s.email = %q{flip@infochimps.org}
-  s.executables = ["broham", "broham-diss", "broham-fuck_all_yall", "broham-host", "broham-hosts_like", "broham-register", "broham-register_as_next", "broham-sup", "broham-sup_yall", "broham-unregister", "broham-unregister-like", "broham-yo"]
+  s.executables = ["broham", "broham-diss", "broham-fuck_all_yall", "broham-get", "broham-hosts", "broham-register", "broham-sup", "broham-unregister", "broham-unregister_like", "broham-word", "broham-yo"]
   s.extra_rdoc_files = [
     "LICENSE",
      "README.textile"
@@ -27,18 +27,20 @@ Gem::Specification.new do |s|
      "bin/broham",
      "bin/broham-diss",
      "bin/broham-fuck_all_yall",
-     "bin/broham-host",
-     "bin/broham-hosts_like",
+     "bin/broham-get",
+     "bin/broham-hosts",
      "bin/broham-register",
-     "bin/broham-register_as_next",
      "bin/broham-sup",
-     "bin/broham-sup_yall",
      "bin/broham-unregister",
-     "bin/broham-unregister-like",
+     "bin/broham-unregister_like",
+     "bin/broham-word",
      "bin/broham-yo",
      "broham.gemspec",
      "lib/broham.rb",
      "lib/broham/script.rb",
+     "lib/broham/sdb.rb",
+     "lib/broham/sdb/active_sdb.rb",
+     "lib/broham/sdb/right_sdb_interface.rb",
      "spec/broham_spec.rb",
      "spec/spec.opts",
      "spec/spec_helper.rb"