commit
cd5ff7a943
@ -0,0 +1,30 @@
|
|||||||
|
# Pleroma: A lightweight social networking server
|
||||||
|
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
||||||
|
# SPDX-License-Identifier: AGPL-3.0-only
|
||||||
|
|
||||||
|
defmodule Pleroma.Web.Plugs.UserTrackingPlug do
|
||||||
|
alias Pleroma.User
|
||||||
|
|
||||||
|
import Plug.Conn, only: [assign: 3]
|
||||||
|
|
||||||
|
@update_interval :timer.hours(24)
|
||||||
|
|
||||||
|
def init(opts), do: opts
|
||||||
|
|
||||||
|
def call(%{assigns: %{user: %User{id: id} = user}} = conn, _) when not is_nil(id) do
|
||||||
|
with true <- needs_update?(user),
|
||||||
|
{:ok, user} <- User.update_last_active_at(user) do
|
||||||
|
assign(conn, :user, user)
|
||||||
|
else
|
||||||
|
_ -> conn
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def call(conn, _), do: conn
|
||||||
|
|
||||||
|
defp needs_update?(%User{last_active_at: nil}), do: true
|
||||||
|
|
||||||
|
defp needs_update?(%User{last_active_at: last_active_at}) do
|
||||||
|
NaiveDateTime.diff(NaiveDateTime.utc_now(), last_active_at, :millisecond) >= @update_interval
|
||||||
|
end
|
||||||
|
end
|
@ -0,0 +1,11 @@
|
|||||||
|
defmodule Pleroma.Repo.Migrations.AddLastActiveAtToUsers do
|
||||||
|
use Ecto.Migration
|
||||||
|
|
||||||
|
def change do
|
||||||
|
alter table(:users) do
|
||||||
|
add(:last_active_at, :naive_datetime)
|
||||||
|
end
|
||||||
|
|
||||||
|
create_if_not_exists(index(:users, [:last_active_at]))
|
||||||
|
end
|
||||||
|
end
|
@ -0,0 +1,58 @@
|
|||||||
|
# Pleroma: A lightweight social networking server
|
||||||
|
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
||||||
|
# SPDX-License-Identifier: AGPL-3.0-only
|
||||||
|
|
||||||
|
defmodule Pleroma.Web.Plugs.UserTrackingPlugTest do
|
||||||
|
use Pleroma.Web.ConnCase, async: true
|
||||||
|
|
||||||
|
import Pleroma.Factory
|
||||||
|
|
||||||
|
alias Pleroma.Web.Plugs.UserTrackingPlug
|
||||||
|
|
||||||
|
test "updates last_active_at for a new user", %{conn: conn} do
|
||||||
|
user = insert(:user)
|
||||||
|
|
||||||
|
assert is_nil(user.last_active_at)
|
||||||
|
|
||||||
|
test_started_at = NaiveDateTime.utc_now() |> NaiveDateTime.truncate(:second)
|
||||||
|
|
||||||
|
%{assigns: %{user: user}} =
|
||||||
|
conn
|
||||||
|
|> assign(:user, user)
|
||||||
|
|> UserTrackingPlug.call(%{})
|
||||||
|
|
||||||
|
assert user.last_active_at >= test_started_at
|
||||||
|
assert user.last_active_at <= NaiveDateTime.truncate(NaiveDateTime.utc_now(), :second)
|
||||||
|
end
|
||||||
|
|
||||||
|
test "doesn't update last_active_at if it was updated recently", %{conn: conn} do
|
||||||
|
last_active_at =
|
||||||
|
NaiveDateTime.utc_now()
|
||||||
|
|> NaiveDateTime.add(-:timer.hours(1), :millisecond)
|
||||||
|
|> NaiveDateTime.truncate(:second)
|
||||||
|
|
||||||
|
user = insert(:user, %{last_active_at: last_active_at})
|
||||||
|
|
||||||
|
%{assigns: %{user: user}} =
|
||||||
|
conn
|
||||||
|
|> assign(:user, user)
|
||||||
|
|> UserTrackingPlug.call(%{})
|
||||||
|
|
||||||
|
assert user.last_active_at == last_active_at
|
||||||
|
end
|
||||||
|
|
||||||
|
test "skips updating last_active_at if user ID is nil", %{conn: conn} do
|
||||||
|
%{assigns: %{user: user}} =
|
||||||
|
conn
|
||||||
|
|> assign(:user, %Pleroma.User{})
|
||||||
|
|> UserTrackingPlug.call(%{})
|
||||||
|
|
||||||
|
assert is_nil(user.last_active_at)
|
||||||
|
end
|
||||||
|
|
||||||
|
test "does nothing if user is not present", %{conn: conn} do
|
||||||
|
%{assigns: assigns} = UserTrackingPlug.call(conn, %{})
|
||||||
|
|
||||||
|
refute Map.has_key?(assigns, :user)
|
||||||
|
end
|
||||||
|
end
|
Loading…
Reference in new issue