xrires.rb
106 lines
| 2.8 KiB
| text/x-ruby
|
RubyLexer
Eric Davis
|
r2376 | require "cgi" | ||
| require "openid/yadis/xri" | ||||
| require "openid/yadis/xrds" | ||||
| require "openid/fetchers" | ||||
| module OpenID | ||||
| module Yadis | ||||
| module XRI | ||||
| class XRIHTTPError < StandardError; end | ||||
| class ProxyResolver | ||||
| DEFAULT_PROXY = 'http://proxy.xri.net/' | ||||
| def initialize(proxy_url=nil) | ||||
| if proxy_url | ||||
| @proxy_url = proxy_url | ||||
| else | ||||
| @proxy_url = DEFAULT_PROXY | ||||
| end | ||||
| @proxy_url += '/' unless @proxy_url.match('/$') | ||||
| end | ||||
| def query_url(xri, service_type=nil) | ||||
| # URI normal form has a leading xri://, but we need to strip | ||||
| # that off again for the QXRI. This is under discussion for | ||||
| # XRI Resolution WD 11. | ||||
| qxri = XRI.to_uri_normal(xri)[6..-1] | ||||
| hxri = @proxy_url + qxri | ||||
| args = {'_xrd_r' => 'application/xrds+xml'} | ||||
| if service_type | ||||
| args['_xrd_t'] = service_type | ||||
| else | ||||
| # don't perform service endpoint selection | ||||
| args['_xrd_r'] += ';sep=false' | ||||
| end | ||||
| return XRI.append_args(hxri, args) | ||||
| end | ||||
| def query(xri, service_types) | ||||
| # these can be query args or http headers, needn't be both. | ||||
| # headers = {'Accept' => 'application/xrds+xml;sep=true'} | ||||
| canonicalID = nil | ||||
| services = service_types.collect { |service_type| | ||||
| url = self.query_url(xri, service_type) | ||||
| begin | ||||
| response = OpenID.fetch(url) | ||||
| rescue | ||||
| raise XRIHTTPError, ["Could not fetch #{xri}", $!] | ||||
| end | ||||
| raise XRIHTTPError, "Could not fetch #{xri}" if response.nil? | ||||
| xrds = Yadis::parseXRDS(response.body) | ||||
| canonicalID = Yadis::get_canonical_id(xri, xrds) | ||||
| Yadis::services(xrds) unless xrds.nil? | ||||
| } | ||||
| # TODO: | ||||
| # * If we do get hits for multiple service_types, we're almost | ||||
| # certainly going to have duplicated service entries and | ||||
| # broken priority ordering. | ||||
| services = services.inject([]) { |flatter, some_services| | ||||
| flatter += some_services unless some_services.nil? | ||||
| } | ||||
| return canonicalID, services | ||||
| end | ||||
| end | ||||
| def self.urlencode(args) | ||||
| a = [] | ||||
| args.each do |key, val| | ||||
| a << (CGI::escape(key) + "=" + CGI::escape(val)) | ||||
| end | ||||
| a.join("&") | ||||
| end | ||||
| def self.append_args(url, args) | ||||
| return url if args.length == 0 | ||||
| # rstrip question marks | ||||
| rstripped = url.dup | ||||
| while rstripped[-1].chr == '?' | ||||
| rstripped = rstripped[0...rstripped.length-1] | ||||
| end | ||||
| if rstripped.index('?') | ||||
| sep = '&' | ||||
| else | ||||
| sep = '?' | ||||
| end | ||||
| return url + sep + XRI.urlencode(args) | ||||
| end | ||||
| end | ||||
| end | ||||
| end | ||||