From 39ccfdc08439401a01c84a4d45488d0145afa937 Mon Sep 17 00:00:00 2001
From: eal <eal@waifu.club>
Date: Tue, 12 Dec 2017 18:35:23 +0200
Subject: [PATCH] Add follow import.

---
 lib/pleroma/web/router.ex                     |  5 +++++
 .../controllers/util_controller.ex            | 20 ++++++++++++++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/lib/pleroma/web/router.ex b/lib/pleroma/web/router.ex
index 6806e8a75..6ec5e3097 100644
--- a/lib/pleroma/web/router.ex
+++ b/lib/pleroma/web/router.ex
@@ -51,6 +51,11 @@ def user_fetcher(username) do
     get "/emoji", UtilController, :emoji
   end
 
+  scope "/api/pleroma", Pleroma.Web.TwitterAPI do
+    pipe_through :authenticated_api
+    post "/follow_import", UtilController, :follow_import
+  end
+
   scope "/oauth", Pleroma.Web.OAuth do
     get "/authorize", OAuthController, :authorize
     post "/authorize", OAuthController, :create_authorization
diff --git a/lib/pleroma/web/twitter_api/controllers/util_controller.ex b/lib/pleroma/web/twitter_api/controllers/util_controller.ex
index de2abd4d1..c76486392 100644
--- a/lib/pleroma/web/twitter_api/controllers/util_controller.ex
+++ b/lib/pleroma/web/twitter_api/controllers/util_controller.ex
@@ -2,7 +2,7 @@ defmodule Pleroma.Web.TwitterAPI.UtilController do
   use Pleroma.Web, :controller
   alias Pleroma.Web
   alias Pleroma.Formatter
-
+  alias Pleroma.Web.ActivityPub.ActivityPub
   alias Pleroma.{Repo, PasswordResetToken, User}
 
   def show_password_reset(conn, %{"token" => token}) do
@@ -73,4 +73,22 @@ def version(conn, _params) do
   def emoji(conn, _params) do
     json conn, Enum.into(Formatter.get_custom_emoji(), %{})
   end
+
+  def follow_import(%{assigns: %{user: user}} = conn, %{"list" => list}) do
+    errors = list
+    |> String.split()
+    |> Enum.map(fn nick ->
+      with %User{} = follower <- User.get_cached_by_ap_id(user.ap_id),
+      %User{} = followed <- User.get_or_fetch_by_nickname(nick),
+      {:ok, follower} <- User.follow(follower, followed),
+      {:ok, _activity} <- ActivityPub.follow(follower, followed) do
+        :ok
+      else
+        _e -> nick
+      end
+    end)
+    |> Enum.reject(fn x -> x == :ok end)
+
+    json conn, %{"failed follows" => errors}
+  end
 end