bluebox-boxcutter 0.0.14

data/lib/bluebox-boxcutter/machine.rb

@@ -0,0 +1,84 @@
+module Boxcutter
+  class Machine
+
+    def self.search(search_string)
+      ids = name_to_id_hash(search_string)
+      [].tap do |res|
+        res << [:name, :id, :url]
+        ids.each_pair do |name, id|
+          res << [ name, id, Boxcutter::Boxpanel.machine_url(id) ]
+        end
+      end
+    end
+
+    # given hostname, return a mash of machine attributes
+    def self.get(name)
+      machine_id = id name
+      machine_info_url = "/private/hosts/show/#{machine_id}.json"
+      json = Boxcutter::Boxpanel.get(machine_info_url)
+      mash = Mash.new( { :key => "value" } )
+      mash.merge! Boxcutter::flatten_hash(JSON::parse(json))
+    end
+
+    # return a hash of {hostname => machine_id}
+    def self.name_to_id_hash(regex_string = '.')
+      Hash.new.tap do |this|
+        regex = Regexp.new regex_string
+        JSON::parse(Boxpanel.get('/private/hosts.json')).each do |i|
+          this[i["hostname"]] = i["id"].to_s if i['hostname'] =~ regex
+        end
+      end
+    end
+
+    # given a hostname, return its id
+    def self.id(name)
+      ids = name_to_id_hash name
+      raise "found more than one host matching '#{name}'" if ids.size > 1
+      raise "unknown machine #{name}" unless ids[name]
+      ids[name]
+    end
+
+    # given a dom and an input field name, return its value, or nil
+    def self.get_input_value(dom, name)
+      elements = dom.css("input##{name}")
+      elements.empty? ? nil : elements.first['value']
+    end
+
+    def self.host_id(machine)
+      machine.id
+    end
+
+    def self.switchport_id(machine)
+      machine.network_interfaces_eth0_id
+    end
+
+    def self.eth0_mac(machine)
+      machine.network_interfaces_eth0_mac_address
+    end
+
+    def self.power_cycle!(machine)
+      body = Boxpanel.post '/private/hosts/do_power_cycle', :body => {:id => Boxcutter::Machine::host_id(machine), :delay => 0 }
+      raise "reboot failed" unless body =~ /job_queue/
+    end
+
+    def self.switch_vlan!(machine, vlan_id)
+      body = Boxpanel.post(
+        "/private/machine_interfaces/submit_switchport_job/#{Boxcutter::Machine::switchport_id(machine)}",
+        :body => {:vlan => 255, :description => machine[:hostname], :enabled => 1, :commit => 'Submit'})
+      raise 'vlan switch failed' unless body =~ Regexp.new(machine[:hostname])
+    end
+
+    def self.clear_hwtoken!(machine)
+      Boxpanel.post("/private/hosts/clear_hwagent_token/#{Boxcutter::Machine::host_id(machine)}", :body => {})
+    end    
+
+    def self.machine_summary(machine)
+      Array.new.tap do |a|
+        a << "Hostname: #{machine[:hostname]}"
+        a << "Customer: #{machine[:customer_business_name]}"
+        a << "Location: #{machine[:location_description]}"
+        a << "Purpose:  #{machine[:purpose]}"
+      end
+    end
+  end
+end

data/lib/bluebox-boxcutter/password.rb

@@ -0,0 +1,37 @@
+
+module Boxcutter
+  module Password
+    def self.search(regexp_string)
+      regexp = Regexp.new regexp_string
+      paths = name_to_path_hash
+      [[:name, :uri, :user, :pass]].tap do |res|
+        name_to_path_hash.each_pair do |name, path|
+          if name =~ regexp
+            pw = get_pw(path)
+            res << [name].concat(pw.values)
+          end
+        end
+      end
+    end
+
+    def self.name_to_path_hash()
+      dom = Boxpanel.get_html '/private/service_password'
+      pw_links = dom.css('a').select { |a| a['onclick'] and a['onclick'] =~ /fetch_password/ }
+      Hash[
+        pw_links.map do |l|
+          [ l.text, l['onclick'].split("'").select { |s| s =~ /service_password\/fetch_password/ }.first ]
+        end
+      ]
+    end
+
+    def self.get_pw(path)
+      dom = Boxpanel.get_html path
+      {
+        :uri => dom.css('a').first['href'],
+        :user => dom.css('input')[0]['value'],
+        :pass => dom.css('input')[1]['value']
+      }
+    end
+
+  end
+end

data/lib/bluebox-boxcutter/razor.rb

@@ -0,0 +1,59 @@
+require 'httparty'
+require 'json'
+
+module Boxcutter
+  module Razor
+
+    ENDPOINT='http://razor.sea03.blueboxgrid.com:80'
+    USER = 'boxpanel'
+    PASS = '4nlpb4IFQxjx'
+
+    class StubbleClient
+      include HTTParty
+      base_uri ENDPOINT
+      basic_auth USER, PASS
+
+      def self.tags()
+        resp = get '/tags'
+        raise 'get /tags failed' unless resp.code == 200
+        JSON.parse resp.body
+      end
+
+      def self.delete_mac(mac)
+        resp = delete '/mac', :body => {:mac => mac }
+        raise 'delete /mac failed' unless resp.code == 200
+      end
+
+      def self.add_mac_to_tag(mac, tag)
+        resp = post '/mac', :body => {:mac => mac, :tag => tag }
+        raise 'post /mac failed' unless resp.code == 200
+      end
+
+      def self.log(mac)
+        resp = get "/mac/#{mac}/log"
+        raise 'get /mac/X/log failed' unless resp.code == 200
+        JSON.parse resp.body
+      end
+    end
+
+    def self.image!(machine, tag_name)
+      raise "unknown razor tag #{tag_name}" unless StubbleClient.tags.include?(tag_name)
+      mac = Boxcutter::Machine::eth0_mac(machine)
+
+      raise "unknown mac" unless mac and mac =~ /[a-f0-9:]+/
+
+      StubbleClient.delete_mac mac
+      StubbleClient.add_mac_to_tag mac, tag_name
+
+      Boxcutter.msg "switching machine into vlan 255."
+      Boxcutter::Machine.switch_vlan! machine, 255
+
+      Boxcutter.msg "initiating razor boot"
+      Boxcutter::Machine.power_cycle! machine
+
+      Boxcutter.msg "removing hardware token from Boxpanel machine record"
+      Boxcutter::Machine.clear_hwtoken! machine      
+    end
+
+  end
+end

data/lib/bluebox-boxcutter/ui.rb

@@ -0,0 +1,53 @@
+require 'colorize'
+require 'terminal-table'
+
+module Boxcutter
+
+  @@colors = true
+  @@yes = false
+  @@quiet = false
+
+  def self.error!(msg)
+    STDERR.puts msg.red
+    exit 1
+  end
+
+  def self.no_colors!
+    @@colors = false
+  end
+
+  def self.quiet!
+    @@quiet = true
+  end
+
+  def self.msg(s)
+    puts (@@colors ? s.green : s) unless @@quiet
+  end
+
+  def self.yes!
+    @@yes = true
+  end
+
+  def self.confirm(prompt)
+    if @@yes
+      yield
+    else
+      puts "#{prompt}  :  are you sure? [y/n]".yellow
+      answer = STDIN.gets.chomp
+      answer == 'y' ? yield : error!('aborting at user request')
+    end
+  end
+
+  def self.table_of_hashes(rows)
+    keys = rows.empty? ? [] : rows.first.keys.map { |k| k.dup }
+    table([keys] + rows.map { |r| r.values })
+  end
+
+  def self.table(rows, headers_present=true)
+    headers = nil
+    headers = rows.shift.map { |x| x.to_s } if headers_present
+    headers.map! { |h| h.dup.cyan } if @@colors
+    Terminal::Table.new(:headings => headers, :rows => rows).to_s
+  end
+
+end

data/lib/bluebox-boxcutter/version.rb

@@ -0,0 +1,3 @@
+module Boxcutter
+  VERSION = '0.0.14'
+end

data/spec/boxcutter_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe Boxcutter do
+
+  it 'should update' do
+    Boxcutter::Git.stub(:upstream_version).and_return "1000.1000.1000"
+    Boxcutter.should_receive(:install)
+    Boxcutter::update
+  end
+
+end

data/spec/fixtures/private/hosts.json

@@ -0,0 +1,4 @@
+[{"hostname":"foo1","id":123},
+{"hostname":"foo2","id":124},
+{"hostname":"bar1","id":125},
+{"hostname":"bar2","id":126}]

data/spec/fixtures/private/hosts/live_search_machines

@@ -0,0 +1,4 @@
+<a href="https://boxpanel.bluebox.net/private/machines/123/edit" class="asset">foo1</a>
+<a href="https://boxpanel.bluebox.net/private/machines/124/edit" class="asset">foo2</a>
+<a href="https://boxpanel.bluebox.net/private/machines/125/edit" class="asset">bar1</a>
+<a href="https://boxpanel.bluebox.net/private/machines/126/edit" class="asset">bar2</a>

data/spec/fixtures/private/hosts/show/123.json

@@ -0,0 +1,13 @@
+{
+    "id": "98765",
+    "network_interfaces": {
+      "eth0": {
+          "mac_address": "11:11:11:11:11",
+          "id": "68686"
+      }
+    },
+    "hostname": "foo1.blah.com",
+    "asset": {
+        "serial": "FFFFFF"
+    }
+}

data/spec/fixtures/private/machines/123/edit

@@ -0,0 +1,18 @@
+<input id="host_hostname" name="host[hostname]" size="30" type="text" value="foo1.blah.com" />
+<input hint="e.g.: 2.0" id="host_cpu_cores" name="host[cpu_cores]" size="8" type="text" value="8.0000" />
+<input id="host_memory_size" name="host[memory_size]" size="8" type="text" value="16 GB" />
+<input id="host_disk_size" name="host[disk_size]" size="8" type="text" value="696.8 GB" />
+<input id="asset_serial" name="asset[serial]" size="30" type="text" value="2t3fi51" />
+
+<form action="/private/hosts/98765" class="form-horizontal" method="post" onsubmit="new Ajax.Request('/private/hosts/98765', {asynchronous:true, evalScripts:true, method:'put', parameters:Form.serialize(this)}); return false;">
+
+<div id="hidden_content_confirm_power_cycle_foo1" style="display: none;"></div><a href="#" onclick="new Ajax.Updater('hidden_content_confirm_power_cycle_foo1', '/private/hosts/confirm_power_cycle/98765', {asynchronous:true, evalScripts:true, method:'get', onComplete:function(request){RedBox.addHiddenContent('hidden_content_confirm_power_cycle_foo1'); }, onLoading:function(request){RedBox.loading(); }}); return false;"><button class="btn btn-danger" name="button" type="submit">Power Cycle</button></a>
+
+<a href="/private/machine_interfaces/switchport_switcher/68686">[Edit Switchport]</a>
+<a href="/private/hosts/setup_dns/1205043">[Setup DNS Records]</a>
+
+<td class="inactiveHeaderLeft" style="padding: 3px; border: 1px solid #EEE;">
+  <div style="float: left;"><span style=" font-weight: bold;">Interface eth0</span> (00:30:48:cc:d6:32)
+    connected to <a href="/private/machines/3750387/edit">dsw01a.sea03</a> interface Et102/1/7
+  </div>
+</td>

data/spec/fixtures/private/service_password

@@ -0,0 +1,13 @@
+<html>
+<body>
+
+  ...
+
+  <a href="#" onclick="new Ajax.Updater('hidden_content_fetch_password_163', '/private/service_password/fetch_password/163', {asynchronous:true, evalScripts:true, onComplete:function(request){RedBox.addHiddenContent('hidden_content_fetch_password_163'); }, onLoading:function(request){RedBox.loading(); }}); return false;">foo number 1</a>
+  <a href="#" onclick="new Ajax.Updater('hidden_content_fetch_password_164', '/private/service_password/fetch_password/164', {asynchronous:true, evalScripts:true, onComplete:function(request){RedBox.addHiddenContent('hidden_content_fetch_password_164'); }, onLoading:function(request){RedBox.loading(); }}); return false;">foo number 2</a>
+  <a href="#" onclick="new Ajax.Updater('hidden_content_fetch_password_165', '/private/service_password/fetch_password/165', {asynchronous:true, evalScripts:true, onComplete:function(request){RedBox.addHiddenContent('hidden_content_fetch_password_165'); }, onLoading:function(request){RedBox.loading(); }}); return false;">kvm9</a>
+
+  ...
+
+</body>
+</html>

data/spec/fixtures/private/service_passwords/fetch_password/163

@@ -0,0 +1,14 @@
+
+<div>
+  <a href="https://some-auth-url" title="some-auth-url" target="_blank">Click Here</a>
+</div>
+
+<div>
+  <strong>Username:</strong>
+  <input style="width: 100%;" type="text" value="some user" />
+</div>
+
+<div>
+  <strong>Password:</strong>
+  <input style="width: 100%;" type="text" value="XXXX" />
+</div>

data/spec/fixtures/private/service_passwords/fetch_password/164

@@ -0,0 +1,14 @@
+
+<div>
+  <a href="https://another-auth-url" title="another-auth-url" target="_blank">Click Here</a>
+</div>
+
+<div>
+  <strong>Username:</strong>
+  <input style="width: 100%;" type="text" value="another user" />
+</div>
+
+<div>
+  <strong>Password:</strong>
+  <input style="width: 100%;" type="text" value="YYYY" />
+</div>

data/spec/fixtures/private/service_passwords/fetch_password/165

@@ -0,0 +1,14 @@
+
+<div>
+  <a href="https://yet-another-auth-url" title="yet-another-auth-url" target="_blank">Click Here</a>
+</div>
+
+<div>
+  <strong>Username:</strong>
+  <input style="width: 100%;" type="text" value="yet-another user" />
+</div>
+
+<div>
+  <strong>Password:</strong>
+  <input style="width: 100%;" type="text" value="ZZZ" />
+</div>

data/spec/git_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe Boxcutter::Git do
+
+  before do
+    Boxcutter::Git::clone
+  end
+
+  after do
+    Boxcutter::Git::rm_clone
+  end
+
+  it 'should get the upstream version' do
+    Boxcutter::Git::upstream_version.should match /[0-9]+.[0-9]+.[0-9]+/
+  end
+
+end

data/spec/machine_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+describe Boxcutter::Machine do
+
+  before do
+    stub_boxpanel_calls!
+  end
+
+  it 'should grab machine ids and names from machine list' do
+    Boxcutter::Machine.name_to_id_hash().should == {
+      "foo1" => "123",
+      "foo2" => "124",
+      "bar1" => "125",
+      "bar2" => "126"
+    }
+  end
+
+  it 'should search machines by regexp' do
+    Boxcutter::Machine.search('bar*').should == [
+      [:name,  :id, :url],
+      ["bar1", "125", "https://boxpanel.bluebox.net/private/machines/125/edit"],
+      ["bar2", "126", "https://boxpanel.bluebox.net/private/machines/126/edit"]
+    ]
+  end
+
+  it 'should provide id given hostname' do
+    Boxcutter::Machine.id('foo1').should == '123'
+  end
+
+  it 'should issue power cycles' do
+    Boxcutter::Boxpanel.should_receive(:post).
+      with('/private/hosts/do_power_cycle', :body => {:id => '98765', :delay => 0 }).
+      and_return('window.location.href = "/private/job_queue/queue";')
+    Boxcutter::Machine.power_cycle!(Boxcutter::Machine.get 'foo1')
+  end
+
+  it 'should switch vlans' do
+    Boxcutter::Boxpanel.should_receive(:post).
+      with('/private/machine_interfaces/submit_switchport_job/68686',
+           :body=>{:vlan=>255, :description=>"foo1.blah.com", :enabled=>1, :commit=>"Submit"}
+      ).
+      and_return('foo1.blah.com')
+    Boxcutter::Machine.switch_vlan!(Boxcutter::Machine.get('foo1'), 255)
+  end
+
+end

data/spec/razor_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+describe Boxcutter::Razor do
+
+  before do
+    stub_boxpanel_calls!
+    Boxcutter::Razor::StubbleClient.stub(:tags).and_return [ 'scientific-6', 'ubuntu-precise' ]
+  end
+
+  it 'should raise on unknown tag name' do
+    expect { Boxcutter::Razor.image!({}, 'bad-tag') }.to raise_error
+  end
+
+  it 'should raise if no mac is provided' do
+    expect { Boxcutter::Razor.image!({}, 'ubuntu-precise') }.to raise_error
+  end
+
+  it 'should prepare razor and vlan, and reboot the machine' do
+    mac = 'aa:bb:cc:dd:ee:ff'
+    machine = Mash.new( { :network_interfaces_eth0_mac_address => mac } )
+    Boxcutter::Razor::StubbleClient.should_receive(:delete_mac).with(mac)
+    Boxcutter::Razor::StubbleClient.should_receive(:add_mac_to_tag).with(mac, 'ubuntu-precise')
+    Boxcutter::Machine.should_receive(:switch_vlan!).with(machine, 255)
+    Boxcutter::Machine.should_receive(:power_cycle!).with(machine)
+    Boxcutter::Machine.should_receive(:clear_hwtoken!).with(machine)
+    Boxcutter::Razor.image! machine, 'ubuntu-precise'
+  end
+end