Justin Edmund
c0922203a7
* Add table for data version and migrate * Modify migration and re-migrate * Create data_version.rb Adds a model for DataVersion * Add aws-sdk-s3 and create aws_service.rb AwsService handles streaming game image files from the Granblue Fantasy server to our S3 instance. * Add importers The Importer libraries take CSV data and import them into the database for each type. We currently support characters, summons and weapons. * Add downloaders Downloaders take Granblue IDs and download images for those items from the Granblue Fantasy server in all relevant sizes. Downloaders can download to disk or stream the file directly to S3. * Create data_importer.rb * Fetches a list of all CSV files present in the updates folder * Checks which have already been imported * Sends unimported data to the appropriate Importer to handle * Create download_manager.rb Creates an appropriate downloader for each Granblue ID it receives * Update download_images.rake Most of this task has been extracted into the Downloader libraries * Update import_data.rake * Create deploy.rake This task is to be run as a post-deploy script. It checks for new unimported data, imports it, then downloads the relevant images to S3 or local disk depending on the parameters provided. * Update credentials.yml.enc * Began working on a README and added example CSVs * Modify importer to handle updates This way we can also add FLBs and other uncaps easier. * Updates only require values that will change When updating a row, fields that don't have a provided value will not be changed * Rebuild search indices in post deploy * Clean up logs with LoggingHelper * More logging adjustments Trying to get a nice-looking output * Change some ASCII characters * Final ASCII changes * Fix issues with Summon and Weapon importers * Finish README for contributing
67 lines
1.7 KiB
Ruby
67 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'aws-sdk-s3'
|
|
|
|
class AwsService
|
|
class ConfigurationError < StandardError; end
|
|
|
|
def initialize
|
|
validate_credentials!
|
|
|
|
@s3_client = Aws::S3::Client.new(
|
|
region: Rails.application.credentials.dig(:aws, :region),
|
|
access_key_id: Rails.application.credentials.dig(:aws, :access_key_id),
|
|
secret_access_key: Rails.application.credentials.dig(:aws, :secret_access_key)
|
|
)
|
|
@bucket = Rails.application.credentials.dig(:aws, :bucket_name)
|
|
rescue KeyError => e
|
|
raise ConfigurationError, "Missing AWS credential: #{e.message}"
|
|
end
|
|
|
|
def upload_stream(io, key)
|
|
@s3_client.put_object(
|
|
bucket: @bucket,
|
|
key: key,
|
|
body: io
|
|
)
|
|
end
|
|
|
|
def file_exists?(key)
|
|
@s3_client.head_object(
|
|
bucket: @bucket,
|
|
key: key
|
|
)
|
|
true
|
|
rescue Aws::S3::Errors::NotFound
|
|
false
|
|
end
|
|
|
|
private
|
|
|
|
def credentials
|
|
@credentials ||= begin
|
|
creds = Rails.application.credentials[:aws]
|
|
raise ConfigurationError, 'AWS credentials not found' unless creds
|
|
|
|
{
|
|
region: creds[:region],
|
|
access_key_id: creds[:access_key_id],
|
|
secret_access_key: creds[:secret_access_key],
|
|
bucket_name: creds[:bucket_name]
|
|
}
|
|
end
|
|
end
|
|
|
|
def validate_credentials!
|
|
missing = []
|
|
creds = Rails.application.credentials[:aws]
|
|
|
|
%i[region access_key_id secret_access_key bucket_name].each do |key|
|
|
missing << key unless creds&.dig(key)
|
|
end
|
|
|
|
return unless missing.any?
|
|
|
|
raise ConfigurationError, "Missing AWS credentials: #{missing.join(', ')}"
|
|
end
|
|
end
|