From c3b6c4479f80f5e27a97c3dd216a641bb0e50d8f Mon Sep 17 00:00:00 2001
From: Eduardo Gurgel <edgurgel@gmail.com>
Date: Sat, 28 Dec 2013 16:03:29 -0300
Subject: [PATCH] Add '/redirect-to'

---
 lib/httparrot.ex                     |  3 ++-
 lib/httparrot/redirect_to_handler.ex | 23 +++++++++++++++++++++++
 test/redirect_to_handler_test.exs    | 25 +++++++++++++++++++++++++
 3 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 lib/httparrot/redirect_to_handler.ex
 create mode 100644 test/redirect_to_handler_test.exs

diff --git a/lib/httparrot.ex b/lib/httparrot.ex
index 4987f8c..077fdcb 100644
--- a/lib/httparrot.ex
+++ b/lib/httparrot.ex
@@ -7,7 +7,8 @@ defmodule HTTParrot do
              {'/user-agent', HTTParrot.UserAgentHandler, []},
              {'/headers', HTTParrot.HeadersHandler, []},
              {'/get', HTTParrot.GetHandler, []},
-             {'/status/:code', HTTParrot.StatusCodeHandler, []} ] }
+             {'/status/:code', HTTParrot.StatusCodeHandler, []},
+             {'/redirect-to', HTTParrot.RedirectToHandler, []} ] }
     ])
     {:ok, port} = :application.get_env(:httparrot, :port)
     {:ok, _} = :cowboy.start_http(:http, 100, [port: port], [env: [dispatch: dispatch], onresponse: &prettify_json/4])
diff --git a/lib/httparrot/redirect_to_handler.ex b/lib/httparrot/redirect_to_handler.ex
new file mode 100644
index 0000000..91907c9
--- /dev/null
+++ b/lib/httparrot/redirect_to_handler.ex
@@ -0,0 +1,23 @@
+defmodule HTTParrot.RedirectToHandler do
+  def init(_transport, _req, _opts) do
+    {:upgrade, :protocol, :cowboy_rest}
+  end
+
+  def allowed_methods(req, state) do
+    {["GET", "HEAD", "OPTIONS"], req, state}
+  end
+
+  def malformed_request(req, state) do
+    {url, req} = :cowboy_req.qs_val("url", req, nil)
+    if url, do: {false, req, url}, else: {true, req, state}
+  end
+
+  def resource_exists(req, state), do: {false, req, state}
+  def previously_existed(req, state), do: {true, req, state}
+
+  def moved_permanently(req, url) do
+    {{true, url}, req, url}
+  end
+
+  def terminate(_, _, _), do: :ok
+end
diff --git a/test/redirect_to_handler_test.exs b/test/redirect_to_handler_test.exs
new file mode 100644
index 0000000..782e5ae
--- /dev/null
+++ b/test/redirect_to_handler_test.exs
@@ -0,0 +1,25 @@
+defmodule HTTParrot.RedirectToHandlerTest do
+  use ExUnit.Case
+  import :meck
+  import HTTParrot.RedirectToHandler
+
+  setup do
+    new :cowboy_req
+  end
+
+  teardown do
+    unload :cowboy_req
+  end
+
+  test "malformed_request returns true if no 'url' is defined" do
+    expect(:cowboy_req, :qs_val, [{["url", :req1, nil], {:url, :req2}}])
+    assert malformed_request(:req1, :state) == {false, :req2, :url}
+    assert validate :cowboy_req
+  end
+
+  test "malformed_request returns false if no 'url' is defined" do
+    expect(:cowboy_req, :qs_val, [{["url", :req1, nil], {nil, :req2}}])
+    assert malformed_request(:req1, :state) == {true, :req2, :state}
+    assert validate :cowboy_req
+  end
+end