include Specinfra::Helper::Set

class Koma::Backend::Base
  attr_reader :host, :options

  def initialize(host, options)
    @host = host
    @options = options
  end

  def out(key = nil)
    out = {}
    keys = if key.nil?
             inventory_keys
           else
             key.split(',')
           end
    keys.each do |k|
      begin
        out[k] = Specinfra.backend.host_inventory[k]
        out[k] = Specinfra.backend.host_inventory[k].inspect if k == 'ec2'
      rescue NotImplementedError
        out[k] = nil
      end
    end
    out
  end

  def inventory_keys
    Specinfra::HostInventory.inventory_keys
  end
end