95 lines
3.0 KiB
Ruby
95 lines
3.0 KiB
Ruby
require 'fog/core'
|
|
require 'fog/xml'
|
|
|
|
module Fog
|
|
module XenServer
|
|
class InvalidLogin < Fog::Errors::Error; end
|
|
class NotFound < Fog::Errors::Error; end
|
|
class RequestFailed < Fog::Errors::Error; end
|
|
|
|
extend Fog::Provider
|
|
|
|
service(:compute, 'Compute')
|
|
|
|
class Connection
|
|
require 'xmlrpc/client'
|
|
attr_reader :credentials
|
|
|
|
def initialize(host, timeout)
|
|
@factory = XMLRPC::Client.new(host, '/')
|
|
@factory.set_parser(NokogiriStreamParser.new)
|
|
@factory.timeout = timeout
|
|
end
|
|
|
|
def find_pool_master( username, password )
|
|
@credentials = authenticate( username, password )
|
|
response = @factory.call('host.get_all_records', @credentials)
|
|
if response['Status'] == "Failure"
|
|
if response['ErrorDescription'][0] == "HOST_IS_SLAVE"
|
|
ip_address = response['ErrorDescription'][1]
|
|
ip_address = ip_address.chomp
|
|
valid = !(IPAddr.new(ip_address) rescue nil).nil?
|
|
if valid
|
|
response['ErrorDescription'][1]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def authenticate( username, password )
|
|
response = @factory.call('session.login_with_password', username.to_s, password.to_s)
|
|
raise Fog::XenServer::InvalidLogin.new unless response["Status"] =~ /Success/
|
|
@credentials = response["Value"]
|
|
end
|
|
|
|
def request(options, *params)
|
|
begin
|
|
parser = options.delete(:parser)
|
|
method = options.delete(:method)
|
|
|
|
if params.empty?
|
|
response = @factory.call(method, @credentials)
|
|
else
|
|
if params.length.eql?(1) and params.first.is_a?(Hash)
|
|
response = @factory.call(method, @credentials, params.first)
|
|
elsif params.length.eql?(2) and params.last.is_a?(Array)
|
|
response = @factory.call(method, @credentials, params.first, params.last)
|
|
else
|
|
response = eval("@factory.call('#{method}', '#{@credentials}', #{params.map {|p| p.is_a?(String) ? "'#{p}'" : p}.join(',')})")
|
|
end
|
|
end
|
|
raise RequestFailed.new("#{method}: " + response["ErrorDescription"].to_s) unless response["Status"].eql? "Success"
|
|
if parser
|
|
parser.parse( response["Value"] )
|
|
response = parser.response
|
|
end
|
|
|
|
response
|
|
end
|
|
end
|
|
end
|
|
|
|
class NokogiriStreamParser < XMLRPC::XMLParser::AbstractStreamParser
|
|
def initialize
|
|
require 'nokogiri/xml/sax/document'
|
|
require 'nokogiri/xml/sax/parser'
|
|
|
|
@parser_class = Class.new(Nokogiri::XML::SAX::Document) do
|
|
|
|
include XMLRPC::XMLParser::StreamParserMixin
|
|
|
|
alias_method :start_element, :startElement
|
|
alias_method :end_element, :endElement
|
|
alias_method :characters, :character
|
|
alias_method :cdata_block, :character
|
|
|
|
def parse(str)
|
|
Nokogiri::XML::SAX::Parser.new(self).parse(str)
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|