Add ondemand flag and api v2 support to floaty ssh

Author: yachub

lib/vmfloaty.rb

@@ -484,7 +484,7 @@ def run # rubocop:disable Metrics/AbcSize
 
         FloatyLogger.info "Can't ssh to multiple hosts; Using #{host_os} only..." if args.length > 1
 
-        service.ssh(verbose, host_os, use_token)
+        service.ssh(verbose, host_os, use_token, options.ondemand)
         exit 0
       end
     end

lib/vmfloaty/service.rb

@@ -87,7 +87,7 @@ def wait_for_request(verbose, requestid)
     @service_object.wait_for_request verbose, requestid, url
   end
 
-  def ssh(verbose, host_os, use_token = true)
+  def ssh(verbose, host_os, use_token = true, ondemand = nil)
     token_value = nil
     if use_token
       begin
@@ -97,7 +97,7 @@ def ssh(verbose, host_os, use_token = true)
         FloatyLogger.info 'Could not get token... requesting vm without a token anyway...'
       end
     end
-    Ssh.ssh(verbose, self, host_os, token_value)
+    Ssh.ssh(verbose, self, host_os, token_value, ondemand)
   end
 
   def query(verbose, hostname)

lib/vmfloaty/ssh.rb

@@ -14,27 +14,46 @@ def self.which(cmd)
     nil
   end
 
-  def self.command_string(verbose, service, host_os, use_token)
+  def self.command_string(verbose, service, host_os, use_token, ondemand = nil)
     ssh_path = which('ssh')
     raise 'Could not determine path to ssh' unless ssh_path
-
-    os_types = {}
+    os_types = Utils.generate_os_hash([host_os])
     os_types[host_os] = 1
 
-    response = service.retrieve(verbose, os_types, use_token)
+    response = service.retrieve(verbose, os_types, use_token, ondemand)
     raise "Could not get vm from #{service.type}:\n #{response}" unless response['ok']
 
     user = /win/.match?(host_os) ? 'Administrator' : 'root'
 
-    hostname = response[host_os]['hostname']
-    hostname = response[host_os]['hostname'][0] if response[host_os]['hostname'].is_a?(Array)
-    hostname = "#{hostname}.#{response['domain']}" unless hostname.end_with?('puppetlabs.net')
+    if ondemand
+      requestid = response['request_id']
+      service.wait_for_request(verbose, requestid)
+      response = service.retrieve(verbose, os_types, use_token, ondemand)
+      hosts = service.check_ondemandvm(verbose, requestid, service.url)
+      if hosts['domain'].nil?
+        hostname = hosts[host_os]['hostname']
+        hostname = hosts[host_os]['hostname'][0] if hosts[host_os]['hostname'].is_a?(Array)
+      else
+        # Provides backwards compatibility with VMPooler API v1
+        hostname = "#{hosts[host_os]['hostname'][0]}.#{hosts['domain']}"
+        hostname = "#{hosts[host_os]['hostname'][0]}.#{hosts['domain']}" if hosts[host_os]['hostname'].is_a?(Array)
+      end
+    else
+      if response['domain'].nil?
+        hostname = response[host_os]['hostname']
+        hostname = response[host_os]['hostname'][0] if response[host_os]['hostname'].is_a?(Array)
+      else
+        # Provides backwards compatibility with VMPooler API v1
+        hostname = "#{response[host_os]['hostname']}.#{response['domain']}"
+        hostname = "#{response[host_os]['hostname'][0]}.#{response['domain']}" if response[host_os]['hostname'].is_a?(Array)
+      end
+    end
 
     "#{ssh_path} #{user}@#{hostname}"
   end
 
-  def self.ssh(verbose, service, host_os, use_token)
-    cmd = command_string(verbose, service, host_os, use_token)
+  def self.ssh(verbose, service, host_os, use_token, ondemand)
+    cmd = command_string(verbose, service, host_os, use_token, ondemand)
     # TODO: Should this respect more ssh settings? Can it be configured
     #       by users ssh config and does this respect those settings?
     Kernel.exec(cmd)

spec/vmfloaty/ssh_spec.rb

@@ -4,24 +4,30 @@
 require 'vmfloaty/ssh'
 
 class ServiceStub
-  def retrieve(_verbose, os_types, _use_token)
+  def retrieve(_verbose, os_types, _use_token, ondemand)
     if os_types.keys[0] == 'abs_host_string'
       return {
         os_types.keys[0] => { 'hostname' => ['abs-hostname.delivery.puppetlabs.net'] },
         'ok' => true
       }
+
+    elsif os_types.keys[0] == 'vmpooler_api_v2_host_string'
+      return {
+        os_types.keys[0] => { 'hostname' => ['vmpooler-v2-hostname.delivery.puppetlabs.net'] },
+        'ok' => true
+      }
     end
 
     {
-      os_types.keys[0] => { 'hostname' => 'vmpooler-hostname' },
+      os_types.keys[0] => { 'hostname' => 'vmpooler-v1-hostname.delivery.puppetlabs.net' },
       'domain' => 'delivery.puppetlabs.net',
       'ok' => true
     }
   end
 
   def type
     return 'abs' if os_types == 'abs_host_string'
-    return 'vmpooler' if os_types == 'vmpooler_host_string'
+    return 'vmpooler' if os_types == 'vmpooler_api_v1_host_string' || os_types == 'vmpooler_api_v2_host_string'
   end
 end
 
@@ -38,12 +44,21 @@ def type
     expect(cmd).to match(/ssh [email protected]/)
   end
 
-  it 'gets a hostname string for vmpooler' do
+  it 'gets a hostname string for vmpooler api v1' do
+    verbose = false
+    service = ServiceStub.new
+    host_os = 'vmpooler_api_v1_host_string'
+    use_token = false
+    cmd = Ssh.command_string(verbose, service, host_os, use_token)
+    expect(cmd).to match(/ssh [email protected]/)
+  end
+
+  it 'gets a hostname string for vmpooler api v2' do
     verbose = false
     service = ServiceStub.new
-    host_os = 'vmpooler_host_string'
+    host_os = 'vmpooler_api_v2_host_string'
     use_token = false
     cmd = Ssh.command_string(verbose, service, host_os, use_token)
-    expect(cmd).to match(/ssh [email protected]/)
+    expect(cmd).to match(/ssh root@vmpooler-v2-hostname.delivery.puppetlabs.net/)
   end
 end