broham 0.0.11 → 0.0.12

data/VERSION

@@ -1 +1 @@
-0.0.11
+0.0.12

data/bin/broham-register_as_next

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

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

@@ -5,14 +5,14 @@
 
 Gem::Specification.new do |s|
   s.name = %q{broham}
-  s.version = "0.0.11"
+  s.version = "0.0.12"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Philip (flip) Kromer"]
   s.date = %q{2010-05-04}
   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-get", "broham-hosts", "broham-register", "broham-sup", "broham-unregister", "broham-unregister_like", "broham-word", "broham-yo"]
+  s.executables = ["broham", "broham-diss", "broham-fuck_all_yall", "broham-get", "broham-hosts", "broham-register", "broham-register_as_next", "broham-sup", "broham-unregister", "broham-unregister_like", "broham-word", "broham-yo", "broham-yo_yo_yo"]
   s.extra_rdoc_files = [
     "LICENSE",
      "README.textile"
@@ -30,11 +30,13 @@ Gem::Specification.new do |s|
      "bin/broham-get",
      "bin/broham-hosts",
      "bin/broham-register",
+     "bin/broham-register_as_next",
      "bin/broham-sup",
      "bin/broham-unregister",
      "bin/broham-unregister_like",
      "bin/broham-word",
      "bin/broham-yo",
+     "bin/broham-yo_yo_yo",
      "broham.gemspec",
      "lib/broham.rb",
      "lib/broham/script.rb",

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 11
-  version: 0.0.11
+  - 12
+  version: 0.0.12
 platform: ruby
 authors: 
 - Philip (flip) Kromer
@@ -52,11 +52,13 @@ executables:
 - broham-get
 - broham-hosts
 - broham-register
+- broham-register_as_next
 - broham-sup
 - broham-unregister
 - broham-unregister_like
 - broham-word
 - broham-yo
+- broham-yo_yo_yo
 extensions: []
 
 extra_rdoc_files: 
@@ -75,11 +77,13 @@ files:
 - bin/broham-get
 - bin/broham-hosts
 - bin/broham-register
+- bin/broham-register_as_next
 - bin/broham-sup
 - bin/broham-unregister
 - bin/broham-unregister_like
 - bin/broham-word
 - bin/broham-yo
+- bin/broham-yo_yo_yo
 - broham.gemspec
 - lib/broham.rb
 - lib/broham/script.rb