1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.UserInviteToken do
11 alias Pleroma.UserInviteToken
13 @type token :: String.t()
15 schema "user_invite_tokens" do
16 field(:token, :string)
17 field(:used, :boolean, default: false)
18 field(:max_use, :integer)
19 field(:expire_at, :date)
20 field(:uses, :integer, default: 0)
21 field(:invite_type, :string)
26 @spec create_invite(map()) :: UserInviteToken.t()
27 def create_invite(params \\ %{}) do
29 |> cast(params, ~w(max_use expire_at)a)
35 defp add_token(changeset) do
36 token = :crypto.strong_rand_bytes(32) |> Base.url_encode64()
37 put_change(changeset, :token, token)
40 defp assign_type(%{changes: %{max_use: _max_use, expire_at: _expire_at}} = changeset) do
41 put_change(changeset, :invite_type, "reusable_date_limited")
44 defp assign_type(%{changes: %{expire_at: _expire_at}} = changeset) do
45 put_change(changeset, :invite_type, "date_limited")
48 defp assign_type(%{changes: %{max_use: _max_use}} = changeset) do
49 put_change(changeset, :invite_type, "reusable")
52 defp assign_type(changeset), do: put_change(changeset, :invite_type, "one_time")
54 @spec list_invites() :: [UserInviteToken.t()]
56 query = from(u in UserInviteToken, order_by: u.id)
60 @spec update_invite!(UserInviteToken.t(), map()) :: UserInviteToken.t() | no_return()
61 def update_invite!(invite, changes) do
62 change(invite, changes) |> Repo.update!()
65 @spec update_invite(UserInviteToken.t(), map()) ::
66 {:ok, UserInviteToken.t()} | {:error, Changeset.t()}
67 def update_invite(invite, changes) do
68 change(invite, changes) |> Repo.update()
71 @spec find_by_token!(token()) :: UserInviteToken.t() | no_return()
72 def find_by_token!(token), do: Repo.get_by!(UserInviteToken, token: token)
74 @spec find_by_token(token()) :: {:ok, UserInviteToken.t()} | nil
75 def find_by_token(token) do
76 with invite <- Repo.get_by(UserInviteToken, token: token) do
81 @spec valid_invite?(UserInviteToken.t()) :: boolean()
82 def valid_invite?(%{invite_type: "one_time"} = invite) do
86 def valid_invite?(%{invite_type: "date_limited"} = invite) do
87 not_overdue_date?(invite) and not invite.used
90 def valid_invite?(%{invite_type: "reusable"} = invite) do
91 invite.uses < invite.max_use and not invite.used
94 def valid_invite?(%{invite_type: "reusable_date_limited"} = invite) do
95 not_overdue_date?(invite) and invite.uses < invite.max_use and not invite.used
98 defp not_overdue_date?(%{expire_at: expire_at} = invite) do
99 Date.compare(Date.utc_today(), expire_at) in [:lt, :eq] ||
100 (update_invite!(invite, %{used: true}) && false)
103 @spec update_usage!(UserInviteToken.t()) :: nil | UserInviteToken.t() | no_return()
104 def update_usage!(%{invite_type: "date_limited"}), do: nil
106 def update_usage!(%{invite_type: "one_time"} = invite),
107 do: update_invite!(invite, %{used: true})
109 def update_usage!(%{invite_type: invite_type} = invite)
110 when invite_type == "reusable" or invite_type == "reusable_date_limited" do
112 uses: invite.uses + 1
116 if changes.uses >= invite.max_use do
117 Map.put(changes, :used, true)
122 update_invite!(invite, changes)