rig 0.3.1

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

@@ -0,0 +1,18 @@
+require 'rig'
+require 'fog'
+
+module Rig
+  module Command
+    module Environment
+      class Destroy < Abstract
+        parameter "NAME", "name of the environment to destroy"
+
+        def execute
+          Rig::Cloud::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

@@ -0,0 +1,30 @@
+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::Cloud::Environment.list
+
+          if simple?
+            print_table(%w{Name}, envs)
+          else
+            rows = []
+            envs.each do |e|
+              env = Rig::Cloud::Environment.find(e)
+              rows << [env.name, env.template, env.region, env.servers.count, env.balancers.count]
+            end
+            print_table(%w{Name Template Region Servers# Balancers#}, 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

@@ -0,0 +1,15 @@
+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/instance/create.rb

@@ -0,0 +1,27 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Instance
+      class Create < Abstract
+        include Options::Config
+
+        include Options::AwsKey
+        include Options::AwsSecret
+        include Options::AwsRegion
+        include Options::AwsAmi
+        include Options::AwsKeypair
+
+        parameter "CLUSTER", "the cluster this instance should belong to"
+        parameter "ENVIRONMENT", "the environment this instance should belong to"
+        parameter "ROLE", "the instance role", :default => 'solo'
+
+        def execute
+          puts "create: ami:#{ami} region:#{region} keypair:#{keypair} cluster:#{cluster} env:#{environment} role:#{role}"
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Main.subcommand 'create', 'Create an instance', Rig::Command::Instance::Create

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

@@ -0,0 +1,20 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Instance
+    class Destroy < Abstract
+      include Options::Config
+
+      parameter "NAME", "the name of the instance to connect to"
+
+      def execute
+        list = Rig::Cloud::Instance.find(name)
+        Rig::Cloud::Instance.destroy(list)
+      end
+    end
+    end
+  end
+end
+
+Rig::Command::Instance::Main.subcommand 'destroy', 'destroy (terminate) instance', Rig::Command::Instance::Destroy

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

@@ -0,0 +1,25 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Instance
+      class List < Abstract
+        include Options::ShowAll
+
+        parameter "[ENV]", "environment"
+
+        def execute
+          connection = Rig::Cloud.compute
+          filters = {}
+          filters["instance-state-name"] = "running" unless showall?
+          filters["tag:Environment"] = env if env
+
+          list = connection.servers.all(filters)
+          instance_list(list)
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Main.subcommand 'list', 'List instances', Rig::Command::Instance::List

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

@@ -0,0 +1,16 @@
+module Rig
+  module Command
+    module Instance
+      class Main < Abstract
+        #subcommand 'list', 'List instances', InstanceList
+        #subcommand 'create', 'Create an instance', InstanceCreate
+        #subcommand 'ssh', 'Connect to instance', InstanceSsh
+        #subcommand 'tag', 'Manage tags on instance', InstanceTag
+        self.default_subcommand = "list"
+      end
+    end
+  end
+end
+
+Rig::Command::Main.subcommand 'instance', 'Commands related to instances', Rig::Command::Instance::Main
+

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

@@ -0,0 +1,32 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Instance
+    class Ssh < Abstract
+      include Options::Config
+
+      parameter "NAME", "the name of the instance to connect to"
+
+      def execute
+        connection = Rig::Cloud.compute
+
+        filters = {}
+        filters["tag:Name"] = name if name
+
+        list = connection.servers.all(filters)
+        if list.count > 1
+          puts "there is more than one instance that matches:"
+          instance_list(list)
+          return
+        end
+
+        server = list.first
+        puts "ssh root@#{server.public_ip_address}"
+      end
+    end
+    end
+  end
+end
+
+Rig::Command::Instance::Main.subcommand 'ssh', 'Connect to instance', Rig::Command::Instance::Ssh

data/lib/rig/command/instance/tag/get.rb

@@ -0,0 +1,41 @@
+module Rig
+  module Command
+    module Instance
+      module Tag
+        class Get < Abstract
+          include Options::Config
+          include Options::Instance
+          include Options::InstanceName
+
+          def execute
+            unless iname || iid
+              raise "Must set either instance name or instance id (-iname or -iid)"
+            end
+
+            connection = Rig::Cloud.compute
+            if iid
+              instance = connection.servers.get(iid)
+            else
+              list = connection.servers.all({"tag:Name" => iname})
+              if list.count > 1
+                puts "there is more than one instance that matches:"
+                instance_list(list)
+                return
+              end
+              instance = list.first
+            end
+
+            unless instance
+              puts "could not find instance"
+              return
+            end
+
+            puts "#{instance.tags.inspect}"
+          end
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Tag::Main.subcommand 'get', 'get tag(s) for this instance', Rig::Command::Instance::Tag::Get

data/lib/rig/command/instance/tag/main.rb

@@ -0,0 +1,14 @@
+require 'rig'
+
+module Rig
+  module Command
+    module Instance
+      module Tag
+        class Main < Abstract
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Main.subcommand 'tag', 'Manage tags on instance', Rig::Command::Instance::Tag::Main

data/lib/rig/command/instance/tag/remove.rb

@@ -0,0 +1,44 @@
+
+module Rig
+  module Command
+    module Instance
+      module Tag
+        class Remove < Abstract
+          include Options::Config
+          include Options::Instance
+          include Options::InstanceName
+
+          parameter "TAG", "tag key to remove", :attribute_name => :raw_tag
+
+          def execute
+            unless iname || iid
+              raise "Must set either instance name or instance id (-iname or -iid)"
+            end
+
+            connection = Rig::Cloud.compute
+            if iid
+              instance = connection.servers.get(iid)
+            else
+              list = connection.servers.all({"tag:Name" => iname})
+              if list.count > 1
+                puts "there is more than one instance that matches:"
+                instance_list(list)
+                return
+              end
+              instance = list.first
+            end
+
+            unless instance
+              puts "could not find instance"
+              return
+            end
+
+            ap connection.delete_tags(instance.id, raw_tag => instance.tags[raw_tag])
+          end
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Tag::Main.subcommand 'rm', 'remove tag(s) for this instance', Rig::Command::Instance::Tag::Remove

data/lib/rig/command/instance/tag/set.rb

@@ -0,0 +1,47 @@
+module Rig
+  module Command
+    module Instance
+      module Tag
+        class Set < Abstract
+          include Options::Config
+          include Options::Instance
+          include Options::InstanceName
+
+          parameter "TAGS ...", "tags of the form key:value", :attribute_name => :raw_tags
+
+          def execute
+            unless iname || iid
+              raise "Must set either instance name or instance id (-iname or -iid)"
+            end
+
+            connection = Rig::Cloud.compute
+            if iid
+              instance = connection.servers.get(iid)
+            else
+              list = connection.servers.all({"tag:Name" => iname})
+              if list.count > 1
+                puts "there is more than one instance that matches:"
+                instance_list(list)
+                return
+              end
+              instance = list.first
+            end
+
+            unless instance
+              puts "could not find instance"
+              return
+            end
+
+            raw_tags.each do |s|
+              (k,v) = s.split(':',2)
+              #tags[k] = v if k && v
+              connection.create_tags(instance.id, k => v)
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+Rig::Command::Instance::Tag::Main.subcommand 'set', 'set tag(s) for this instance', Rig::Command::Instance::Tag::Set

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

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

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

@@ -0,0 +1,14 @@
+module Rig
+  module Command
+    module Keypair
+      class List < Abstract
+        def execute
+          connection = Rig::Cloud.compute
+          list = connection.key_pairs.all
+          puts list.inspect
+        end
+      end
+    end
+  end
+end
+Rig::Command::Keypair::Main.subcommand 'list', 'list keypairs', Rig::Command::Keypair::List

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

@@ -0,0 +1,11 @@
+
+module Rig
+  module Command
+    module Keypair
+      class Main < Abstract
+        self.default_subcommand = "list"
+      end
+    end
+  end
+end
+Rig::Command::Main.subcommand 'keypair', 'Commands related to keypairs', Rig::Command::Keypair::Main

data/lib/rig/command/main.rb

@@ -0,0 +1,48 @@
+
+module Rig
+  module Command
+    class Main < Abstract
+      option %w{--config}, :flag, "print configuration and exit" do |c|
+        puts "Rig Version: #{Rig::Version::STRING}"
+        puts Rig.config.to_yaml
+        exit(0)
+      end
+
+      option %w{--init}, :flag, "initialize the ~/.rig configuration file" do |o|
+        dir = ENV['RIG_CONFIG'] ? File.dirname(ENV['RIG_CONFIG']) : ENV['HOME']
+
+        unless dir
+          puts "directory could not be found: tried environment variables RIG_CONFIG (#{ENV['RIG_CONFIG']}) and HOME (#{ENV['HOME']}"
+          exit(1)
+        end
+
+        file = "#{dir}/.rig"
+        puts "[initializing] #{file}"
+        data = <<-EOF
+rig:
+  provider: AWS
+  key: AKIAIWUGNGSUZWW5XVCQ
+  secret: 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
+  ami: ami-8baa73e2
+  region: us-east-1
+  keypair: inqcloud-dev
+
+  dns:
+    zone: inqlabs.com
+
+  # if you're using chef, set this to the knife.rb configuration file
+  # this will allow Rig to manage the chef nodes, environments and such.
+  #chef:
+  #   knife: ~/work/chef/.chef/knife.rb
+  # TODO: Figure out the same thing for puppet
+
+  # eventually we should be able to put default options and such in the config file
+
+        EOF
+        #puts data
+        File.open(file, 'w') { |f| f.write(data) }
+        exit(0)
+      end
+    end
+  end
+end