knife-google 1.3.1 → 2.0.0

data/lib/chef/knife/cloud/google_service_helpers.rb

@@ -0,0 +1,62 @@
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2016 Chef Software, 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.
+#
+
+class Chef::Knife::Cloud
+  module GoogleServiceHelpers
+    REQUIRED_KEYS = [:gce_project, :gce_zone]
+
+    def create_service_instance
+      Chef::Knife::Cloud::GoogleService.new(
+        project:       locate_config_value(:gce_project),
+        zone:          locate_config_value(:gce_zone),
+        wait_time:     locate_config_value(:request_timeout),
+        refresh_rate:  locate_config_value(:request_refresh_rate),
+        max_pages:     locate_config_value(:max_pages),
+        max_page_size: locate_config_value(:max_page_size)
+      )
+    end
+
+    def check_for_missing_config_values!(*keys)
+      keys_to_check = REQUIRED_KEYS + keys
+
+      missing = keys_to_check.select { |x| locate_config_value(x).nil? }
+
+      unless missing.empty?
+        message = "The following required parameters are missing: #{missing.join(', ')}"
+        ui.error(message)
+        raise message
+      end
+    end
+
+    def private_ip_for(server)
+      server.network_interfaces.first.network_ip
+    rescue NoMethodError
+      "unknown"
+    end
+
+    def public_ip_for(server)
+      server.network_interfaces.first.access_configs.first.nat_ip
+    rescue NoMethodError
+      "unknown"
+    end
+
+    def valid_disk_size?(size)
+      size.between?(10, 10_000)
+    end
+  end
+end

data/lib/chef/knife/cloud/google_service_options.rb

@@ -0,0 +1,58 @@
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2016 Chef Software, 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.
+#
+
+class Chef::Knife::Cloud
+  module GoogleServiceOptions
+    def self.included(includer)
+      includer.class_eval do
+        option :gce_project,
+          long:        "--gce-project PROJECT",
+          description: "Name of the Google Cloud project to use"
+
+        option :gce_zone,
+          short:       "-Z ZONE",
+          long:        "--gce-zone ZONE",
+          description: "Name of the Google Compute Engine zone to use"
+
+        option :gce_max_pages,
+          long:        "--gce-max-pages NUMPAGES",
+          description: "Maximum number of pages to request for paginated listing requests, defaults to 20",
+          default:     20,
+          proc:        proc { |pages| pages.to_i }
+
+        option :gce_max_page_size,
+          long:        "--gce-max-page-size NUMPAGES",
+          description: "Maximum number of items per page to request for paginated listing requests, defaults to 100",
+          default:     100,
+          proc:        proc { |items| items.to_i }
+
+        option :request_refresh_rate,
+          long:        "--request-refresh-rate SECS",
+          description: "Number of seconds to sleep between each check of the request status, defaults to 2",
+          default:     2,
+          proc:        proc { |secs| secs.to_i }
+
+        option :request_timeout,
+          long:        "--request-timeout SECS",
+          description: "Number of seconds to wait for a request to complete, defaults to 600",
+          default:     600,
+          proc:        proc { |secs| secs.to_i }
+      end
+    end
+  end
+end

data/lib/chef/knife/google_disk_create.rb

@@ -1,4 +1,8 @@
-# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# Author:: Paul Rossman (<paulrossman@google.com>)
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright 2015-2016 Google Inc., Chef Software, 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.
@@ -11,58 +15,50 @@
 # 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/google_base'
 
-class Chef
-  class Knife
-    class GoogleDiskCreate < Knife
+require "chef/knife"
+require "chef/knife/cloud/command"
+require "chef/knife/cloud/google_service"
+require "chef/knife/cloud/google_service_helpers"
+require "chef/knife/cloud/google_service_options"
 
-      include Knife::GoogleBase
+class Chef::Knife::Cloud
+  class GoogleDiskCreate < Command
+    include GoogleServiceHelpers
+    include GoogleServiceOptions
 
-      banner "knife google disk create NAME --gce-disk-size N -Z ZONE (options)"
+    banner "knife google disk create NAME --gce-disk-size N (options)"
 
-      deps do
-        require 'google/compute'
-      end
+    option :disk_size,
+      long:        "--gce-disk-size SIZE",
+      description: "Size of the persistent disk between 10 and 10000 GB, specified in GB; default is '10' GB",
+      default:     10,
+      proc:        proc { |size| size.to_i }
 
-      option :zone,
-        :short => "-Z ZONE",
-        :long => "--gce-zone ZONE",
-        :description => "The Zone for this disk",
-        :required => true
+    option :disk_type,
+      long:        "--gce-disk-type TYPE",
+      description: "Disk type to use to create the disk. Possible values are 'pd-standard', 'pd-ssd' and 'local-ssd'; default is 'pd-standard'",
+      default:     "pd-standard"
 
-      option :disk_size,
-        :long => "--gce-disk-size SIZE",
-        :description => "Size of the persistent disk between 1 and 10000 GB, specified in GB; default is '1' GB",
-        :default => "1"
+    option :disk_source,
+      long:        "--gce-disk-source_image IMAGE_URL",
+      description: "GCE disk source image to use when creating disk, such as projects/centos-cloud/global/images/centos-7-v20160216; optional, if not supplied, a blank disk will be created",
+      default:     nil
 
-      def run
-        $stdout.sync = true
-        unless @name_args.size > 0
-          ui.error("Please provide the name of the new disk")
-          exit 1
-        end
-
-        begin
-          zone = client.zones.get(config[:zone])
-        rescue Google::Compute::ResourceNotFound
-          ui.error("Zone '#{config[:zone]}' not found")
-          exit 1
-        end
+    def validate_params!
+      check_for_missing_config_values!(:disk_size, :disk_type)
+      raise "Please specify a disk name." unless @name_args.first
+      raise "Disk size must be between 10 and 10,000" unless valid_disk_size?(locate_config_value(:disk_size))
+      super
+    end
 
-        begin
-          disk_size = config[:disk_size].to_i
-          raise if !disk_size.between?(1, 10000)
-        rescue
-          ui.error("Size of the persistent disk must be between 1 and 10000 GB.")
-          exit 1
-        end
+    def execute_command
+      name = @name_args.first
+      size = locate_config_value(:disk_size)
+      type = locate_config_value(:disk_type)
+      src  = locate_config_value(:disk_source)
 
-        zone_operation = client.disks.create(:name => @name_args.first,
-                                             :sizeGb => config[:disk_size],
-                                             :zone => zone.name)
-      end
+      service.create_disk(name, size, type, src)
     end
   end
 end

data/lib/chef/knife/google_disk_delete.rb

@@ -1,4 +1,8 @@
-# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# Author:: Paul Rossman (<paulrossman@google.com>)
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright 2015-2016 Google Inc., Chef Software, 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.
@@ -11,50 +15,28 @@
 # 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/google_base'
-
-class Chef
-  class Knife
-    class GoogleDiskDelete < Knife
 
-      include Knife::GoogleBase
+require "chef/knife"
+require "chef/knife/cloud/command"
+require "chef/knife/cloud/google_service"
+require "chef/knife/cloud/google_service_helpers"
+require "chef/knife/cloud/google_service_options"
 
-      banner "knife google disk delete NAME -Z ZONE (options)"
+class Chef::Knife::Cloud
+  class GoogleDiskDelete < Command
+    include GoogleServiceHelpers
+    include GoogleServiceOptions
 
-      deps do
-        require 'google/compute'
-      end
+    banner "knife google disk delete NAME [NAME] (options)"
 
-      option :zone,
-        :short => "-Z ZONE",
-        :long => "--gce-zone ZONE",
-        :description => "The Zone for this disk",
-        :required => true
-
-      def run
-        unless @name_args.size > 0
-          ui.error("Please provide the name of the disk to be deleted")
-          exit 1
-        end
-
-        begin
-          zone = client.zones.get(config[:zone])
-        rescue Google::Compute::ResourceNotFound
-          ui.error("Zone '#{config[:zone]}' not found")
-          exit 1
-        end
+    def validate_params!
+      check_for_missing_config_values!
+      raise "You must specify at least one disk to delete." if @name_args.empty?
+      super
+    end
 
-        begin
-          disk = client.disks.get(:zone=>zone.name, :disk=>@name_args.first)
-          ui.confirm("Are you sure you want to delete disk '#{zone.name}:#{disk.name}'")
-          zone_operation = client.disks.delete(:zone=>zone.name, :disk=>disk.name)
-          ui.warn("Disk '#{zone.name}:#{disk.name}' deleted")
-        rescue Google::Compute::ResourceNotFound
-          ui.error("Disk '#{zone.name}:#{@name_args.first}' not found")
-          exit 1
-        end
-      end
+    def execute_command
+      @name_args.each { |disk| service.delete_disk(disk) }
     end
   end
 end

data/lib/chef/knife/google_disk_list.rb

@@ -1,4 +1,8 @@
-# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# Author:: Paul Rossman (<paulrossman@google.com>)
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright 2015-2016 Google Inc., Chef Software, 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.
@@ -11,65 +15,67 @@
 # 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/google_base'
 
-class Chef
-  class Knife
-    class GoogleDiskList < Knife
+require "chef/knife"
+require "chef/knife/cloud/list_resource_command"
+require "chef/knife/cloud/google_service"
+require "chef/knife/cloud/google_service_helpers"
+require "chef/knife/cloud/google_service_options"
 
-      include Knife::GoogleBase
+class Chef::Knife::Cloud
+  class GoogleDiskList < ResourceListCommand
+    include GoogleServiceHelpers
+    include GoogleServiceOptions
 
-      banner "knife google disk list -Z ZONE (options)"
+    banner "knife google disk list"
+
+    def validate_params!
+      check_for_missing_config_values!
+      super
+    end
 
-      option :zone,
-        :short => "-Z ZONE",
-        :long => "--gce-zone ZONE",
-        :description => "The Zone for disk listing",
-        :required => true
+    def before_exec_command
+      @columns_with_info = [
+        { label: "Zone",         key: "name" },
+        { label: "Status",       key: "status", value_callback: method(:format_status_value) },
+        { label: "Size (GB)", key: "size_gb" },
+        { label: "Type",         key: "type", value_callback: method(:format_disk_type) },
+        { label: "Source Image", key: "source_image", value_callback: method(:format_source_image) },
+        { label: "Attached To",  key: "users", value_callback: method(:format_users) },
+      ]
 
-      def run
-        $stdout.sync = true
+      @sort_by_field = "name"
+    end
+
+    def query_resource
+      service.list_disks
+    end
 
-        begin
-          zone = client.zones.get(config[:zone])
-        rescue Google::Compute::ResourceNotFound
-          ui.error("Zone '#{config[:zone]}' not found")
-          exit 1
-        end
+    def format_status_value(status)
+      status = status.downcase
+      status_color = if status == "ready"
+                       :green
+                     else
+                       :red
+                     end
 
-        disk_list = [
-          ui.color("name", :bold),
-          ui.color('zone', :bold),
-          ui.color('source snapshot', :bold),
-          ui.color('size (in GB)', :bold),
-          ui.color('status', :bold)].flatten.compact
+      ui.color(status, status_color)
+    end
+
+    def format_disk_type(type)
+      type.split("/").last
+    end
+
+    def format_source_image(source)
+      return "unknown" if source.nil? || source.empty?
+
+      source.split("/").last(4).join("/")
+    end
 
-        output_column_count = disk_list.length
+    def format_users(users)
+      return "none" if users.nil? || users.empty?
 
-        client.disks.list(:zone=>zone.name).each do |disk|
-          disk_list << disk.name
-          disk_list << selflink2name(disk.zone)
-          if disk.source_snapshot.nil?
-            disk_list << " "
-          else
-            disk_list << selflink2name(disk.source_snapshot)
-          end
-          disk_list << disk.size_gb
-          disk_list << begin
-            status = disk.status.downcase
-            case status
-            when 'stopping', 'stopped', 'terminated'
-              ui.color(status, :red)
-            when 'requested', 'provisioning', 'staging'
-              ui.color(status, :yellow)
-            else
-              ui.color(status, :green)
-            end
-          end
-        end
-        ui.info(ui.list(disk_list, :uneven_columns_across, output_column_count))
-      end
+      users.map { |user| user.split("/").last(5).join("/") }.join(", ")
     end
   end
 end

data/lib/chef/knife/google_project_quotas.rb

@@ -0,0 +1,59 @@
+#
+# Author:: Paul Rossman (<paulrossman@google.com>)
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright 2015-2016 Google Inc., Chef Software, 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"
+require "chef/knife/cloud/list_resource_command"
+require "chef/knife/cloud/google_service"
+require "chef/knife/cloud/google_service_helpers"
+require "chef/knife/cloud/google_service_options"
+
+class Chef::Knife::Cloud
+  class GoogleProjectQuotas < ResourceListCommand
+    include GoogleServiceHelpers
+    include GoogleServiceOptions
+
+    banner "knife google project quotas"
+
+    def validate_params!
+      check_for_missing_config_values!
+      super
+    end
+
+    def before_exec_command
+      @columns_with_info = [
+        { label: "Quota", key: "metric", value_callback: method(:format_name) },
+        { label: "Limit", key: "limit", value_callback: method(:format_number) },
+        { label: "Usage", key: "usage", value_callback: method(:format_number) },
+      ]
+
+      @sort_by_field = "metric"
+    end
+
+    def query_resource
+      service.list_project_quotas
+    end
+
+    def format_name(name)
+      name.split("_").map { |x| x.capitalize }.join(" ")
+    end
+
+    def format_number(number)
+      number % 1 == 0 ? number.to_i.to_s : number.to_s
+    end
+  end
+end