knife-sce 0.2.0

data/lib/chef/knife/sce_base.rb

@@ -0,0 +1,128 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+class String
+  def is_number?
+    true if Float(self) rescue false
+  end
+end
+
+require 'chef/knife'
+
+class Chef
+  class Knife
+    module SceBase
+
+      # :nodoc:
+      # Would prefer to do this in a rational way, but can't be done b/c of
+      # Mixlib::CLI's design :(
+      def self.included(includer)
+        includer.class_eval do
+
+          deps do
+            require 'fog'
+            require 'readline'
+            require 'chef/json_compat'
+          end
+
+          option :ibm_username,
+            :short => "-A USERNAME",
+            :long => "--ibm-username USERNAME",
+            :description => "Your IBM SCE username",
+            :proc => Proc.new { |key| Chef::Config[:knife][:ibm_username] = key }
+
+          option :ibm_password,
+            :short => "-K PASSWORD",
+            :long => "--ibm-password PASSWORD",
+            :description => "Your SCE password",
+            :proc => Proc.new { |key| Chef::Config[:knife][:ibm_password] = key }
+        end
+      end
+
+      def connection
+        @connection ||= begin
+          connection = Fog::Compute.new(
+            :provider => 'IBM',
+            :ibm_username => Chef::Config[:knife][:ibm_username],
+            :ibm_password => Chef::Config[:knife][:ibm_password]
+          )
+        end
+      end
+      
+      def connection_storage
+        @connection_storage ||= begin
+          connection_storage = Fog::Storage.new(
+            :provider => 'IBM',
+            :ibm_username => Chef::Config[:knife][:ibm_username],
+            :ibm_password => Chef::Config[:knife][:ibm_password]
+          )
+        end
+      end
+
+      def locate_config_value(key)
+        key = key.to_sym
+        config[key] || Chef::Config[:knife][key]
+      end
+      
+      def datacenter_id
+        id = nil
+        if locate_config_value(:datacenter).to_s.is_number?
+          location = connection.locations.get( locate_config_value(:datacenter) )
+          if !location.nil?
+            id = location.id.to_s
+          else
+            ui.error("Location ID #{locate_config_value(:datacenter)} is invalid.  Use knife sce location list to learn what IDs or textual locations are available.")
+            exit 1
+          end
+        else
+          connection.locations.all.each do |location|
+            if location.name.to_s.split(",").first.downcase.eql?( locate_config_value(:datacenter).downcase )
+              id = location.id.to_s
+            end
+          end
+        end
+        
+        if id.nil?
+          ui.error("Location #{locate_config_value(:datacenter)} is invalid.  Use knife sce location list to learn what IDs or textual locations are available.")
+          exit 1
+        end
+        
+        id
+      end
+
+      def msg_pair(label, value, color=:cyan)
+        if value && !value.to_s.empty?
+          puts "#{ui.color(label, color)}: #{value}"
+        end
+      end
+      
+      def validate!(keys=[:ibm_username, :ibm_password])
+        errors = []
+        keys.each do |k|
+          pretty_key = k.to_s.gsub(/_/, ' ').gsub(/\w+/){ |w| (w =~ /(ssh)|(aws)/i) ? w.upcase  : w.capitalize }
+          if Chef::Config[:knife][k].nil?
+            errors << "You did not provide a valid '#{pretty_key}' value."
+          end
+        end
+        if errors.each{|e| ui.error(e)}.any?
+          exit 1
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/sce_image_describe.rb

@@ -0,0 +1,82 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceImageDescribe < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce image describe IMAGE_ID [IMAGE_ID]"
+
+      def run
+
+        validate!
+
+        @name_args.each do |image_id|
+            
+          begin
+            @image = connection.images.get(image_id)
+          rescue Excon::Errors::InternalServerError => e
+            if image_id.is_number?
+              ui.error e.inspect
+              exit 1
+            end
+            # is not a number and we received an error, ignore, API likes numbers only, we try to fetch the image by the name
+          end
+          
+          if @image.nil?
+            connection.images.all.each do |i|
+              if i.name.to_s == image_id
+                @image = i
+              end
+            end
+          end
+          
+          msg_pair("Image ID", @image.id.to_s)
+          msg_pair("Name", @image.name.to_s)
+          msg_pair("Location", connection.locations.get(@image.location).name.to_s)
+          msg_pair("Description", @image.description.to_s)
+          msg_pair("Visbility", @image.visibility.to_s)
+          msg_pair("Platform", @image.platform.to_s)
+          msg_pair("Architecture", @image.architecture.to_s)
+          msg_pair("Owner", @image.owner.to_s)
+          msg_pair("State", @image.state.to_s)
+          msg_pair("Manifest", @image.manifest.to_s)
+          msg_pair("Product codes", ":")
+          @image.product_codes.each do |pc|
+            msg_pair("  -> ", pc)
+          end
+          msg_pair("Supported instance types", ":")
+          @image.supported_instance_types.each do |sit|
+            msg_pair("  -> ", sit.id.to_s)
+            msg_pair("     ", sit.label.to_s)
+            msg_pair("     ", sit.detail.to_s)
+            msg_pair("     ", "Price: #{sit.price['rate']}#{sit.price['currencyCode']}/#{sit.price['pricePerQuantity']}#{sit.price['unitOfMeasure']}")
+          end
+          msg_pair("Documentation", @image.documentation.to_s)
+          
+          puts "\n"
+
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/sce_instance_data.rb

@@ -0,0 +1,55 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+
+class Chef
+  class Knife
+    class SceInstanceData < Knife
+      
+      deps do
+        require 'chef/json_compat'
+      end
+
+      banner "knife sce instance data (options)"
+
+      option :edit,
+        :short => "-e",
+        :long => "--edit",
+        :description => "Edit the instance data"
+
+      option :run_list,
+        :short => "-r RUN_LIST",
+        :long => "--run-list RUN_LIST",
+        :description => "Comma separated list of roles/recipes to apply",
+        :proc => lambda { |o| o.split(/[\s,]+/) },
+        :default => []
+
+      def run
+        data = {
+          "chef_server" => Chef::Config[:chef_server_url],
+          "validation_client_name" => Chef::Config[:validation_client_name],
+          "validation_key" => IO.read(Chef::Config[:validation_key]),
+          "attributes" => { "run_list" => config[:run_list] }
+        }
+        data = edit_data(data) if config[:edit]
+        ui.output(data)
+      end
+    end
+  end
+end
+

data/lib/chef/knife/sce_key_create.rb

@@ -0,0 +1,58 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceKeyCreate < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce key create KEYNAME"
+      
+      def run!(key_name)
+        @key = connection.keys.create(:name => key_name)
+        @key
+      end
+      
+      def run
+        
+        $stdout.sync = true
+
+        validate!
+        
+        @key = run!(config[:name])
+        
+        msg_pair("Name", @key.name.to_s)
+        msg_pair("Default", (@key.default ? "Yes" : "No"))
+        msg_pair("Key", @key.public_key.to_s)
+        
+      end
+      
+      def validate!
+        
+        super
+        
+        raise "No key name specified." if @name_args.length == 0
+        config[:name] = @name_args[0]
+        
+      end
+      
+    end
+  end
+end

data/lib/chef/knife/sce_key_delete.rb

@@ -0,0 +1,66 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceKeyDelete < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce key delete KEYNAME"
+      
+      def run!(key)
+        key.destroy
+      end
+      
+      def run
+        
+        $stdout.sync = true
+
+        validate!
+        
+        @key = connection.keys.get(config[:name])
+        
+        raise "Key #{config[:name]} does not exist." if @key.nil?
+        
+        msg_pair("Name", @key.name.to_s)
+        msg_pair("Instances", @key.instance_ids.join(", ").to_s)
+        msg_pair("Default", (@key.default ? "Yes" : "No"))
+        
+        puts "\n"
+        confirm("Do you really want to delete this key")
+        
+        run!(@key)
+        
+        ui.warn("Deleted key #{@key.name.to_s}")
+        
+      end
+      
+      def validate!
+        
+        super
+        
+        raise "No key name specified." if @name_args.length == 0
+        config[:name] = @name_args[0]
+        
+      end
+      
+    end
+  end
+end

data/lib/chef/knife/sce_key_get.rb

@@ -0,0 +1,61 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceKeyGet < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce key get KEYNAME"
+      
+      def run!(key_name)
+        @key = connection.keys.get(key_name)
+        @key
+      end
+      
+      def run
+        
+        $stdout.sync = true
+
+        validate!
+        
+        @key = run!(config[:name])
+        
+        raise "Key #{config[:name]} does not exist." if @key.nil?
+        
+        msg_pair("Name", @key.name.to_s)
+        msg_pair("Instances", @key.instance_ids.join(", ").to_s)
+        msg_pair("Default", (@key.default ? "Yes" : "No"))
+        msg_pair("Key", @key.public_key.to_s)
+        
+      end
+      
+      def validate!
+        
+        super
+        
+        raise "No key name specified." if @name_args.length == 0
+        config[:name] = @name_args[0]
+        
+      end
+      
+    end
+  end
+end

data/lib/chef/knife/sce_key_list.rb

@@ -0,0 +1,61 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceKeyList < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce key list"
+      
+      def run!
+        connection.keys.all
+      end
+      
+      def run
+        
+        $stdout.sync = true
+
+        validate!
+        
+        keys = run!
+        
+        key_list = [
+          ui.color('Name', :bold),
+          ui.color('Default', :bold),
+          ui.color("Modified at", :bold),
+          ui.color('Instances', :bold)
+        ].flatten.compact
+        
+        output_column_count = key_list.length
+        
+        keys.each do |k|
+          key_list << k.name.to_s
+          key_list << (k.default ? "*" : " ")
+          key_list << Time.at(k.modified_at/1000).to_datetime.to_s
+          key_list << k.instance_ids.join(", ").to_s
+        end
+        
+        puts ui.list(key_list, :uneven_columns_across, output_column_count)
+        
+      end
+    end
+  end
+end

data/lib/chef/knife/sce_location_list.rb

@@ -0,0 +1,105 @@
+#
+# Author:: Rad Gruchalski (<radek@gruchalski.com>)
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/sce_base'
+
+class Chef
+  class Knife
+    class SceLocationList < Knife
+
+      include Knife::SceBase
+
+      banner "knife sce location list (options)"
+      
+      option :description,
+        :short => "-L",
+        :long => "--with-description",
+        :boolean => true,
+        :default => false,
+        :description => "Display description of the location"
+        
+      option :capabilities,
+        :short => "-C",
+        :long => "--with-capabilities",
+        :boolean => true,
+        :default => false,
+        :description => "Display capabilities"
+      
+      def run!
+        connection.locations.all
+      end
+      
+      def run
+        $stdout.sync = true
+
+        validate!
+        
+        location_list = [
+          ui.color('Location ID', :bold),
+          ui.color("Name", :bold),
+          ui.color('Location', :bold),
+          if config[:description]
+            ui.color('Description', :bold)
+          end,
+          if config[:capabilities]
+            ui.color('Capabilities', :bold)
+          end
+        ].flatten.compact
+        
+        output_column_count = location_list.length
+        
+        locations = run!
+        
+        locations.each do |location|
+          location_list << location.id.to_s
+          location_list << location.name.to_s
+          location_list << location.location.to_s
+          if config[:description]
+            location_list << location.description.to_s
+          end
+          if config[:capabilities]
+            if location.capabilities.length > 0
+              tabs = "\t"
+              tabs = "#{tabs}\t" if location.capabilities[0]["id"].to_s.length < 25
+              location_list << "#{location.capabilities[0]["id"].to_s}:#{tabs}#{location.capabilities[0]["entries"].to_s}"
+            else
+              location_list << " "
+            end
+          end
+          
+          if config[:capabilities]
+            (1...location.capabilities.length).each do |index|
+              location_list << " "
+              location_list << " "
+              location_list << " "
+              if config[:description]
+                location_list << " "
+              end
+              tabs = "\t"
+              tabs = "#{tabs}\t" if location.capabilities[index]["id"].to_s.length < 25
+              location_list << "#{location.capabilities[index]["id"].to_s}:#{tabs}#{location.capabilities[index]["entries"].to_s}"
+            end
+          end
+          
+        end
+        
+        puts ui.list(location_list, :uneven_columns_across, output_column_count)
+
+      end
+    end
+  end
+end