defmodule Pleroma.Web.MastodonAPI.MastodonAPI do import Ecto.Query import Ecto.Changeset alias Pleroma.Activity alias Pleroma.Notification alias Pleroma.Pagination alias Pleroma.ScheduledActivity alias Pleroma.User alias Pleroma.Web.CommonAPI def follow(follower, followed, params \\ %{}) do options = cast_params(params) reblogs = options[:reblogs] result = if not User.following?(follower, followed) do CommonAPI.follow(follower, followed) else {:ok, follower, followed, nil} end with {:ok, follower, followed, _} <- result do reblogs |> case do false -> CommonAPI.hide_reblogs(follower, followed) _ -> CommonAPI.show_reblogs(follower, followed) end |> case do {:ok, follower} -> {:ok, follower} _ -> {:ok, follower} end end end def get_followers(user, params \\ %{}) do user |> User.get_followers_query() |> Pagination.fetch_paginated(params) end def get_friends(user, params \\ %{}) do user |> User.get_friends_query() |> Pagination.fetch_paginated(params) end def get_notifications(user, params \\ %{}) do options = cast_params(params) user |> Notification.for_user_query() |> restrict(:exclude_types, options) |> Pagination.fetch_paginated(params) end def get_scheduled_activities(user, params \\ %{}) do user |> ScheduledActivity.for_user_query() |> Pagination.fetch_paginated(params) end defp cast_params(params) do param_types = %{ exclude_types: {:array, :string}, reblogs: :boolean } changeset = cast({%{}, param_types}, params, Map.keys(param_types)) changeset.changes end defp restrict(query, :exclude_types, %{exclude_types: mastodon_types = [_ | _]}) do ap_types = mastodon_types |> Enum.map(&Activity.from_mastodon_notification_type/1) |> Enum.filter(& &1) query |> where([q, a], not fragment("? @> ARRAY[?->>'type']::varchar[]", ^ap_types, a.data)) end defp restrict(query, _, _), do: query end