knife-cloudstack-fog 0.3.3 → 0.3.4

data/lib/chef/knife/cloudstack_publicip_list.rb

@@ -1,57 +1,57 @@
-# Author:: Takashi Kanai (<anikundesu@gmail.com>)
-# Copyright:: Copyright (c) 2012 IDC Frontier 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/cloudstack_base'
-
-class Chef
-  class Knife
-    class CloudstackPublicipList < Knife
-
-      include Knife::CloudstackBase
-
-      banner "knife cloudstack publicip list"
-
-      def run
-        $stdout.sync = true
-
-        validate!
-
-        publicip_list = [
-          ui.color('ID', :bold),
-          ui.color('ipaddress', :bold),
-          ui.color('isSourceNAT', :bold),
-          ui.color('isStaticNAT', :bold),
-          ui.color('VirtualMachineDisplayName', :bold)
-        ]
-        response = connection.list_public_ip_addresses['listpublicipaddressesresponse']
-          if publicips = response['publicipaddress']
-            publicips.each do |publicip|
-              publicip_list << publicip['id'].to_s
-              publicip_list << publicip['ipaddress'].to_s
-              publicip_list << publicip['issourcenat'].to_s
-              publicip_list << publicip['isstaticnat'].to_s
-              publicip_list << publicip['virtualmachinedisplayname'].to_s
-            end
-          end
-        puts ui.list(publicip_list, :columns_across, 5)
-
-      end
-
-    end
-  end
-end
+# Author:: Takashi Kanai (<anikundesu@gmail.com>)
+# Copyright:: Copyright (c) 2012 IDC Frontier 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/cloudstack_base'
+
+class Chef
+  class Knife
+    class CloudstackPublicipList < Knife
+
+      include Knife::CloudstackBase
+
+      banner "knife cloudstack publicip list"
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        publicip_list = [
+          ui.color('ID', :bold),
+          ui.color('ipaddress', :bold),
+          ui.color('isSourceNAT', :bold),
+          ui.color('isStaticNAT', :bold),
+          ui.color('VirtualMachineDisplayName', :bold)
+        ]
+        response = connection.list_public_ip_addresses['listpublicipaddressesresponse']
+          if publicips = response['publicipaddress']
+            publicips.each do |publicip|
+              publicip_list << publicip['id'].to_s
+              publicip_list << publicip['ipaddress'].to_s
+              publicip_list << publicip['issourcenat'].to_s
+              publicip_list << publicip['isstaticnat'].to_s
+              publicip_list << publicip['virtualmachinedisplayname'].to_s
+            end
+          end
+        puts ui.list(publicip_list, :columns_across, 5)
+
+      end
+
+    end
+  end
+end

data/lib/chef/knife/cloudstack_securitygroup_list.rb

@@ -1,124 +1,124 @@
-# Author:: Jeff Moody (<jmoody@datapipe.com>)
-# Copyright:: Copyright (c) 2012 Datapipe
-# 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/cloudstack_base'
-
-class Chef
-  class Knife
-    class CloudstackSecuritygroupList < Knife
-
-      include Knife::CloudstackBase
-
-      banner "knife cloudstack securitygroup list (options)"
-      option :groupid,
-             :short => "-g GROUPID",
-             :long => "--groupid GROUPID",
-             :description => "List the rules contained within a Security Group, specified by ID",
-             :default => 'none'
-      option :groupname,
-             :short => "-G GROUPNAME",
-             :long => "--groupname GROUPNAME",
-             :description => "List the rules contained within a Security Group, specified by name.",
-             :default => 'none'
-
-      def sg_details_list(securitygroup_list, securitygroup_details, groups, options={})
-        temp = groups
-        if groupid = options[:groupid]
-          temp.reject!{|g| g['id'] != groupid.to_i}
-        end
-        if groupname = options[:groupname]
-          temp.reject!{|g| g['name'] != groupname}
-        end
-        
-        temp.each do |securitygroup|
-          securitygroup_list << securitygroup['id'].to_s
-          securitygroup_list << securitygroup['name'].to_s
-          securitygroup_list << securitygroup['description'].to_s
-          if securitygroup['ingressrule'].nil?
-            securitygroup_details << ' '
-          else
-            securitygroup['ingressrule'].each do |ingressrule|
-              rule_details = []
-              securitygroup_details << ingressrule['protocol'].to_s
-              securitygroup_details << ingressrule['startport'].to_s
-              securitygroup_details << ingressrule['endport'].to_s
-              if ingressrule['cidr'].nil?
-                rule_details << ingressrule['securitygroupname'].to_s
-                rule_details << ingressrule['account'].to_s
-              else
-                rule_details << ingressrule['cidr'].to_s
-              end
-              securitygroup_details << rule_details.join(", ")
-            end
-          end
-        end
-      end
-            
-      def run
-        $stdout.sync = true
-
-        validate!
-        groupid = locate_config_value(:groupid)
-        groupname = locate_config_value(:groupname)
-
-        if (groupid == 'none' and groupname == 'none')
-          securitygroup_list = [
-            ui.color('ID', :bold),
-            ui.color('Name', :bold),
-            ui.color('Description', :bold)
-          ]
-          response = connection.list_security_groups['listsecuritygroupsresponse']
-            if securitygroups = response['securitygroup']
-              securitygroups.each do |securitygroup|
-                securitygroup_list << securitygroup['id'].to_s
-                securitygroup_list << securitygroup['name'].to_s
-                securitygroup_list << securitygroup['description'].to_s
-              end
-            end
-          puts ui.list(securitygroup_list, :columns_across, 3)
-        else
-          securitygroup_details = [
-            ui.color('Protocol', :bold),
-            ui.color('Start Port', :bold),
-            ui.color('End Port', :bold),
-            ui.color('Restricted To', :bold)
-          ]
-          securitygroup_list = [
-            ui.color('ID', :bold),
-            ui.color('Name', :bold),
-            ui.color('Description', :bold)
-            ]
-                        
-            if response = connection.list_security_groups['listsecuritygroupsresponse']
-              if securitygroups = response['securitygroup']
-                filters = {}
-                filters[:groupid] = groupid unless groupid == 'none'
-                filters[:groupname] = groupname unless groupname == 'none'
-                sg_details_list(securitygroup_list, securitygroup_details, securitygroups, filters)
-                
-                puts ui.list(securitygroup_list, :columns_across, 3)
-                puts ui.list(securitygroup_details, :columns_across, 4)
-              end
-            end
-            
-        end
-      end
-       
-    end
-  end
-end
+# Author:: Jeff Moody (<jmoody@datapipe.com>)
+# Copyright:: Copyright (c) 2012 Datapipe
+# 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/cloudstack_base'
+
+class Chef
+  class Knife
+    class CloudstackSecuritygroupList < Knife
+
+      include Knife::CloudstackBase
+
+      banner "knife cloudstack securitygroup list (options)"
+      option :groupid,
+             :short => "-g GROUPID",
+             :long => "--groupid GROUPID",
+             :description => "List the rules contained within a Security Group, specified by ID",
+             :default => 'none'
+      option :groupname,
+             :short => "-G GROUPNAME",
+             :long => "--groupname GROUPNAME",
+             :description => "List the rules contained within a Security Group, specified by name.",
+             :default => 'none'
+
+      def sg_details_list(securitygroup_list, securitygroup_details, groups, options={})
+        temp = groups
+        if groupid = options[:groupid]
+          temp.reject!{|g| g['id'] != groupid.to_i}
+        end
+        if groupname = options[:groupname]
+          temp.reject!{|g| g['name'] != groupname}
+        end
+
+        temp.each do |securitygroup|
+          securitygroup_list << securitygroup['id'].to_s
+          securitygroup_list << securitygroup['name'].to_s
+          securitygroup_list << securitygroup['description'].to_s
+          if securitygroup['ingressrule'].nil?
+            securitygroup_details << ' '
+          else
+            securitygroup['ingressrule'].each do |ingressrule|
+              rule_details = []
+              securitygroup_details << ingressrule['protocol'].to_s
+              securitygroup_details << ingressrule['startport'].to_s
+              securitygroup_details << ingressrule['endport'].to_s
+              if ingressrule['cidr'].nil?
+                rule_details << ingressrule['securitygroupname'].to_s
+                rule_details << ingressrule['account'].to_s
+              else
+                rule_details << ingressrule['cidr'].to_s
+              end
+              securitygroup_details << rule_details.join(", ")
+            end
+          end
+        end
+      end
+
+      def run
+        $stdout.sync = true
+
+        validate!
+        groupid = locate_config_value(:groupid)
+        groupname = locate_config_value(:groupname)
+
+        if (groupid == 'none' and groupname == 'none')
+          securitygroup_list = [
+            ui.color('ID', :bold),
+            ui.color('Name', :bold),
+            ui.color('Description', :bold)
+          ]
+          response = connection.list_security_groups['listsecuritygroupsresponse']
+            if securitygroups = response['securitygroup']
+              securitygroups.each do |securitygroup|
+                securitygroup_list << securitygroup['id'].to_s
+                securitygroup_list << securitygroup['name'].to_s
+                securitygroup_list << securitygroup['description'].to_s
+              end
+            end
+          puts ui.list(securitygroup_list, :columns_across, 3)
+        else
+          securitygroup_details = [
+            ui.color('Protocol', :bold),
+            ui.color('Start Port', :bold),
+            ui.color('End Port', :bold),
+            ui.color('Restricted To', :bold)
+          ]
+          securitygroup_list = [
+            ui.color('ID', :bold),
+            ui.color('Name', :bold),
+            ui.color('Description', :bold)
+            ]
+
+            if response = connection.list_security_groups['listsecuritygroupsresponse']
+              if securitygroups = response['securitygroup']
+                filters = {}
+                filters[:groupid] = groupid unless groupid == 'none'
+                filters[:groupname] = groupname unless groupname == 'none'
+                sg_details_list(securitygroup_list, securitygroup_details, securitygroups, filters)
+
+                puts ui.list(securitygroup_list, :columns_across, 3)
+                puts ui.list(securitygroup_details, :columns_across, 4)
+              end
+            end
+
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/cloudstack_server_create.rb

@@ -1,320 +1,320 @@
-# Author:: Jeff Moody (<jmoody@datapipe.com>), Takashi Kanai (<anikundesu@gmail.com>)
-# Copyright:: Copyright (c) 2012 Datapipe,  Copyright (c) 2012 IDC Frontier 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/bootstrap'
-Chef::Knife::Bootstrap.load_deps
-require 'socket'
-require 'net/ssh/multi'
-require 'chef/json_compat'
-require 'chef/knife/cloudstack_base'
-
-class Chef
-  class Knife
-    class CloudstackServerCreate < Knife
-      
-      include Knife::CloudstackBase
-      
-      banner "knife cloudstack server create -s SERVICEID -t TEMPLATEID -z ZONEID (options)"
-      
-      option  :cloudstack_serviceid,
-              :short => "-s SERVICEID",
-              :long => "--serviceid SERVICEID",
-              :description => "The CloudStack service offering ID."
-
-      option  :cloudstack_templateid,
-              :short => "-t TEMPLATEID",
-              :long => "--templateid TEMPLATEID",
-              :description => "The CloudStack template ID for the server."
-
-      option  :cloudstack_zoneid,
-              :short => "-z ZONEID",
-              :long => "--zoneid ZONE",
-              :description => "The CloudStack zone ID for the server."
-
-      option  :cloudstack_networkids,
-              :short => "-w NETWORKIDS",
-              :long => "--networkids NETWORKIDS",
-              :description => "Comma separated list of CloudStack network IDs.",
-              :proc => lambda { |n| n.split(/[\s,]+/) },
-              :default => []
-
-      option  :cloudstack_groupids,
-              :short => "-g SECURITYGROUPIDS",
-              :long => "--groupids SECURITYGROUPIDS",
-              :description => "Comma separated list of CloudStack Security Group IDs.",
-              :proc => lambda { |n| n.split(/[\s,]+/) },
-              :default => []
-
-      option  :cloudstack_groupnames,
-              :short => "-G SECURITYGROUPNAMES",
-              :long => "--groupnames SECURITYGROUPNAMES",
-              :description => "Comma separated list of CloudStack Security Group names. Each group name must be encapuslated in quotes if it contains whitespace.",
-              :proc => lambda { |n| n.split(/[\s,]+/) },
-              :default => []
-
-      option  :distro,
-              :short => "-d DISTRO",
-              :long => "--distro DISTRO",
-              :description => "Bootstrap a distro using a template; default is 'chef-full'",
-              :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  :ssh_user,
-              :short => "-x USERNAME",
-              :long => "--ssh-user USERNAME",
-              :description => "The ssh username",
-              :default => 'root'
-
-      option  :ssh_password,
-              :short => "-P PASSWORD",
-              :long => "--ssh-password PASSWORD",
-              :description => "The ssh password"
-      
-      option  :identity_file,        
-              :short => "-i PRIVATE_KEY_FILE",
-              :long => "--identity-file PRIVATE_KEY_FILE",
-              :description => "The Private key file for authenticating SSH session. --keypair option is also needed."
-                      
-      option  :server_name,
-              :short => "-N NAME",
-              :long => "--display-name NAME",
-              :description => "The instance display name"
-      
-      option  :host_name,
-              :short => "-H NAME",
-              :long => "--hostname NAME",
-              :description => "The instance host name"
-      
-      option  :keypair,
-              :short => "-k KEYPAIR",
-              :long => "--keypair KEYPAIR",
-              :description => "The CloudStack Key Pair to use for SSH key authentication."
-              
-      option  :diskoffering,
-              :short => "-D DISKOFFERINGID",
-              :long => "--diskoffering DISKOFFERINGID",
-              :description => "Specifies either the Disk Offering ID for the ROOT disk for an ISO template, or a DATA disk."
-
-      option  :size,
-              :short => "-Z SIZE",
-              :long => "--size SIZE",
-              :description => "Specifies the arbitrary Disk Size for DATADISK volume in GB. Must be passed with custom size Disk Offering ID."
-        
-      def bootstrap_for_node(host, user, password)
-        Chef::Log.debug("Bootstrap host: #{host}")
-        Chef::Log.debug("Bootstrap user: #{user}")
-        Chef::Log.debug("Bootstrap pass: #{password}")
-        bootstrap = Chef::Knife::Bootstrap.new
-        bootstrap.name_args = host
-        bootstrap.config[:run_list] = config[:run_list]
-        bootstrap.config[:ssh_user] = user
-        bootstrap.config[:ssh_password] = password
-        bootstrap.config[:identity_file] = locate_config_value(:identity_file)
-        bootstrap.config[:chef_node_name] = config[:server_name] if config[:server_name]
-        bootstrap.config[:prerelease] = config[:prerelease]
-        bootstrap.config[:bootstrap_version] = locate_config_value(:bootstrap_version)
-        bootstrap.config[:distro] = locate_config_value(:distro)
-        bootstrap.config[:use_sudo] = true
-        bootstrap.config[:template_file] = locate_config_value(:template_file)
-        bootstrap.config[:environment] = config[:environment]
-        # may be needed for vpc_mode
-        bootstrap.config[:no_host_key_verify] = config[:no_host_key_verify]
-        bootstrap
-      end
-      
-      def tcp_test_ssh(hostname)
-        print("#{ui.color(".", :magenta)}")
-        tcp_socket = TCPSocket.new(hostname, 22)
-        readable = IO.select([tcp_socket], nil, nil, 5)
-        if readable
-          Chef::Log.debug("\nsshd accepting connections on #{hostname}, banner is #{tcp_socket.gets}\n")
-          yield
-          true
-        else
-          false
-        end
-        
-        rescue Errno::ETIMEDOUT
-          false
-        rescue Errno::EPERM
-          false
-        rescue Errno::ECONNREFUSED
-          sleep 2
-          false
-        rescue Errno::EHOSTUNREACH
-          sleep 2
-          false
-        rescue Errno::ENETUNREACH
-          sleep 30
-          false
-        ensure
-        tcp_socket && tcp_socket.close
-      end
-
-      def run
-        $stdout.sync = true
-        
-        options = {}
-
-        options['zoneid'] = locate_config_value(:cloudstack_zoneid)
-        options['templateid'] = locate_config_value(:cloudstack_templateid)
-
-        if locate_config_value(:cloudstack_serviceid) != nil
-          options['serviceofferingid'] = locate_config_value(:cloudstack_serviceid)
-        end
-        
-        if locate_config_value(:server_name) != nil
-          options['displayname'] = locate_config_value(:server_name)
-        end
-
-        if locate_config_value(:host_name) != nil
-          options['name'] = locate_config_value(:host_name)
-        end
-        
-        network_ids = []
-        if locate_config_value(:cloudstack_networkids) != []
-          cs_networkids = locate_config_value(:cloudstack_networkids)
-          cs_networkids.each do |id|
-            network_ids.push(id)
-          end
-          options['networkids'] = network_ids
-        end
-        
-        security_groups = []
-        if locate_config_value(:cloudstack_groupids) != []
-          cs_groupids = locate_config_value(:cloudstack_groupids)
-          cs_groupids.each do |id|
-            security_groups.push(id)
-          end
-          options['securitygroupids'] = security_groups
-        elsif locate_config_value(:cloudstack_groupnames) != []
-          cs_groupnames = locate_config_value(:cloudstack_groupnames)
-          cs_groupnames.each do |name|
-            security_groups.push(name)
-          end
-          options['securitygroupnames'] = security_groups
-        end
-        
-        if locate_config_value(:keypair) != nil
-          options['keypair'] = locate_config_value(:keypair)
-        end
-        
-        if locate_config_value(:diskoffering) != nil
-          options['diskofferingid'] = locate_config_value(:diskoffering)
-        end
-        
-        if locate_config_value(:size) != nil
-          options['size'] = locate_config_value(:size)
-        end
-        
-        Chef::Log.debug("Options: #{options} \n")
-
-        server = connection.deploy_virtual_machine(options)
-        jobid = server['deployvirtualmachineresponse'].fetch('jobid')
-
-        server_start = connection.query_async_job_result('jobid'=>jobid)
-
-        Chef::Log.debug("Job ID: #{jobid} \n")
-
-        print "#{ui.color("Waiting for server", :magenta)}"
-        while server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 0
-          print "#{ui.color(".", :magenta)}"
-          sleep(15)
-          server_start = connection.query_async_job_result('jobid'=>jobid)
-          Chef::Log.debug("Server_Start: #{server_start} \n")
-        end
-        puts "\n\n"
-
-        if server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 2
-          errortext = server_start['queryasyncjobresultresponse'].fetch('jobresult').fetch('errortext')
-          puts "#{ui.color("ERROR! Job failed with #{errortext}", :red)}"
-        end
-
-        if server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 1 
-
-          Chef::Log.debug("Job ID: #{jobid} \n")
-          Chef::Log.debug("Options: #{options} \n")
-          server_start = connection.query_async_job_result('jobid'=>jobid)
-          Chef::Log.debug("Server_Start: #{server_start} \n")
-          
-          server_info = server_start['queryasyncjobresultresponse']['jobresult']['virtualmachine']
-          
-          server_name = server_info['displayname']
-          server_id = server_info['name']
-          server_serviceoffering = server_info['serviceofferingname']
-          server_template = server_info['templatename']
-          if server_info['password'] != nil
-            ssh_password = server_info['password']
-          else
-            ssh_password = locate_config_value(:ssh_password)
-          end
-          
-          ssh_user = locate_config_value(:ssh_user)
-          
-          public_ip = nil
-          
-          if server_info['nic'].size > 0
-            public_ip = server_info['nic'].first['ipaddress']
-          end
-          
-          puts "\n\n"
-          puts "#{ui.color("Name", :cyan)}: #{server_name}"
-          puts "#{ui.color("Public IP", :cyan)}: #{public_ip}"
-          puts "#{ui.color("Username", :cyan)}: #{ssh_user}"
-          puts "#{ui.color("Password", :cyan)}: #{ssh_password}"
-          
-          print "\n#{ui.color("Waiting for sshd", :magenta)}"
-          
-          print("#{ui.color(".", :magenta)}") until tcp_test_ssh(public_ip) { sleep @initial_sleep_delay ||= 10; puts("done") }
-
-          puts("#{ui.color("Waiting for password/keys to sync.", :magenta)}")
-          sleep 15
-          
-          bootstrap_for_node(public_ip, ssh_user, ssh_password).run
-
-          Chef::Log.debug("#{server_info}")
-          
-          puts "\n"
-          puts "#{ui.color("Instance Name", :green)}: #{server_name}"
-          puts "#{ui.color("Instance ID", :green)}: #{server_id}"
-          puts "#{ui.color("Service Offering", :green)}: #{server_serviceoffering}"
-          puts "#{ui.color("Template", :green)}: #{server_template}"
-          puts "#{ui.color("Public IP Address", :green)}: #{public_ip}"
-          puts "#{ui.color("User", :green)}: #{ssh_user}"
-          puts "#{ui.color("Password", :green)}: #{ssh_password}"
-          puts "#{ui.color("Environment", :green)}: #{config[:environment] || '_default'}"
-          puts "#{ui.color("Run List", :green)}: #{config[:run_list].join(', ')}" 
-        end
- 
-      end
-      
-    end
-  end
-end
+# Author:: Jeff Moody (<jmoody@datapipe.com>), Takashi Kanai (<anikundesu@gmail.com>)
+# Copyright:: Copyright (c) 2012 Datapipe,  Copyright (c) 2012 IDC Frontier 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/bootstrap'
+Chef::Knife::Bootstrap.load_deps
+require 'socket'
+require 'net/ssh/multi'
+require 'chef/json_compat'
+require 'chef/knife/cloudstack_base'
+
+class Chef
+  class Knife
+    class CloudstackServerCreate < Knife
+
+      include Knife::CloudstackBase
+
+      banner "knife cloudstack server create -s SERVICEID -t TEMPLATEID -z ZONEID (options)"
+
+      option  :cloudstack_serviceid,
+              :short => "-s SERVICEID",
+              :long => "--serviceid SERVICEID",
+              :description => "The CloudStack service offering ID."
+
+      option  :cloudstack_templateid,
+              :short => "-t TEMPLATEID",
+              :long => "--templateid TEMPLATEID",
+              :description => "The CloudStack template ID for the server."
+
+      option  :cloudstack_zoneid,
+              :short => "-z ZONEID",
+              :long => "--zoneid ZONE",
+              :description => "The CloudStack zone ID for the server."
+
+      option  :cloudstack_networkids,
+              :short => "-w NETWORKIDS",
+              :long => "--networkids NETWORKIDS",
+              :description => "Comma separated list of CloudStack network IDs.",
+              :proc => lambda { |n| n.split(/[\s,]+/) },
+              :default => []
+
+      option  :cloudstack_groupids,
+              :short => "-g SECURITYGROUPIDS",
+              :long => "--groupids SECURITYGROUPIDS",
+              :description => "Comma separated list of CloudStack Security Group IDs.",
+              :proc => lambda { |n| n.split(/[\s,]+/) },
+              :default => []
+
+      option  :cloudstack_groupnames,
+              :short => "-G SECURITYGROUPNAMES",
+              :long => "--groupnames SECURITYGROUPNAMES",
+              :description => "Comma separated list of CloudStack Security Group names. Each group name must be encapuslated in quotes if it contains whitespace.",
+              :proc => lambda { |n| n.split(/[\s,]+/) },
+              :default => []
+
+      option  :distro,
+              :short => "-d DISTRO",
+              :long => "--distro DISTRO",
+              :description => "Bootstrap a distro using a template; default is 'chef-full'",
+              :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  :ssh_user,
+              :short => "-x USERNAME",
+              :long => "--ssh-user USERNAME",
+              :description => "The ssh username",
+              :default => 'root'
+
+      option  :ssh_password,
+              :short => "-P PASSWORD",
+              :long => "--ssh-password PASSWORD",
+              :description => "The ssh password"
+
+      option  :identity_file,
+              :short => "-i PRIVATE_KEY_FILE",
+              :long => "--identity-file PRIVATE_KEY_FILE",
+              :description => "The Private key file for authenticating SSH session. --keypair option is also needed."
+
+      option  :server_name,
+              :short => "-N NAME",
+              :long => "--display-name NAME",
+              :description => "The instance display name"
+
+      option  :host_name,
+              :short => "-H NAME",
+              :long => "--hostname NAME",
+              :description => "The instance host name"
+
+      option  :keypair,
+              :short => "-k KEYPAIR",
+              :long => "--keypair KEYPAIR",
+              :description => "The CloudStack Key Pair to use for SSH key authentication."
+
+      option  :diskoffering,
+              :short => "-D DISKOFFERINGID",
+              :long => "--diskoffering DISKOFFERINGID",
+              :description => "Specifies either the Disk Offering ID for the ROOT disk for an ISO template, or a DATA disk."
+
+      option  :size,
+              :short => "-Z SIZE",
+              :long => "--size SIZE",
+              :description => "Specifies the arbitrary Disk Size for DATADISK volume in GB. Must be passed with custom size Disk Offering ID."
+
+      def bootstrap_for_node(host, user, password)
+        Chef::Log.debug("Bootstrap host: #{host}")
+        Chef::Log.debug("Bootstrap user: #{user}")
+        Chef::Log.debug("Bootstrap pass: #{password}")
+        bootstrap = Chef::Knife::Bootstrap.new
+        bootstrap.name_args = host
+        bootstrap.config[:run_list] = config[:run_list]
+        bootstrap.config[:ssh_user] = user
+        bootstrap.config[:ssh_password] = password
+        bootstrap.config[:identity_file] = locate_config_value(:identity_file)
+        bootstrap.config[:chef_node_name] = config[:server_name] if config[:server_name]
+        bootstrap.config[:prerelease] = config[:prerelease]
+        bootstrap.config[:bootstrap_version] = locate_config_value(:bootstrap_version)
+        bootstrap.config[:distro] = locate_config_value(:distro)
+        bootstrap.config[:use_sudo] = true
+        bootstrap.config[:template_file] = locate_config_value(:template_file)
+        bootstrap.config[:environment] = config[:environment]
+        # may be needed for vpc_mode
+        bootstrap.config[:no_host_key_verify] = config[:no_host_key_verify]
+        bootstrap
+      end
+
+      def tcp_test_ssh(hostname)
+        print("#{ui.color(".", :magenta)}")
+        tcp_socket = TCPSocket.new(hostname, 22)
+        readable = IO.select([tcp_socket], nil, nil, 5)
+        if readable
+          Chef::Log.debug("\nsshd accepting connections on #{hostname}, banner is #{tcp_socket.gets}\n")
+          yield
+          true
+        else
+          false
+        end
+
+        rescue Errno::ETIMEDOUT
+          false
+        rescue Errno::EPERM
+          false
+        rescue Errno::ECONNREFUSED
+          sleep 2
+          false
+        rescue Errno::EHOSTUNREACH
+          sleep 2
+          false
+        rescue Errno::ENETUNREACH
+          sleep 30
+          false
+        ensure
+        tcp_socket && tcp_socket.close
+      end
+
+      def run
+        $stdout.sync = true
+
+        options = {}
+
+        options['zoneid'] = locate_config_value(:cloudstack_zoneid)
+        options['templateid'] = locate_config_value(:cloudstack_templateid)
+
+        if locate_config_value(:cloudstack_serviceid) != nil
+          options['serviceofferingid'] = locate_config_value(:cloudstack_serviceid)
+        end
+
+        if locate_config_value(:server_name) != nil
+          options['displayname'] = locate_config_value(:server_name)
+        end
+
+        if locate_config_value(:host_name) != nil
+          options['name'] = locate_config_value(:host_name)
+        end
+
+        network_ids = []
+        if locate_config_value(:cloudstack_networkids) != []
+          cs_networkids = locate_config_value(:cloudstack_networkids)
+          cs_networkids.each do |id|
+            network_ids.push(id)
+          end
+          options['networkids'] = network_ids
+        end
+
+        security_groups = []
+        if locate_config_value(:cloudstack_groupids) != []
+          cs_groupids = locate_config_value(:cloudstack_groupids)
+          cs_groupids.each do |id|
+            security_groups.push(id)
+          end
+          options['securitygroupids'] = security_groups
+        elsif locate_config_value(:cloudstack_groupnames) != []
+          cs_groupnames = locate_config_value(:cloudstack_groupnames)
+          cs_groupnames.each do |name|
+            security_groups.push(name)
+          end
+          options['securitygroupnames'] = security_groups
+        end
+
+        if locate_config_value(:keypair) != nil
+          options['keypair'] = locate_config_value(:keypair)
+        end
+
+        if locate_config_value(:diskoffering) != nil
+          options['diskofferingid'] = locate_config_value(:diskoffering)
+        end
+
+        if locate_config_value(:size) != nil
+          options['size'] = locate_config_value(:size)
+        end
+
+        Chef::Log.debug("Options: #{options} \n")
+
+        server = connection.deploy_virtual_machine(options)
+        jobid = server['deployvirtualmachineresponse'].fetch('jobid')
+
+        server_start = connection.query_async_job_result('jobid'=>jobid)
+
+        Chef::Log.debug("Job ID: #{jobid} \n")
+
+        print "#{ui.color("Waiting for server", :magenta)}"
+        while server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 0
+          print "#{ui.color(".", :magenta)}"
+          sleep(15)
+          server_start = connection.query_async_job_result('jobid'=>jobid)
+          Chef::Log.debug("Server_Start: #{server_start} \n")
+        end
+        puts "\n\n"
+
+        if server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 2
+          errortext = server_start['queryasyncjobresultresponse'].fetch('jobresult').fetch('errortext')
+          puts "#{ui.color("ERROR! Job failed with #{errortext}", :red)}"
+        end
+
+        if server_start['queryasyncjobresultresponse'].fetch('jobstatus') == 1
+
+          Chef::Log.debug("Job ID: #{jobid} \n")
+          Chef::Log.debug("Options: #{options} \n")
+          server_start = connection.query_async_job_result('jobid'=>jobid)
+          Chef::Log.debug("Server_Start: #{server_start} \n")
+
+          server_info = server_start['queryasyncjobresultresponse']['jobresult']['virtualmachine']
+
+          server_name = server_info['displayname']
+          server_id = server_info['name']
+          server_serviceoffering = server_info['serviceofferingname']
+          server_template = server_info['templatename']
+          if server_info['password'] != nil
+            ssh_password = server_info['password']
+          else
+            ssh_password = locate_config_value(:ssh_password)
+          end
+
+          ssh_user = locate_config_value(:ssh_user)
+
+          public_ip = nil
+
+          if server_info['nic'].size > 0
+            public_ip = server_info['nic'].first['ipaddress']
+          end
+
+          puts "\n\n"
+          puts "#{ui.color("Name", :cyan)}: #{server_name}"
+          puts "#{ui.color("Public IP", :cyan)}: #{public_ip}"
+          puts "#{ui.color("Username", :cyan)}: #{ssh_user}"
+          puts "#{ui.color("Password", :cyan)}: #{ssh_password}"
+
+          print "\n#{ui.color("Waiting for sshd", :magenta)}"
+
+          print("#{ui.color(".", :magenta)}") until tcp_test_ssh(public_ip) { sleep @initial_sleep_delay ||= 10; puts("done") }
+
+          puts("#{ui.color("Waiting for password/keys to sync.", :magenta)}")
+          sleep 15
+
+          bootstrap_for_node(public_ip, ssh_user, ssh_password).run
+
+          Chef::Log.debug("#{server_info}")
+
+          puts "\n"
+          puts "#{ui.color("Instance Name", :green)}: #{server_name}"
+          puts "#{ui.color("Instance ID", :green)}: #{server_id}"
+          puts "#{ui.color("Service Offering", :green)}: #{server_serviceoffering}"
+          puts "#{ui.color("Template", :green)}: #{server_template}"
+          puts "#{ui.color("Public IP Address", :green)}: #{public_ip}"
+          puts "#{ui.color("User", :green)}: #{ssh_user}"
+          puts "#{ui.color("Password", :green)}: #{ssh_password}"
+          puts "#{ui.color("Environment", :green)}: #{config[:environment] || '_default'}"
+          puts "#{ui.color("Run List", :green)}: #{config[:run_list].join(', ')}"
+        end
+
+      end
+
+    end
+  end
+end