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: user_id} = _user, id) 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 = from(
  51       u in User,
  52       where: u.follower_address in ^following
  53     )
  54     {:ok, Repo.all(q)}
  55   end
  56
  57   def rename(%Pleroma.List{} = list, title) do
  58     list
  59     |> title_changeset(%{title: title})
  60     |> Repo.update()
  61   end
  62
  63   def create(title, %User{} = creator) do
  64     list = %Pleroma.List{user_id: creator.id, title: title}
  65     Repo.insert(list)
  66   end
  67
  68   # TODO check that user is following followed
  69   def follow(%Pleroma.List{following: following} = list, %User{} = followed) do
  70     update_follows(list, %{following: Enum.uniq([followed.follower_address | following])})
  71   end
  72
  73   def unfollow(%Pleroma.List{following: following} = list, %User{} = unfollowed) do
  74     update_follows(list, %{following: List.delete(following, unfollowed.follower_address)})
  75   end
  76
  77   def delete(%Pleroma.List{} = list) do
  78     Repo.delete(list)
  79   end
  80
  81   def update_follows(%Pleroma.List{} = list, attrs) do
  82     list
  83     |> follow_changeset(attrs)
  84     |> Repo.update()
  85   end
  86 end