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 get_file(_) do
   7     {:ok, {:static_dir, upload_path()}}
   8   end
   9
  10   def put_file(name, uuid, tmpfile, _content_type, opts) do
  11     upload_folder = get_upload_path(uuid, opts.dedupe)
  12
  13     File.mkdir_p!(upload_folder)
  14
  15     result_file = Path.join(upload_folder, name)
  16
  17     if File.exists?(result_file) do
  18       File.rm!(tmpfile)
  19     else
  20       File.cp!(tmpfile, result_file)
  21     end
  22
  23     {:ok, {:file, get_url(name, uuid, opts.dedupe)}}
  24   end
  25
  26   def upload_path do
  27     Pleroma.Config.get!([__MODULE__, :uploads])
  28   end
  29
  30   defp get_upload_path(uuid, should_dedupe) do
  31     if should_dedupe do
  32       upload_path()
  33     else
  34       Path.join(upload_path(), uuid)
  35     end
  36   end
  37
  38   defp get_url(name, uuid, should_dedupe) do
  39     if should_dedupe do
  40       :cow_uri.urlencode(name)
  41     else
  42       Path.join(uuid, :cow_uri.urlencode(name))
  43     end
  44   end
  45 end