Move http request into its own module

apps/epp_proxy/include/epp_proxy.hrl

@@ -15,5 +15,15 @@
         headers       % [{"User-Agent", <<"EPP proxy">>}, {"Other", <<"Header">>}]
        }).
 
+%% Unified version of the two records above. Depending on the method,
+%% it either encodes parameters into multipart form or query parameters and url.
+-record(epp_unified_request,
+       {method,       % get
+        url,          % "https://example.com/some-url"
+        params,       % {[{<<"msg">>, <<"Some">>}, {<<"code">>, <<"2001">>}]}
+        cookies,      % [<<"session=SomeSession; Version=1">>]
+        headers       % [{"User-Agent", <<"EPP proxy">>}, {"Other", <<"Header">>}]
+       }).
+
 -type epp_request() :: #epp_request{}.
 -type epp_error_request() :: #epp_error_request{}.

apps/epp_proxy/src/epp_http_client.erl

@@ -4,7 +4,7 @@
 
 -behaviour(epp_http_client_behaviour).
 
--export([request/1, error_request/1]).
+-export([request/1, error_request/1, request_builder/1]).
 
 %% Callback API
 request(#epp_request{} = Request) ->
@@ -21,6 +21,9 @@ error_request(#epp_error_request{} = Request) ->
     {ok, Body} = hackney:body(ClientRef),
     {Status, Body}.
 
+request_builder(Map) ->
+    request_from_map(Map).
+
 %% Private API
 -spec handle_args(epp_request()) -> list().
 handle_args(#epp_request{method=get, url=URL, headers=Headers, body="",
@@ -36,3 +39,66 @@ handle_error_args(#epp_error_request{method=get, url=URL, headers=Headers,
     QueryString = hackney_url:qs(Params),
     CompleteURL = [URL, <<"?">>, QueryString],
     [get, CompleteURL, Headers, "", [{cookie, Cookies}, insecure]].
+
+
+%% Map request and return values.
+request_from_map(#{command := "error", session_id := SessionId,
+                   code := Code, message := Message, headers:=Headers,
+                   cl_trid := ClTRID}) ->
+    URL = epp_router:route_request("error"),
+    RequestMethod = epp_router:request_method("error"),
+    Cookie = hackney_cookie:setcookie("session", SessionId, []),
+    QueryParams = query_params(Code, Message, ClTRID),
+    Headers=Headers,
+    Request = #epp_error_request{url=URL,
+                           method=RequestMethod,
+                           query_params=QueryParams,
+                           cookies=[Cookie],
+                           headers=Headers},
+    lager:info("Error Request from map: [~p]~n", [Request]),
+    Request;
+request_from_map(#{command := Command, session_id := SessionId,
+          raw_frame := RawFrame, headers:=Headers, cl_trid := ClTRID}) ->
+    URL = epp_router:route_request(Command),
+    RequestMethod = epp_router:request_method(Command),
+    Cookie = hackney_cookie:setcookie("session", SessionId, []),
+    Body = request_body(Command, RawFrame, ClTRID),
+    Headers=Headers,
+    Request = #epp_request{url=URL,
+                           method=RequestMethod,
+                           body=Body,
+                           cookies=[Cookie],
+                           headers=Headers},
+    lager:info("Request from map: [~p]~n", [Request]),
+    Request;
+request_from_map(#{command := Command, session_id := SessionId,
+          raw_frame := RawFrame, common_name := CommonName,
+          client_cert := ClientCert, peer_ip := PeerIp, cl_trid := ClTRID}) ->
+    URL = epp_router:route_request(Command),
+    RequestMethod = epp_router:request_method(Command),
+    Cookie = hackney_cookie:setcookie("session", SessionId, []),
+    Body = request_body(Command, RawFrame, ClTRID),
+    Headers = [{"SSL_CLIENT_CERT", ClientCert},
+               {"SSL_CLIENT_S_DN_CN", CommonName},
+               {"User-Agent", <<"EPP proxy">>},
+               {"X-Forwarded-for", epp_util:readable_ip(PeerIp)}],
+    Request = #epp_request{url=URL,
+                           method=RequestMethod,
+                           body=Body,
+                           cookies=[Cookie],
+                           headers=Headers},
+    lager:info("Request from map: [~p]~n", [Request]),
+    Request.
+
+%% Return form data or an empty list.
+request_body("hello", _, _) ->
+    "";
+request_body(_Command, RawFrame, nomatch) ->
+    {multipart, [{<<"raw_frame">>, RawFrame}]};
+request_body(_Command, RawFrame, ClTRID) ->
+    {multipart, [{<<"raw_frame">>, RawFrame}, {<<"clTRID">>, ClTRID}]}.
+
+query_params(Code, Message, nomatch) ->
+    [{<<"code">>, Code}, {<<"msg">>, Message}];
+query_params(Code, Message, ClTRID) ->
+    [{<<"code">>, Code}, {<<"msg">>, Message}, {<<"clTRID">>, ClTRID}].

apps/epp_proxy/src/epp_http_client_behaviour.erl

@@ -7,3 +7,5 @@
 %% Abstract module for http client behaviour. It should call EPP HTTP server
 %% and return a response back to the caller.
 -callback request(epp_request()) -> http_response().
+-callback error_request(epp_error_request()) -> http_response().
+-callback request_builder(map()) -> epp_request() | epp_error_request().

apps/epp_proxy/src/epp_tls_worker.erl

@@ -22,7 +22,7 @@
                 headers }).
 
 init(Socket) ->
-    lager:info("Created a worker process"),
+    lager:info("Created a worker process: [~p]", [self()]),
     SessionId = epp_util:session_id(self()),
     {ok, #state{socket=Socket, session_id=SessionId}}.
 
@@ -40,7 +40,7 @@ handle_cast(greeting, State = #state{socket=Socket,
                                      session_id=SessionId,
                                      headers=Headers}) ->
 
-    Request = request_from_map(#{command => "hello",
+    Request = epp_http_client:request_builder(#{command => "hello",
                                                 session_id => SessionId,
                                                 raw_frame => "",
                                                 headers => Headers,
@@ -51,6 +51,9 @@ handle_cast(greeting, State = #state{socket=Socket,
     frame_to_socket(Body, Socket),
     gen_server:cast(self(), process_command),
     {noreply, State#state{socket=Socket, session_id=SessionId}};
+
+%% Main loop of processing commands. Ends the connection when command is logout.
+%%
 handle_cast(process_command,
             State = #state{socket=Socket,session_id=SessionId,
                            headers=Headers}) ->
@@ -58,7 +61,7 @@ handle_cast(process_command,
 
     case parse_frame(RawFrame) of
         #valid_frame{command=Command, cl_trid=ClTRID} ->
-            Request = request_from_map(#{command => Command,
+            Request = epp_http_client:request_builder(#{command => Command,
                                                         session_id => SessionId,
                                                         raw_frame => RawFrame,
                                                         headers => Headers,
@@ -67,7 +70,7 @@ handle_cast(process_command,
             {_Status, Body} = epp_http_client:request(Request);
         #invalid_frame{message=Message, code=Code, cl_trid=ClTRID} ->
             Command = "error",
-            Request = request_from_map(#{command => Command,
+            Request = epp_http_client:request_builder(#{command => Command,
                                                         session_id => SessionId,
                                                         headers => Headers,
                                                         code => Code,
@@ -197,7 +200,6 @@ frame_from_socket(Socket, State) ->
 
     Frame = case read_frame(Socket, Length) of
         {ok, FrameData} ->
-            io:format("~p~n", [FrameData]),
             FrameData;
         {error, _FrameDetails} ->
             {stop, normal, State}