git.squeep.com Git - akkoma/blob - lib/pleroma/web/pleroma_api/controllers/user_import_controller.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.PleromaAPI.UserImportController do
   6   use Pleroma.Web, :controller
   7
   8   require Logger
   9
  10   alias Pleroma.Plugs.OAuthScopesPlug
  11   alias Pleroma.User
  12
  13   plug(OAuthScopesPlug, %{scopes: ["follow", "write:follows"]} when action == :follow)
  14   plug(OAuthScopesPlug, %{scopes: ["follow", "write:blocks"]} when action == :blocks)
  15   plug(OAuthScopesPlug, %{scopes: ["follow", "write:mutes"]} when action == :mutes)
  16
  17   def follow(conn, %{"list" => %Plug.Upload{path: path}}) do
  18     follow(conn, %{"list" => File.read!(path)})
  19   end
  20
  21   def follow(%{assigns: %{user: follower}} = conn, %{"list" => list}) do
  22     identifiers =
  23       list
  24       |> String.split("\n")
  25       |> Enum.map(&(&1 |> String.split(",") |> List.first()))
  26       |> List.delete("Account address")
  27       |> Enum.map(&(&1 |> String.trim() |> String.trim_leading("@")))
  28       |> Enum.reject(&(&1 == ""))
  29
  30     User.Import.follow_import(follower, identifiers)
  31     json(conn, "job started")
  32   end
  33
  34   def blocks(conn, %{"list" => %Plug.Upload{path: path}}) do
  35     blocks(conn, %{"list" => File.read!(path)})
  36   end
  37
  38   def blocks(%{assigns: %{user: blocker}} = conn, %{"list" => list}) do
  39     User.Import.blocks_import(blocker, prepare_user_identifiers(list))
  40     json(conn, "job started")
  41   end
  42
  43   def mutes(conn, %{"list" => %Plug.Upload{path: path}}) do
  44     mutes(conn, %{"list" => File.read!(path)})
  45   end
  46
  47   def mutes(%{assigns: %{user: user}} = conn, %{"list" => list}) do
  48     User.Import.mutes_import(user, prepare_user_identifiers(list))
  49     json(conn, "job started")
  50   end
  51
  52   defp prepare_user_identifiers(list) do
  53     list
  54     |> String.split()
  55     |> Enum.map(&String.trim_leading(&1, "@"))
  56   end
  57 end