knife-profitbricks 1.0.0

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new
+
+task :default => :spec
+task :test => :spec

data/knife-profitbricks.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'knife-profitbricks/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "knife-profitbricks"
+  spec.version       = Knife::ProfitBricks::VERSION
+  spec.authors       = ["Ethan Devenport"]
+  spec.email         = ["ethand@stackpointcloud.com"]
+  spec.summary       = %q{Chef Knife plugin for ProfitBricks platform}
+  spec.description   = %q{Official Chef Knife plugin for ProfitBricks platform using REST API}
+  spec.homepage      = "https://github.com/profitbricks/knife-profitbricks"
+  spec.license       = "Apache"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "profitbricks-sdk-ruby", "~> 1.0"
+  spec.add_runtime_dependency "chef", "~> 12"
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake", "~> 10.4"
+  spec.add_development_dependency "rspec", "~> 3.2"
+end

data/lib/chef/knife/profitbricks_base.rb

@@ -0,0 +1,56 @@
+require 'chef/knife'
+
+class Chef
+  class Knife
+    module ProfitbricksBase
+      def self.included(includer)
+        includer.class_eval do
+          deps do
+            require 'profitbricks'
+          end
+
+          option :profitbricks_username,
+            short: '-u USERNAME',
+            long: '--username USERNAME',
+            description: 'Your ProfitBricks username',
+            proc: proc { |username| Chef::Config[:knife][:profitbricks_username] = username }
+
+          option :profitbricks_password,
+            short: '-p PASSWORD',
+            long: '--password PASSWORD',
+            description: 'Your ProfitBricks password',
+            proc: proc { |password| Chef::Config[:knife][:profitbricks_password] = password }
+
+          option :profitbricks_url,
+            short: '-U URL',
+            long: '--url URL',
+            description: 'The ProfitBricks API URL',
+            proc: proc { |url| Chef::Config[:knife][:profitbricks_url] = url }
+        end
+      end
+
+      def connection
+        ProfitBricks.configure do |config|
+          config.username = Chef::Config[:knife][:profitbricks_username]
+          config.password = Chef::Config[:knife][:profitbricks_password]
+          config.url = Chef::Config[:knife][:profitbricks_url]
+          config.debug = Chef::Config[:knife][:profitbricks_debug] || false
+          config.global_classes = false
+        end
+      end
+
+      def msg_pair(label, value, color = :cyan)
+        if value && !value.to_s.empty?
+          puts "#{ui.color(label, color)}: #{value}"
+        end
+      end
+    end
+  end
+end
+
+# compact method will remove nil values from Hash
+class Hash
+  def compact
+    delete_if { |_k, v| v.nil? }
+  end
+end

data/lib/chef/knife/profitbricks_datacenter_create.rb

@@ -0,0 +1,51 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksDatacenterCreate < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks datacenter create (options)'
+
+      option :name,
+             short: '-n NAME',
+             long: '--name NAME',
+             description: 'Name of the data center'
+
+      option :description,
+             short: '-D DESCRIPTION',
+             long: '--description DESCRIPTION',
+             description: 'Description of the data center'
+
+      option :location,
+             short: '-l LOCATION',
+             long: '--location LOCATION',
+             description: 'Location of the data center',
+             proc: proc { |location| Chef::Config[:knife][:location] = location }
+
+      def run
+        $stdout.sync = true
+
+        print "#{ui.color('Creating data center...', :magenta)}"
+
+        connection
+        datacenter = ProfitBricks::Datacenter.create(
+          name: config[:name],
+          description: config[:description],
+          location: config[:location]
+        )
+
+        dot = ui.color('.', :magenta)
+        datacenter.wait_for { print dot; ready? }
+        datacenter.reload
+
+        puts "\n"
+        puts "#{ui.color('ID', :cyan)}: #{datacenter.id}"
+        puts "#{ui.color('Name', :cyan)}: #{datacenter.properties['name']}"
+        puts "#{ui.color('Description', :cyan)}: #{datacenter.properties['description']}"
+        puts "#{ui.color('Location', :cyan)}: #{datacenter.properties['location']}"
+        puts 'done'
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_datacenter_delete.rb

@@ -0,0 +1,44 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksDatacenterDelete < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks datacenter delete DATACENTER_ID' \
+             ' [DATACENTER_ID] (options)'
+
+      option :purge,
+             short: '-p',
+             long: '--purge',
+             boolean: true,
+             default: false,
+             description: 'Recursively delete the datacenter and all' \
+                          ' corresponding resources under the datacenter.'
+
+      def run
+        connection
+        @name_args.each do |datacenter_id|
+          begin
+            datacenter = ProfitBricks::Datacenter.get(datacenter_id)
+          rescue Excon::Errors::NotFound
+            ui.error("Data center ID #{datacenter_id} not found. Skipping.")
+            next
+          end
+
+          msg_pair('ID', datacenter.id)
+          msg_pair('Name', datacenter.properties['name'])
+          msg_pair('Description', datacenter.properties['description'])
+          msg_pair('Location', datacenter.properties['location'])
+          msg_pair('Version', datacenter.properties['version'])
+
+          puts "\n"
+          confirm('Do you really want to delete this data center')
+
+          datacenter.delete
+          ui.warn("Deleted data center #{datacenter.id}")
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_datacenter_list.rb

@@ -0,0 +1,32 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksDatacenterList < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks datacenter list'
+
+      def run
+        $stdout.sync = true
+        datacenter_list = [
+          ui.color('ID', :bold),
+          ui.color('Name', :bold),
+          ui.color('Description', :bold),
+          ui.color('Location', :bold),
+          ui.color('Version', :bold)
+        ]
+        connection
+        ProfitBricks::Datacenter.list.each do |datacenter|
+          datacenter_list << datacenter.id
+          datacenter_list << datacenter.properties['name']
+          datacenter_list << (datacenter.properties['description'] == nil ? '' : datacenter.properties['description'])
+          datacenter_list << datacenter.properties['location']
+          datacenter_list << datacenter.properties['version'].to_s
+        end
+
+        puts ui.list(datacenter_list, :uneven_columns_across, 5)
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_firewall_create.rb

@@ -0,0 +1,121 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksFirewallCreate < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks firewall create (options)'
+
+      option :datacenter_id,
+             short: '-D DATACENTER_UUID',
+             long: '--datacenter-id DATACENTER_UUID',
+             description: 'UUID of the data center',
+             proc: proc { |datacenter_id| Chef::Config[:knife][:datacenter_id] = datacenter_id }
+
+      option :server_id,
+             short: '-S SERVER_UUID',
+             long: '--server-id SERVER_UUID',
+             description: 'UUID of the server'
+
+      option :nic_id,
+             short: '-N NIC_UUID',
+             long: '--nic-id NIC_UUID',
+             description: 'UUID of the NIC'
+
+      option :name,
+             short: '-n NAME',
+             long: '--name NAME',
+             description: 'Name of the NIC'
+
+      option :protocol,
+             short: '-P PROTOCOL',
+             long: '--protocol PROTOCOL',
+             default: 'TCP',
+             description: 'The protocol of the firewall rule (TCP, UDP, ICMP,' \
+                          ' ANY)'
+
+      option :sourcemac,
+             short: '-m MAC',
+             long: '--source-mac MAC',
+             description: 'Only traffic originating from the respective MAC' \
+                          ' address is allowed'
+
+      option :sourceip,
+             short: '-I IP',
+             long: '--source-ip IP',
+             description: 'Only traffic originating from the respective IPv4' \
+                          ' address is allowed; null allows all source IPs'
+
+      option :targetip,
+             long: '--target-ip IP',
+             description: 'In case the target NIC has multiple IP addresses,' \
+                          ' only traffic directed to the respective IP' \
+                          ' address of the NIC is allowed; null value allows' \
+                          ' all target IPs'
+
+      option :portrangestart,
+             short: '-p PORT',
+             long: '--port-range-start PORT',
+             description: 'Defines the start range of the allowed port(s)'
+
+      option :portrangeend,
+             short: '-t PORT',
+             long: '--port-range-end PORT',
+             description: 'Defines the end range of the allowed port(s)'
+
+      option :icmptype,
+             long: '--icmp-type INT',
+             description: 'Defines the allowed type (from 0 to 254) if the' \
+                          ' protocol ICMP is chosen; null allows all types'
+
+      option :icmpcode,
+             long: '--icmp-code INT',
+             description: 'Defines the allowed code (from 0 to 254) if the' \
+                          ' protocol ICMP is chosen; null allows all codes'
+
+      def run
+        $stdout.sync = true
+
+        print "#{ui.color('Creating firewall...', :magenta)}"
+
+        params = {
+          name: config[:name],
+          protocol: config[:protocol],
+          sourceMac: config[:sourcemac],
+          sourceIp: config[:sourceip],
+          targetIp: config[:targetip],
+          portRangeStart: config[:portrangestart],
+          portRangeEnd: config[:portrangeend],
+          icmpType: config[:icmptype],
+          icmpCode: config[:icmpcode]
+        }
+
+        connection
+        firewall = ProfitBricks::Firewall.create(
+          config[:datacenter_id],
+          config[:server_id],
+          config[:nic_id],
+          params.compact
+        )
+
+        dot = ui.color('.', :magenta)
+        firewall.wait_for { print dot; ready? }
+        firewall.reload
+
+        puts "\n"
+        puts "#{ui.color('ID', :cyan)}: #{firewall.id}"
+        puts "#{ui.color('Name', :cyan)}: #{firewall.properties['name']}"
+        puts "#{ui.color('Protocol', :cyan)}: #{firewall.properties['protocol']}"
+        puts "#{ui.color('Source MAC', :cyan)}: #{firewall.properties['sourceMac']}"
+        puts "#{ui.color('Source IP', :cyan)}: #{firewall.properties['sourceIp']}"
+        puts "#{ui.color('Target IP', :cyan)}: #{firewall.properties['targetIp']}"
+        puts "#{ui.color('Port Range Start', :cyan)}: #{firewall.properties['portRangeStart']}"
+        puts "#{ui.color('Port Range End', :cyan)}: #{firewall.properties['portRangeEnd']}"
+        puts "#{ui.color('ICMP Type', :cyan)}: #{firewall.properties['icmpType']}"
+        puts "#{ui.color('ICMP Code', :cyan)}: #{firewall.properties['icmpCode']}"
+        puts 'done'
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_firewall_delete.rb

@@ -0,0 +1,58 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksFirewallDelete < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks firewall delete FIREWALL_UUID [FIREWALL_UUID] (options)'
+
+      option :datacenter_id,
+             short: '-D DATACENTER_UUID',
+             long: '--datacenter-id DATACENTER_UUID',
+             description: 'The UUID of the data center',
+             proc: proc { |datacenter_id| Chef::Config[:knife][:datacenter_id] = datacenter_id }
+
+      option :server_id,
+             short: '-S SERVER_UUID',
+             long: '--server-id SERVER_UUID',
+             description: 'The UUID of the server'
+
+      option :nic_id,
+             short: '-N NIC_UUID',
+             long: '--nic-id NIC_UUID',
+             description: 'UUID of the NIC'
+
+      def run
+        connection
+        @name_args.each do |firewall_id|
+          begin
+            firewall = ProfitBricks::Firewall.get(config[:datacenter_id],
+                                                  config[:server_id],
+                                                  config[:nic_id],
+                                                  firewall_id)
+          rescue Excon::Errors::NotFound
+            ui.error("Firewall ID #{firewall_id} not found. Skipping.")
+            next
+          end
+
+          msg_pair('ID', firewall.id)
+          msg_pair('Name', firewall.properties['name'])
+          msg_pair('Protocol', firewall.properties['protocol'])
+          msg_pair('Source MAC', firewall.properties['sourceMac'])
+          msg_pair('Source IP', firewall.properties['sourceIp'])
+          msg_pair('Target IP', firewall.properties['targetIp'])
+          msg_pair('Port Range Start', firewall.properties['portRangeStart'])
+          msg_pair('Port Range End', firewall.properties['portRangeEnd'])
+          msg_pair('ICMP Type', firewall.properties['icmpType'])
+          msg_pair('ICMP Code', firewall.properties['icmpCode'])
+
+          confirm('Do you really want to delete this firewall rule')
+
+          firewall.delete
+          ui.warn("Deleted firewall rule #{firewall.id}")
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/profitbricks_firewall_list.rb

@@ -0,0 +1,59 @@
+require 'chef/knife/profitbricks_base'
+
+class Chef
+  class Knife
+    class ProfitbricksFirewallList < Knife
+      include Knife::ProfitbricksBase
+
+      banner 'knife profitbricks firewall list (options)'
+
+      option :datacenter_id,
+             short: '-D DATACENTER_UUID',
+             long: '--datacenter-id DATACENTER_UUID',
+             description: 'UUID of the data center',
+             proc: proc { |datacenter_id| Chef::Config[:knife][:datacenter_id] = datacenter_id }
+
+      option :server_id,
+             short: '-S SERVER_UUID',
+             long: '--server-id SERVER_UUID',
+             description: 'The UUID of the server'
+
+      option :nic_id,
+             short: '-N NIC_UUID',
+             long: '--nic-id NIC_UUID',
+             description: 'UUID of the NIC'
+
+      def run
+        $stdout.sync = true
+        firewall_list = [
+          ui.color('ID', :bold),
+          ui.color('Name', :bold),
+          ui.color('Protocol', :bold),
+          ui.color('Source MAC', :bold),
+          ui.color('Source IP', :bold),
+          ui.color('Target IP', :bold),
+          ui.color('Port Range Start', :bold),
+          ui.color('Port Range End', :bold),
+          ui.color('ICMP Type', :bold),
+          ui.color('ICMP CODE', :bold)
+        ]
+        connection
+
+        ProfitBricks::Firewall.list(config[:datacenter_id], config[:server_id], config[:nic_id]).each do |firewall|
+          firewall_list << firewall.id
+          firewall_list << firewall.properties['name']
+          firewall_list << firewall.properties['protocol'].to_s
+          firewall_list << firewall.properties['sourceMac'].to_s
+          firewall_list << firewall.properties['sourceIp'].to_s
+          firewall_list << firewall.properties['targetIp'].to_s
+          firewall_list << firewall.properties['portRangeStart'].to_s
+          firewall_list << firewall.properties['portRangeEnd'].to_s
+          firewall_list << firewall.properties['icmpType'].to_s
+          firewall_list << firewall.properties['icmpCode'].to_s
+        end
+
+        puts ui.list(firewall_list, :uneven_columns_across, 10)
+      end
+    end
+  end
+end