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

   1 defmodule Pleroma.Filter do
   2   use Ecto.Schema
   3   import Ecto.{Changeset, Query}
   4   alias Pleroma.{User, Repo, Activity}
   5
   6   schema "filters" do
   7     belongs_to(:user, Pleroma.User)
   8     field(:filter_id, :integer)
   9     field(:hide, :boolean, default: false)
  10     field(:whole_word, :boolean, default: true)
  11     field(:phrase, :string)
  12     field(:context, {:array, :string})
  13     field(:expires_at, :utc_datetime)
  14
  15     timestamps()
  16   end
  17
  18   def get(id, %{id: user_id} = _user) do
  19     query =
  20       from(
  21         f in Pleroma.Filter,
  22         where: f.filter_id == ^id,
  23         where: f.user_id == ^user_id
  24       )
  25
  26     Repo.one(query)
  27   end
  28
  29   def get_filters(%Pleroma.User{id: user_id} = user) do
  30     query =
  31       from(
  32         f in Pleroma.Filter,
  33         where: f.user_id == ^user_id
  34       )
  35
  36     Repo.all(query)
  37   end
  38
  39   def create(%Pleroma.Filter{} = filter) do
  40     Repo.insert(filter)
  41   end
  42
  43   def delete(%Pleroma.Filter{id: filter_key} = filter) when is_number(filter_key) do
  44     Repo.delete(filter)
  45   end
  46
  47   def delete(%Pleroma.Filter{id: filter_key} = filter) when is_nil(filter_key) do
  48     %Pleroma.Filter{id: id} = get(filter.filter_id, %{id: filter.user_id})
  49
  50     filter
  51     |> Map.put(:id, id)
  52     |> Repo.delete()
  53   end
  54
  55   def update(%Pleroma.Filter{} = filter) do
  56     destination = Map.from_struct(filter)
  57
  58     Pleroma.Filter.get(filter.filter_id, %{id: filter.user_id})
  59     |> cast(destination, [:phrase, :context, :hide, :expires_at, :whole_word])
  60     |> Repo.update()
  61   end
  62 end