tecnh-chef 0.9.9.vpc

data/lib/chef/knife/status.rb

@@ -0,0 +1,70 @@
+#
+# Author:: Ian Meyer (<ianmmeyer@gmail.com>)
+# Copyright:: Copyright (c) 2010 Ian Meyer
+# 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'
+require 'highline'
+require 'chef/search/query'
+
+class Chef
+  class Knife
+    class Status < Knife
+
+      banner "knife status"
+
+      def highline
+        @h ||= HighLine.new
+      end
+
+      def run
+        all_nodes = []
+        Chef::Search::Query.new.search(:node, '*:*') do |node|
+          all_nodes << node
+        end
+        all_nodes.sort { |n1, n2| n1["ohai_time"] <=> n2["ohai_time"] }.each do |node|
+          # 59 seconds
+          # 1000 hours
+          # date = DateTime.parse(Time.at(node["ohai_time"]).to_s)
+          hours, minutes, seconds = time_difference_in_hms(node["ohai_time"])
+          hours_text   = "#{hours} hour#{hours == 1 ? ' ' : 's'}"
+          minutes_text = "#{minutes} minute#{minutes == 1 ? ' ' : 's'}"
+          if hours > 24
+            highline.say("<%= color('#{hours_text}', RED) %> ago, #{node['fqdn']} checked in as a #{node['platform']} #{node['platform_version']} node.")
+          elsif hours > 1
+            highline.say("<%= color('#{hours_text}', YELLOW) %> ago, #{node['fqdn']} checked in as a #{node['platform']} #{node['platform_version']} node.")
+          elsif hours == 0
+            highline.say("<%= color('#{minutes_text}', GREEN) %> ago, #{node['fqdn']} checked in as a #{node['platform']} #{node['platform_version']} node.")
+          end
+        end
+
+      end
+
+      # :nodoc:
+      # TODO: this is duplicated from StatusHelper in the Webui. dedup.
+      def time_difference_in_hms(unix_time)
+        now = Time.now.to_i
+        difference = now - unix_time.to_i
+        hours = (difference / 3600).to_i
+        difference = difference % 3600
+        minutes = (difference / 60).to_i
+        seconds = (difference % 60)
+        return [hours, minutes, seconds]
+      end
+
+    end
+  end
+end

data/lib/chef/knife/terremark_server_create.rb

@@ -0,0 +1,152 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# 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'
+require 'json'
+require 'tempfile'
+
+class Chef
+  class Knife
+    class TerremarkServerCreate < Knife
+
+      banner "knife terremark server create NAME [RUN LIST...] (options)"
+
+      option :terremark_password,
+        :short => "-K PASSWORD",
+        :long => "--terremark-password PASSWORD",
+        :description => "Your terremark password",
+        :proc => Proc.new { |key| Chef::Config[:knife][:terremark_password] = key } 
+
+      option :terremark_username,
+        :short => "-A USERNAME",
+        :long => "--terremark-username USERNAME",
+        :description => "Your terremark username",
+        :proc => Proc.new { |username| Chef::Config[:knife][:terremark_username] = username } 
+
+      option :terremark_service,
+        :short => "-S SERVICE",
+        :long => "--terremark-service SERVICE",
+        :description => "Your terremark service name",
+        :proc => Proc.new { |service| Chef::Config[:knife][:terremark_service] = service } 
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+        require 'net/scp'
+
+        server_name = @name_args[0]
+
+        terremark = Fog::Terremark.new(
+          :terremark_username => Chef::Config[:knife][:terremark_username],
+          :terremark_password => Chef::Config[:knife][:terremark_password],
+          :terremark_service  => Chef::Config[:knife][:terremark_service] || :vcloud
+        )
+
+        $stdout.sync = true
+
+        puts "Instantiating vApp #{h.color(server_name, :bold)}"
+        vapp_id = terremark.instantiate_vapp_template(server_name).body['href'].split('/').last
+
+        deploy_task_id = terremark.deploy_vapp(vapp_id).body['href'].split('/').last
+        print "Waiting for deploy task [#{h.color(deploy_task_id, :bold)}]"
+        terremark.tasks.get(deploy_task_id).wait_for { print "."; ready? }
+        print "\n"
+
+        power_on_task_id = terremark.power_on(vapp_id).body['href'].split('/').last
+        print "Waiting for power on task [#{h.color(power_on_task_id, :bold)}]"
+        terremark.tasks.get(power_on_task_id).wait_for { print "."; ready? }
+        print "\n"
+
+        private_ip = terremark.get_vapp(vapp_id).body['IpAddress']
+        ssh_internet_service = terremark.create_internet_service(terremark.default_vdc_id, 'SSH', 'TCP', 22).body
+        ssh_internet_service_id = ssh_internet_service['Id']
+        public_ip = ssh_internet_service['PublicIpAddress']['Name']
+        public_ip_id = ssh_internet_service['PublicIpAddress']['Id']
+        ssh_node_service_id = terremark.add_node_service(ssh_internet_service_id, private_ip, 'SSH', 22).body['Id']
+
+        puts "\nBootstrapping #{h.color(server_name, :bold)}..."
+        password = terremark.get_vapp_template(12).body['Description'].scan(/\npassword: (.*)\n/).first.first
+
+        command =  <<EOH
+bash -c '
+echo nameserver 208.67.222.222 > /etc/resolv.conf
+echo nameserver 208.67.220.220 >> /etc/resolv.conf
+
+if [ ! -f /usr/bin/chef-client ]; then
+  apt-get update
+  apt-get install -y ruby ruby1.8-dev build-essential wget libruby-extras libruby1.8-extras
+  cd /tmp
+  wget http://rubyforge.org/frs/download.php/69365/rubygems-1.3.6.tgz
+  tar xvf rubygems-1.3.6.tgz
+  cd rubygems-1.3.6
+  ruby setup.rb
+  cp /usr/bin/gem1.8 /usr/bin/gem
+  gem install chef ohai --no-rdoc --no-ri --verbose
+fi
+
+mkdir -p /etc/chef
+
+(
+cat <<'EOP'
+#{IO.read(Chef::Config[:validation_key])}
+EOP
+) > /etc/chef/validation.pem
+
+(
+cat <<'EOP'
+log_level        :info
+log_location     STDOUT
+chef_server_url  "#{Chef::Config[:chef_server_url]}" 
+validation_client_name "#{Chef::Config[:validation_client_name]}"
+EOP
+) > /etc/chef/client.rb
+
+(
+cat <<'EOP'
+#{{ "run_list" => @name_args[1..-1] }.to_json}
+EOP
+) > /etc/chef/first-boot.json
+
+/usr/bin/chef-client -j /etc/chef/first-boot.json'
+EOH
+
+        begin
+          ssh = Chef::Knife::Ssh.new
+          ssh.name_args = [ public_ip, "sudo #{command}" ]
+          ssh.config[:ssh_user] = "vcloud"
+          ssh.config[:manual] = true
+          ssh.config[:password] = password
+          ssh.password = password
+          ssh.run
+        rescue Errno::ETIMEDOUT
+          puts "Timed out on bootstrap, re-trying. Hit CTRL-C to abort."
+          puts "You probably need to log in to Terremark and powercycle #{h.color(@name_args[0], :bold)}"
+          retry
+        end
+
+      end
+    end
+  end
+end
+

data/lib/chef/knife/terremark_server_delete.rb

@@ -0,0 +1,87 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# 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'
+require 'json'
+
+class Chef
+  class Knife
+    class TerremarkServerDelete < Knife
+
+      banner "knife terremark server delete SERVER (options)"
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+
+        terremark = Fog::Terremark.new(
+          :terremark_username => Chef::Config[:knife][:terremark_username],
+          :terremark_password => Chef::Config[:knife][:terremark_password],
+          :terremark_service  => Chef::Config[:knife][:terremark_service] || :vcloud
+        )
+
+        $stdout.sync = true
+
+        vapp_id = terremark.servers.detect {|server| server.name == @name_args[0]}.id
+        confirm("Do you really want to delete server ID #{vapp_id} named #{@name_args[0]}")
+
+        puts "Cleaning up internet services..."
+        private_ip = terremark.servers.get(vapp_id).ip_address
+        internet_services = terremark.get_internet_services(terremark.default_vdc_id).body['InternetServices']
+        public_ip_usage = {}
+        internet_services.each do |internet_service|
+          public_ip_address = internet_service['PublicIpAddress']['Name']
+          public_ip_usage[public_ip_address] ||= []
+          public_ip_usage[public_ip_address] << internet_service['Id']
+        end
+        internet_services.each do |internet_service|
+          node_services = terremark.get_node_services(internet_service['Id']).body['NodeServices']
+          node_services.delete_if do |node_service|
+            if node_service['IpAddress'] == private_ip
+              terremark.delete_node_service(node_service['Id'])
+            end
+          end
+          if node_services.empty?
+            terremark.delete_internet_service(internet_service['Id'])
+            public_ip_usage.each_value {|internet_services| internet_services.delete(internet_service['Id'])}
+            if public_ip_usage[internet_service['PublicIpAddress']['Name']].empty?
+              terremark.delete_public_ip(internet_service['PublicIpAddress']['Id'])
+            end
+          end
+        end
+
+        power_off_task_id = terremark.power_off(vapp_id).body['href'].split('/').last
+        print "Waiting for power off task [#{h.color(power_off_task_id, :bold)}]"
+        terremark.tasks.get(power_off_task_id).wait_for { print '.'; ready? }
+        print "\n"
+
+        print "Deleting vApp #{h.color(vapp_id, :bold)}"
+        delete_vapp_task_id = terremark.delete_vapp(vapp_id).headers['Location'].split('/').last
+        terremark.tasks.get(delete_vapp_task_id).wait_for { print '.'; ready? }
+        print "\n"
+        
+        Chef::Log.warn("Deleted server #{@name_args[0]}")
+      end
+    end
+  end
+end
+

data/lib/chef/knife/terremark_server_list.rb

@@ -0,0 +1,77 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# 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'
+require 'json'
+require 'tempfile'
+
+class Chef
+  class Knife
+    class TerremarkServerList < Knife
+
+      banner "knife terremark server list (options)"
+
+      option :terremark_password,
+        :short => "-K PASSWORD",
+        :long => "--terremark-password PASSWORD",
+        :description => "Your terremark password",
+        :proc => Proc.new { |key| Chef::Config[:knife][:terremark_password] = key } 
+
+      option :terremark_username,
+        :short => "-A USERNAME",
+        :long => "--terremark-username USERNAME",
+        :description => "Your terremark username",
+        :proc => Proc.new { |username| Chef::Config[:knife][:terremark_username] = username } 
+
+      option :terremark_service,
+        :short => "-S SERVICE",
+        :long => "--terremark-service SERVICE",
+        :description => "Your terremark service name",
+        :proc => Proc.new { |service| Chef::Config[:knife][:terremark_service] = service } 
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+
+        server_name = @name_args[0]
+
+        terremark = Fog::Terremark.new(
+          :terremark_username => Chef::Config[:knife][:terremark_username],
+          :terremark_password => Chef::Config[:knife][:terremark_password],
+          :terremark_service  => Chef::Config[:knife][:terremark_service] || :vcloud
+        )
+
+        $stdout.sync = true
+
+        server_list = [ h.color('ID', :bold), h.color('Name', :bold) ]
+        terremark.servers.all.each do |server|
+          server_list << server.id.to_s
+          server_list << server.name
+        end
+        puts h.list(server_list, :columns_across, 2)
+
+      end
+    end
+  end
+end
+
+

data/lib/chef/log.rb

@@ -0,0 +1,61 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: AJ Christensen (<@aj@opscode.com>)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# 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 'logger'
+require 'mixlib/log'
+
+class Chef
+  class Log
+    extend Mixlib::Log
+    
+    class << self
+      attr_accessor :verbose
+      attr_reader :verbose_logger
+      protected :verbose_logger
+      
+      def verbose
+        !(@verbose_logger.nil?)
+      end
+
+      def verbose=(value)
+        if value
+          @verbose_logger ||= Logger.new(STDOUT)
+          @verbose_logger.level = self.logger.level
+          @verbose_logger.formatter = self.logger.formatter
+        else
+          @verbose_logger = nil
+        end
+        self.verbose
+      end
+      
+      def method_missing(method_symbol, *args)
+        self.verbose_logger.send(method_symbol, *args) if self.verbose
+        logger.send(method_symbol, *args)
+      end
+    end  
+
+    class Formatter
+      def self.show_time=(*args)
+        Mixlib::Log::Formatter.show_time = *args
+      end
+    end
+    
+  end
+end
+