Refactor upload, get url from modules

To get the url and base url, we had a function in upload.ex.
This included a `case` to check on what module is used.
This means that adding a new upload module implies changing code besides just adding a new module.

Here we try to get the logic properly into the respective modules.
This doesn't add new functionality, but should make it easier to add new uploaders in the future.

lib/pleroma/upload.ex

@@ -86,7 +86,7 @@ def store(upload, opts \\ []) do
          {_, true} <-
            {:description_limit,
             String.length(description) <= Pleroma.Config.get([:instance, :description_limit])},
-         {:ok, url_spec} <- Pleroma.Uploaders.Uploader.put_file(opts.uploader, upload) do
+         {:ok, file} <- Pleroma.Uploaders.Uploader.put_file(opts.uploader, upload) do
       {:ok,
        %{
          "id" => Utils.generate_object_id(),
@@ -96,7 +96,7 @@ def store(upload, opts \\ []) do
            %{
              "type" => "Link",
              "mediaType" => upload.content_type,
-             "href" => url_from_spec(upload, opts.base_url, url_spec)
+             "href" => get_url(upload, file)
            }
            |> Maps.put_if_present("width", upload.width)
            |> Maps.put_if_present("height", upload.height)
@@ -142,8 +142,7 @@ defp get_opts(opts) do
       size_limit: Keyword.get(opts, :size_limit, size_limit),
       uploader: Keyword.get(opts, :uploader, Pleroma.Config.get([__MODULE__, :uploader])),
       filters: Keyword.get(opts, :filters, Pleroma.Config.get([__MODULE__, :filters])),
-      description: Keyword.get(opts, :description),
-      base_url: base_url()
+      description: Keyword.get(opts, :description)
     }
   end
 
@@ -215,7 +214,9 @@ defp tempfile_for_image(data) do
     tmp_path
   end
 
-  defp url_from_spec(%__MODULE__{name: name}, base_url, {:file, path}) do
+  defp get_url(%__MODULE__{name: name}, {:file, path}) do
+    base_url = base_url()
+
     path =
       URI.encode(path, &char_unescaped?/1) <>
         if Pleroma.Config.get([__MODULE__, :link_name], false) do
@@ -228,42 +229,10 @@ defp url_from_spec(%__MODULE__{name: name}, base_url, {:file, path}) do
     |> Path.join()
   end
 
-  defp url_from_spec(_upload, _base_url, {:url, url}), do: url
+  defp get_url(_upload, {:url, url}), do: url
 
   def base_url do
     uploader = Config.get([Pleroma.Upload, :uploader])
-    upload_base_url = Config.get([Pleroma.Upload, :base_url])
-    public_endpoint = Config.get([uploader, :public_endpoint])
-
-    case uploader do
-      Pleroma.Uploaders.Local ->
-        upload_base_url || Pleroma.Web.Endpoint.url() <> "/media/"
-
-      Pleroma.Uploaders.S3 ->
-        bucket = Config.get([Pleroma.Uploaders.S3, :bucket])
-        truncated_namespace = Config.get([Pleroma.Uploaders.S3, :truncated_namespace])
-        namespace = Config.get([Pleroma.Uploaders.S3, :bucket_namespace])
-
-        bucket_with_namespace =
-          cond do
-            !is_nil(truncated_namespace) ->
-              truncated_namespace
-
-            !is_nil(namespace) ->
-              namespace <> ":" <> bucket
-
-            true ->
-              bucket
-          end
-
-        if public_endpoint do
-          Path.join([public_endpoint, bucket_with_namespace])
-        else
-          Path.join([upload_base_url, bucket_with_namespace])
-        end
-
-      _ ->
-        public_endpoint || upload_base_url || Pleroma.Web.Endpoint.url() <> "/media/"
-    end
+    uploader.base_url()
   end
 end

lib/pleroma/uploaders/local.ex

@@ -4,6 +4,7 @@
 
 defmodule Pleroma.Uploaders.Local do
   @behaviour Pleroma.Uploaders.Uploader
+  alias Pleroma.Config
 
   @impl true
   def get_file(_) do
@@ -33,7 +34,7 @@ def put_file(upload) do
   end
 
   def upload_path do
-    Pleroma.Config.get!([__MODULE__, :uploads])
+    Config.get!([__MODULE__, :uploads])
   end
 
   @impl true
@@ -46,4 +47,9 @@ def delete_file(path) do
       {:error, posix_error} -> {:error, to_string(posix_error)}
     end
   end
+
+  @impl true
+  def base_url() do
+    Config.get([Pleroma.Upload, :base_url]) || Pleroma.Web.Endpoint.url() <> "/media/"
+  end
 end

lib/pleroma/uploaders/s3.ex

@@ -67,6 +67,33 @@ def delete_file(file) do
     end
   end
 
+  @impl true
+  def base_url() do
+    upload_base_url = Config.get([Pleroma.Upload, :base_url])
+    public_endpoint = Config.get([Pleroma.Uploaders.S3, :public_endpoint])
+    bucket = Config.get([Pleroma.Uploaders.S3, :bucket])
+    truncated_namespace = Config.get([Pleroma.Uploaders.S3, :truncated_namespace])
+    namespace = Config.get([Pleroma.Uploaders.S3, :bucket_namespace])
+
+    bucket_with_namespace =
+      cond do
+        !is_nil(truncated_namespace) ->
+          truncated_namespace
+
+        !is_nil(namespace) ->
+          namespace <> ":" <> bucket
+
+        true ->
+          bucket
+      end
+
+    if public_endpoint do
+      Path.join([public_endpoint, bucket_with_namespace])
+    else
+      Path.join([upload_base_url, bucket_with_namespace])
+    end
+  end
+
   @regex Regex.compile!("[^0-9a-zA-Z!.*/'()_-]")
   def strict_encode(name) do
     String.replace(name, @regex, "-")

lib/pleroma/uploaders/uploader.ex

@@ -40,6 +40,8 @@ defmodule Pleroma.Uploaders.Uploader do
 
   @callback delete_file(file :: String.t()) :: :ok | {:error, String.t()}
 
+  @callback base_url() :: String.t()
+
   @callback http_callback(Plug.Conn.t(), Map.t()) ::
               {:ok, Plug.Conn.t()}
               | {:ok, Plug.Conn.t(), file_spec()}

test/pleroma/upload_test.exs

@@ -41,6 +41,8 @@ def http_callback(conn, _params),
         do: {:ok, conn, {:file, "post-process-file.jpg"}}
 
       def put_file(upload), do: TestUploaderBase.put_file(upload, __MODULE__)
+
+      def base_url(), do: "http://localhost:4001/media/"
     end
 
     setup do: [uploader: TestUploaderSuccess]