floatyhelper 1.4 → 2.0.1

data/lib/floatyhelper/vm.rb

@@ -1,19 +1,55 @@
+# This is for interfacing with floaty (or vmpooler/ABS directly if needed) to
+# request, query, or modify VMs on the service, and managing snapshots in the
+# yaml file.
 require 'floatyhelper/groups'
 require 'floatyhelper/hosts'
-require 'floatyhelper/conf'
+require 'floatyhelper/config'
+require 'colorize'
+require 'net/http'
+require 'json'
 
 class VM
+  def self.pooled_platforms_default
+    [
+      'centos-7-x86_64',
+      'centos-8-x86_64',
+      'oracle-7-x86_64',
+      'redhat-7-x86_64',
+      'redhat-8-x86_64',
+      'redhat-fips-7-x86_64',
+      'scientific-7-x86_64',
+      'sles-12-x86_64',
+      'ubuntu-1804-x86_64',
+    ]
+  end
+
+  def self.find_pooled_platforms
+    begin # rubocop:disable Style/RedundantBegin
+      result = Net::HTTP.get('vmpooler.delivery.puppetlabs.net','/status')
+      result = JSON.parse(result)
+      # Techinally, 'max > 0' tells you if it's a pooled platform, but if
+      # the pool is empty, we'll want to fall back to ABS anyway.
+      result['pools'].select { |_pool, info| info['ready'].positive? }.map { |pool, _info| pool.gsub('-pixa4','') }
+    rescue StandardError
+      # Not a great practice to swallow all errors, but this list is probably
+      # pretty stable, so let's just pass along the default.
+      puts 'Error looking up pooled platforms from vmpooler.delivery.puppetlabs.net. Using default pooled platform list instead.'.yellow
+      pooled_platforms_default
+    end
+  end
 
+  ### VM Management ###
   def self.destroy(id)
     hosts = Hosts.get_hosts_from_id(id)
-    Groups.delete_tag(id) if Groups.is_tag?(id)
+    Groups.delete_tag(id) if Groups.tag?(id)
     Groups.delete_all if id == 'all'
     hosts = hosts.select { |host| alive(host) }
-    puts `floaty delete #{hosts.join(',')}` unless hosts.empty?
+    puts Floaty.floaty_cmd("delete #{hosts.join(',')} --service vmpooler") unless hosts.empty?
   end
 
   def self.query(host)
-    eval(`floaty query #{host}`)
+    output = Floaty.floaty_cmd("query #{host} --service vmpooler")
+    Floaty.parse_floaty_json_output(output)
   end
 
   def self.get_current_lifetime(host)
@@ -21,58 +57,70 @@ class VM
     status['ok'] ? status[host]['lifetime'] : nil
   end
 
-  def self.alive(host, query=nil)
+  def self.alive(host, query = nil)
     query ||= query(host)
     query['ok'] && query[host]['state'] == 'running'
   end
 
-  def self.increaselife(id, amount)
-    amount = 100 if amount.nil?
+  def self.increaselife(id, amount = nil)
+    amount ||= Config.get_config_setting('increaselife').to_i
+    return if amount < 1
     hosts = Hosts.get_hosts_from_id(id)
     hosts.each do |host|
-      if lifetime = get_current_lifetime(host)
+      if (lifetime = get_current_lifetime(host))
         lifetime += amount
-        print "#{host} to #{lifetime} hours: "
-        puts `floaty modify #{host} --lifetime #{lifetime}`.split("\n")[0]
+        output = Floaty.floaty_cmd("modify #{host} --lifetime #{lifetime} --service vmpooler")
+        if output =~ /Successfully modified/
+          puts "#{host} lifetime set to #{lifetime} hours".green
+        else
+          puts "Error setting VM lifetime: #{output}".red
+        end
       else
         puts "Could not query host #{host}".red
       end
     end
   end
 
-  def self.snapshot(id, snaptag, clr=false)
-    snaptag = 'Blank Snaptag' unless snaptag
+  ### Snapshots ###
+  def self.snapshot(id, snaptag)
+    clr = Config.get_config_setting('vertical_snapshot_status')
+    clr = clr.to_s.downcase == 'true'
+    snaptag ||= 'Blank Snaptag'
     hosts = Hosts.get_hosts_from_id(id)
     shas = {}
+    # There's probably a better way to do this...
+    puts `tput clear` if clr
+
     hosts.each do |host|
-      print "#{host}: "
-      result = `floaty snapshot #{host}`
-      message = result.split("\n")[0]
-      answer = eval(result.sub(message,''))
+      result = Floaty.floaty_cmd("snapshot #{host} --service vmpooler")
+      answer = Floaty.parse_floaty_json_output(result)
       sha = answer[host]['snapshot']
-      puts sha
+      puts "#{host}: #{sha}"
       shas[host] = sha
     end
 
-    data = Conf.load_data
+    data = Config.load_data
     data['snapshots'] ||= {}
     data['snapshots'][id] ||= {}
     data['snapshots'][id][snaptag] = shas
-    Conf.write_data(data)
+    Config.write_data(data)
 
-    puts `tput clear` if clr
+    puts
     puts 'Waiting for snapshots to appear in floaty query...'
     alldone = false
-    while !alldone do
-      puts `tput cup 1` if clr
+    until alldone
+      puts `tput cup 4` if clr
       alldone = true
       print "\r" unless clr
       hosts.each do |host|
-        answer = eval(`floaty query #{host}`)[host]
+        answer = query(host)[host]
         done = answer.keys.include?('snapshots') && answer['snapshots'].include?(shas[host])
-        status = done ? 'Done' : 'Wait'
-        print "* #{host}: #{status} *" unless clr
-        puts "* #{host}: #{status} *" if clr
+        status = done ? 'Done'.green : 'Wait'.yellow
+        if clr
+          puts "* %s #{status} *" % "#{host}:".ljust(16)
+        else
+          print "* %s #{status} *" % "#{host}:".ljust(16)
+        end
         alldone &= done
       end
       sleep(1)
@@ -81,31 +129,41 @@ class VM
   end
 
   def self.getsnapshot(id, snaptag)
-    data = Conf.load_data
+    data = Config.load_data
     data['snapshots'][id][snaptag]
   end
 
   def self.snaptag_exists?(id, snaptag)
-    data = Conf.load_data
-    exists = data['snapshots'].keys.include?(id) ? data['snapshots'][id].keys.include?(snaptag) : false
+    data = Config.load_data
+    data['snapshots'].keys.include?(id) ? data['snapshots'][id].keys.include?(snaptag) : false
+  end
+
+  def self.snaplist(tag)
+    data = Config.load_data
+    return [] if !data['snapshots'].keys.include?(tag)
+    data['snapshots'][tag].keys
   end
 
   def self.revert(id, snaptag)
-    snaptag = 'Blank Snaptag' unless snaptag
+    snaptag ||= 'Blank Snaptag'
     shas = VM.getsnapshot(id, snaptag)
     shas.each do |host, sha|
       print "#{host}: #{sha} --- "
-      puts `floaty revert #{host} #{sha}`
+      puts Floaty.floaty_cmd("revert #{host} #{sha} --service vmpooler")
     end
     puts 'Waiting 10 seconds for revert to take effect'
     sleep(10)
   end
 
-  def self.get_vm(platform='centos-7-x86_64')
-    response = `floaty get #{platform} 2>&1`
-    raise "Error obtaining a VM: #{response}" if response.include?('error')
-    return response.split(' ')[1].split('.')[0]
+  ### Get a VM from floaty ###
+  def self.get_vm(platform: 'centos-7-x86_64', force_abs: false)
+    if !find_pooled_platforms.include?(platform.gsub('-pixa4','')) || force_abs
+      response = Floaty.floaty_cmd("get #{platform} --service abs --priority 1", use_pty: true)
+      response.chomp.split('- ')[1].split[0].split('.')[0]
+    else
+      response = Floaty.floaty_cmd("get #{platform} --service vmpooler")
+      raise "Error obtaining a VM: #{response}" if response.include?('error')
+      response.split[1].split('.')[0]
+    end
   end
-
 end
-