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