From 2f19e015b691940c644735285c312a958b9d18e6 Mon Sep 17 00:00:00 2001
From: Eduardo Gurgel <edgurgel@gmail.com>
Date: Mon, 30 Dec 2013 17:23:14 -0300
Subject: [PATCH] Add '/post'

TODO:
- [] File/upload data
- [] Invalid JSON
---
 lib/httparrot.ex              |  1 +
 lib/httparrot/post_handler.ex | 37 ++++++++++++++++++++++++++++++++
 test/post_handler_test.exs    | 40 +++++++++++++++++++++++++++++++++++
 3 files changed, 78 insertions(+)
 create mode 100644 lib/httparrot/post_handler.ex
 create mode 100644 test/post_handler_test.exs

diff --git a/lib/httparrot.ex b/lib/httparrot.ex
index 52a73ef..cdd5a50 100644
--- a/lib/httparrot.ex
+++ b/lib/httparrot.ex
@@ -7,6 +7,7 @@ defmodule HTTParrot do
              {'/user-agent', HTTParrot.UserAgentHandler, []},
              {'/headers', HTTParrot.HeadersHandler, []},
              {'/get', HTTParrot.GetHandler, []},
+             {'/post', HTTParrot.PostHandler, []},
              {'/status/:code', HTTParrot.StatusCodeHandler, []},
              {'/redirect-to', HTTParrot.RedirectToHandler, []},
              {'/html', :cowboy_static, {:priv_file, :httparrot, "html.html"}} ] }
diff --git a/lib/httparrot/post_handler.ex b/lib/httparrot/post_handler.ex
new file mode 100644
index 0000000..c3e4d30
--- /dev/null
+++ b/lib/httparrot/post_handler.ex
@@ -0,0 +1,37 @@
+defmodule HTTParrot.PostHandler do
+  alias HTTParrot.GeneralRequestInfo
+
+  def init(_transport, _req, _opts) do
+    {:upgrade, :protocol, :cowboy_rest}
+  end
+
+  def allowed_methods(req, state) do
+    {["POST"], req, state}
+  end
+  def content_types_accepted(req, state) do
+    {[{{"application", "json", :*}, :post_json},
+      {{"application", "x-www-form-urlencoded", :*}, :post_form}], req, state}
+  end
+
+  def post_form(req, state) do
+    {:ok, body, req} = :cowboy_req.body_qs(req)
+    post(req, [form: body, data: "", json: nil])
+  end
+
+  def post_json(req, state) do
+    {:ok, body, req} = :cowboy_req.body(req)
+    post(req, [form: [{}], data: body, json: JSEX.decode!(body)])
+  end
+
+  defp post(req, body) do
+    {info, req} = GeneralRequestInfo.retrieve(req)
+    req = :cowboy_req.set_resp_body(response(info, body), req)
+    {true, req, nil}
+  end
+
+  defp response(info, body) do
+    info ++ body |> JSEX.encode!
+  end
+
+  def terminate(_, _, _), do: :ok
+end
diff --git a/test/post_handler_test.exs b/test/post_handler_test.exs
new file mode 100644
index 0000000..91ec36e
--- /dev/null
+++ b/test/post_handler_test.exs
@@ -0,0 +1,40 @@
+defmodule HTTParrot.PostHandlerTest do
+  use ExUnit.Case
+  import :meck
+  import HTTParrot.PostHandler
+
+  setup do
+    new HTTParrot.GeneralRequestInfo
+    new JSEX
+    new :cowboy_req
+  end
+
+  teardown do
+    unload HTTParrot.GeneralRequestInfo
+    unload JSEX
+    unload :cowboy_req
+  end
+
+  test "returns json with general info and POST form data" do
+    expect(:cowboy_req, :body_qs, 1, {:ok, :body_qs, :req2})
+    expect(:cowboy_req, :set_resp_body, [{[:response, :req3], :req4}])
+    expect(HTTParrot.GeneralRequestInfo, :retrieve, 1, {[:info], :req3})
+    expect(JSEX, :encode!, [{[[:info, {:form, :body_qs}, {:data, ""}, {:json, nil}]], :response}])
+
+    assert post_form(:req1, :state) == {true, :req4, nil}
+
+    assert validate HTTParrot.GeneralRequestInfo
+  end
+
+  test "returns json with general info and POST JSON body data" do
+    expect(:cowboy_req, :body, 1, {:ok, :body, :req2})
+    expect(:cowboy_req, :set_resp_body, [{[:response, :req3], :req4}])
+    expect(HTTParrot.GeneralRequestInfo, :retrieve, 1, {[:info], :req3})
+    expect(JSEX, :decode!, 1, :decoded_json)
+    expect(JSEX, :encode!, [{[[:info, {:form, [{}]}, {:data, :body}, {:json, :decoded_json}]], :response}])
+
+    assert post_json(:req1, :state) == {true, :req4, nil}
+
+    assert validate HTTParrot.GeneralRequestInfo
+  end
+end