From a3e470659734282a728a1b98b3c3ea6b509ceb63 Mon Sep 17 00:00:00 2001
From: "byte[]" <byteslice@airmail.cc>
Date: Mon, 16 Dec 2019 21:49:17 -0500
Subject: [PATCH] track usages

---
 lib/philomena_web/plugs/reload_user_plug.ex | 33 +++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/lib/philomena_web/plugs/reload_user_plug.ex b/lib/philomena_web/plugs/reload_user_plug.ex
index abaf4794..72f12de7 100644
--- a/lib/philomena_web/plugs/reload_user_plug.ex
+++ b/lib/philomena_web/plugs/reload_user_plug.ex
@@ -1,7 +1,12 @@
 defmodule PhilomenaWeb.ReloadUserPlug do
+  alias Plug.Conn
   alias Pow.Plug
   alias Philomena.Users
 
+  alias Philomena.UserIps.UserIp
+  alias Philomena.UserFingerprints.UserFingerprint
+  alias Philomena.Repo
+
   def init(opts), do: opts
 
   def call(conn, _opts) do
@@ -12,9 +17,37 @@ defmodule PhilomenaWeb.ReloadUserPlug do
         conn
 
       user ->
+        spawn fn -> update_usages(conn, user) end
         reloaded_user = Users.get_by(id: user.id)
 
         Plug.assign_current_user(conn, reloaded_user, config)
     end
   end
+
+  # TODO: move this to a background server instead of spawning
+  # off for every connection
+  defp update_usages(conn, user) do
+    conn = Conn.fetch_cookies(conn)
+
+    {:ok, ip} = EctoNetwork.INET.cast(conn.remote_ip)
+    fp = conn.cookies["_ses"]
+
+    if ip do
+      Repo.insert_all(
+        UserIp,
+        [%{user_id: user.id, ip: ip, uses: 1}],
+        conflict_target: [:user_id, :ip],
+        on_conflict: [inc: [uses: 1]]
+      )
+    end
+
+    if fp do
+      Repo.insert_all(
+        UserFingerprint,
+        [%{user_id: user.id, fingerprint: fp, uses: 1}],
+        conflict_target: [:user_id, :fingerprint],
+        on_conflict: [inc: [uses: 1]]
+      )
+    end
+  end
 end