46 lines
		
	
	
	
		
			1.1 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			46 lines
		
	
	
	
		
			1.1 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
defmodule Pleroma.Uploaders.S3 do
 | 
						|
  @behaviour Pleroma.Uploaders.Uploader
 | 
						|
  require Logger
 | 
						|
 | 
						|
  # The file name is re-encoded with S3's constraints here to comply with previous links with less strict filenames
 | 
						|
  def get_file(file) do
 | 
						|
    config = Pleroma.Config.get([__MODULE__])
 | 
						|
 | 
						|
    {:ok,
 | 
						|
     {:url,
 | 
						|
      Path.join([
 | 
						|
        Keyword.fetch!(config, :public_endpoint),
 | 
						|
        Keyword.fetch!(config, :bucket),
 | 
						|
        strict_encode(URI.decode(file))
 | 
						|
      ])}}
 | 
						|
  end
 | 
						|
 | 
						|
  def put_file(upload = %Pleroma.Upload{}) do
 | 
						|
    config = Pleroma.Config.get([__MODULE__])
 | 
						|
    bucket = Keyword.get(config, :bucket)
 | 
						|
 | 
						|
    {:ok, file_data} = File.read(upload.tempfile)
 | 
						|
 | 
						|
    s3_name = strict_encode(upload.path)
 | 
						|
 | 
						|
    op =
 | 
						|
      ExAws.S3.put_object(bucket, s3_name, file_data, [
 | 
						|
        {:acl, :public_read},
 | 
						|
        {:content_type, upload.content_type}
 | 
						|
      ])
 | 
						|
 | 
						|
    case ExAws.request(op) do
 | 
						|
      {:ok, _} ->
 | 
						|
        {:ok, {:file, s3_name}}
 | 
						|
 | 
						|
      error ->
 | 
						|
        Logger.error("#{__MODULE__}: #{inspect(error)}")
 | 
						|
        {:error, "S3 Upload failed"}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  @regex Regex.compile!("[^0-9a-zA-Z!.*/'()_-]")
 | 
						|
  def strict_encode(name) do
 | 
						|
    String.replace(name, @regex, "-")
 | 
						|
  end
 | 
						|
end
 |