knife-profitbricks 1.2.1 → 2.0.1

data/lib/chef/knife/profitbricks_ipfailover_add.rb

@@ -0,0 +1,60 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksFailoverAdd < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks ipfailover add (options)'
+
+      option :datacenter_id,
+             short: '-D DATACENTER_ID',
+             long: '--datacenter-id DATACENTER_ID',
+             description: 'Name of the data center',
+             proc: proc { |datacenter_id| Chef::Config[:knife][:datacenter_id] = datacenter_id }
+
+      option :lan_id,
+             short: '-l LAN_ID',
+             long: '--lan-id LAN_ID',
+             description: 'Lan ID'
+      option :ip,
+             short: '-i',
+             long: '--ip',
+             description: 'IP to be added to IP failover group'
+
+      option :nic_id,
+             short: '-n',
+             long: '--nic-id',
+             description: 'NIC to be added to IP failover group'
+
+      def run
+        $stdout.sync = true
+        validate_required_params(%i[datacenter_id lan_id ip nic_id], Chef::Config[:knife])
+
+        connection
+
+        lan = ProfitBricks::LAN.get(Chef::Config[:knife][:datacenter_id], Chef::Config[:knife][:lan_id])
+
+        failover_ips = lan.properties[:ipFailover]
+        failover_ips ||= []
+        ip_failover = {}
+        ip_failover['ip'] = Chef::Config[:knife][:ip]
+        ip_failover['nicUuid'] = Chef::Config[:knife][:nic_id]
+
+        failover_ips.push(ip_failover)
+
+        lan.update(ipFailover: failover_ips)
+        lan.wait_for { ready? }
+        lan.reload
+
+        puts "\n"
+        puts "#{ui.color('ID', :cyan)}: #{lan.id}"
+        puts "#{ui.color('Name', :cyan)}: #{lan.properties['name']}"
+        puts "#{ui.color('Public', :cyan)}: #{lan.properties['public']}"
+        puts "#{ui.color('IP Failover', :cyan)}: #{lan.properties['ipFailover']}"
+
+        puts 'done'
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_ipfailover_remove.rb

@@ -0,0 +1,60 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksFailoverRemove < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks ipfailover remove (options)'
+
+      option :datacenter_id,
+             short: '-D DATACENTER_ID',
+             long: '--datacenter-id DATACENTER_ID',
+             description: 'Name of the data center',
+             proc: proc { |datacenter_id| Chef::Config[:knife][:datacenter_id] = datacenter_id }
+
+      option :lan_id,
+             short: '-l LAN_ID',
+             long: '--lan-id LAN_ID',
+             description: 'Lan ID'
+      option :ip,
+             short: '-i',
+             long: '--ip',
+             description: 'IP to be removed from IP failover group'
+
+      option :nic_id,
+             short: '-n',
+             long: '--nic-id',
+             description: 'NIC to be removed from IP failover group'
+
+      def run
+        $stdout.sync = true
+        validate_required_params(%i[datacenter_id lan_id ip nic_id], Chef::Config[:knife])
+
+        connection
+
+        lan = ProfitBricks::LAN.get(Chef::Config[:knife][:datacenter_id], Chef::Config[:knife][:lan_id])
+
+        ipfailover = lan.properties['ipFailover']
+
+        ipfailover.each_with_index do |value, index|
+          if value['nicUuid'] == Chef::Config[:knife][:nic_id] && value['ip'] == Chef::Config[:knife][:ip]
+            ipfailover.delete_at(index)
+          end
+        end
+
+        lan.update(ipFailover: ipfailover)
+        lan.wait_for { ready? }
+        lan.reload
+
+        puts "\n"
+        puts "#{ui.color('ID', :cyan)}: #{lan.id}"
+        puts "#{ui.color('Name', :cyan)}: #{lan.properties['name']}"
+        puts "#{ui.color('Public', :cyan)}: #{lan.properties['public']}"
+        puts "#{ui.color('IP Failover', :cyan)}: #{lan.properties['ipFailover']}"
+
+        puts 'done'
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_lan_create.rb

@@ -26,16 +26,18 @@ class Chef
              description: 'Boolean indicating if the LAN faces the public ' \
                           'Internet or not; defaults to false'
 
+
       def run
         $stdout.sync = true
+        validate_required_params(%i(datacenter_id), Chef::Config[:knife])
 
         print "#{ui.color('Creating LAN...', :magenta)}"
 
         connection
         lan = ProfitBricks::LAN.create(
-          config[:datacenter_id],
-          name: config[:name],
-          public: config[:public]
+          Chef::Config[:knife][:datacenter_id],
+          name: Chef::Config[:knife][:name],
+          public: Chef::Config[:knife][:public]
         )
 
         dot = ui.color('.', :magenta)

data/lib/chef/knife/profitbricks_lan_list.rb

@@ -22,7 +22,7 @@ class Chef
         ]
         connection
 
-        ProfitBricks::LAN.list(config[:datacenter_id]).each do |lan|
+        ProfitBricks::LAN.list(Chef::Config[:knife][:datacenter_id]).each do |lan|
           lan_list << lan.id
           lan_list << lan.properties['name']
           lan_list << lan.properties['public'].to_s

data/lib/chef/knife/profitbricks_nic_create.rb

@@ -16,8 +16,7 @@ class Chef
       option :server_id,
              short: '-S SERVER_ID',
              long: '--server-id SERVER_ID',
-             description: 'Name of the server',
-             required: true
+             description: 'Name of the server'
 
       option :name,
              short: '-n NAME',
@@ -40,35 +39,34 @@ class Chef
       option :lan,
              short: '-l ID',
              long: '--lan ID',
-             description: 'The LAN ID the NIC will reside on; if the LAN ID does not exist it will be created',
-             required: true
+             description: 'The LAN ID the NIC will reside on; if the LAN ID does not exist it will be created'
 
       option :nat,
              long: '--nat',
              boolean: true | false,
-             description: 'Set to enable NAT on the NIC',
-             required: false
+             description: 'Set to enable NAT on the NIC'
 
       def run
         $stdout.sync = true
+        validate_required_params(%i(datacenter_id server_id lan), Chef::Config[:knife])
 
         print "#{ui.color('Creating nic...', :magenta)}"
 
         params = {
-          name: config[:name],
-          ips: config[:ips],
-          dhcp: config[:dhcp],
-          lan: config[:lan]
+          name: Chef::Config[:knife][:name],
+          ips: Chef::Config[:knife][:ips],
+          dhcp: Chef::Config[:knife][:dhcp],
+          lan: Chef::Config[:knife][:lan]
         }
 
-        if config[:nat]
-          params[:nat] = config[:nat]
+        if Chef::Config[:knife][:nat]
+          params[:nat] = Chef::Config[:knife][:nat]
         end
 
         connection
         nic = ProfitBricks::NIC.create(
-          config[:datacenter_id],
-          config[:server_id],
+          Chef::Config[:knife][:datacenter_id],
+          Chef::Config[:knife][:server_id],
           params.compact
         )
 

data/lib/chef/knife/profitbricks_nic_delete.rb

@@ -16,14 +16,15 @@ class Chef
       option :server_id,
              short: '-S SERVER_ID',
              long: '--server-id SERVER_ID',
-             description: 'The ID of the server assigned the NIC',
-             required: true
+             description: 'The ID of the server assigned the NIC'
 
       def run
         connection
+        validate_required_params(%i(datacenter_id server_id), Chef::Config[:knife])
+
         @name_args.each do |nic_id|
           begin
-            nic = ProfitBricks::NIC.get(config[:datacenter_id], config[:server_id], nic_id)
+            nic = ProfitBricks::NIC.get(Chef::Config[:knife][:datacenter_id], Chef::Config[:knife][:server_id], nic_id)
           rescue Excon::Errors::NotFound
             ui.error("NIC ID #{nic_id} not found. Skipping.")
             next

data/lib/chef/knife/profitbricks_nic_list.rb

@@ -16,11 +16,12 @@ class Chef
       option :server_id,
              short: '-S SERVER_ID',
              long: '--server-id SERVER_ID',
-             description: 'The ID of the server assigned the NIC',
-             required: true
+             description: 'The ID of the server assigned the NIC'
 
       def run
         $stdout.sync = true
+        validate_required_params(%i(datacenter_id server_id), Chef::Config[:knife])
+
         nic_list = [
           ui.color('ID', :bold),
           ui.color('Name', :bold),
@@ -31,7 +32,7 @@ class Chef
         ]
         connection
 
-        ProfitBricks::NIC.list(config[:datacenter_id], config[:server_id]).each do |nic|
+        ProfitBricks::NIC.list(Chef::Config[:knife][:datacenter_id], Chef::Config[:knife][:server_id]).each do |nic|
           nic_list << nic.id
           nic_list << nic.properties['name']
           nic_list << nic.properties['ips'].to_s

data/lib/chef/knife/profitbricks_server_create.rb

@@ -16,14 +16,12 @@ class Chef
       option :name,
              short: '-n NAME',
              long: '--name NAME',
-             description: 'Name of the server',
-             required: true
+             description: 'Name of the server'
 
       option :cores,
              short: '-C CORES',
              long: '--cores CORES',
-             description: 'The number of processor cores',
-             required: true
+             description: 'The number of processor cores'
 
       option :cpufamily,
              short: '-f CPU_FAMILY',
@@ -34,8 +32,7 @@ class Chef
       option :ram,
              short: '-r RAM',
              long: '--ram RAM',
-             description: 'The amount of RAM in MB',
-             required: true
+             description: 'The amount of RAM in MB'
 
       option :availabilityzone,
              short: '-a AVAILABILITY_ZONE',
@@ -53,27 +50,28 @@ class Chef
 
       def run
         $stdout.sync = true
+        validate_required_params(%i(datacenter_id name cores ram), Chef::Config[:knife])
 
         print "#{ui.color('Creating server...', :magenta)}"
         params = {
-          name: config[:name],
-          cores: config[:cores],
-          cpuFamily: config[:cpufamily],
-          ram: config[:ram],
-          availabilityZone: config[:availabilityzone]
+          name: Chef::Config[:knife][:name],
+          cores: Chef::Config[:knife][:cores],
+          cpuFamily: Chef::Config[:knife][:cpufamily],
+          ram: Chef::Config[:knife][:ram],
+          availabilityZone: Chef::Config[:knife][:availabilityzone]
         }
 
-        if config.key?(:bootcdrom)
-          params[:bootCdrom] = { id: config[:bootcdrom] }
+        if Chef::Config[:knife][:bootcdrom]
+          params[:bootCdrom] = { id: Chef::Config[:knife][:bootcdrom] }
         end
 
-        if config.key?(:bootvolume)
-          params[:bootVolume] = { id: config[:bootvolume] }
+        if Chef::Config[:knife][:bootvolume]
+          params[:bootVolume] = { id: Chef::Config[:knife][:bootvolume] }
         end
 
         connection
         server = ProfitBricks::Server.create(
-          config[:datacenter_id],
+          Chef::Config[:knife][:datacenter_id],
           params.compact
         )
 

data/lib/chef/knife/profitbricks_server_delete.rb

@@ -17,7 +17,7 @@ class Chef
         connection
         @name_args.each do |server_id|
           begin
-            server = ProfitBricks::Server.get(config[:datacenter_id], server_id)
+            server = ProfitBricks::Server.get(Chef::Config[:knife][:datacenter_id], server_id)
           rescue Excon::Errors::NotFound
             ui.error("Server ID #{server_id} not found. Skipping.")
             next

data/lib/chef/knife/profitbricks_server_list.rb

@@ -27,7 +27,7 @@ class Chef
         ]
         connection
 
-        ProfitBricks::Server.list(config[:datacenter_id]).each do |server|
+        ProfitBricks::Server.list(Chef::Config[:knife][:datacenter_id]).each do |server|
           server_list << server.id
           server_list << server.properties['name']
           server_list << server.properties['cores'].to_s

data/lib/chef/knife/profitbricks_server_reboot.rb

@@ -17,7 +17,7 @@ class Chef
         connection
         @name_args.each do |server_id|
           begin
-            server = ProfitBricks::Server.get(config[:datacenter_id], server_id)
+            server = ProfitBricks::Server.get(Chef::Config[:knife][:datacenter_id], server_id)
           rescue Excon::Errors::NotFound
             ui.error("Server ID #{server_id} not found. Skipping.")
             next

data/lib/chef/knife/profitbricks_server_start.rb

@@ -17,7 +17,7 @@ class Chef
         connection
         @name_args.each do |server_id|
           begin
-            server = ProfitBricks::Server.get(config[:datacenter_id], server_id)
+            server = ProfitBricks::Server.get(Chef::Config[:knife][:datacenter_id], server_id)
           rescue Excon::Errors::NotFound
             ui.error("Server ID #{server_id} not found. Skipping.")
             next

data/lib/chef/knife/profitbricks_server_stop.rb

@@ -17,7 +17,7 @@ class Chef
         connection
         @name_args.each do |server_id|
           begin
-            server = ProfitBricks::Server.get(config[:datacenter_id], server_id)
+            server = ProfitBricks::Server.get(Chef::Config[:knife][:datacenter_id], server_id)
           rescue Excon::Errors::NotFound
             ui.error("Server ID #{server_id} not found. Skipping.")
             next

data/lib/chef/knife/profitbricks_volume_attach.rb

@@ -16,21 +16,21 @@ class Chef
       option :server_id,
              short: '-S SERVER_ID',
              long: '--server-id SERVER_ID',
-             description: 'The ID of the server',
-             required: true
+             description: 'The ID of the server'
 
       def run
+        validate_required_params(%i(datacenter_id server_id), Chef::Config[:knife])
+
         connection
         @name_args.each do |volume_id|
-          volume = ProfitBricks::Volume.get(config[:datacenter_id], nil, volume_id)
-          volume
+          volume = ProfitBricks::Volume.get(Chef::Config[:knife][:datacenter_id], nil, volume_id)
 
           if volume.nil?
             ui.error("Volume ID #{volume_id} not found. Skipping.")
             next
           end
 
-          volume.attach(config[:server_id])
+          volume.attach(Chef::Config[:knife][:server_id])
           ui.msg("Volume #{volume_id} attached to server")
         end
       end

data/lib/chef/knife/profitbricks_volume_create.rb

@@ -21,8 +21,7 @@ class Chef
       option :size,
              short: '-S SIZE',
              long: '--size SIZE',
-             description: 'The size of the volume in GB',
-             required: true
+             description: 'The size of the volume in GB'
 
       option :bus,
              short: '-b BUS',
@@ -34,6 +33,10 @@ class Chef
              long: '--image ID',
              description: 'The image or snapshot ID'
 
+       option :imagealias,
+              long: '--image-alias IMAGE_ALIAS',
+              description: '(required) The image alias'
+
       option :imagepassword,
              short: '-P PASSWORD',
              long: '--image-password PASSWORD',
@@ -42,15 +45,14 @@ class Chef
       option :type,
              short: '-t TYPE',
              long: '--type TYPE',
-             description: 'The disk type (HDD OR SSD)',
-             required: true
+             description: 'The disk type (HDD OR SSD)'
 
       option :licencetype,
              short: '-l LICENCE',
              long: '--licence-type LICENCE',
              description: 'The licence type of the volume (LINUX, WINDOWS, UNKNOWN, OTHER)'
 
-      option :sshkeys, 
+      option :sshkeys,
              short: '-K SSHKEY[,SSHKEY,...]',
              long: '--ssh-keys SSHKEY1,SSHKEY2,...',
              description: 'A list of public SSH keys to include',
@@ -64,33 +66,51 @@ class Chef
 
       def run
         $stdout.sync = true
+        validate_required_params(%i(datacenter_id name type size), Chef::Config[:knife])
+
+        if !Chef::Config[:knife][:image] && !Chef::Config[:knife][:imagealias]
+          ui.error("Either '--image' or '--image-alias' parameter must be provided")
+          exit(1)
+        end
+
+        if !Chef::Config[:knife][:sshkeys] && !Chef::Config[:knife][:imagepassword]
+          ui.error("Either '--image-password' or '--ssh-keys' parameter must be provided")
+          exit(1)
+        end
 
         print "#{ui.color('Creating volume...', :magenta)}"
 
         params = {
-          name: config[:name],
-          size: config[:size],
-          bus: config[:bus] || 'VIRTIO',
-          image: config[:image],
-          type: config[:type],
-          licenceType: config[:licencetype],
+          name: Chef::Config[:knife][:name],
+          size: Chef::Config[:knife][:size],
+          bus: Chef::Config[:knife][:bus] || 'VIRTIO',
+          type: Chef::Config[:knife][:type],
+          licenceType: Chef::Config[:knife][:licencetype],
         }
 
-        if config[:sshkeys]
-          params[:sshKeys] = config[:sshkeys]
+        if Chef::Config[:knife][:image]
+          params[:image] = Chef::Config[:knife][:image]
+        end
+
+        if Chef::Config[:knife][:imagealias]
+          params[:imageAlias] = Chef::Config[:knife][:imagealias]
+        end
+
+        if Chef::Config[:knife][:sshkeys]
+          params[:sshKeys] = Chef::Config[:knife][:sshkeys]
         end
 
-        if config[:imagepassword]
-          params[:imagePassword] = config[:imagepassword]
+        if Chef::Config[:knife][:imagepassword]
+          params[:imagePassword] = Chef::Config[:knife][:imagepassword]
         end
 
-        if config[:volume_availability_zone]
-          params[:availabilityZone] = config[:volume_availability_zone]
+        if Chef::Config[:knife][:volume_availability_zone]
+          params[:availabilityZone] = Chef::Config[:knife][:volume_availability_zone]
         end
 
         connection
         volume = ProfitBricks::Volume.create(
-          config[:datacenter_id],
+          Chef::Config[:knife][:datacenter_id],
           params.compact
         )