rig 0.4.5 → 0.5.0

data/lib/rig/chef.rb

@@ -1,59 +0,0 @@
-require 'rubygems'
-
-require 'chef'
-require 'chef/config'
-require 'chef/api_client'
-require 'chef/node'
-require 'chef/environment'
-
-module Rig
-  module Chef
-    class << self
-      def configure
-        @chef ||= ::Chef::Config.from_file(File.expand_path(Rig.config[:chef][:knife]))
-      end
-
-      def environment_create(name)
-        configure
-        env = ::Chef::Environment.new
-        env.name(name)
-        env.description("created by Rig")
-        env.save
-        true
-      rescue => e
-        puts "*** chef exception: #{e.message}"
-        false
-      end
-
-      def environment_destroy(name)
-        configure
-        env = ::Chef::Environment.load(name) rescue nil
-        env.destroy if env
-        true
-      rescue => e
-        puts "*** chef exception: #{e.message}"
-        false
-      end
-
-      def client_delete(name)
-        configure
-        # TODO: better error handling
-        client = ::Chef::ApiClient.load(name) rescue nil
-        client.destroy if client
-      end
-
-      def client_list
-        configure
-        # TODO: better error handling
-        ::Chef::ApiClient.list
-      end
-
-      def node_delete(name)
-        configure
-        # TODO: better error handling
-        node = ::Chef::Node.load(name) rescue nil
-        node.destroy if node
-      end
-    end
-  end
-end

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

@@ -1,16 +0,0 @@
-require 'rig'
-
-module Rig
-  module Command
-    module Balancer
-      class Destroy < Abstract
-        parameter "NAME", "name of load balancer"
-        def execute
-          Rig::Model::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

@@ -1,30 +0,0 @@
-require 'rig'
-
-module Rig
-  module Command
-    module Balancer
-      class List < Abstract
-        def execute
-          list = Rig::Model::Balancer.all
-          list.each do |lb|
-            puts lb.id
-            puts "- listeners"
-            lb.listeners.each do |l|
-              puts "  - #{l.protocol}:#{l.lb_port} => #{l.instance_protocol}:#{l.instance_port}"
-            end
-            puts "- instances"
-            lb.instances.each do |inst|
-              puts "  - #{inst}"
-            end
-            puts "- zones"
-            lb.availability_zones.each do |z|
-              puts "  - #{z}"
-            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/listener.rb

@@ -1,31 +0,0 @@
-module Rig
-  module Command
-    module Listener
-      class Main < Abstract
-        subcommand "add", "add a listener to a balancer" do
-          parameter "BALANCER", "balancer name"
-          parameter "LISTENER", "listener data PROTO:PORT/PROTO:PORT"
-          parameter "[CERT]", "arn for server certificate"
-          parameter "[POLICY] ...", "policy names"
-
-          def execute
-            (from, to) = listener.split('/')
-            #newlist    = elb.listeners.create()
-            Rig::Model::Balancer.add_listener(balancer, from, to, cert, policy_list)
-          end
-        end
-
-        subcommand "rm", "remove a listener from a balancer" do
-          parameter "BALANCER", "balancer name"
-          parameter "LISTENER", "listener data PROTO:PORT (just the elb side)"
-
-          def execute
-            Rig::Model::Balancer.destroy_listener(balancer, listener)
-          end
-        end
-      end
-    end
-  end
-end
-
-Rig::Command::Balancer::Main.subcommand "listener", "manage balancer listeners (protocols and ports", Rig::Command::Listener::Main

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

@@ -1,40 +0,0 @@
-require 'rig'
-require 'rest-client'
-
-module Rig
-  module Command
-    module Balancer
-      class Main < Abstract
-        self.default_subcommand = 'list'
-        subcommand 'test', 'test a load balancer' do
-          option %w{-c --count}, "COUNT", "number of times to loop", :default => 100
-          option %w{-p --poll}, "POLL", "poll time", :default => 10
-          parameter "NAME", "name of the load balancer to test"
-
-          def execute
-            elb = Rig::Model::Balancer.find(name)
-            raise "elb #{name} not found" unless elb
-
-            dns = elb.dns_name
-            count.times do |i|
-              puts "### loop #{i}"
-              ips = `host #{dns}`.split(/\n/).collect { |e| e.split.last }
-              ips.each do |ip|
-                val = begin
-                  RestClient.get("http://#{ip}")
-                rescue => e
-                  e.message
-                end
-                print ".. #{ip} - #{val}"
-                puts
-              end
-              sleep poll
-            end
-          end
-        end
-      end
-    end
-  end
-end
-
-Rig::Command::Main.subcommand 'balancer', 'view load balancers', Rig::Command::Balancer::Main

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

@@ -1,19 +0,0 @@
-require 'awesome_print'
-
-module Rig
-  module Command
-    module Balancer
-      class View < Abstract
-        include Options::ShowAll
-
-        parameter "NAME", "name of instance"
-
-        def execute
-          item = Rig::Model::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

@@ -1,19 +0,0 @@
-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::Model::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

@@ -1,16 +0,0 @@
-require 'rig'
-
-module Rig
-  module Command
-    module Dns
-      class Destroy < Abstract
-        parameter "NAME", "name of load balancer"
-        def execute
-          Rig::Model::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

@@ -1,20 +0,0 @@
-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::Model::Dns.destroy(name)
-          Rig::Model::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

@@ -1,21 +0,0 @@
-require 'rig'
-
-module Rig
-  module Command
-    module Dns
-      class List < Abstract
-        def execute
-          zone    = Rig.get_config(:dns_zone)
-          records = Rig::Model::Dns.records("#{zone}")
-          rows    = []
-          records.each do |record|
-            rows << [zone, record.type, record.ttl, record.name, [*record.value].join(",")]
-          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

@@ -1,13 +0,0 @@
-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

@@ -1,21 +0,0 @@
-require 'rig'
-require 'fog'
-
-module Rig
-  module Command
-    module Environment
-      class Create < Abstract
-        parameter "NAME", "the name of the environment"
-        parameter "TEMPLATE", "the template to use (solo, multi, etc)", :default => "solo"
-
-        def execute
-          #connection = Rig::Connection.compute
-          env = Rig::Model::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

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

@@ -1,18 +0,0 @@
-require 'rig'
-require 'fog'
-
-module Rig
-  module Command
-    module Environment
-      class Destroy < Abstract
-        parameter "NAME", "name of the environment to destroy"
-
-        def execute
-          Rig::Model::Environment.destroy(name)
-        end
-      end
-    end
-  end
-end
-
-Rig::Command::Environment::Main.subcommand 'destroy', 'destroy environment', Rig::Command::Environment::Destroy

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

@@ -1,31 +0,0 @@
-require 'rig'
-require 'fog'
-
-module Rig
-  module Command
-    module Environment
-      class List < Abstract
-
-        option %w{-s --simple}, :flag, "just show list of names"
-
-        def execute
-          envs = Rig::Model::Environment.list
-
-          if simple?
-            print_table(%w{Name}, envs)
-          else
-            rows = []
-            envs.each do |e|
-              env = Rig::Model::Environment.find(e)
-              zones = env.servers.collect {|e| e.availability_zone }
-              rows << [env.name, env.template, env.region, env.servers.count, env.balancers.count, env.protected?, zones]
-            end
-            print_table(%w{Name Template Region Servers# Balancers# Protected? AZs}, rows)
-          end
-        end
-      end
-    end
-  end
-end
-
-Rig::Command::Environment::Main.subcommand 'list', 'List environments', Rig::Command::Environment::List

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

@@ -1,15 +0,0 @@
-require 'clamp'
-require 'rig'
-
-module Rig
-  module Command
-    module Environment
-      class Main < Abstract
-        self.default_subcommand = "list"
-      end
-    end
-  end
-end
-
-Rig::Command::Main.subcommand 'environment', 'Commands related to environments', Rig::Command::Environment::Main
-

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

@@ -1,22 +0,0 @@
-module Rig
-  module Command
-    module Environment
-      class Protect < Abstract
-        option ["-u","--undo"], :flag, "disable protection instead of enabling"
-        parameter "NAME", "the environment to protect"
-
-        def execute
-          env = Rig::Model::Environment.find(name)
-          raise "Environment #{name} not found" if env.nil?
-          connection = Rig::Connection.compute
-          val = undo? ? "false" : "true"
-
-          env.servers.each do |s|
-            connection.create_tags(s.id, "Protected" => val)
-          end
-        end
-      end
-    end
-  end
-end
-Rig::Command::Environment::Main.subcommand "protect", "[un]set protection on environment, preventing termination", Rig::Command::Environment::Protect

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

@@ -1,60 +0,0 @@
-require 'rig'
-require 'awesome_print'
-
-module Rig
-  module Command
-    module Instance
-      class Create < Abstract
-        option %w{-i --image}, "IMAGE", "the image / ami to use, default: your account image"
-        option %w{-f --flavor}, "FLAVOR", "the flavor / size to use"
-        option %w{-k --keypair}, "KEYPAIR", "the key pair / name to use, default: your account keypair"
-        option %w{-g --groups}, "GROUPS", "comma separated list of groups, default: your account groups"
-        option %w{-r --region}, "GROUPS", "the region, default: your account region"
-        option %w{--add}, :flag, "add this to the existing instances of this role and environment"
-        option "--[no-]chef", :flag, "use chef to bootstrap", :default => true
-
-        parameter "ENVIRONMENT", "the environment this instance should belong to"
-        parameter "ROLE", "the instance role", :default => 'solo'
-
-        def execute
-          name = role
-          # figure out if there are others of this role, if there are add a number to name to distinguish.
-          n    = "#{name}.#{environment}.env"
-          chef = Rig.config[:chef] ? true : false
-
-          unless flavor
-            raise "must specify flavor"
-          end
-
-          o = {
-              :image_id  => image,
-              :flavor_id => flavor,
-              :key_name  => keypair,
-              :groups    => groups.split(','),
-              :user_data => Rig::Model::Userdata.create(n, role, environment, :chef => chef)
-          }
-          tags = {'Name' => n, 'Environment' => environment, 'Role' => role}
-          server = Rig::Model::Instance.create(o, tags)
-          puts "created: #{server.id}"
-        rescue => e
-          puts "error: #{e.message}"
-          raise Clamp::HelpWanted, self
-        end
-
-        def default_image
-          Rig.account[:image].is_a?(Hash) ? Rig.account[:image].first.last : Rig.account[:image]
-        end
-
-        def default_keypair
-          Rig.account[:keypair]
-        end
-
-        def default_groups
-          Rig.account[:groups].join(",")
-        end
-      end
-    end
-  end
-end
-
-Rig::Command::Instance::Main.subcommand 'create', 'Create an instance', Rig::Command::Instance::Create