class Jerakia::Client

Constants

VERSION

Attributes

config[R]

Public Class Methods

config_file() click to toggle source
# File lib/jerakia/client.rb, line 27
def self.config_file
  [
    File.join(ENV['HOME'] || '', '.jerakia', 'jerakia.yaml'),
    '/etc/jerakia/jerakia.yaml'
  ]. each do |filename|
    return filename if File.exists?(filename)
  end
  return nil
end
load_config_from_file() click to toggle source
# File lib/jerakia/client.rb, line 37
def self.load_config_from_file
  filename = config_file
  confighash = {}
  return {} unless filename
  configdata = YAML.load(File.read(filename))
  if configdata['client'].is_a?(Hash)
    configdata['client'].each do |k, v|
      confighash[k.to_sym] = v
    end
    return confighash
  end
  return {}
end
new(opts={}) click to toggle source
# File lib/jerakia/client.rb, line 17
def initialize(opts={})
  config_file_opts = self.class.load_config_from_file
  @config = default_config.merge(
    config_file_opts.merge(
      opts.reject { |k,v| v.nil? }
    )
  )
  @token = nil
end

Public Instance Methods

default_config() click to toggle source
# File lib/jerakia/client.rb, line 51
def default_config
  {
  :host => 'localhost',
  :port => 9843,
  :api  => 'v1',
  :proto => 'http',
  :content_type => 'json',
  }
end
default_lookup_options() click to toggle source
# File lib/jerakia/client.rb, line 61
def default_lookup_options
  {
    :policy => :default,
    :lookup_type => :first,
  }
end
encode_request(request) click to toggle source
# File lib/jerakia/client.rb, line 102
def encode_request(request)
  if not @config.key?(:content_type) or @config[:content_type] == 'json'
    request.add_field('Content-Type', 'application/json')
    if request.key?('body')
      request.body.to_json
    end
  elsif @config[:content_type] == 'msgpack'
    request.add_field('Content-Type', 'application/x-msgpack')
    if request.key?('body')
      request.body.to_msgpack
    end
  else
    raise Jerakia::Client::Error, "Invalid setting \"#{@config[:content_type]}\" for \":content_type\" in config - supported is either json or msgpack."
    exit(false)
  end
  return request
end
get(url_path, params={}) click to toggle source
# File lib/jerakia/client.rb, line 120
def get(url_path, params={})
  uri = URI.parse(url_address +  url_path)
  uri.query = URI.encode_www_form(params)
  request = Net::HTTP::Get.new(uri.to_s)
  request = encode_request(request)
  return http_send(request)
end
http_send(request) click to toggle source
# File lib/jerakia/client.rb, line 79
def http_send(request)
  request.add_field("X-Authentication",  token)
  response = Net::HTTP.new(@config[:host], @config[:port]).start do |http|
    http.request(request)
  end

  case response.code
  when "200"
    if not @config.key?(:content_type) or @config[:content_type] == 'json'
      return JSON.parse(response.body)
    else @config[:content_type] == 'msgpack'
      return MessagePack.unpack(response.body)
    end
  when "401"
    raise Jerakia::Client::AuthorizationError, "Request not authorized"
  when "501"
    raise Jerakia::Client::ScopeNotFoundError, "Scope data not found" if response.body =~ /No scope data found/
    raise Jerakia::Client::Error, response.body
  else
    raise Jerakia::Client::Error, response.body
  end
end
put(url_path, params) click to toggle source
# File lib/jerakia/client.rb, line 128
def put(url_path, params)
  uri = URI.parse(url_address + url_path)
  request = Net::HTTP::Put.new(uri.path)
  request = set_content_type(request)
  request.body = params
  request = encode_request(request)
  return http_send(request)
end
token() click to toggle source
# File lib/jerakia/client.rb, line 68
def token
  jerakia_token = ENV['JERAKIA_TOKEN'] || @config[:token]
  @token ||= jerakia_token
  raise Jerakia::Client::Error, "No authorization token available" if @token.nil?
  @token
end
url_address() click to toggle source
# File lib/jerakia/client.rb, line 75
def url_address
  "#{@config[:proto]}://#{@config[:host]}:#{@config[:port]}"
end