mbailey-chef 0.9.12.1

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/knife/windows_bootstrap.rb

@@ -0,0 +1,154 @@
+#
+# Author:: Doug MacEachern <dougm@vmware.com>
+# Copyright:: Copyright (c) 2010 VMware, 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 'fileutils'
+require 'chef/knife/bootstrap.rb'
+
+class Chef
+  class Knife
+    class WindowsBootstrap < Chef::Knife::Bootstrap
+
+      banner "knife windows bootstrap FQDN [RUN LIST...] (options)"
+
+      option :user,
+        :short => "-x USERNAME",
+        :long  => "--user USERNAME",
+        :description => "The Windows username"
+
+      option :password,
+        :short => "-P PASSWORD",
+        :long  => "--password PASSWORD",
+        :description => "The Windows password"
+
+      option :chef_node_name,
+        :short => "-N NAME",
+        :long => "--node-name NAME",
+        :description => "The Chef node name for your new node"
+
+      option :distro,
+        :short => "-d DISTRO",
+        :long => "--distro DISTRO",
+        :description => "Bootstrap a distro using a template",
+        :default => "windows-gems"
+
+      option :template_file,
+        :long => "--template-file TEMPLATE",
+        :description => "Full path to location of template to use",
+        :default => false
+
+      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(",") },
+        :default => []
+
+      def is_mounted
+        @net_use ||= Chef::Util::Windows::NetUse.new(@admin_share)
+        begin
+          @net_use.get_info
+          return true
+        rescue
+          return false
+        end
+      end
+
+      def mount_admin_share
+        if @add_mount && !is_mounted
+          use = {
+            :remote => @admin_share, :local => '',
+            :username => config[:user], :password => config[:password]
+          }
+          @net_use.add(use)
+          if is_mounted
+            Chef::Log.info("Mounted #{@admin_share} for copying files")
+          else
+            Chef::Log.fatal("Failed to mount #{@admin_share}")
+            exit 1
+          end
+        end
+      end
+
+      def unmount_admin_share
+        if @add_mount && is_mounted
+          Chef::Log.debug("Unmounting #{@admin_share}")
+          @net_use.delete
+        end
+      end
+
+      def psexec(args)
+        cmd = ['psexec', @unc_path, "-h", "-w", 'c:\chef\tmp']
+        if config[:user]
+          cmd << "-u" << config[:user]
+        end
+        if config[:password]
+          cmd << "-p" << config[:password]
+        end
+        cmd << args
+        cmd = cmd.join(' ')
+        Chef::Log.debug("system #{cmd}")
+        system(cmd)
+      end
+
+      def run
+        require 'chef/util/windows/net_use'
+
+        if @name_args.first == nil
+          Chef::Log.error("Must pass a node name/ip to windows bootstrap")
+          exit 1
+        end
+
+        config[:server_name] = @name_args.first
+        if Chef::Config[:http_proxy]
+          uri = URI.parse(Chef::Config[:http_proxy])
+          config[:proxy] = "#{uri.host}:#{uri.port}"
+        end
+
+        @unc_path = "\\\\#{config[:server_name]}"
+        @admin_share = "#{@unc_path}\\c$"
+        path = "#{@admin_share}\\chef"
+        etc = "#{path}\\etc"
+        tmp = "#{path}\\tmp"
+
+        $stdout.sync = true
+
+        command = render_template(load_template(config[:bootstrap_template]))
+
+        Chef::Log.info("Bootstrapping Chef on #{config[:server_name]}")
+
+        @add_mount = config[:user] != nil && !is_mounted
+        mount_admin_share
+
+        begin
+          [etc, tmp, "#{path}\\log"].each do |dir|
+            unless File.exists?(dir)
+              Chef::Log.debug("mkdir_p #{dir}")
+              FileUtils.mkdir_p(dir)
+            end
+          end
+          File.open("#{tmp}\\bootstrap.bat", 'w') {|f| f.write(command) }
+          FileUtils.cp(File.join(File.dirname(__FILE__), 'bootstrap', 'client-install.vbs'), tmp)
+          FileUtils.cp(Chef::Config[:validation_key], etc)
+          psexec("cmd.exe /c bootstrap.bat")
+        ensure
+          unmount_admin_share
+        end
+      end
+    end
+  end
+end