X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fmix%2Ftasks%2Fpleroma%2Frelay.ex;h=c32fcdc33b57216f79db1b2dcbc7e54c6e34cda3;hb=5a405bdadfbad41f08b609925300255b856c2f79;hp=f4b7ff6a0bb34bfe739d865a09eb9ef50f06d330;hpb=5436aaad4f5c60c93c35a827d7e7fb93903b9e0b;p=akkoma

diff --git a/lib/mix/tasks/pleroma/relay.ex b/lib/mix/tasks/pleroma/relay.ex
index f4b7ff6a0..c32fcdc33 100644
--- a/lib/mix/tasks/pleroma/relay.ex
+++ b/lib/mix/tasks/pleroma/relay.ex
@@ -1,43 +1,59 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
 defmodule Mix.Tasks.Pleroma.Relay do
   use Mix.Task
+  import Mix.Pleroma
   alias Pleroma.Web.ActivityPub.Relay
-  alias Mix.Tasks.Pleroma.Common
 
   @shortdoc "Manages remote relays"
-  @moduledoc """
-  Manages remote relays
-
-  ## Follow a remote relay
+  @moduledoc File.read!("docs/docs/administration/CLI_tasks/relay.md")
 
-  ``mix pleroma.relay follow <relay_url>``
+  def run(["follow", target]) do
+    start_pleroma()
 
-  Example: ``mix pleroma.relay follow  https://example.org/relay``
+    with {:ok, _activity} <- Relay.follow(target) do
+      # put this task to sleep to allow the genserver to push out the messages
+      :timer.sleep(500)
+    else
+      {:error, e} -> shell_error("Error while following #{target}: #{inspect(e)}")
+    end
+  end
 
-  ## Unfollow a remote relay
+  def run(["unfollow", target | rest]) do
+    start_pleroma()
 
-  ``mix pleroma.relay unfollow <relay_url>``
+    {options, [], []} =
+      OptionParser.parse(
+        rest,
+        strict: [force: :boolean],
+        aliases: [f: :force]
+      )
 
-  Example: ``mix pleroma.relay unfollow https://example.org/relay``
-  """
-  def run(["follow", target]) do
-    Common.start_pleroma()
+    force = Keyword.get(options, :force, false)
 
-    with {:ok, activity} <- Relay.follow(target) do
+    with {:ok, _activity} <- Relay.unfollow(target, %{force: force}) do
       # put this task to sleep to allow the genserver to push out the messages
       :timer.sleep(500)
     else
-      {:error, e} -> Mix.shell().error("Error while following #{target}: #{inspect(e)}")
+      {:error, e} -> shell_error("Error while following #{target}: #{inspect(e)}")
     end
   end
 
-  def run(["unfollow", target]) do
-    Common.start_pleroma()
+  def run(["list"]) do
+    start_pleroma()
 
-    with {:ok, activity} <- Relay.unfollow(target) do
-      # put this task to sleep to allow the genserver to push out the messages
-      :timer.sleep(500)
+    with {:ok, list} <- Relay.list() do
+      Enum.each(list, &print_relay_url/1)
     else
-      {:error, e} -> Mix.shell().error("Error while following #{target}: #{inspect(e)}")
+      {:error, e} -> shell_error("Error while fetching relay subscription list: #{inspect(e)}")
     end
   end
+
+  defp print_relay_url(%{followed_back: false} = relay) do
+    shell_info("#{relay.actor} - no Accept received (relay didn't follow back)")
+  end
+
+  defp print_relay_url(relay), do: shell_info(relay.actor)
 end