knife-cloudstack 0.0.13 → 0.0.14

data/lib/chef/knife/cs_user_list.rb

@@ -0,0 +1,93 @@
+#
+# Author:: Sander Botman (<sbotman@schubergphilis.com>)
+# Copyright:: Copyright (c) 2013 Sander Botman.
+# 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/cs_base'
+require 'chef/knife/cs_baselist'
+      
+module KnifeCloudstack
+  class CsUserList < Chef::Knife
+
+    include Chef::Knife::KnifeCloudstackBase
+    include Chef::Knife::KnifeCloudstackBaseList
+
+    deps do
+      require 'knife-cloudstack/connection'
+      Chef::Knife.load_deps
+    end
+
+    banner "knife cs user list (options)"
+
+    option :listall,
+           :long => "--listall",
+           :description => "List all users",
+           :boolean => true
+
+    option :keyword,
+           :long => "--keyword KEY",
+           :description => "List by keyword"
+
+    def run
+      validate_base_options
+
+      if locate_config_value(:fields)
+        object_list = []
+        locate_config_value(:fields).split(',').each { |n| object_list << ui.color(("#{n}").strip, :bold) }
+      else
+        object_list = [
+          ui.color('Account', :bold),
+          ui.color('Type', :bold),
+          ui.color('State', :bold),
+          ui.color('Domain', :bold),
+          ui.color('Username', :bold),
+          ui.color('First', :bold),
+          ui.color('Last', :bold)
+        ]
+      end
+
+      columns = object_list.count
+      object_list = [] if locate_config_value(:noheader)
+
+      connection_result = connection.list_object(
+        "listUsers",
+        "user",
+        locate_config_value(:filter),
+        locate_config_value(:listall),
+        locate_config_value(:keyword)
+      )
+
+      output_format(connection_result)
+
+      connection_result.each do |r|
+       if locate_config_value(:fields)
+          locate_config_value(:fields).downcase.split(',').each { |n| object_list << ((r[("#{n}").strip]).to_s || 'N/A') }
+        else
+          object_list << r['account'].to_s 
+          object_list << r['accounttype'].to_s
+          object_list << r['state'].to_s
+          object_list << r['domain'].to_s
+          object_list << r['username'].to_s
+          object_list << r['firstname'].to_s
+          object_list << r['lastname'].to_s
+        end
+      end
+      puts ui.list(object_list, :uneven_columns_across, columns)
+      list_object_fields(connection_result) if locate_config_value(:fieldlist)
+    end
+
+  end
+end

data/lib/chef/knife/cs_volume_list.rb

@@ -0,0 +1,94 @@
+#
+# Author:: Sander Botman (<sbotman@schubergphilis.com>)
+# Copyright:: Copyright (c) 2013 Sander Botman.
+# 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/cs_base'
+require 'chef/knife/cs_baselist'
+
+module KnifeCloudstack
+  class CsVolumeList < Chef::Knife
+
+    include Chef::Knife::KnifeCloudstackBase
+    include Chef::Knife::KnifeCloudstackBaseList
+
+    deps do
+      require 'knife-cloudstack/connection'
+      Chef::Knife.load_deps
+    end
+
+    banner "knife cs volume list (options)"
+
+    option :listall,
+           :long => "--listall",
+           :description => "List all volumes",
+           :boolean => true
+
+    option :name,
+           :long => "--name NAME",
+           :description => "Specify volume name to list"
+
+    option :keyword,
+           :long => "--keyword KEY",
+           :description => "List by keyword"
+
+    def run
+      validate_base_options
+
+      object_list = []
+      if locate_config_value(:fields)
+        locate_config_value(:fields).split(',').each { |n| object_list << ui.color(("#{n}").strip, :bold) }
+      else
+        object_list << ui.color('Name', :bold)
+        object_list << ui.color('Account', :bold) unless locate_config_value(:cloudstack_project)
+        object_list << ui.color('Domain', :bold)
+        object_list << ui.color('State', :bold)
+        object_list << ui.color('VMName', :bold)
+        object_list << ui.color('VMState', :bold)
+      end
+
+      columns = object_list.count
+      object_list = [] if locate_config_value(:noheader)
+
+      connection_result = connection.list_object(
+        "listVolumes",
+        "volume",
+        locate_config_value(:filter),
+        locate_config_value(:listall),
+        locate_config_value(:keyword),
+        locate_config_value(:name)
+      )
+
+      output_format(connection_result)
+
+      connection_result.each do |r|
+       if locate_config_value(:fields)
+          locate_config_value(:fields).downcase.split(',').each { |n| object_list << ((r[("#{n}").strip]).to_s || 'N/A') }
+        else
+          object_list << r['name'].to_s
+          object_list << r['account'].to_s unless locate_config_value(:cloudstack_project)
+          object_list << r['domain'].to_s
+          object_list << r['state'].to_s
+          object_list << r['vmname'].to_s
+          object_list << r['vmstate'].to_s
+        end
+      end
+      puts ui.list(object_list, :uneven_columns_across, columns)
+      list_object_fields(connection_result) if locate_config_value(:fieldlist)
+    end
+
+  end
+end

data/lib/chef/knife/cs_zone_list.rb

@@ -1,6 +1,8 @@
 #
 # Author:: Ryan Holmes (<rholmes@edmunds.com>)
+# Author:: Sander Botman (<sbotman@schubergphilis.com>)
 # Copyright:: Copyright (c) 2011 Edmunds, Inc.
+# Copyright:: Copyright (c) 2013 Sander Botman.
 # License:: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,62 +18,79 @@
 # limitations under the License.
 #
 
-require 'chef/knife'
+require 'chef/knife/cs_base'
+require 'chef/knife/cs_baselist'
 
 module KnifeCloudstack
   class CsZoneList < Chef::Knife
 
+    include Chef::Knife::KnifeCloudstackBase
+    include Chef::Knife::KnifeCloudstackBaseList
+
     deps do
       require 'knife-cloudstack/connection'
+      require 'chef/knife'
+      Chef::Knife.load_deps
     end
 
     banner "knife cs zone list (options)"
 
-    option :cloudstack_url,
-           :short => "-U URL",
-           :long => "--cloudstack-url URL",
-           :description => "The CloudStack endpoint URL",
-           :proc => Proc.new { |url| Chef::Config[:knife][:cloudstack_url] = url }
-
-    option :cloudstack_api_key,
-           :short => "-A KEY",
-           :long => "--cloudstack-api-key KEY",
-           :description => "Your CloudStack API key",
-           :proc => Proc.new { |key| Chef::Config[:knife][:cloudstack_api_key] = key }
+    option :keyword,
+           :long => "--keyword KEY",
+           :description => "List by keyword"
 
-    option :cloudstack_secret_key,
-           :short => "-K SECRET",
-           :long => "--cloudstack-secret-key SECRET",
-           :description => "Your CloudStack secret key",
-           :proc => Proc.new { |key| Chef::Config[:knife][:cloudstack_secret_key] = key }
+    option :index,
+           :long => "--index",
+           :description => "Add index numbers to the output",
+           :boolean => true
 
     def run
+      validate_base_options
 
-      connection = CloudstackClient::Connection.new(
-          locate_config_value(:cloudstack_url),
-          locate_config_value(:cloudstack_api_key),
-          locate_config_value(:cloudstack_secret_key)
-      )
+      object_list = []
+      object_list << ui.color('Index', :bold) if locate_config_value(:index)
 
-      zone_list = [
-          ui.color('Name', :bold),
+      if locate_config_value(:fields)
+        object_list = []
+        locate_config_value(:fields).split(',').each { |n| object_list << ui.color(("#{n}").strip, :bold) }
+      else
+        [
+ 	  ui.color('Name', :bold),
           ui.color('Network Type', :bold),
           ui.color('Security Groups', :bold)
-      ]
-
-      zones = connection.list_zones
-      zones.each do |z|
-        zone_list << z['name']
-        zone_list << z['networktype']
-        zone_list << z['securitygroupsenabled'].to_s
+        ].each { |field| object_list << field }
       end
-      puts ui.list(zone_list, :columns_across, 3)
 
-    end
+      columns = object_list.count
+      object_list = [] if locate_config_value(:noheader)
 
-    def locate_config_value(key)
-      key = key.to_sym
-      Chef::Config[:knife][key] || config[key]
+      connection_result = connection.list_object(
+        "listZones",
+        "zone", 
+        locate_config_value(:filter),
+	false,
+        locate_config_value(:keyword)
+      )
+
+      output_format(connection_result)
+
+      index_num = 0
+      connection_result.each do |r|
+        if locate_config_value(:index)
+          index_num += 1
+          object_list << index_num.to_s
+        end
+
+        if locate_config_value(:fields)
+          locate_config_value(:fields).downcase.split(',').each { |n| object_list << ((r[("#{n}").strip]).to_s || 'N/A') }
+        else
+          object_list << r['name'].to_s
+          object_list << r['networktype'].to_s
+          object_list << r['securitygroupsenabled'].to_s
+        end
+      end
+      puts ui.list(object_list, :uneven_columns_across, columns)
+      list_object_fields(connection_result) if locate_config_value(:fieldlist)
     end
 
   end

data/lib/knife-cloudstack/connection.rb

@@ -1,6 +1,8 @@
 #
 # Author:: Ryan Holmes (<rholmes@edmunds.com>)
 # Author:: KC Braunschweig (<kcbraunschweig@gmail.com>)
+# Author:: Sander Botman (<sbotman@schubergphilis.com>)
+# Author:: Frank Breedijk (<fbreedijk@schubergphilis.com>)
 # Copyright:: Copyright (c) 2011 Edmunds, Inc.
 # License:: Apache License, Version 2.0
 #
@@ -24,6 +26,8 @@ require 'uri'
 require 'cgi'
 require 'net/http'
 require 'json'
+require 'highline/import'
+require 'knife-cloudstack/string_to_regexp'
 
 module CloudstackClient
   class Connection
@@ -85,7 +89,7 @@ module CloudstackClient
       if ip_addr
         return ip_addr['ipaddress']
       end
-      nic['ipaddress']
+      nic['ipaddress'] || []
     end
 
     ##
@@ -112,6 +116,31 @@ module CloudstackClient
       end
     end
 
+    ## 
+    # List all the objects based on the command that is specified.
+    
+    def list_object(command, json_result, filter=nil, listall=nil, keyword=nil, name=nil, templatefilter=nil)
+      params = {
+          'command' => command
+      }
+      params['listall'] = true if listall || name || keyword unless listall == false
+      params['keyword'] = keyword if keyword
+      params['name'] = name if name
+
+      if templatefilter
+        template = 'featured'
+        template = templatefilter.downcase if ["featured","self","self-executable","executable","community"].include?(templatefilter.downcase)
+        params['templateFilter'] = template
+      end
+
+      json = send_request(params)
+      Chef::Log.debug("JSON (list_object) result: #{json}")
+
+      result = json["#{json_result}"] || []
+      result = data_filter(result, filter) if filter
+      result
+    end
+
     ##
     # Lists all the servers in your account.
 
@@ -119,9 +148,6 @@ module CloudstackClient
       params = {
           'command' => 'listVirtualMachines'
       }
-      # if @project_id
-      #   params['projectId'] = @project_id
-      # end
       json = send_request(params)
       json['virtualmachine'] || []
     end
@@ -129,7 +155,7 @@ module CloudstackClient
     ##
     # Deploys a new server using the specified parameters.
 
-    def create_server(host_name, service_name, template_name, zone_name=nil, network_names=[])
+    def create_server(host_name, service_name, template_name, zone_name=nil, network_names=[], extra_params)
 
       if host_name then
         if get_server(host_name) then
@@ -184,9 +210,8 @@ module CloudstackClient
           'zoneId' => zone['id'],
           'networkids' => network_ids.join(',')
       }
-      # if @project_id
-      #   params['projectId'] = @project_id
-      # end
+
+      params.merge!(extra_params) if extra_params
 
       params['name'] = host_name if host_name
 
@@ -365,7 +390,8 @@ module CloudstackClient
     #Fetch project with the specified name
     def get_project(name)
       params = {
-        'command' => 'listProjects'
+        'command' => 'listProjects',
+	'listall' => true
       }
 
       json = send_request(params)
@@ -380,6 +406,22 @@ module CloudstackClient
       nil
     end
 
+    ##
+    # Filter data on regex or just on string
+
+    def data_filter(data, filters)
+      filters.split(',').each do |filter|
+        field = filter.split(':').first.strip.downcase
+        search = filter.split(':').last.strip
+        if search =~ /^\/.*\/?/
+          data = data.find_all { |k| k["#{field}"].to_s =~ search.to_regexp } if field && search
+        else
+          data = data.find_all { |k| k["#{field}"].to_s == "#{search}" } if field && search
+        end
+      end
+      data
+    end
+
 
     ##
     # Finds the network with the specified name.
@@ -388,9 +430,6 @@ module CloudstackClient
       params = {
           'command' => 'listNetworks'
       }
-      # if @project_id
-      #   params['projectId'] = @project_id
-      # end
       json = send_request(params)
 
       networks = json['network']
@@ -414,9 +453,6 @@ module CloudstackClient
           'isDefault' => true,
           'zoneid' => zone
       }
-      # if @project_id
-      #   params['projectId'] = @project_id
-      # end
       json = send_request(params)
 
       networks = json['network']
@@ -513,7 +549,7 @@ module CloudstackClient
       params = { 'command' => 'listPublicIpAddresses'}
 
       json = send_request(params)
-      return json['publicipaddress']
+      return json['publicipaddress'] || []
     end
     ##
     # Acquires and associates a public IP to an account.
@@ -564,6 +600,17 @@ module CloudstackClient
       send_async_request(params)
     end
 
+    def create_firewall_rule(ipaddress_id, protocol, start_port, end_port, cidr_list)
+      params = {
+        'command' => 'createFirewallRule',
+        'ipaddressId' => ipaddress_id,
+        'protocol' => protocol,
+        'startport' =>  start_port,
+        'endport' => end_port,
+        'cidrlist' => cidr_list
+      }
+      send_async_request(params)
+    end
 
     ##
     # Disassociates an ip address from the account.
@@ -634,15 +681,15 @@ module CloudstackClient
 
       params_arr = []
       params.sort.each { |elem|
-        params_arr << elem[0].to_s + '=' + elem[1].to_s
+        params_arr << elem[0].to_s + '=' + CGI.escape(elem[1].to_s).gsub('+', '%20').gsub(' ','%20')
       }
       data = params_arr.join('&')
-      encoded_data = URI.encode(data.downcase).gsub('+', '%20').gsub(',', '%2c')
-      signature = OpenSSL::HMAC.digest('sha1', @secret_key, encoded_data)
+      signature = OpenSSL::HMAC.digest('sha1', @secret_key, data.downcase)
       signature = Base64.encode64(signature).chomp
       signature = CGI.escape(signature)
 
       url = "#{@api_url}?#{data}&signature=#{signature}"
+      Chef::Log.debug("URL: #{url}")
       uri = URI.parse(url)
       http = Net::HTTP.new(uri.host, uri.port)
       http.use_ssl = @use_ssl
@@ -651,9 +698,15 @@ module CloudstackClient
       response = http.request(request)
 
       if !response.is_a?(Net::HTTPOK) then
-        puts "Error #{response.code}: #{response.message}"
-        puts JSON.pretty_generate(JSON.parse(response.body))
-        puts "URL: #{url}"
+        case response.code
+        when "432"
+          puts "\n" 
+          puts "Error #{response.code}: Your account does not have the right to execute this command or the command does not exist."
+        else
+          puts "Error #{response.code}: #{response.message}"
+          puts JSON.pretty_generate(JSON.parse(response.body))
+          puts "URL: #{url}"
+        end
         exit 1
       end