pleroma

media_controller.ex (3061B)

# Pleroma: A lightweight social networking server
# Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Web.MastodonAPI.MediaController do
  use Pleroma.Web, :controller
  alias Pleroma.Object
  alias Pleroma.User
  alias Pleroma.Web.ActivityPub.ActivityPub
  alias Pleroma.Web.Plugs.OAuthScopesPlug
  action_fallback(Pleroma.Web.MastodonAPI.FallbackController)
  plug(Majic.Plug, [pool: Pleroma.MajicPool] when action in [:create, :create2])
  plug(Pleroma.Web.ApiSpec.CastAndValidate, replace_params: false)
  plug(OAuthScopesPlug, %{scopes: ["read:media"]} when action == :show)
  plug(OAuthScopesPlug, %{scopes: ["write:media"]} when action != :show)
  defdelegate open_api_operation(action), to: Pleroma.Web.ApiSpec.MediaOperation
  @doc "POST /api/v1/media"
  def create(
        %{assigns: %{user: user}, private: %{open_api_spex: %{body_params: %{file: file} = data}}} =
          conn,
        _
      ) do
    with {:ok, object} <-
           ActivityPub.upload(
             file,
             actor: User.ap_id(user),
             description: Map.get(data, :description)
           ) do
      attachment_data = Map.put(object.data, "id", object.id)
      render(conn, "attachment.json", %{attachment: attachment_data})
    end
  end
  def create(_conn, _data), do: {:error, :bad_request}
  @doc "POST /api/v2/media"
  def create2(
        %{assigns: %{user: user}, private: %{open_api_spex: %{body_params: %{file: file} = data}}} =
          conn,
        _
      ) do
    with {:ok, object} <-
           ActivityPub.upload(
             file,
             actor: User.ap_id(user),
             description: Map.get(data, :description)
           ) do
      attachment_data = Map.put(object.data, "id", object.id)
      render(conn, "attachment.json", %{attachment: attachment_data})
    end
  end
  def create2(_conn, _data), do: {:error, :bad_request}
  @doc "PUT /api/v1/media/:id"
  def update(
        %{
          assigns: %{user: user},
          private: %{
            open_api_spex: %{body_params: %{description: description}, params: %{id: id}}
          }
        } = conn,
        _
      ) do
    with %Object{} = object <- Object.get_by_id(id),
         :ok <- Object.authorize_access(object, user),
         {:ok, %Object{data: data}} <- Object.update_data(object, %{"name" => description}) do
      attachment_data = Map.put(data, "id", object.id)
      render(conn, "attachment.json", %{attachment: attachment_data})
    end
  end
  def update(conn, data), do: show(conn, data)
  @doc "GET /api/v1/media/:id"
  def show(%{assigns: %{user: user}, private: %{open_api_spex: %{params: %{id: id}}}} = conn, _) do
    with %Object{data: data, id: object_id} = object <- Object.get_by_id(id),
         :ok <- Object.authorize_access(object, user) do
      attachment_data = Map.put(data, "id", object_id)
      render(conn, "attachment.json", %{attachment: attachment_data})
    end
  end
  def show(_conn, _data), do: {:error, :bad_request}
end