forked from AkkomaGang/akkoma
60 lines
1.4 KiB
Elixir
60 lines
1.4 KiB
Elixir
defmodule Pleroma.Bookmark do
|
|
use Ecto.Schema
|
|
|
|
import Ecto.Changeset
|
|
import Ecto.Query
|
|
|
|
alias Pleroma.Activity
|
|
alias Pleroma.Bookmark
|
|
alias Pleroma.FlakeId
|
|
alias Pleroma.Repo
|
|
alias Pleroma.User
|
|
|
|
@type t :: %__MODULE__{}
|
|
|
|
schema "bookmarks" do
|
|
belongs_to(:user, User, type: FlakeId)
|
|
belongs_to(:activity, Activity, type: FlakeId)
|
|
|
|
timestamps()
|
|
end
|
|
|
|
@spec create(FlakeId.t(), FlakeId.t()) :: {:ok, Bookmark.t()} | {:error, Changeset.t()}
|
|
def create(user_id, activity_id) do
|
|
attrs = %{
|
|
user_id: user_id,
|
|
activity_id: activity_id
|
|
}
|
|
|
|
%Bookmark{}
|
|
|> cast(attrs, [:user_id, :activity_id])
|
|
|> validate_required([:user_id, :activity_id])
|
|
|> unique_constraint(:activity_id, name: :bookmarks_user_id_activity_id_index)
|
|
|> Repo.insert()
|
|
end
|
|
|
|
@spec for_user_query(FlakeId.t()) :: Ecto.Query.t()
|
|
def for_user_query(user_id) do
|
|
Bookmark
|
|
|> where(user_id: ^user_id)
|
|
|> join(:inner, [b], activity in assoc(b, :activity))
|
|
|> preload([b, a], activity: a)
|
|
end
|
|
|
|
def get(user_id, activity_id) do
|
|
Bookmark
|
|
|> where(user_id: ^user_id)
|
|
|> where(activity_id: ^activity_id)
|
|
|> Repo.one()
|
|
end
|
|
|
|
@spec destroy(FlakeId.t(), FlakeId.t()) :: {:ok, Bookmark.t()} | {:error, Changeset.t()}
|
|
def destroy(user_id, activity_id) do
|
|
from(b in Bookmark,
|
|
where: b.user_id == ^user_id,
|
|
where: b.activity_id == ^activity_id
|
|
)
|
|
|> Repo.one()
|
|
|> Repo.delete()
|
|
end
|
|
end
|