1 files changed, 107 insertions, 0 deletions
diff --git a/lib/nickserver/em_server.rb b/lib/nickserver/em_server.rb
new file mode 100644
index 0000000..faace4b
--- /dev/null
+++ b/lib/nickserver/em_server.rb
@@ -0,0 +1,107 @@
+require 'eventmachine'
+silence_warnings do
+  require 'evma_httpserver'
+end
+
+module Nickserver
+  class EmServer < EM::Connection
+    include EM::HttpServer
+
+    def self.start(options = {})
+      EventMachine.run do
+        EM.start_server options[:host], options[:port], Nickserver::EmServer
+      end
+    end
+
+    def post_init
+      super
+      no_environment_strings
+    end
+
+    def process_http_request
+      uid = get_uid_from_request
+      if uid.nil?
+        send_not_found
+      elsif uid !~ EmailAddress
+        send_error("Not a valid address")
+      else
+        send_key(uid)
+      end
+    rescue RuntimeError => exc
+      puts "Error: #{exc}"
+      puts exc.backtrace
+      send_error(exc.to_s)
+    end
+
+    private
+
+    def send_error(msg = "not supported")
+      send_response(status: 500, content: "500 #{msg}\n")
+    end
+
+    def send_not_found(msg = "Not Found")
+      send_response(status: 404, content: "404 #{msg}\n")
+    end
+
+    def send_response(opts = {})
+      options = {status: 200, content_type: 'text/plain', content: ''}.merge(opts)
+      response = EM::DelegatedHttpResponse.new(self)
+      response.status = options[:status]
+      response.content_type options[:content_type]
+      response.content = options[:content]
+      silence_warnings do
+        response.send_response
+      end
+    end
+
+    def get_uid_from_request
+      if @http_query_string
+        params = CGI.parse(@http_query_string)
+      elsif @http_post_content
+        params = CGI.parse(@http_post_content)
+      end
+      if params && params["address"] && params["address"].any?
+        return params["address"].first
+      else
+        return nil
+      end
+    end
+
+    def send_key(uid)
+      if local_address?(uid)
+        source = Nickserver::CouchDB::Source.new(adapter)
+      else
+        source = Nickserver::Hkp::Source.new(adapter)
+      end
+      source.query(uid) do |response|
+        send_response(status: response.status, content: response.content)
+      end
+    end
+
+    #
+    # Return true if the user address is for a user of this service provider.
+    # e.g. if the provider is example.org, then alice@example.org returns true.
+    #
+    # If 'domain' is not configured, we rely on the Host header of the HTTP request.
+    #
+    def local_address?(uid)
+      uid_domain = uid.sub(/^.*@(.*)$/, "\\1")
+      if Config.domain
+        return uid_domain == Config.domain
+      else
+        # no domain configured, use Host header
+        host_header = @http_headers.split(/\0/).grep(/^Host: /).first
+        if host_header.nil?
+          send_error("HTTP request must include a Host header.")
+        else
+          host = host_header.split(':')[1].strip.sub(/^nicknym\./, '')
+          return uid_domain == host
+        end
+      end
+    end
+
+    def adapter
+      @adapter ||= Nickserver::Adapters::EmHttp.new
+    end
+  end
+end