knife-digital_ocean 2.2.0 → 2.3.0

data/lib/chef/knife/digital_ocean_domain_record_destroy.rb

@@ -21,14 +21,14 @@ class Chef
       banner 'knife digital_ocean domain record destroy (options)'
 
       option :domain,
-        :short       => '-D NAME',
-        :long        => '--domain-id NAME',
-        :description => 'The domain name'
+             short: '-D NAME',
+             long: '--domain-id NAME',
+             description: 'The domain name'
 
       option :record,
-        :short       => '-R ID',
-        :long        => '--record-id ID',
-        :description => 'The record id'
+             short: '-R ID',
+             long: '--record-id ID',
+             description: 'The record id'
 
       def run
         $stdout.sync = true
@@ -36,19 +36,18 @@ class Chef
         validate!
 
         unless locate_config_value(:domain)
-          ui.error("Domain cannot be empty. => -D <domain-name>")
+          ui.error('Domain cannot be empty. => -D <domain-name>')
           exit 1
         end
 
         unless locate_config_value(:record)
-          ui.error("Record cannot be empty. => -R <record-id>")
+          ui.error('Record cannot be empty. => -R <record-id>')
           exit 1
         end
 
         result = client.domain_records.delete for_domain: locate_config_value(:domain), id: locate_config_value(:record)
-        ui.error JSON.parse(result)['message'] rescue 'OK'
+        ui.info 'OK' if result == true or ui.error JSON.parse(result)['message']
       end
-
     end
   end
 end

data/lib/chef/knife/digital_ocean_domain_record_edit.rb

@@ -21,29 +21,29 @@ class Chef
       banner 'knife digital_ocean domain record edit (options)'
 
       option :domain,
-        :short       => '-D NAME',
-        :long        => '--domain-id NAME',
-        :description => 'The domain name'
+             short: '-D NAME',
+             long: '--domain-id NAME',
+             description: 'The domain name'
 
       option :record,
-        :short       => '-R ID',
-        :long        => '--record-id ID',
-        :description => 'The record id'
+             short: '-R ID',
+             long: '--record-id ID',
+             description: 'The record id'
 
       option :type,
-        :short       => '-T RECORD TYPE',
-        :long        => '--type RECORD TYPE',
-        :description => 'The type of record'
+             short: '-T RECORD TYPE',
+             long: '--type RECORD TYPE',
+             description: 'The type of record'
 
       option :name,
-        :short       => '-N RECORD NAME',
-        :long        => '--name RECORD NAME',
-        :description => 'The record name'
+             short: '-N RECORD NAME',
+             long: '--name RECORD NAME',
+             description: 'The record name'
 
       option :data,
-        :short       => '-a DATA',
-        :long        => '--data DATA',
-        :description => 'The record data'
+             short: '-a DATA',
+             long: '--data DATA',
+             description: 'The record data'
 
       def run
         $stdout.sync = true
@@ -51,27 +51,27 @@ class Chef
         validate!
 
         unless locate_config_value(:domain)
-          ui.error("Domain cannot be empty. => -D <domain-name>")
+          ui.error('Domain cannot be empty. => -D <domain-name>')
           exit 1
         end
 
         unless locate_config_value(:record)
-          ui.error("Record cannot be empty. => -R <record-id>")
+          ui.error('Record cannot be empty. => -R <record-id>')
           exit 1
         end
 
         unless locate_config_value(:type)
-          ui.error("Record type cannot be empty. => -T <record-type>")
+          ui.error('Record type cannot be empty. => -T <record-type>')
           exit 1
         end
 
         unless locate_config_value(:name)
-          ui.error("Record name cannot be empty. => -N <record-name>")
+          ui.error('Record name cannot be empty. => -N <record-name>')
           exit 1
         end
 
         unless locate_config_value(:data)
-          ui.error("Record data cannot be empty. => -d <data>")
+          ui.error('Record data cannot be empty. => -d <data>')
           exit 1
         end
 
@@ -81,7 +81,7 @@ class Chef
           data: locate_config_value(:data)
         )
         result = client.domain_records.update domain_record, for_domain: locate_config_value(:domain), id: locate_config_value(:record)
-        ui.error JSON.parse(result)['message'] rescue 'OK'
+        ui.info 'OK' if result == true or ui.error JSON.parse(result)['message']
       end
     end
   end

data/lib/chef/knife/digital_ocean_domain_record_list.rb

@@ -19,9 +19,9 @@ class Chef
       banner 'knife digital_ocean domain record list (options)'
 
       option :name,
-        :short       => '-D NAME',
-        :long        => '--domain-name NAME',
-        :description => 'The domain name'
+             short: '-D NAME',
+             long: '--domain-name NAME',
+             description: 'The domain name'
 
       def run
         $stdout.sync = true
@@ -29,7 +29,7 @@ class Chef
         validate!
 
         unless locate_config_value(:name)
-          ui.error("Domain Name cannot be empty. => -D <domain-name>")
+          ui.error('Domain Name cannot be empty. => -D <domain-name>')
           exit 1
         end
 
@@ -48,7 +48,6 @@ class Chef
           domains_list << domain.data.to_s
         end
 
-
         puts ui.list(domains_list, :uneven_columns_across, 4)
       end
     end

data/lib/chef/knife/digital_ocean_droplet_create.rb

@@ -206,13 +206,12 @@ class Chef
                                           ipv6: locate_config_value(:ipv6)
                                          )
 
-
         server = client.droplets.create(droplet)
 
-        # if client.droplets.find(id: server.id).status != 'in-progress'
-        #   ui.error("Droplet could not be started #{server.inspect}")
-        #   exit 1
-        # end
+        if client.droplets.find(id: server.id).status != 'new'
+          ui.error("Droplet could not be started #{server.inspect}")
+          exit 1
+        end
 
         puts "Droplet creation for #{locate_config_value(:server_name)} started. Droplet-ID is #{server.id}"
 

data/lib/chef/knife/digital_ocean_droplet_destroy.rb

@@ -29,7 +29,6 @@ class Chef
              long: '--all',
              description: '!WARNING! UNRECOVERABLE Destroy all droplets.'
 
-
       def run
         $stdout.sync = true
 
@@ -52,15 +51,13 @@ class Chef
         end
 
         if locate_config_value(:all)
-          droplets_ids = client.droplets.all.map do |droplet|
-            droplet.id
-          end
+          droplets_ids = client.droplets.all.map(&:id)
         end
 
         droplets_ids.each do |id|
-          puts "Delete droplet with id: #{id}"
+          ui.info "Delete droplet with id: #{id}"
           result = client.droplets.delete(id: id)
-          puts JSON.parse(result)['message'] rescue 'OK'
+          ui.info 'OK' if result == true or ui.error JSON.parse(result)['message']
         end
       end
     end

data/lib/chef/knife/digital_ocean_droplet_power.rb

@@ -0,0 +1,66 @@
+# 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/digital_ocean_base'
+
+class Chef
+  class Knife
+   class DigitalOceanDropletPower < Knife
+      include Knife::DigitalOceanBase
+
+      banner 'knife digital_ocean droplet power (options)'
+
+      option :action,
+             short: '-a ACTION',
+             long: '--action ACTION',
+             description: 'Power Action On/Off'
+
+      option :id,
+             short: '-I ID',
+             long: '--droplet-id ID',
+             description: 'Droplet ID'
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        unless locate_config_value(:action)
+          ui.error('Action cannot be empty. => -a <action>')
+          exit 1
+        end
+
+        unless locate_config_value(:id)
+          ui.error('ID cannot be empty. => -I <id>')
+          exit 1
+        end
+
+        case locate_config_value(:action)
+        when /(on)/i
+          result = client.droplet_actions.power_on(droplet_id: locate_config_value(:id))
+        when /(off)/i
+          result = client.droplet_actions.power_off(droplet_id: locate_config_value(:id))
+        else
+          ui.error 'Bad Action: Use on/off.'
+          exit 1
+        end
+
+        unless result.class == DropletKit::Action
+          ui.error JSON.parse(result)['message']
+          exit 1
+        end
+
+        wait_for_status(result)
+      end
+    end
+  end
+end

data/lib/chef/knife/digital_ocean_droplet_powercycle.rb

@@ -0,0 +1,48 @@
+# 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/digital_ocean_base'
+
+class Chef
+  class Knife
+   class DigitalOceanDropletPowercycle < Knife
+      include Knife::DigitalOceanBase
+
+      banner 'knife digital_ocean droplet powercycle (options)'
+
+      option :id,
+             short: '-I ID',
+             long: '--droplet-id ID',
+             description: 'Droplet ID'
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        unless locate_config_value(:id)
+          ui.error('ID cannot be empty. => -I <id>')
+          exit 1
+        end
+
+        result = client.droplet_actions.power_cycle(droplet_id: locate_config_value(:id))
+
+        unless result.class == DropletKit::Action
+          ui.error JSON.parse(result)['message']
+          exit 1
+        end
+
+        wait_for_status(result, status: 'active')
+      end
+    end
+  end
+end

data/lib/chef/knife/digital_ocean_droplet_reboot.rb

@@ -0,0 +1,48 @@
+# 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/digital_ocean_base'
+
+class Chef
+  class Knife
+   class DigitalOceanDropletReboot < Knife
+      include Knife::DigitalOceanBase
+
+      banner 'knife digital_ocean droplet reboot (options)'
+
+      option :id,
+             short: '-I ID',
+             long: '--droplet-id ID',
+             description: 'Droplet ID'
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        unless locate_config_value(:id)
+          ui.error('ID cannot be empty. => -I <id>')
+          exit 1
+        end
+
+        result = client.droplet_actions.reboot(droplet_id: locate_config_value(:id))
+
+        unless result.class == DropletKit::Action
+          ui.error JSON.parse(result)['message']
+          exit 1
+        end
+
+        wait_for_status(result)
+      end
+    end
+  end
+end

data/lib/chef/knife/digital_ocean_droplet_rebuild.rb

@@ -0,0 +1,58 @@
+# 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/digital_ocean_base'
+
+class Chef
+  class Knife
+   class DigitalOceanDropletRebuild < Knife
+      include Knife::DigitalOceanBase
+
+      banner 'knife digital_ocean droplet rebuild (options)'
+
+      option :image,
+             short: '-i IMAGE',
+             long: '--image-id IMAGE',
+             description: 'Image ID'
+
+      option :id,
+             short: '-I ID',
+             long: '--droplet-id ID',
+             description: 'Droplet ID'
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        unless locate_config_value(:image)
+          ui.error('Image ID cannot be empty. => -i <image-id>')
+          exit 1
+        end
+
+        unless locate_config_value(:id)
+          ui.error('ID cannot be empty. => -I <id>')
+          exit 1
+        end
+
+        result = client.droplet_actions.rebuild(droplet_id: locate_config_value(:id), image: locate_config_value(:image))
+
+        unless result.class == DropletKit::Action
+          ui.error JSON.parse(result)['message']
+          exit 1
+        end
+
+        wait_for_status(result, status: 'active')
+      end
+    end
+  end
+end

data/lib/chef/knife/digital_ocean_droplet_rename.rb

@@ -0,0 +1,58 @@
+# 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/digital_ocean_base'
+
+class Chef
+  class Knife
+   class DigitalOceanDropletRename < Knife
+      include Knife::DigitalOceanBase
+
+      banner 'knife digital_ocean droplet rename (options)'
+
+      option :name,
+             short: '-N NAME',
+             long: '--droplet-name NAME',
+             description: 'Name of droplet'
+
+      option :id,
+             short: '-I ID',
+             long: '--droplet-id ID',
+             description: 'Droplet ID'
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        unless locate_config_value(:name)
+          ui.error('Name cannot be empty. => -N <name>')
+          exit 1
+        end
+
+        unless locate_config_value(:id)
+          ui.error('ID cannot be empty. => -I <id>')
+          exit 1
+        end
+
+        result = client.droplet_actions.rename(droplet_id: locate_config_value(:id), name: locate_config_value(:name))
+
+        unless result.class == DropletKit::Action
+          ui.error JSON.parse(result)['message']
+          exit 1
+        end
+
+        wait_for_status(result)
+      end
+    end
+  end
+end