astroband/ruby-stellar-sdk

require "tomlrb"
require "uri"
require "faraday"
require "json"

module Stellar
  class Federation
    def self.lookup(federated_name)
      _, domain = federated_name.split("*")
      if domain.nil?
        raise InvalidFederationAddress.new
      end

      domain_req = Faraday.new("https://#{domain}/.well-known/stellar.toml").get

      unless domain_req.status == 200
        raise InvalidStellarDomain, "Domain does not contain stellar.toml file"
      end

      fed_server_url = Tomlrb.parse(domain_req.body)["FEDERATION_SERVER"]
      if fed_server_url.nil?
        raise InvalidStellarTOML, "Invalid Stellar TOML file"
      end

      unless fed_server_url&.match?(URI::DEFAULT_PARSER.make_regexp)
        raise InvalidFederationURL, "Invalid Federation Server URL"
      end

      lookup_req = Faraday.new(fed_server_url).get { |req|
        req.params[:q] = federated_name
        req.params[:type] = "name"
      }

      unless lookup_req.status == 200
        raise AccountNotFound, "Account not found"
      end

      JSON.parse(lookup_req.body)["account_id"]
    end
  end

  class AccountNotFound < StandardError; end

  class InvalidStellarTOML < StandardError; end

  class InvalidFederationAddress < StandardError; end

  class InvalidStellarDomain < StandardError; end

  class InvalidFederationURL < StandardError; end
end