knife-vcenter 1.0.0

data/lib/chef/knife/vcenter_cluster_list.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterClusterList < ResourceListCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+
+        banner 'knife vcenter cluster list'
+
+        def before_exec_command
+          @columns_with_info = [
+            { label: 'ID',    key: 'cluster' },
+            { label: 'Name',  key: 'name' },
+            { label: 'DRS?', key: 'drs_enabled', value_callback: method(:format_boolean) },
+            { label: 'HA?', key: 'ha_enabled', value_callback: method(:format_boolean)},
+          ]
+
+          @sort_by_field = 'name'
+        end
+
+        def query_resource
+          # Call service to get the list of hosts from vcenter
+          service.list_clusters
+        end
+
+        def format_boolean(status)
+          status_color = status ? :green : :red
+          ui.color(status ? "True" : "False", status_color)
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_datacenter_list.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterDatacenterList < ResourceListCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+
+        banner 'knife vcenter datacenter list'
+
+        def before_exec_command
+          @columns_with_info = [
+            { label: 'ID',    key: 'datacenter' },
+            { label: 'Name',  key: 'name' },
+          ]
+
+          @sort_by_field = 'name'
+        end
+
+        def query_resource
+          # Call service to get the list of hosts from vcenter
+          service.list_datacenters
+        end        
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_host_list.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterHostList < ResourceListCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+
+        banner 'knife vcenter host list'
+
+        def before_exec_command
+          @columns_with_info = [
+            { label: 'ID',    key: 'host' },
+            { label: 'Name',  key: 'name' },
+            { label: 'Power State', key: 'power_state', value_callback: method(:format_power_status) },
+            { label: 'Connection State', key: 'connection_state'},
+          ]
+
+          @sort_by_field = 'name'
+        end
+
+        def query_resource
+          # Call service to get the list of hosts from vcenter
+          service.list_hosts
+        end
+
+        def format_power_status(status)
+          status_check = status.value
+          status_color = case status_check
+                          when 'POWERED_OFF'
+                            :red
+                          when 'POWERED_ON'
+                            :green
+                          when 'SUSPENDED'
+                            :yellow
+                          end
+
+          ui.color(status.value, status_color)
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_vm_clone.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/server/create_command'
+require 'chef/knife/cloud/server/create_options'
+require 'chef/knife/cloud/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterVmClone < Chef::Knife::Cloud::ServerCreateCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+        include ServerCreateOptions
+
+        banner 'knife vcenter vm clone NAME'
+
+        option :template,
+               long:        "--template NAME",
+               description: "Name of VM or template to use to clone machine from"
+
+        option :targethost,
+               long:        "--targethost HOST",
+               description: "Host that the machine should be created on"
+
+        option :datacenter,
+               long:        "--datacenter NAME",
+               description: "The datacenter for vSphere"
+
+        option :disable_power_on,
+               long:        "--disable-power-on",
+               boolean:     true,
+               default:     false
+
+        option :folder,
+               long:        "--folder NAME",
+               description: "Folder to deploy the new machine into"
+
+        def validate_params!
+          super
+
+          if @name_args.empty?
+            ui.error('You must provide the name of the new machine')
+          end
+
+          check_for_missing_config_values!(:template, :targethost, :datacenter)
+        end
+
+        def before_exec_command
+          super
+
+          @create_options = {
+            name:         @name_args[0],
+            type:         "clone",
+            template:     locate_config_value(:template),
+            targethost:   locate_config_value(:targethost),
+            datacenter:   locate_config_value(:datacenter),
+            poweron:      !locate_config_value(:disable_power_on),
+            folder:       locate_config_value(:folder)
+          }
+
+        end
+
+        def before_bootstrap
+          super
+
+          config[:chef_node_name] = locate_config_value(:chef_node_name) ? locate_config_value(:chef_node_name) : server.name
+
+          # determine the IP address to use to bootstrap the machine with chef
+          config[:bootstrap_ip_address] = hostname_for_server
+        end
+
+        def hostname_for_server
+          ipaddress = service.ipaddress
+
+          ipaddress.nil? ? server.name : ipaddress
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_vm_create.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/server/create_command'
+require 'chef/knife/cloud/server/create_options'
+require 'chef/knife/cloud/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterVmCreate < Chef::Knife::Cloud::ServerCreateCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+        include ServerCreateOptions
+
+        banner 'knife vcenter vm create NAME'
+
+        option :targethost,
+               long:        "--targethost HOST",
+               description: "vCenter host on which the new VM should be created"
+
+        option :folder,
+               long:        "--folder FOLDER",
+               description: "Folder in which the machine will reside"
+
+        option :datastore,
+               long:        "--datastore DATASTORE",
+               description: "Datastore to be used for the disks etc"
+
+        option :resource_pool,
+               long:        "--resource_pool RESOURCEPOOOL",
+               description: "Resource Pool to create the machine"
+
+
+        def validate_params!
+          super
+
+          if @name_args.empty?
+            ui.error('You must provide the name of the new machine')
+          end
+
+          check_for_missing_config_values!(:targethost, :datastore, :folder)
+        end
+
+        def before_exec_command
+          super
+
+          @create_options = {
+            name:         @name_args[0],
+            type:         "create",
+            targethost:   locate_config_value(:targethost),
+            folder:       locate_config_value(:folder),
+            datastore:    locate_config_value(:datastore),
+            resource_pool: locate_config_value(:resource_pool)
+          }
+
+          puts @create_options
+        end
+
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_vm_delete.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/server/delete_options'
+require 'chef/knife/cloud/server/delete_command'
+require 'chef/knife/cloud/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterVmDelete < ServerDeleteCommand
+        include ServerDeleteOptions
+        include VcenterServiceOptions
+        include VcenterServiceHelpers
+
+        banner 'knife vcenter vm delete NAME [NAME] (options)'
+
+        # rubocop:disable Style/GuardClause
+        def validate_params!
+          if @name_args.empty?
+            ui.error('You must supply the name of the virtual machine to delete.')
+            exit(1) if @name_args.empty?
+          end
+        end
+
+        def execute_command
+          @name_args.each do |name|
+            service.delete_vm(name)
+            delete_from_chef(name)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/vcenter_vm_list.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2017 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/server/list_command'
+require 'chef/knife/cloud/server/list_options'
+require 'chef/knife/cloud/vcenter_service'
+require 'chef/knife/cloud/vcenter_service_helpers'
+require 'chef/knife/cloud/vcenter_service_options'
+
+class Chef
+  class Knife
+    class Cloud
+      class VcenterVmList < Chef::Knife::Cloud::ServerListCommand
+        include VcenterServiceHelpers
+        include VcenterServiceOptions
+
+        banner 'knife vcenter vm list'
+
+        def before_exec_command
+          @columns_with_info = [
+            { label: 'ID',    key: 'vm' },
+            { label: 'Name',  key: 'name' },
+            { label: 'Power State', key: 'power_state', value_callback: method(:format_power_status) },
+            { label: 'CPU Count', key: 'cpu_count'},
+            { label: 'RAM Size (MB)', key: 'memory_size_MiB', value_callback: method(:format_memory_value) },
+          ]
+
+          @sort_by_field = 'name'
+        end
+
+        def format_power_status(status)
+          status_check = status.value
+          status_color = case status_check
+                          when 'POWERED_OFF'
+                            :red
+                          when 'POWERED_ON'
+                            :green
+                          when 'SUSPENDED'
+                            :yellow
+                          end
+
+          ui.color(status.value, status_color)
+        end
+
+        def format_memory_value(value)
+          value.to_s.reverse.gsub(/...(?=.)/,'\&,').reverse
+        end
+      end
+    end
+  end
+end