mirror of
https://github.com/philomena-dev/philomena.git
synced 2024-11-23 20:18:00 +01:00
58 lines
1.4 KiB
Elixir
58 lines
1.4 KiB
Elixir
defmodule PhilomenaWeb.ReloadUserPlug do
|
|
alias Plug.Conn
|
|
alias Pow.Plug
|
|
alias Philomena.Users
|
|
|
|
alias Philomena.UserIps.UserIp
|
|
alias Philomena.UserFingerprints.UserFingerprint
|
|
alias Philomena.Repo
|
|
import Ecto.Query
|
|
|
|
def init(opts), do: opts
|
|
|
|
def call(conn, _opts) do
|
|
config = Plug.fetch_config(conn)
|
|
|
|
case Plug.current_user(conn, config) do
|
|
nil ->
|
|
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
|
|
update = update(UserIp, inc: [uses: 1], set: [updated_at: fragment("now()")])
|
|
|
|
Repo.insert_all(
|
|
UserIp,
|
|
[%{user_id: user.id, ip: ip, uses: 1}],
|
|
conflict_target: [:user_id, :ip],
|
|
on_conflict: update
|
|
)
|
|
end
|
|
|
|
if fp do
|
|
update = update(UserFingerprint, inc: [uses: 1], set: [updated_at: fragment("now()")])
|
|
|
|
Repo.insert_all(
|
|
UserFingerprint,
|
|
[%{user_id: user.id, fingerprint: fp, uses: 1}],
|
|
conflict_target: [:user_id, :fingerprint],
|
|
on_conflict: update
|
|
)
|
|
end
|
|
end
|
|
end
|