rig 0.3.1

data/lib/rig/cloud/instance.rb

@@ -0,0 +1,64 @@
+require 'fog'
+
+module Rig
+  module Cloud
+    class Instance
+      class << self
+        def all(filters={})
+          Rig::Cloud.compute.servers.all(filters)
+        end
+
+        def running
+          self.all({"instance-state-name"=>"running"})
+        end
+
+        def find(id)
+          if id =~ /^i-/
+            Rig::Cloud.compute.servers.find(id)
+          else
+            # must be tag:Name
+            Rig::Cloud.compute.servers.all({"tag:Name"=>id})
+          end
+        end
+
+        def find_by_environment(env)
+          Rig::Cloud.compute.servers.all({"tag:Environment"=>env})
+        end
+
+        def create(opts, tags={})
+          fog = Rig::Cloud.compute
+          server = fog.servers.create(opts)
+          fog.create_tags(server.id, tags) if tags.count > 0
+          puts ".. created: #{server.id}"
+          server
+        end
+
+        def destroy(list)
+          list = [*list]
+
+          if list.count > 0
+            compute = Rig::Cloud.compute
+            ids = list.map {|e| e.kind_of?(String) ? e : e.id}
+            puts ".. destroying: #{ids.inspect}"
+
+            # if we've got a chef config, use it
+            if Rig.config.chef
+              list.each do |e|
+                Rig::Chef.client_delete(e.tags['Name']) if e.tags['Name']
+                Rig::Chef.node_delete(e.tags['Name']) if e.tags['Name']
+              end
+            end
+
+            # destroy the instances
+            list.each do |e|
+              e.destroy
+            end
+
+            tags = %w{Name Environment Role}.inject({}) {|h, e| h[e] = nil; h }
+            compute.create_tags(ids, tags)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rig/cloud/userdata.rb

@@ -0,0 +1,111 @@
+require 'erubis'
+
+module Rig
+  module Cloud
+    class << self
+      def userdata(name, role, environment, opts={ })
+        data = {
+            :name         => name,
+            :role         => role,
+            :environment  => environment,
+            :zone         => Rig.config.dns.zone,
+            :dependencies => %w{ ruby1.9.3 ruby1.9.1-dev rubygems ruby-switch libopenssl-ruby1.9.1 rdoc1.9.1 ri1.9.1 irb1.9.1
+                                 build-essential wget ssl-cert curl },
+            :gems         => %w{chef},
+            :chef         => false,
+        }.merge(opts)
+        template  = <<-EOF
+#!/bin/bash
+set -e -x
+exec > ~/userdata.log 2>&1
+export DEBIAN_FRONTEND=noninteractive
+apt-add-repository -y ppa:brightbox/ruby-ng-experimental
+
+apt-get update && apt-get upgrade -y
+apt-get -y install <%= @dependencies.join(' ') %>
+mkdir -p /var/log/chef
+mkdir -p /var/backups/chef
+mkdir -p /var/run/chef
+mkdir -p /var/cache/chef
+mkdir -p /var/lib/chef
+mkdir /etc/chef
+
+# set up hostname
+hostname <%= @name %>.<%= @zone %>
+echo "<%= @name %>.<%= @zone %>" > /etc/hostname
+echo "127.0.0.1 <%= @name %>.<%= @zone %> <%= @name.split('.')[0] %> <%= @name %> # entered by userdata script" >> /etc/hosts
+
+gem install rubygems-update --no-ri --no-rdoc
+update_rubygems
+
+<% @gems.each do |gem| %>
+gem install <%= gem %> --no-ri --no-rdoc
+<% end %>
+
+cat <<FACTS > /etc/facts.yml
+inq_name: <%= @name %>
+inq_role: <%= @role %>
+inq_environment: <%= @environment %>
+FACTS
+
+cat <<JSON > /etc/chef/bootstrap.json
+{
+  "run_list": [
+    "role[<%= @role %>]"
+  ]
+}
+JSON
+
+cat <<CLIENT > /etc/chef/client.rb
+log_level               :info
+log_location            "/var/log/chef/client.log"
+ssl_verify_mode         :verify_none
+validation_client_name  "chef-validator"
+validation_key          "/etc/chef/validation.pem"
+client_key              "/etc/chef/client.pem"
+chef_server_url         "http://chef.inqlabs.com:4000";
+file_cache_path         "/var/cache/chef"
+file_backup_path        "/var/backups/chef"
+pid_file                "/var/run/chef/client.pid"
+node_name               "<%= @name %>";
+Chef::Log::Formatter.show_time = true
+CLIENT
+
+cat <<VALID > /etc/chef/validation.pem
+-----BEGIN RSA PRIVATE KEY-----
+MIIEpgIBAAKCAQEAxFST0blraLl7mac91QzSxNNZQt7A4Mo4O297e+r6F5b1+Eae
+RNAH9w4p9sctLttIT8rI77hWrgS04f6mbamSaY+pAyWje+MX2s5YXizEML1Fdyr6
+hV1n/vIMOmhcVwGbF3fg6BKwjur7A1dS+aOM1sEYi77hfbY4NHhLCQiR4v2m4saC
+C9SPX7uNIzKvGB2P/5U+HNPaJuo6GAUsrOrTA/YB53Bza4DPMpyp9+Hf/ZR9l6Hd
+4tCAp07dUNZ/LKKL4AmsnCqlT5EiYFIsaq2ezzPnS9t2FlFV60a2opbdEvq/QLbz
+ouH6H+/4n/+BY9bVBCg1oBXHlLwbZmNl7Xl4wQIDAQABAoIBAQCFDIAkZ5C3psNx
+bJoFkKwhQM1D+OsjK0eSV8mKP8J+TAqi11CYd5Z+QbYEjBQ9pdNKZ7VmZzkusvfW
+E5m7xWf4a6fw/wfchBh9assN/y5xjULPMGwYByXb6zKIPoMdX3Q4IkBjZY8LO4ki
+SYJstP5T5wX4ZkIvzFYokMDg/VLITUZPnxu+OunFF/vfl7mFOIrHtCj6LSpCk5xW
+NsQqJWzK6/xjKH8KweWkl+E7k/ILUetXuCjKhj2XjiYUbzRH12e5jJIsEXRKWw8I
+Iw5I8N4fWaXbTmoI+NUV+24BqwOfCH0RjoZXcc6LOxmxKDd8a00H3bCYj9GbpLed
+oFcCY2OFAoGBAPSj4u2sSI1uMWNMivTbEYC5pYeltJgRYByazFHj5mHtPF7aDs2a
+IWBfaaHbRIk7d9mIGYOrEiFFpDi5zWYo1gEbwqqOVw4cxL6coVA5mUMdZFDSGcpL
+jwCIp9Yz92mOvygJEQosBIaMaBNTVY2GJPEo8Uhp0OstMDxIR9YrxdwjAoGBAM1y
+atxCx8Z7Mua33x6gPxyyQVnqU/Hvpa9uTF0laAcwarsNlXczyfw/5TNJsEyTzbE1
+6yJdm5GDS7L6whprx20SCHgpfBcAK6UGMMYsRvRz8kThLI7jgsXQjNXO8MNvaYwO
+z9uVp+i1OVN0jURPwu6NsHcduSCP4asZwBIIOoPLAoGBAJdZQM5rlKMy9gkJ1Lnx
+Qd3SfK7z/EtMDJsFxbwv81rIK14/J4NZhKabwL0q6za/fMEuqyJDVr8Q161kfcdj
+IiqjYwajcwY/FBUcz8vTJpdX+rTplp0Jq9nXsYYatkkuz+JT4Z/aZZ5cBRNXrgkV
+YYt1GfkDxgWI+luBGDhO8mTjAoGBAJ5j3dreaWKuTLnvfEcCSGX5T8TgkcNufRBZ
+Oup9EkyomUCI4ni6JrkhEbWO1CrVsLk01ojyHiP7U3szND5mZaPDSWBKq26YJQ3f
+Te8a58FDOr670zx+Tq3vcQTNvHUQChF+weWK5Z7VSNL8goTt8V01Q50x/y4Jf3QI
+MDo2udYtAoGBAKk596lHuH5/LEjT5hfVrbarSvUjlXswJusZGnwbtEUERY7abUbG
+5mtI+Xme2q6T0Wjf6N977ecNYAi5VwpUzoziZUPeSCQUlRWSyX2o6+fVBngOFpLZ
+tRv2ngXlc2sYh9RbnlWSdaK0g+U7V8l2wSsQjcPQ1J2vbIGjwBHDSzjc
+-----END RSA PRIVATE KEY-----
+VALID
+
+chef-client -N <%= @name %> <%= @chef && @environment ? "-E " + @environment : "" %> --once -j /etc/chef/bootstrap.json 2>&1
+        EOF
+        erb           = Erubis::Eruby.new(template)
+        out = erb.evaluate(data)
+      end
+    end
+  end
+end

data/lib/rig/command.rb

@@ -0,0 +1,42 @@
+require 'rig'
+require 'clamp'
+require 'fog'
+require 'yaml'
+
+require 'rig/cloud'
+
+require 'rig/command/abstract'
+require 'rig/command/options'
+
+require 'rig/command/main'
+
+require 'rig/command/environment/main'
+require 'rig/command/environment/list'
+require 'rig/command/environment/create'
+require 'rig/command/environment/destroy'
+
+require 'rig/command/balancer/main'
+require 'rig/command/balancer/list'
+require 'rig/command/balancer/view'
+require 'rig/command/balancer/destroy'
+
+require 'rig/command/dns/main'
+require 'rig/command/dns/list'
+require 'rig/command/dns/create'
+require 'rig/command/dns/edit'
+require 'rig/command/dns/destroy'
+
+require 'rig/command/instance/main'
+require 'rig/command/instance/ssh'
+require 'rig/command/instance/list'
+require 'rig/command/instance/create'
+require 'rig/command/instance/destroy'
+require 'rig/command/instance/view'
+
+require 'rig/command/instance/tag/main'
+require 'rig/command/instance/tag/get'
+require 'rig/command/instance/tag/set'
+require 'rig/command/instance/tag/remove'
+
+require 'rig/command/keypair/main'
+require 'rig/command/keypair/list'

data/lib/rig/command/abstract.rb

@@ -0,0 +1,36 @@
+require 'terminal-table/import'
+
+module Rig
+  module Command
+    class Abstract < Clamp::Command
+      option ["-n","--test"], :flag, "use Fog.mock! to test command without creating / destroying cloud resources" do |o|
+        Fog.mock!
+      end
+
+      def print_table(columns, rows)
+        puts table(columns, *rows)
+      end
+
+      def instance_list(list)
+        rows = []
+        list.each do |server|
+          name = server.tags['Name']
+          env = server.tags['Environment']
+          role = server.tags['Role']
+          ip = server.public_ip_address.to_s
+          flavor = server.flavor_id.to_s
+          id = server.id.to_s
+          state = server.state.to_s.downcase
+
+          rows << [name, env, role, ip, flavor, id, state]
+        end
+
+        print_table(%w{Name Environment Role PublicIP Size InstanceID State}, rows)
+      end
+
+      def execute
+        super
+      end
+    end
+  end
+end

data/lib/rig/command/balancer/destroy.rb

@@ -0,0 +1,16 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Balancer
+      class Destroy < Abstract
+        parameter "NAME", "name of load balancer"
+        def execute
+          Rig::Cloud::Balancer.destroy(name)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Balancer::Main.subcommand 'destroy', 'destroy load balancer', Rig::Command::Balancer::Destroy

data/lib/rig/command/balancer/list.rb

@@ -0,0 +1,21 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Balancer
+      class List < Abstract
+        def execute
+          list = Rig::Cloud::Balancer.all
+          list.each do |lb|
+            puts lb.id
+            lb.instances.each do |inst|
+              puts "- #{inst}"
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Balancer::Main.subcommand 'list', 'list load balancers', Rig::Command::Balancer::List

data/lib/rig/command/balancer/main.rb

@@ -0,0 +1,13 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Balancer
+      class Main < Abstract
+        self.default_subcommand = 'list'
+      end
+    end
+  end
+end
+
+Rig::Command::Main.subcommand 'balancer', 'view load balancers', Rig::Command::Balancer::Main

data/lib/rig/command/balancer/view.rb

@@ -0,0 +1,19 @@
+require 'awesome_print'
+
+module Rig
+  module Command
+    module Balancer
+      class View < Abstract
+        include Options::ShowAll
+
+        parameter "NAME", "name of instance"
+
+        def execute
+          item = Rig::Cloud::Balancer.find(name)
+          puts item.attributes.to_yaml
+        end
+      end
+    end
+  end
+end
+Rig::Command::Balancer::Main.subcommand 'view', 'view instances', Rig::Command::Balancer::View

data/lib/rig/command/dns/create.rb

@@ -0,0 +1,19 @@
+require 'rig'
+require 'ipaddress'
+
+module Rig
+  module Command
+    module Dns
+      class Create < Abstract
+        parameter 'NAME', 'the name to create'
+        parameter 'VALUE', 'where the name points to'
+
+        def execute
+          Rig::Cloud::Dns.create(name, value)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Dns::Main.subcommand 'create', 'create dns name', Rig::Command::Dns::Create

data/lib/rig/command/dns/destroy.rb

@@ -0,0 +1,16 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Dns
+      class Destroy < Abstract
+        parameter "NAME", "name of load balancer"
+        def execute
+          Rig::Cloud::Dns.destroy(name)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Dns::Main.subcommand 'destroy', 'destroy dns record', Rig::Command::Dns::Destroy

data/lib/rig/command/dns/edit.rb

@@ -0,0 +1,20 @@
+require 'rig'
+require 'ipaddress'
+
+module Rig
+  module Command
+    module Dns
+      class Edit < Abstract
+        parameter 'NAME', 'the name to create'
+        parameter 'VALUE', 'where the name points to'
+
+        def execute
+          Rig::Cloud::Dns.destroy(name)
+          Rig::Cloud::Dns.create(name, value)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Dns::Main.subcommand 'edit', 'edit dns name', Rig::Command::Dns::Edit

data/lib/rig/command/dns/list.rb

@@ -0,0 +1,21 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Dns
+      class List < Abstract
+        def execute
+          zone    = Rig.config.dns.zone
+          records = Rig::Cloud::Dns.records("#{zone}")
+          rows    = []
+          records.each do |record|
+            rows << [zone, record.type, record.ttl, record.name, record.value]
+          end
+          print_table(%w{Domain Type TTL Name Value}, rows)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Dns::Main.subcommand 'list', 'list dns', Rig::Command::Dns::List

data/lib/rig/command/dns/main.rb

@@ -0,0 +1,13 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Dns
+      class Main < Abstract
+        self.default_subcommand = 'list'
+      end
+    end
+  end
+end
+
+Rig::Command::Main.subcommand 'dns', 'view dns zones', Rig::Command::Dns::Main

data/lib/rig/command/environment/create.rb

@@ -0,0 +1,24 @@
+require 'rig'
+require 'fog'
+
+module Rig
+  module Command
+    module Environment
+      class Create < Abstract
+        include Options::AwsKey
+        include Options::AwsSecret
+
+        parameter "NAME", "the name of the environment"
+        parameter "TEMPLATE", "the template to use (solo, multi, etc)", :default => "solo"
+
+        def execute
+          #connection = Rig::Cloud.compute
+          env = Rig::Cloud::Environment.create(name, template.to_sym)
+          #ap env.servers
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Environment::Main.subcommand 'create', 'create environment', Rig::Command::Environment::Create