vmfloaty/lib/vmfloaty/pooler.rb

require 'faraday'
require 'vmfloaty/http'
require 'json'

class Pooler
  def self.list(verbose, url, os_filter=nil)
    conn = Http.get_conn(verbose, url)

    response = conn.get '/vm'
    response_body = JSON.parse(response.body)

    if os_filter
      hosts = response_body.select { |i| i[/#{os_filter}/] }
    else
      hosts = response_body
    end

    hosts
  end

  def self.retrieve(verbose, os_type, token, url)
    conn = Http.get_conn(verbose, url)
    if token
      conn.headers['X-AUTH-TOKEN'] = token
    end

    os_string = ""
    os_type.each do |os,num|
      num.times do |i|
        os_string << os+"+"
      end
    end

    os_string = os_string.chomp("+")

    if os_string.size == 0
      STDERR.puts "No request was made, os hash specified no vms #{os_type}"
      exit 1
    end

    response = conn.post "/vm/#{os_string}"

    res_body = JSON.parse(response.body)
    res_body
  end

  def self.modify(verbose, url, hostname, token, lifetime, tags)
    modify_body = {}
    if lifetime
      modify_body['lifetime'] = lifetime
    end
    if tags
      modify_body['tags'] = tags
    end

    conn = Http.get_conn(verbose, url)
    conn.headers['X-AUTH-TOKEN'] = token

    response = conn.put do |req|
      req.url "/vm/#{hostname}"
      req.body = modify_body
    end
    res_body = JSON.parse(response.body)
    res_body
  end

  def self.delete(verbose, url, hostnames, token)
    if hostnames.nil?
      STDERR.puts "You did not provide any hosts to delete"
      exit 1
    end

    hosts = hostnames.split(',')
    conn = Http.get_conn(verbose, url)

    if token
      conn.headers['X-AUTH-TOKEN'] = token
    end

    hosts.each do |host|
      puts "Scheduling host #{host} for deletion"
      response = conn.delete "/vm/#{host}"
      res_body = JSON.parse(response.body)
      if res_body['ok']
        puts "Deletion for vm #{host} successfully scheduled"
      else
        STDERR.puts "There was a problem with your request for vm #{host}"
      end
    end
  end

  def self.status(verbose, url)
    conn = Http.get_conn(verbose, url)

    response = conn.get '/status'
    res_body = JSON.parse(response.body)
    res_body
  end

  def self.summary(verbose, url)
    conn = Http.get_conn(verbose, url)

    response = conn.get '/summary'
    res_body = JSON.parse(response.body)
    res_body
  end

  def self.query(verbose, url, hostname)
    conn = Http.get_conn(verbose, url)

    response = conn.get "/vm/#{hostname}"
    res_body = JSON.parse(response.body)

    res_body
  end

  def self.snapshot(verbose, url, hostname, token)
    conn = Http.get_conn(verbose, url)
    conn.headers['X-AUTH-TOKEN'] = token

    response = conn.post "/vm/#{hostname}/snapshot"
    res_body = JSON.parse(response.body)
    res_body
  end

  def self.revert(verbose, url, hostname, token, snapshot_sha)
    conn = Http.get_conn(verbose, url)
    conn.headers['X-AUTH-TOKEN'] = token

    response = conn.post "/vm/#{hostname}/snapshot/#{snapshot}"
    res_body = JSON.parse(response.body)
    res_body
  end
end