git.squeep.com Git - akkoma/blob - lib/pleroma/list.ex

   1 defmodule Pleroma.List do
   2   use Ecto.Schema
   3   import Ecto.{Changeset, Query}
   4   alias Pleroma.{User, Repo}
   5
   6   schema "lists" do
   7     belongs_to(:user, Pleroma.User)
   8     field(:title, :string)
   9     field(:following, {:array, :string}, default: [])
  10
  11     timestamps()
  12   end
  13
  14   def title_changeset(list, attrs \\ %{}) do
  15     list
  16     |> cast(attrs, [:title])
  17     |> validate_required([:title])
  18   end
  19
  20   def follow_changeset(list, attrs \\ %{}) do
  21     list
  22     |> cast(attrs, [:following])
  23     |> validate_required([:following])
  24   end
  25
  26   def for_user(user, opts) do
  27     query =
  28       from(
  29         l in Pleroma.List,
  30         where: l.user_id == ^user.id,
  31         order_by: [desc: l.id],
  32         limit: 50
  33       )
  34
  35     Repo.all(query)
  36   end
  37
  38   def get(id, %{id: user_id} = _user) do
  39     query =
  40       from(
  41         l in Pleroma.List,
  42         where: l.id == ^id,
  43         where: l.user_id == ^user_id
  44       )
  45
  46     Repo.one(query)
  47   end
  48
  49   def get_following(%Pleroma.List{following: following} = list) do
  50     q =
  51       from(
  52         u in User,
  53         where: u.follower_address in ^following
  54       )
  55
  56     {:ok, Repo.all(q)}
  57   end
  58
  59   def rename(%Pleroma.List{} = list, title) do
  60     list
  61     |> title_changeset(%{title: title})
  62     |> Repo.update()
  63   end
  64
  65   def create(title, %User{} = creator) do
  66     list = %Pleroma.List{user_id: creator.id, title: title}
  67     Repo.insert(list)
  68   end
  69
  70   def follow(%Pleroma.List{following: following} = list, %User{} = followed) do
  71     update_follows(list, %{following: Enum.uniq([followed.follower_address | following])})
  72   end
  73
  74   def unfollow(%Pleroma.List{following: following} = list, %User{} = unfollowed) do
  75     update_follows(list, %{following: List.delete(following, unfollowed.follower_address)})
  76   end
  77
  78   def delete(%Pleroma.List{} = list) do
  79     Repo.delete(list)
  80   end
  81
  82   def update_follows(%Pleroma.List{} = list, attrs) do
  83     list
  84     |> follow_changeset(attrs)
  85     |> Repo.update()
  86   end
  87 end