git.squeep.com Git - akkoma/blob - lib/pleroma/uploaders/local.ex

   1 defmodule Pleroma.Uploaders.Local do
   2   @behaviour Pleroma.Uploaders.Uploader
   3
   4   alias Pleroma.Web
   5
   6   def put_file(name, uuid, tmpfile, _content_type, should_dedupe) do
   7     upload_folder = get_upload_path(uuid, should_dedupe)
   8     url_path = get_url(name, uuid, should_dedupe)
   9
  10     File.mkdir_p!(upload_folder)
  11
  12     result_file = Path.join(upload_folder, name)
  13
  14     if File.exists?(result_file) do
  15       File.rm!(tmpfile)
  16     else
  17       File.cp!(tmpfile, result_file)
  18     end
  19
  20     {:ok, url_path}
  21   end
  22
  23   def upload_path do
  24     settings = Application.get_env(:pleroma, Pleroma.Uploaders.Local)
  25     Keyword.fetch!(settings, :uploads)
  26   end
  27
  28   defp get_upload_path(uuid, should_dedupe) do
  29     if should_dedupe do
  30       upload_path()
  31     else
  32       Path.join(upload_path(), uuid)
  33     end
  34   end
  35
  36   defp get_url(name, uuid, should_dedupe) do
  37     if should_dedupe do
  38       url_for(:cow_uri.urlencode(name))
  39     else
  40       url_for(Path.join(uuid, :cow_uri.urlencode(name)))
  41     end
  42   end
  43
  44   defp url_for(file) do
  45     settings = Application.get_env(:pleroma, Pleroma.Uploaders.Local)
  46
  47     Keyword.get(settings, :uploads_url)
  48     |> String.replace("{{file}}", file)
  49     |> String.replace("{{base_url}}", Web.base_url())
  50   end
  51 end