Contents

begin
  require 'aws-sdk'
rescue LoadError
  raise LoadError.new("Missing required 'aws-sdk'.  Please 'gem install "\
                      "aws-sdk' and require it in your application, or "\
                      "add: gem 'aws-sdk' to your Gemfile.")
end

module SitemapGenerator
  # Class for uploading the sitemaps to an S3 bucket using the plain AWS SDK gem
  class AwsSdkAdapter
    # @param [String] bucket name of the S3 bucket
    # @param [Hash]   opts   alternate means of configuration other than ENV
    # @option opts  [String] :aws_access_key_id instead of ENV['AWS_ACCESS_KEY_ID']
    # @option opts  [String] :aws_region instead of ENV['AWS_REGION']
    # @option opts  [String] :aws_secret_access_key instead of ENV['AWS_SECRET_ACCESS_KEY']
    # @option opts  [String] :path use this prefix on the object key instead of 'sitemaps/'
    def initialize(bucket, opts = {})
      @bucket = bucket

      @aws_access_key_id = opts[:aws_access_key_id] || ENV['AWS_ACCESS_KEY_ID']
      @aws_region = opts[:aws_region] || ENV['AWS_REGION']
      @aws_secret_access_key = opts[:aws_secret_access_key] || ENV['AWS_SECRET_ACCESS_KEY']

      @path = opts[:path] || 'sitemaps/'
    end

    # Call with a SitemapLocation and string data
    def write(location, raw_data)
      SitemapGenerator::FileAdapter.new.write(location, raw_data)

      credentials = Aws::Credentials.new(@aws_access_key_id, @aws_secret_access_key)
      s3 = Aws::S3::Resource.new(credentials: credentials, region: @aws_region)

      s3_object_key = "#{@path}#{location.path_in_public}"
      s3_object = s3.bucket(@bucket).object(s3_object_key)

      content_type = location[:compress] ? 'application/x-gzip' : 'application/xml'
      s3_object.upload_file(location.path,
                            acl: 'public-read',
                            cache_control: 'private, max-age=0, no-cache',
                            content_type: content_type)
    end
  end
end

Version data entries

4 entries across 4 versions & 2 rubygems

Version	Path
enju_leaf-1.2.1	vendor/bundle/ruby/2.3/gems/sitemap_generator-5.3.1/lib/sitemap_generator/adapters/aws_sdk_adapter.rb
sitemap_generator-5.3.1	lib/sitemap_generator/adapters/aws_sdk_adapter.rb
sitemap_generator-5.3.0	lib/sitemap_generator/adapters/aws_sdk_adapter.rb
sitemap_generator-5.2.0	lib/sitemap_generator/adapters/aws_sdk_adapter.rb