knife-azure 1.0.0

data/lib/chef/knife/azure_server_create.rb

@@ -0,0 +1,283 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2010-2011 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 File.expand_path('../azure_base', __FILE__)
+
+class Chef
+  class Knife
+    class AzureServerCreate < Knife
+
+      include Knife::AzureBase
+
+      deps do
+        require 'readline'
+        require 'chef/json_compat'
+        require 'chef/knife/bootstrap'
+        Chef::Knife::Bootstrap.load_deps
+      end
+
+      banner "knife azure server create (options)"
+
+      attr_accessor :initial_sleep_delay
+
+      option :chef_node_name,
+        :short => "-N NAME",
+        :long => "--node-name NAME",
+        :description => "The Chef node name for your new node"
+
+      option :ssh_user,
+        :short => "-x USERNAME",
+        :long => "--ssh-user USERNAME",
+        :description => "The ssh username"
+
+      option :ssh_password,
+        :short => "-P PASSWORD",
+        :long => "--ssh-password PASSWORD",
+        :description => "The ssh password"
+
+      option :identity_file,
+        :short => "-i IDENTITY_FILE",
+        :long => "--identity-file IDENTITY_FILE",
+        :description => "The SSH identity file used for authentication"
+
+      option :prerelease,
+        :long => "--prerelease",
+        :description => "Install the pre-release chef gems"
+
+      option :bootstrap_version,
+        :long => "--bootstrap-version VERSION",
+        :description => "The version of Chef to install",
+        :proc => Proc.new { |v| Chef::Config[:knife][:bootstrap_version] = v }
+
+      option :distro,
+        :short => "-d DISTRO",
+        :long => "--distro DISTRO",
+        :description => "Bootstrap a distro using a template",
+        :proc => Proc.new { |d| Chef::Config[:knife][:distro] = d },
+        :default => "chef-full"
+
+      option :template_file,
+        :long => "--template-file TEMPLATE",
+        :description => "Full path to location of template to use",
+        :proc => Proc.new { |t| Chef::Config[:knife][:template_file] = t },
+        :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(/[\s,]+/) },
+        :default => []
+
+      option :no_host_key_verify,
+        :long => "--no-host-key-verify",
+        :description => "Disable host key verification",
+        :boolean => true,
+        :default => false
+
+      option :hosted_service_name,
+        :short => "-s NAME",
+        :long => "--hosted-service-name NAME",
+        :description => "specifies the name for the hosted service"
+
+      option :role_name,
+        :short => "-R name",
+        :long => "--role-name NAME",
+        :description => "specifies the name for the virtual machine"
+
+      option :host_name,
+        :short => "-H NAME",
+        :long => "--host-name NAME",
+        :description => "specifies the host name for the virtual machine"
+
+      option :media_location_prefix,
+        :short => "-m PREFIX",
+        :long => "--media-location-prefix PREFIX",
+        :description => "user account name (used for constructing os disk media link)"
+
+      option :os_disk_name,
+        :short => "-o DISKNAME",
+        :long => "--os-disk-name DISKNAME",
+        :description => "unique name for specifying os disk (optional)"
+
+      option :source_image,
+        :short => "-I IMAGE",
+        :long => "--source-image IMAGE",
+        :description => "disk image name to use to create virtual machine"
+
+      option :role_size,
+        :short => "-z SIZE",
+        :long => "--role-size SIZE",
+        :description => "size of virtual machine (ExtraSmall, Small, Medium, Large, ExtraLarge)"
+
+      option :tcp_endpoints,
+        :short => "-t PORT_LIST",
+        :long => "--tcp-endpoints PORT_LIST",
+        :description => "Comma separated list of TCP local and public ports to open i.e. '80:80,433:5000'"
+
+      option :udp_endpoints,
+        :short => "-u PORT_LIST",
+        :long => "--udp-endpoints PORT_LIST",
+        :description => "Comma separated list of UDP local and public ports to open i.e. '80:80,433:5000'"
+
+
+      def tcp_test_ssh(fqdn, sshport)
+        tcp_socket = TCPSocket.new(fqdn, sshport)
+        readable = IO.select([tcp_socket], nil, nil, 5)
+        if readable
+          Chef::Log.debug("sshd accepting connections on #{fqdn}, banner is #{tcp_socket.gets}")
+          yield
+          true
+        else
+          false
+        end
+      rescue SocketError
+        sleep 2
+        false
+      rescue Errno::ETIMEDOUT
+        false
+      rescue Errno::EPERM
+        false
+      rescue Errno::ECONNREFUSED
+        sleep 2
+        false
+        # This happens on EC2 quite often
+      rescue Errno::EHOSTUNREACH
+        sleep 2
+        false
+      ensure
+        tcp_socket && tcp_socket.close
+      end
+
+      def parameter_test
+        details = Array.new
+        details << ui.color('name', :bold, :blue)
+        details << ui.color('Chef::Config', :bold, :blue)
+        details << ui.color('config', :bold, :blue)
+        details << ui.color('winner is', :bold, :blue)
+        [
+          :azure_subscription_id, 
+          :azure_pem_file, 
+          :azure_host_name,
+          :hosted_service_name, 
+          :role_name, 
+          :host_name, 
+          :ssh_user, 
+          :ssh_password, 
+          :media_location_prefix, 
+          :source_image, 
+          :role_size
+        ].each do |key|
+          key = key.to_sym
+          details << key.to_s
+          details << Chef::Config[:knife][key].to_s
+          details << config[key].to_s
+          details << locate_config_value(key)
+        end 
+        puts ui.list(details, :columns_across, 4)
+      end
+      def run
+        $stdout.sync = true
+
+        Chef::Log.info("validating...")
+        validate!
+
+        Chef::Log.info("creating...")
+        server = connection.deploys.create(create_server_def)
+
+        puts("\n")
+
+        unless server && server.sshipaddress && server.sshport
+          Chef::Log.fatal("server not created")
+          exit 1
+        end
+
+        fqdn = server.sshipaddress
+        port = server.sshport
+
+        print "\n#{ui.color("Waiting for sshd on #{fqdn}:#{port}", :magenta)}"
+
+        print(".") until tcp_test_ssh(fqdn,port) {
+          sleep @initial_sleep_delay ||= 10
+          puts("done")
+        }
+
+        sleep 15
+
+        bootstrap_for_node(server,fqdn,port).run
+
+        puts "\n"
+      end
+
+      def bootstrap_for_node(server,fqdn,port)
+        bootstrap = Chef::Knife::Bootstrap.new
+        bootstrap.name_args = [fqdn]
+        bootstrap.config[:run_list] = config[:run_list]
+        bootstrap.config[:ssh_user] = locate_config_value(:ssh_user)
+        bootstrap.config[:ssh_password] = locate_config_value(:ssh_password)
+        bootstrap.config[:ssh_port] = port
+        bootstrap.config[:identity_file] = locate_config_value(:identity_file)
+        bootstrap.config[:chef_node_name] = locate_config_value(:chef_node_name) || server.name
+        bootstrap.config[:prerelease] = locate_config_value(:prerelease)
+        bootstrap.config[:bootstrap_version] = locate_config_value(:bootstrap_version)
+        bootstrap.config[:distro] = locate_config_value(:distro)
+        bootstrap.config[:use_sudo] = true unless locate_config_value(:ssh_user) == 'root'
+        bootstrap.config[:template_file] = config[:template_file]
+        bootstrap.config[:environment] = locate_config_value(:environment)
+        # may be needed for vpc_mode
+        bootstrap.config[:no_host_key_verify] = config[:no_host_key_verify]
+        bootstrap
+      end
+
+      def validate!
+        super([
+              :azure_subscription_id, 
+              :azure_pem_file, 
+              :azure_host_name,
+              :hosted_service_name, 
+              :role_name, 
+              :host_name, 
+              :ssh_user, 
+              :ssh_password, 
+              :media_location_prefix, 
+              :source_image, 
+              :role_size
+        ])
+      end
+
+      def create_server_def
+        server_def = {
+          :hosted_service_name => locate_config_value(:hosted_service_name), 
+          :role_name => locate_config_value(:role_name), 
+          :host_name => locate_config_value(:host_name), 
+          :ssh_user => locate_config_value(:ssh_user), 
+          :ssh_password => locate_config_value(:ssh_password), 
+          :media_location_prefix => locate_config_value(:media_location_prefix), 
+          :os_disk_name => locate_config_value(:os_disk_name), 
+          :source_image => locate_config_value(:source_image), 
+          :role_size => locate_config_value(:role_size),
+          :tcp_endpoints => locate_config_value(:tcp_endpoints),
+          :udp_endpoints => locate_config_value(:udp_endpoints)
+        }
+        server_def
+      end
+    end
+  end
+end

data/lib/chef/knife/azure_server_delete.rb

@@ -0,0 +1,103 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2009-2011 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 File.expand_path('../azure_base', __FILE__)
+
+# These two are needed for the '--purge' deletion case
+require 'chef/node'
+require 'chef/api_client'
+
+class Chef
+  class Knife
+    class AzureServerDelete < Knife
+
+      include Knife::AzureBase
+
+      banner "knife azure server delete SERVER [SERVER] (options)"
+
+      option :purge,
+        :short => "-P",
+        :long => "--purge",
+        :boolean => true,
+        :default => false,
+        :description => "Destroy corresponding node and client on the Chef Server, in addition to destroying the EC2 node itself.  Assumes node and client have the same name as the server (if not, add the '--node-name' option)."
+
+      option :chef_node_name,
+        :short => "-N NAME",
+        :long => "--node-name NAME",
+        :description => "The name of the node and client to delete, if it differs from the server name.  Only has meaning when used with the '--purge' option."
+
+      # Extracted from Chef::Knife.delete_object, because it has a
+      # confirmation step built in... By specifying the '--purge'
+      # flag (and also explicitly confirming the server destruction!)
+      # the user is already making their intent known.  It is not
+      # necessary to make them confirm two more times.
+      def destroy_item(klass, name, type_name)
+        begin
+          object = klass.load(name)
+          object.destroy
+          ui.warn("Deleted #{type_name} #{name}")
+        rescue Net::HTTPServerException
+          ui.warn("Could not find a #{type_name} named #{name} to delete!")
+        end
+      end
+
+      def run
+
+        validate!
+
+        @name_args.each do |name|
+
+          begin
+            server = connection.roles.find(name)
+
+            puts "\n"
+            msg_pair('Service', server.hostedservicename)
+            msg_pair('Deployment', server.deployname)
+            msg_pair('Role', server.name)
+            msg_pair('Size', server.size)
+            msg_pair('SSH Ip Address', server.sshipaddress)
+            msg_pair('SSH Port', server.sshport)
+
+            puts "\n"
+            confirm("Do you really want to delete this server")
+
+            connection.roles.delete(name)
+
+            puts "\n"
+            ui.warn("Deleted server #{server.name}")
+
+            if config[:purge]
+              thing_to_delete = config[:chef_node_name] || name
+              destroy_item(Chef::Node, thing_to_delete, "node")
+              destroy_item(Chef::ApiClient, thing_to_delete, "client")
+            else
+              ui.warn("Corresponding node and client for the #{name} server were not deleted and remain registered with the Chef Server")
+            end
+
+          rescue NoMethodError
+            ui.error("Could not locate server '#{name}'.  Please verify it was provisioned.")
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/azure_server_describe.rb

@@ -0,0 +1,85 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2010-2011 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 File.expand_path('../azure_base', __FILE__)
+
+class Chef
+  class Knife
+    class AzureServerDescribe < Knife
+
+      include Knife::AzureBase
+
+      banner "knife azure server describe ROLE [ROLE]"
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        @name_args.each do |name|
+          role = connection.roles.find name
+          puts ''
+          if (role)
+            details = Array.new
+            details << ui.color('Role name', :bold, :blue)
+            details << role.name
+            details << ui.color('Status', :bold, :blue)
+            details << role.status
+            details << ui.color('Size', :bold, :blue)
+            details << role.size
+            details << ui.color('Hosted service name', :bold, :blue)
+            details << role.hostedservicename
+            details << ui.color('Deployment name', :bold, :blue)
+            details << role.deployname
+            details << ui.color('Host name', :bold, :blue)
+            details << role.hostname
+            details << ui.color('SSH', :bold, :blue)
+            details << role.sshipaddress + ':' + role.sshport
+            puts ui.list(details, :columns_across, 2)
+            if role.tcpports.length > 0 || role.udpports.length > 0
+              details.clear
+              details << ui.color('Ports open', :bold, :blue)
+              details << ui.color('Local port', :bold, :blue)
+              details << ui.color('IP', :bold, :blue)
+              details << ui.color('Public port', :bold, :blue)
+              if role.tcpports.length > 0
+                role.tcpports.each do |port|
+                  details << 'tcp'
+                  details << port['LocalPort']
+                  details << port['Vip']
+                  details << port['PublicPort']
+                end
+              end
+              if role.udpports.length > 0
+                role.udpports.each do |port|
+                  details << 'udp'
+                  details << port['LocalPort']
+                  details << port['Vip']
+                  details << port['PublicPort']
+                end
+              end
+              puts ui.list(details, :columns_across, 4)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/azure_server_list.rb

@@ -0,0 +1,70 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2010-2011 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 File.expand_path('../azure_base', __FILE__)
+
+class Chef
+  class Knife
+    class AzureServerList < Knife
+
+      include Knife::AzureBase
+
+      banner "knife azure server list (options)"
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        server_list = [
+          ui.color('Status', :bold),
+          ui.color('Service', :bold),
+          ui.color('Deployment', :bold),
+          ui.color('Role', :bold),
+          ui.color('Host', :bold),
+          ui.color('SSH IP', :bold),
+          ui.color('SSH Port', :bold)
+        ]
+        items = connection.roles.all
+        items.each do |server|
+          server_list << begin
+                           state = server.status.to_s.downcase
+                           case state
+                           when 'shutting-down','terminated','stopping','stopped'
+                             ui.color(state, :red)
+                           when 'pending'
+                             ui.color(state, :yellow)
+                           else
+                             ui.color('ready', :green)
+                           end
+          end
+          server_list << server.hostedservicename.to_s
+          server_list << server.deployname.to_s
+          server_list << server.name.to_s
+          server_list << server.hostname.to_s
+          server_list << server.sshipaddress.to_s
+          server_list << server.sshport.to_s
+        end
+        puts ''
+        puts ui.list(server_list, :columns_across, 7)
+      end
+    end
+  end
+end

data/lib/knife-azure/version.rb

@@ -0,0 +1,7 @@
+module Knife
+  module Azure
+    VERSION = "1.0.0"
+    MAJOR, MINOR, TINY = VERSION.split('.')
+  end
+end
+