openjaf/cenit

require 'aws-sdk'

Aws.config[:ssl_verify_peer] = false

module Cenit
  module FileStore
    module AwsS3Default
      extend Base
      extend self

      def label
        'AWS-S3-Default'
      end

      ###
      # Save file content.
      def save(file, data, options)
        file.instance_variable_set(:@_store_io, nil)
        opts = {
          content_type: file[:contentType],
          metadata: {
            filename: file[:filename]
          }
        }

        object(file) do |obj|
          if data.is_a?(String)
            obj.put(opts.merge(body: file.decode(data)))
          else
            obj.upload_file(data.path, opts)
          end
        end

        unless options[:skip_public_read_status]
          status =
            if options.key?(:public_read)
              options[:public_read]
            else
              file.orm_model.data_type.public_read
            end
          set_public_read(file, status) if status
        end
      end

      ###
      # Read file content.
      def read_from_store(file, len)
        unless (store_io = file.instance_variable_get(:@_store_io))
          store_io = object(file) do |obj|
            if obj.exists?
              obj.get.body
            else
              StringIO.new
            end
          end
          file.instance_variable_set(:@_store_io, store_io)
        end
        store_io.seek(file.cursor)
        store_io.read(len)
      end

      ###
      # Remove file from amazon
      def destroy(file)
        file.instance_variable_set(:@_store_io, nil)
        object(file) { |obj| obj.delete if obj.exists? }
      end

      ###
      # Returns client connection.
      def client
        @client ||= Aws::S3::Client.new(
          region: Cenit.aws_s3_region,
          access_key_id: ENV['AWS_ACCESS_KEY_ID'],
          secret_access_key: ENV['AWS_SECRET_ACCESS_KEY']
        )
      end

      ###
      # Returns the S3 resource.
      def resource
        @resource ||= Aws::S3::Resource.new(client: client)
      end

      ###
      # Returns bucket reference for a given file
      def bucket(file)
        unless (_bucket = file.instance_variable_get(:@_aws_s3_bucket))
          _bucket = resource.bucket(bucket_name(file))
          _bucket.create unless _bucket.exists?
        end
        _bucket
      end

      ###
      # Returns file object reference
      def object(file, &block)
        obj = bucket(file).object(object_key(file))
        if block
          block.call(obj)
        else
          obj
        end
      end

      def set_public_read(file, status)
        acl = status ? 'public-read' : 'private'
        client.put_object_acl(acl: acl, bucket: bucket_name(file), key: object_key(file))
      end

      def tenant_id
        Account.current ? Account.current.id.to_s : :default
      end

      def bucket_name(file)
        "#{Cenit.aws_s3_bucket_prefix}"
      end

      def object_key(file)
        "#{tenant_id}/#{file.orm_model.data_type.id}/#{Digest::MD5.hexdigest(file.id.to_s)}"
      end

      def public_url(file)
        "https://s3-#{Cenit.aws_s3_region}.amazonaws.com/#{bucket_name(file)}/#{object_key(file)}"
      end

      def stored?(file)
        object(file).exists?
      end
    end
  end
end