get ip address from x-forwarded-for header request, #153

lib/hits.ex

@@ -133,9 +133,22 @@ defmodule Hits do
   end
 
   def get_user_ip_address(conn) do
-    IO.inspect("############ GET IP ADDRESS ##############")
-    IO.inspect(conn.req_headers)
-    IO.inspect("##########################################")
-    Enum.join(Tuple.to_list(conn.remote_ip), ".")
+    get_x_forwarded_for_ip(conn.req_headers) ||
+      Enum.join(Tuple.to_list(conn.remote_ip), ".")
+  end
+
+  defp get_x_forwarded_for_ip(headers) do
+    case Enum.find(headers, fn {k, _} -> k == "x-forwarded-for" end) do
+      # header doesn't exist
+      nil ->
+        nil
+
+      # header exists and can contains multiple ips
+      {_, ips} ->
+        ips
+        |> String.split(",")
+        |> List.first()
+        |> String.trim()
+    end
   end
 end

lib/hits_web/controllers/page_controller.ex

@@ -5,7 +5,7 @@ defmodule HitsWeb.PageController do
     render(conn, "index.html")
   end
 
-  def error(conn, %{"user" => user, "repository" => repository} = params) do
+  def error(conn, %{"user" => user, "repository" => repository}) do
     render(conn, "error.html", user: user, repository: repository)
   end
 end

test/hits_web/controllers/hit_controller_test.exs

@@ -12,6 +12,16 @@ defmodule HitsWeb.HitControllerTest do
     assert res.resp_body =~ Hits.make_badge(1)
   end
 
+  test "GET /user1/repo1.svg", %{conn: conn} do
+    res =
+      put_req_header(conn, "user-agent", "Hackintosh")
+      |> put_req_header("x-forwarded-for", "127.0.0.1, 127.0.0.2")
+      |> put_req_header("accept-language", "en-GB,en;q=0.5")
+      |> get("/user1/repo1.svg")
+
+    assert res.resp_body =~ Hits.make_badge(1)
+  end
+
   test "test counter increments! GET /totes/amaze.svg", %{conn: conn} do
     put_req_header(conn, "user-agent", "Hackintosh")
     |> put_req_header("accept-language", "en-GB,en;q=0.5")