vmfloaty 0.7.9 → 0.8.0

data/lib/vmfloaty.rb

@@ -5,11 +5,13 @@ require 'commander'
 require 'colorize'
 require 'json'
 require 'pp'
+require 'uri'
 require 'vmfloaty/auth'
 require 'vmfloaty/pooler'
 require 'vmfloaty/version'
 require 'vmfloaty/conf'
 require 'vmfloaty/utils'
+require 'vmfloaty/service'
 require 'vmfloaty/ssh'
 
 class Vmfloaty
@@ -17,27 +19,26 @@ class Vmfloaty
 
   def run
     program :version, Vmfloaty::VERSION
-    program :description, 'A CLI helper tool for Puppet Labs vmpooler to help you stay afloat'
+    program :description, 'A CLI helper tool for Puppet Labs VM poolers to help you stay afloat'
 
     config = Conf.read_config
 
     command :get do |c|
       c.syntax = 'floaty get os_type0 os_type1=x ox_type2=y [options]'
       c.summary = 'Gets a vm or vms based on the os argument'
-      c.description = 'A command to retrieve vms from vmpooler. Can either be a single vm, or multiple with the `=` syntax.'
+      c.description = 'A command to retrieve vms from a pooler service. Can either be a single vm, or multiple with the `=` syntax.'
       c.example 'Gets a few vms', 'floaty get centos=3 debian --user brian --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
+      c.option '--service STRING', String, 'Configured pooler service name'
       c.option '--user STRING', String, 'User to authenticate with'
-      c.option '--url STRING', String, 'URL of vmpooler'
-      c.option '--token STRING', String, 'Token for vmpooler'
+      c.option '--url STRING', String, 'URL of pooler service'
+      c.option '--token STRING', String, 'Token for pooler service'
       c.option '--notoken', 'Makes a request without a token'
       c.option '--force', 'Forces vmfloaty to get requested vms'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        token = options.token || config['token']
-        user = options.user ||= config['user']
-        url = options.url ||= config['url']
-        no_token = options.notoken
+        service = Service.new(options, config)
+        use_token = !options.notoken
         force = options.force
 
         if args.empty?
@@ -48,98 +49,51 @@ class Vmfloaty
         os_types = Utils.generate_os_hash(args)
 
         max_pool_request = 5
-        large_pool_requests = os_types.select{|k,v| v > max_pool_request}
+        large_pool_requests = os_types.select{|_,v| v > max_pool_request}
         if ! large_pool_requests.empty? and ! force
           STDERR.puts "Requesting vms over #{max_pool_request} requires a --force flag."
           STDERR.puts "Try again with `floaty get --force`"
           exit 1
         end
 
-        unless os_types.empty?
-          if no_token
-            begin
-              response = Pooler.retrieve(verbose, os_types, nil, url)
-            rescue MissingParamError
-              STDERR.puts e
-              STDERR.puts "See `floaty get --help` for more information on how to get VMs."
-            rescue AuthError => e
-              STDERR.puts e
-              exit 1
-            end
-            puts Utils.format_hosts(response)
-            exit 0
-          else
-            unless token
-              puts "No token found. Retrieving a token..."
-              if !user
-                STDERR.puts "You did not provide a user to authenticate to vmpooler with"
-                exit 1
-              end
-              pass = password "Enter your vmpooler password please:", '*'
-              begin
-                token = Auth.get_token(verbose, url, user, pass)
-              rescue TokenError => e
-                STDERR.puts e
-                exit 1
-              end
-
-              puts "\nToken retrieved!"
-              puts token
-            end
-
-            begin
-              response = Pooler.retrieve(verbose, os_types, token, url)
-            rescue MissingParamError
-              STDERR.puts e
-              STDERR.puts "See `floaty get --help` for more information on how to get VMs."
-            rescue AuthError => e
-              STDERR.puts e
-              exit 1
-            end
-            puts Utils.format_hosts(response)
-            exit 0
-          end
-        else
+        if os_types.empty?
           STDERR.puts "No operating systems provided to obtain. See `floaty get --help` for more information on how to get VMs."
           exit 1
         end
+
+        response = service.retrieve(verbose, os_types, use_token)
+        puts Utils.format_hosts(response)
       end
     end
 
     command :list do |c|
       c.syntax = 'floaty list [options]'
       c.summary = 'Shows a list of available vms from the pooler or vms obtained with a token'
-      c.description = 'List will either show all vm templates available in vmpooler, or with the --active flag it will list vms obtained with a vmpooler token.'
+      c.description = 'List will either show all vm templates available in pooler service, or with the --active flag it will list vms obtained with a pooler service token.'
       c.example 'Filter the list on centos', 'floaty list centos --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
+      c.option '--service STRING', String, 'Configured pooler service name'
       c.option '--active', 'Prints information about active vms for a given token'
-      c.option '--token STRING', String, 'Token for vmpooler'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--token STRING', String, 'Token for pooler service'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
+        service = Service.new(options, config)
         filter = args[0]
-        url = options.url ||= config['url']
-        token = options.token || config['token']
-        active = options.active
 
-        if active
+        if options.active
           # list active vms
-          begin
-            running_vms = Utils.get_all_token_vms(verbose, url, token)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
-          rescue Exception => e
-            STDERR.puts e
-            exit 1
-          end
-
-          if ! running_vms.nil?
-            Utils.prettyprint_hosts(running_vms, verbose, url)
+          running_vms = service.list_active(verbose)
+          host = URI.parse(service.url).host
+          if running_vms.empty?
+            puts "You have no running VMs on #{host}"
+          else
+            puts "Your VMs on #{host}:"
+            Utils.pretty_print_hosts(verbose, service, running_vms)
           end
         else
           # list available vms from pooler
-          os_list = Pooler.list(verbose, url, filter)
+          os_list = service.list(verbose, filter)
           puts os_list
         end
       end
@@ -148,139 +102,74 @@ class Vmfloaty
     command :query do |c|
       c.syntax = 'floaty query hostname [options]'
       c.summary = 'Get information about a given vm'
-      c.description = 'Given a hostname from vmpooler, vmfloaty with query vmpooler to get various details about the vm.'
+      c.description = 'Given a hostname from the pooler service, vmfloaty with query the service to get various details about the vm.'
       c.example 'Get information about a sample host', 'floaty query hostname --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
+        service = Service.new(options, config)
         hostname = args[0]
 
-        query_req = Pooler.query(verbose, url, hostname)
+        query_req = service.query(verbose, hostname)
         pp query_req
       end
     end
 
     command :modify do |c|
       c.syntax = 'floaty modify hostname [options]'
-      c.summary = 'Modify a vms tags, time to live, and disk space'
-      c.description = 'This command makes modifications to the virtual machines state in vmpooler. You can either append tags to the vm, increase how long it stays active for, or increase the amount of disk space.'
+      c.summary = 'Modify a VM\'s tags, time to live, disk space, or reservation reason'
+      c.description = 'This command makes modifications to the virtual machines state in the pooler service. You can either append tags to the vm, increase how long it stays active for, or increase the amount of disk space.'
       c.example 'Modifies myhost1 to have a TTL of 12 hours and adds a custom tag', 'floaty modify myhost1 --lifetime 12 --url https://myurl --token mytokenstring --tags \'{"tag":"myvalue"}\''
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
-      c.option '--token STRING', String, 'Token for vmpooler'
-      c.option '--lifetime INT', Integer, 'VM TTL (Integer, in hours)'
-      c.option '--disk INT', Integer, 'Increases VM disk space (Integer, in gb)'
-      c.option '--tags STRING', String, 'free-form VM tagging (json)'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
+      c.option '--token STRING', String, 'Token for pooler service'
+      c.option '--lifetime INT', Integer, 'VM TTL (Integer, in hours) [vmpooler only]'
+      c.option '--disk INT', Integer, 'Increases VM disk space (Integer, in gb) [vmpooler only]'
+      c.option '--tags STRING', String, 'free-form VM tagging (json) [vmpooler only]'
+      c.option '--reason STRING', String, 'VM reservation reason [nspooler only]'
       c.option '--all', 'Modifies all vms acquired by a token'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
+        service = Service.new(options, config)
         hostname = args[0]
-        lifetime = options.lifetime
-        disk = options.disk
-        tags = JSON.parse(options.tags) if options.tags
-        token = options.token || config['token']
         modify_all = options.all
 
-        running_vms = nil
-
-        if modify_all
-          begin
-            running_vms = Utils.get_all_token_vms(verbose, url, token)
-          rescue Exception => e
-            STDERR.puts e
-          end
-        elsif hostname.include? ","
-          running_vms = hostname.split(",")
+        if hostname.nil? and !modify_all
+          STDERR.puts "ERROR: Provide a hostname or specify --all."
+          exit 1
         end
-
-        if lifetime || tags
-          # all vms
-          if !running_vms.nil?
+        running_vms = modify_all ? service.list_active(verbose) : hostname.split(",")
+
+        tags = options.tags ? JSON.parse(options.tags) : nil
+        modify_hash = {
+            lifetime: options.lifetime,
+            disk: options.disk,
+            tags: tags,
+            reason: options.reason
+        }
+        modify_hash.delete_if { |_, value| value.nil? }
+
+        unless modify_hash.empty?
+          ok = true
+          modified_hash = {}
+          running_vms.each do |vm|
             begin
-              modify_hash = {}
-              modify_flag = true
-
-              running_vms.each do |vm|
-                modify_hash[vm] = Pooler.modify(verbose, url, vm, token, lifetime, tags)
-              end
-
-              modify_hash.each do |hostname,status|
-                if status == false
-                  STDERR.puts "Could not modify #{hostname}."
-                  modify_flag = false
-                end
-              end
-
-              if modify_flag
-                puts "Successfully modified all vms. Use `floaty list --active` to see the results."
-              end
-            rescue Exception => e
-              STDERR.puts e
-              exit 1
-            end
-          else
-            # Single Vm
-            begin
-              modify_req = Pooler.modify(verbose, url, hostname, token, lifetime, tags)
-            rescue TokenError => e
+              modified_hash[vm] = service.modify(verbose, vm, modify_hash)
+            rescue ModifyError => e
               STDERR.puts e
-              exit 1
-            end
-
-            if modify_req["ok"]
-              puts "Successfully modified vm #{hostname}."
-            else
-              STDERR.puts "Could not modify given host #{hostname} at #{url}."
-              puts modify_req
-              exit 1
+              ok = false
             end
           end
-        end
-
-        if disk
-          # all vms
-          if !running_vms.nil?
-            begin
-              modify_hash = {}
-              modify_flag = true
-
-              running_vms.each do |vm|
-                modify_hash[vm] = Pooler.disk(verbose, url, vm, token, disk)
-              end
-
-              modify_hash.each do |hostname,status|
-                if status == false
-                  STDERR.puts "Could not update disk space on #{hostname}."
-                  modify_flag = false
-                end
-              end
-
-              if modify_flag
-                puts "Successfully made request to update disk space on all vms."
-              end
-            rescue Exception => e
-              STDERR.puts e
-              exit 1
-            end
-          else
-            # single vm
-            begin
-              disk_req = Pooler.disk(verbose, url, hostname, token, disk)
-            rescue TokenError => e
-              STDERR.puts e
-              exit 1
-            end
-
-            if disk_req["ok"]
-              puts "Successfully made request to update disk space of vm #{hostname}."
+          if ok
+            if modify_all
+              puts "Successfully modified all VMs."
             else
-              STDERR.puts "Could not modify given host #{hostname} at #{url}."
-              puts disk_req
-              exit 1
+              puts "Successfully modified VM #{hostname}."
             end
+            puts "Use `floaty list --active` to see the results."
           end
         end
       end
@@ -289,95 +178,99 @@ class Vmfloaty
     command :delete do |c|
       c.syntax = 'floaty delete hostname,hostname2 [options]'
       c.summary = 'Schedules the deletion of a host or hosts'
-      c.description = 'Given a comma separated list of hostnames, or --all for all vms, vmfloaty makes a request to vmpooler to schedule the deletion of those vms.'
+      c.description = 'Given a comma separated list of hostnames, or --all for all vms, vmfloaty makes a request to the pooler service to schedule the deletion of those vms.'
       c.example 'Schedules the deletion of a host or hosts', 'floaty delete myhost1,myhost2 --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
+      c.option '--service STRING', String, 'Configured pooler service name'
       c.option '--all', 'Deletes all vms acquired by a token'
       c.option '-f', 'Does not prompt user when deleting all vms'
-      c.option '--token STRING', String, 'Token for vmpooler'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--token STRING', String, 'Token for pooler service'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
+        service = Service.new(options, config)
         hostnames = args[0]
-        token = options.token || config['token']
-        url = options.url ||= config['url']
         delete_all = options.all
         force = options.f
 
-        if delete_all
-          # get vms with token
-          begin
-            running_vms = Utils.get_all_token_vms(verbose, url, token)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
-          rescue Exception => e
-            STDERR.puts e
-            exit 1
-          end
+        failures = []
+        successes = []
 
-          if ! running_vms.nil?
-            Utils.prettyprint_hosts(running_vms, verbose, url)
-            # query y/n
+        if delete_all
+          running_vms = service.list_active(verbose)
+          if running_vms.empty?
+            STDERR.puts "You have no running VMs."
+          else
+            Utils.pretty_print_hosts(verbose, service, running_vms)
+            # Confirm deletion
             puts
-
-            if force
-              ans = true
-            else
-              ans = agree("Delete all VMs associated with token #{token}? [y/N]")
+            confirmed = true
+            unless force
+              confirmed = agree('Delete all these VMs? [y/N]')
             end
-
-            if ans
-              # delete vms
-              puts "Scheduling all vms for for deletion"
-              response = Pooler.delete(verbose, url, running_vms, token)
-              response.each do |host,vals|
-                if vals['ok'] == false
-                  STDERR.puts "There was a problem with your request for vm #{host}."
-                  STDERR.puts vals
+            if confirmed
+              response = service.delete(verbose, running_vms)
+              response.each do |hostname, result|
+                if result['ok']
+                  successes << hostname
+                else
+                  failures << hostname
                 end
               end
             end
           end
-
-          exit 0
-        end
-
-        if hostnames.nil?
+        elsif hostnames || args
+          hostnames = hostnames.split(',')
+          results = service.delete(verbose, hostnames)
+          results.each do |hostname, result|
+            if result['ok']
+              successes << hostname
+            else
+              failures << hostname
+            end
+          end
+        else
           STDERR.puts "You did not provide any hosts to delete"
           exit 1
-        else
-          hosts = hostnames.split(',')
-          begin
-            Pooler.delete(verbose, url, hosts, token)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
+        end
+
+        unless failures.empty?
+          STDERR.puts 'Unable to delete the following VMs:'
+          failures.each do |hostname|
+            STDERR.puts "- #{hostname}"
           end
+          STDERR.puts 'Check `floaty list --active`; Do you need to specify a different service?'
+        end
 
-          puts "Schedulered vmpooler to delete vms #{hosts}."
-          exit 0
+        unless successes.empty?
+          puts unless failures.empty?
+          puts 'Scheduled the following VMs for deletion:'
+          successes.each do |hostname|
+            puts "- #{hostname}"
+          end
         end
+
+        exit 1 unless failures.empty?
       end
     end
 
     command :snapshot do |c|
       c.syntax = 'floaty snapshot hostname [options]'
       c.summary = 'Takes a snapshot of a given vm'
-      c.description = 'Will request a snapshot be taken of the given hostname in vmpooler. This command is known to take a while depending on how much load is on vmpooler.'
+      c.description = 'Will request a snapshot be taken of the given hostname in the pooler service. This command is known to take a while depending on how much load is on the pooler service.'
       c.example 'Takes a snapshot for a given host', 'floaty snapshot myvm.example.com --url http://vmpooler.example.com --token a9znth9dn01t416hrguu56ze37t790bl'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
-      c.option '--token STRING', String, 'Token for vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
+      c.option '--token STRING', String, 'Token for pooler service'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
+        service = Service.new(options, config)
         hostname = args[0]
-        token = options.token ||= config['token']
 
         begin
-          snapshot_req = Pooler.snapshot(verbose, url, hostname, token)
-        rescue TokenError => e
+          snapshot_req = service.snapshot(verbose, hostname)
+        rescue TokenError, ModifyError => e
           STDERR.puts e
           exit 1
         end
@@ -390,17 +283,17 @@ class Vmfloaty
     command :revert do |c|
       c.syntax = 'floaty revert hostname snapshot [options]'
       c.summary = 'Reverts a vm to a specified snapshot'
-      c.description = 'Given a snapshot SHA, vmfloaty will request a revert to vmpooler to go back to a previous snapshot.'
+      c.description = 'Given a snapshot SHA, vmfloaty will request a revert to the pooler service to go back to a previous snapshot.'
       c.example 'Reverts to a snapshot for a given host', 'floaty revert myvm.example.com n4eb4kdtp7rwv4x158366vd9jhac8btq --url http://vmpooler.example.com --token a9znth9dn01t416hrguu56ze37t790bl'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
-      c.option '--token STRING', String, 'Token for vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
+      c.option '--token STRING', String, 'Token for pooler service'
       c.option '--snapshot STRING', String, 'SHA of snapshot'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
+        service = Service.new(options, config)
         hostname = args[0]
-        token = options.token || config['token']
         snapshot_sha = args[1] || options.snapshot
 
         if args[1] && options.snapshot
@@ -408,8 +301,8 @@ class Vmfloaty
         end
 
         begin
-          revert_req = Pooler.revert(verbose, url, hostname, token, snapshot_sha)
-        rescue TokenError => e
+          revert_req = service.revert(verbose, hostname, snapshot_sha)
+        rescue TokenError, ModifyError => e
           STDERR.puts e
           exit 1
         end
@@ -420,42 +313,37 @@ class Vmfloaty
 
     command :status do |c|
       c.syntax = 'floaty status [options]'
-      c.summary = 'Prints the status of pools in vmpooler'
-      c.description = 'Makes a request to vmpooler to request the information about vm pools and how many are ready to be used, what pools are empty, etc.'
-      c.example 'Gets the current vmpooler status', 'floaty status --url http://vmpooler.example.com'
+      c.summary = 'Prints the status of pools in the pooler service'
+      c.description = 'Makes a request to the pooler service to request the information about vm pools and how many are ready to be used, what pools are empty, etc.'
+      c.example 'Gets the current pooler service status', 'floaty status --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.option '--json', 'Prints status in JSON format'
-      c.action do |args, options|
+      c.action do |_, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
-
-        status = Pooler.status(verbose, url)
-        message = status['status']['message']
-        pools = status['pools']
-
+        service = Service.new(options, config)
         if options.json
-          pp status
+          pp service.status(verbose)
         else
-          Utils.prettyprint_status(status, message, pools, verbose)
+          Utils.pretty_print_status(verbose, service)
         end
-
-        exit status['status']['ok']
       end
     end
 
     command :summary do |c|
       c.syntax = 'floaty summary [options]'
-      c.summary = 'Prints a summary of vmpooler'
-      c.description = 'Gives a very detailed summary of information related to vmpooler.'
-      c.example 'Gets the current day summary of vmpooler', 'floaty summary --url http://vmpooler.example.com'
+      c.summary = 'Prints a summary of a pooler service'
+      c.description = 'Gives a very detailed summary of information related to the pooler service.'
+      c.example 'Gets the current day summary of the pooler service', 'floaty summary --url http://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
-      c.action do |args, options|
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
+      c.action do |_, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
+        service = Service.new(options, config)
 
-        summary = Pooler.summary(verbose, url)
+        summary = service.summary(verbose)
         pp summary
         exit 0
       end
@@ -464,54 +352,45 @@ class Vmfloaty
     command :token do |c|
       c.syntax = 'floaty token <get delete status> [options]'
       c.summary = 'Retrieves or deletes a token or checks token status'
-      c.description = 'This command is used to manage your vmpooler token. Through the various options, you are able to get a new token, delete an existing token, and request a tokens status.'
+      c.description = 'This command is used to manage your pooler service token. Through the various options, you are able to get a new token, delete an existing token, and request a tokens status.'
       c.example 'Gets a token from the pooler', 'floaty token get'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.option '--user STRING', String, 'User to authenticate with'
-      c.option '--token STRING', String, 'Token for vmpooler'
+      c.option '--token STRING', String, 'Token for pooler service'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
+        service = Service.new(options, config)
         action = args.first
-        url = options.url ||= config['url']
-        token = args[1] ||= options.token ||= config['token']
-        user = options.user ||= config['user']
-
-        case action
-        when "get"
-          pass = password "Enter your vmpooler password please:", '*'
-          begin
-            token = Auth.get_token(verbose, url, user, pass)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
-          end
-          puts token
-          exit 0
-        when "delete"
-          pass = password "Enter your vmpooler password please:", '*'
-          begin
-            result = Auth.delete_token(verbose, url, user, pass, token)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
-          end
-          puts result
-          exit 0
-        when "status"
-          begin
-            status = Auth.token_status(verbose, url, token)
-          rescue TokenError => e
-            STDERR.puts e
-            exit 1
+
+        begin
+          case action
+            when 'get'
+              token = service.get_new_token(verbose)
+              puts token
+            when 'delete'
+              result = service.delete_token(verbose, options.token)
+              puts result
+            when 'status'
+              token_value = options.token
+              if token_value.nil?
+                token_value = args[1]
+              end
+              status = service.token_status(verbose, token_value)
+              puts status
+            when nil
+              STDERR.puts 'No action provided'
+              exit 1
+            else
+              STDERR.puts "Unknown action: #{action}"
+              exit 1
           end
-          puts status
-          exit 0
-        when nil
-          STDERR.puts "No action provided"
-        else
-          STDERR.puts "Unknown action: #{action}"
+        rescue TokenError => e
+          STDERR.puts e
+          exit 1
         end
+        exit 0
       end
     end
 
@@ -521,16 +400,15 @@ class Vmfloaty
       c.description = 'This command simply will grab a vm template that was requested, and then ssh the user into the machine all at once.'
       c.example 'SSHs into a centos vm', 'floaty ssh centos7 --url https://vmpooler.example.com'
       c.option '--verbose', 'Enables verbose output'
-      c.option '--url STRING', String, 'URL of vmpooler'
+      c.option '--service STRING', String, 'Configured pooler service name'
+      c.option '--url STRING', String, 'URL of pooler service'
       c.option '--user STRING', String, 'User to authenticate with'
-      c.option '--token STRING', String, 'Token for vmpooler'
+      c.option '--token STRING', String, 'Token for pooler service'
       c.option '--notoken', 'Makes a request without a token'
       c.action do |args, options|
         verbose = options.verbose || config['verbose']
-        url = options.url ||= config['url']
-        token = options.token ||= config['token']
-        user = options.user ||= config['user']
-        no_token = options.notoken
+        service = Service.new(options, config)
+        use_token = !options.notoken
 
         if args.empty?
           STDERR.puts "No operating systems provided to obtain. See `floaty ssh --help` for more information on how to get VMs."
@@ -539,25 +417,11 @@ class Vmfloaty
 
         host_os = args.first
 
-        if !no_token && !token
-          puts "No token found. Retrieving a token..."
-          if !user
-            STDERR.puts "You did not provide a user to authenticate to vmpooler with"
-            exit 1
-          end
-          pass = password "Enter your vmpooler password please:", '*'
-          begin
-            token = Auth.get_token(verbose, url, user, pass)
-          rescue TokenError => e
-            STDERR.puts e
-            STDERR.puts 'Could not get token...requesting vm without a token anyway...'
-          else
-            puts "\nToken retrieved!"
-            puts token
-          end
+        if args.length > 1
+          STDERR.puts "Can't ssh to multiple hosts; Using #{host_os} only..."
         end
 
-        Ssh.ssh(verbose, host_os, token, url)
+        service.ssh(verbose, host_os, use_token)
         exit 0
       end
     end
@@ -574,7 +438,7 @@ class Vmfloaty
       EOF
       c.example 'Gets path to bash tab completion script', 'floaty completion --shell bash'
       c.option '--shell STRING', String, 'Shell to request completion script for'
-      c.action do |args, options|
+      c.action do |_, options|
         shell = (options.shell || 'bash').downcase.strip
         completion_file = File.expand_path(File.join('..', '..', 'extras', 'completions', "floaty.#{shell}"), __FILE__)