51 lines
		
	
	
	
		
			1.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			51 lines
		
	
	
	
		
			1.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
defmodule Pleroma.Web.MastodonAPI.MastodonAPI do
 | 
						|
  import Ecto.Query
 | 
						|
  import Ecto.Changeset
 | 
						|
 | 
						|
  alias Pleroma.Activity
 | 
						|
  alias Pleroma.Notification
 | 
						|
  alias Pleroma.Pagination
 | 
						|
  alias Pleroma.User
 | 
						|
 | 
						|
  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
 | 
						|
 | 
						|
  defp cast_params(params) do
 | 
						|
    param_types = %{
 | 
						|
      exclude_types: {:array, :string}
 | 
						|
    }
 | 
						|
 | 
						|
    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
 |