350 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			350 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
# Pleroma: A lightweight social networking server
 | 
						|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
 | 
						|
# SPDX-License-Identifier: AGPL-3.0-only
 | 
						|
 | 
						|
defmodule Pleroma.Web.MastodonAPI.AccountView do
 | 
						|
  use Pleroma.Web, :view
 | 
						|
 | 
						|
  alias Pleroma.FollowingRelationship
 | 
						|
  alias Pleroma.User
 | 
						|
  alias Pleroma.UserRelationship
 | 
						|
  alias Pleroma.Web.CommonAPI.Utils
 | 
						|
  alias Pleroma.Web.MastodonAPI.AccountView
 | 
						|
  alias Pleroma.Web.MediaProxy
 | 
						|
 | 
						|
  def render("index.json", %{users: users} = opts) do
 | 
						|
    relationships_opt =
 | 
						|
      cond do
 | 
						|
        Map.has_key?(opts, :relationships) ->
 | 
						|
          opts[:relationships]
 | 
						|
 | 
						|
        is_nil(opts[:for]) ->
 | 
						|
          UserRelationship.view_relationships_option(nil, [])
 | 
						|
 | 
						|
        true ->
 | 
						|
          UserRelationship.view_relationships_option(opts[:for], users)
 | 
						|
      end
 | 
						|
 | 
						|
    opts = Map.put(opts, :relationships, relationships_opt)
 | 
						|
 | 
						|
    users
 | 
						|
    |> render_many(AccountView, "show.json", opts)
 | 
						|
    |> Enum.filter(&Enum.any?/1)
 | 
						|
  end
 | 
						|
 | 
						|
  def render("show.json", %{user: user} = opts) do
 | 
						|
    if User.visible_for?(user, opts[:for]),
 | 
						|
      do: do_render("show.json", opts),
 | 
						|
      else: %{}
 | 
						|
  end
 | 
						|
 | 
						|
  def render("mention.json", %{user: user}) do
 | 
						|
    %{
 | 
						|
      id: to_string(user.id),
 | 
						|
      acct: user.nickname,
 | 
						|
      username: username_from_nickname(user.nickname),
 | 
						|
      url: User.profile_url(user)
 | 
						|
    }
 | 
						|
  end
 | 
						|
 | 
						|
  def render("relationship.json", %{user: nil, target: _target}) do
 | 
						|
    %{}
 | 
						|
  end
 | 
						|
 | 
						|
  def render(
 | 
						|
        "relationship.json",
 | 
						|
        %{user: %User{} = reading_user, target: %User{} = target} = opts
 | 
						|
      ) do
 | 
						|
    user_relationships = get_in(opts, [:relationships, :user_relationships])
 | 
						|
    following_relationships = get_in(opts, [:relationships, :following_relationships])
 | 
						|
 | 
						|
    follow_state =
 | 
						|
      if following_relationships do
 | 
						|
        user_to_target_following_relation =
 | 
						|
          FollowingRelationship.find(following_relationships, reading_user, target)
 | 
						|
 | 
						|
        User.get_follow_state(reading_user, target, user_to_target_following_relation)
 | 
						|
      else
 | 
						|
        User.get_follow_state(reading_user, target)
 | 
						|
      end
 | 
						|
 | 
						|
    followed_by =
 | 
						|
      if following_relationships do
 | 
						|
        case FollowingRelationship.find(following_relationships, target, reading_user) do
 | 
						|
          %{state: "accept"} -> true
 | 
						|
          _ -> false
 | 
						|
        end
 | 
						|
      else
 | 
						|
        User.following?(target, reading_user)
 | 
						|
      end
 | 
						|
 | 
						|
    # NOTE: adjust UserRelationship.view_relationships_option/2 on new relation-related flags
 | 
						|
    %{
 | 
						|
      id: to_string(target.id),
 | 
						|
      following: follow_state == "accept",
 | 
						|
      followed_by: followed_by,
 | 
						|
      blocking:
 | 
						|
        UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :block,
 | 
						|
          reading_user,
 | 
						|
          target,
 | 
						|
          &User.blocks_user?(&1, &2)
 | 
						|
        ),
 | 
						|
      blocked_by:
 | 
						|
        UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :block,
 | 
						|
          target,
 | 
						|
          reading_user,
 | 
						|
          &User.blocks_user?(&1, &2)
 | 
						|
        ),
 | 
						|
      muting:
 | 
						|
        UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :mute,
 | 
						|
          reading_user,
 | 
						|
          target,
 | 
						|
          &User.mutes?(&1, &2)
 | 
						|
        ),
 | 
						|
      muting_notifications:
 | 
						|
        UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :notification_mute,
 | 
						|
          reading_user,
 | 
						|
          target,
 | 
						|
          &User.muted_notifications?(&1, &2)
 | 
						|
        ),
 | 
						|
      subscribing:
 | 
						|
        UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :inverse_subscription,
 | 
						|
          target,
 | 
						|
          reading_user,
 | 
						|
          &User.subscribed_to?(&2, &1)
 | 
						|
        ),
 | 
						|
      requested: follow_state == "pending",
 | 
						|
      domain_blocking: User.blocks_domain?(reading_user, target),
 | 
						|
      showing_reblogs:
 | 
						|
        not UserRelationship.exists?(
 | 
						|
          user_relationships,
 | 
						|
          :reblog_mute,
 | 
						|
          reading_user,
 | 
						|
          target,
 | 
						|
          &User.muting_reblogs?(&1, &2)
 | 
						|
        ),
 | 
						|
      endorsed: false
 | 
						|
    }
 | 
						|
  end
 | 
						|
 | 
						|
  def render("relationships.json", %{user: user, targets: targets} = opts) do
 | 
						|
    relationships_opt =
 | 
						|
      cond do
 | 
						|
        Map.has_key?(opts, :relationships) ->
 | 
						|
          opts[:relationships]
 | 
						|
 | 
						|
        is_nil(opts[:for]) ->
 | 
						|
          UserRelationship.view_relationships_option(nil, [])
 | 
						|
 | 
						|
        true ->
 | 
						|
          UserRelationship.view_relationships_option(user, targets)
 | 
						|
      end
 | 
						|
 | 
						|
    render_opts = %{as: :target, user: user, relationships: relationships_opt}
 | 
						|
    render_many(targets, AccountView, "relationship.json", render_opts)
 | 
						|
  end
 | 
						|
 | 
						|
  defp do_render("show.json", %{user: user} = opts) do
 | 
						|
    user = User.sanitize_html(user, User.html_filter_policy(opts[:for]))
 | 
						|
    display_name = user.name || user.nickname
 | 
						|
 | 
						|
    image = User.avatar_url(user) |> MediaProxy.url()
 | 
						|
    header = User.banner_url(user) |> MediaProxy.url()
 | 
						|
 | 
						|
    following_count =
 | 
						|
      if !user.hide_follows_count or !user.hide_follows or opts[:for] == user do
 | 
						|
        user.following_count || 0
 | 
						|
      else
 | 
						|
        0
 | 
						|
      end
 | 
						|
 | 
						|
    followers_count =
 | 
						|
      if !user.hide_followers_count or !user.hide_followers or opts[:for] == user do
 | 
						|
        user.follower_count || 0
 | 
						|
      else
 | 
						|
        0
 | 
						|
      end
 | 
						|
 | 
						|
    bot = user.actor_type in ["Application", "Service"]
 | 
						|
 | 
						|
    emojis =
 | 
						|
      (user.source_data["tag"] || [])
 | 
						|
      |> Enum.filter(fn %{"type" => t} -> t == "Emoji" end)
 | 
						|
      |> Enum.map(fn %{"icon" => %{"url" => url}, "name" => name} ->
 | 
						|
        %{
 | 
						|
          "shortcode" => String.trim(name, ":"),
 | 
						|
          "url" => MediaProxy.url(url),
 | 
						|
          "static_url" => MediaProxy.url(url),
 | 
						|
          "visible_in_picker" => false
 | 
						|
        }
 | 
						|
      end)
 | 
						|
 | 
						|
    relationship =
 | 
						|
      render("relationship.json", %{
 | 
						|
        user: opts[:for],
 | 
						|
        target: user,
 | 
						|
        relationships: opts[:relationships]
 | 
						|
      })
 | 
						|
 | 
						|
    %{
 | 
						|
      id: to_string(user.id),
 | 
						|
      username: username_from_nickname(user.nickname),
 | 
						|
      acct: user.nickname,
 | 
						|
      display_name: display_name,
 | 
						|
      locked: user.locked,
 | 
						|
      created_at: Utils.to_masto_date(user.inserted_at),
 | 
						|
      followers_count: followers_count,
 | 
						|
      following_count: following_count,
 | 
						|
      statuses_count: user.note_count,
 | 
						|
      note: user.bio || "",
 | 
						|
      url: User.profile_url(user),
 | 
						|
      avatar: image,
 | 
						|
      avatar_static: image,
 | 
						|
      header: header,
 | 
						|
      header_static: header,
 | 
						|
      emojis: emojis,
 | 
						|
      fields: user.fields,
 | 
						|
      bot: bot,
 | 
						|
      source: %{
 | 
						|
        note: (user.bio || "") |> String.replace(~r(<br */?>), "\n") |> Pleroma.HTML.strip_tags(),
 | 
						|
        sensitive: false,
 | 
						|
        fields: user.raw_fields,
 | 
						|
        pleroma: %{
 | 
						|
          discoverable: user.discoverable,
 | 
						|
          actor_type: user.actor_type
 | 
						|
        }
 | 
						|
      },
 | 
						|
 | 
						|
      # Pleroma extension
 | 
						|
      pleroma: %{
 | 
						|
        confirmation_pending: user.confirmation_pending,
 | 
						|
        tags: user.tags,
 | 
						|
        hide_followers_count: user.hide_followers_count,
 | 
						|
        hide_follows_count: user.hide_follows_count,
 | 
						|
        hide_followers: user.hide_followers,
 | 
						|
        hide_follows: user.hide_follows,
 | 
						|
        hide_favorites: user.hide_favorites,
 | 
						|
        relationship: relationship,
 | 
						|
        skip_thread_containment: user.skip_thread_containment,
 | 
						|
        background_image: image_url(user.background) |> MediaProxy.url()
 | 
						|
      }
 | 
						|
    }
 | 
						|
    |> maybe_put_role(user, opts[:for])
 | 
						|
    |> maybe_put_settings(user, opts[:for], opts)
 | 
						|
    |> maybe_put_notification_settings(user, opts[:for])
 | 
						|
    |> maybe_put_settings_store(user, opts[:for], opts)
 | 
						|
    |> maybe_put_chat_token(user, opts[:for], opts)
 | 
						|
    |> maybe_put_activation_status(user, opts[:for])
 | 
						|
    |> maybe_put_follow_requests_count(user, opts[:for])
 | 
						|
    |> maybe_put_allow_following_move(user, opts[:for])
 | 
						|
    |> maybe_put_unread_conversation_count(user, opts[:for])
 | 
						|
  end
 | 
						|
 | 
						|
  defp username_from_nickname(string) when is_binary(string) do
 | 
						|
    hd(String.split(string, "@"))
 | 
						|
  end
 | 
						|
 | 
						|
  defp username_from_nickname(_), do: nil
 | 
						|
 | 
						|
  defp maybe_put_follow_requests_count(
 | 
						|
         data,
 | 
						|
         %User{id: user_id} = user,
 | 
						|
         %User{id: user_id}
 | 
						|
       ) do
 | 
						|
    count =
 | 
						|
      User.get_follow_requests(user)
 | 
						|
      |> length()
 | 
						|
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:follow_requests_count], count)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_follow_requests_count(data, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_settings(
 | 
						|
         data,
 | 
						|
         %User{id: user_id} = user,
 | 
						|
         %User{id: user_id},
 | 
						|
         _opts
 | 
						|
       ) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:source, :privacy], user.default_scope)
 | 
						|
    |> Kernel.put_in([:source, :pleroma, :show_role], user.show_role)
 | 
						|
    |> Kernel.put_in([:source, :pleroma, :no_rich_text], user.no_rich_text)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_settings(data, _, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_settings_store(data, %User{} = user, %User{}, %{
 | 
						|
         with_pleroma_settings: true
 | 
						|
       }) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:pleroma, :settings_store], user.pleroma_settings_store)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_settings_store(data, _, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_chat_token(data, %User{id: id}, %User{id: id}, %{
 | 
						|
         with_chat_token: token
 | 
						|
       }) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:pleroma, :chat_token], token)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_chat_token(data, _, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_role(data, %User{show_role: true} = user, _) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:pleroma, :is_admin], user.is_admin)
 | 
						|
    |> Kernel.put_in([:pleroma, :is_moderator], user.is_moderator)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_role(data, %User{id: user_id} = user, %User{id: user_id}) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in([:pleroma, :is_admin], user.is_admin)
 | 
						|
    |> Kernel.put_in([:pleroma, :is_moderator], user.is_moderator)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_role(data, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_notification_settings(data, %User{id: user_id} = user, %User{id: user_id}) do
 | 
						|
    Kernel.put_in(data, [:pleroma, :notification_settings], user.notification_settings)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_notification_settings(data, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_allow_following_move(data, %User{id: user_id} = user, %User{id: user_id}) do
 | 
						|
    Kernel.put_in(data, [:pleroma, :allow_following_move], user.allow_following_move)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_allow_following_move(data, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_activation_status(data, user, %User{is_admin: true}) do
 | 
						|
    Kernel.put_in(data, [:pleroma, :deactivated], user.deactivated)
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_activation_status(data, _, _), do: data
 | 
						|
 | 
						|
  defp maybe_put_unread_conversation_count(data, %User{id: user_id} = user, %User{id: user_id}) do
 | 
						|
    data
 | 
						|
    |> Kernel.put_in(
 | 
						|
      [:pleroma, :unread_conversation_count],
 | 
						|
      user.unread_conversation_count
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  defp maybe_put_unread_conversation_count(data, _, _), do: data
 | 
						|
 | 
						|
  defp image_url(%{"url" => [%{"href" => href} | _]}), do: href
 | 
						|
  defp image_url(_), do: nil
 | 
						|
end
 |