hensei-api/app/services/preview_service/image_fetcher_service.rb
Justin Edmund e3a44ca0d5
Implement embed images (#173)
* Add mini_magick and rufus-scheduler

* Expose attributes and add sigs to AwsService

* Get Party ready for preview state

* Added new fields for preview state and generated_at timestamp
* Add preview state enum to model
* Add preview_relevant_changes? after_commit hook

* Add jobs for generating and cleaning up party previews

* Add new endpoints to PartiesController

* `preview` shows the preview and queues it up for generation if it doesn't exist yet
* `regenerate_preview` allows the party owner to force regeneration of previews

* Schedule jobs

* Stalled jobs are checked every 5 minutes
* Failed jobs are retried every hour
* Old preview jobs are cleaned up daily

* Add the preview service

This is where the bulk of the work is. This service renders out the preview images bit by bit. Currently we render the party name, creator, job icon, and weapon grid.

This includes signatures and some fonts.
2025-01-18 09:08:15 -08:00

69 lines
1.7 KiB
Ruby

# app/services/image_fetcher_service.rb
module PreviewService
class ImageFetcherService
def initialize(aws_service)
@aws_service = aws_service
@tempfiles = []
end
def fetch_s3_image(key, folder = nil)
full_key = folder ? "#{folder}/#{key}" : key
temp_file = create_temp_file
download_from_s3(full_key, temp_file)
create_mini_magick_image(temp_file)
rescue => e
handle_fetch_error(e, full_key)
end
def fetch_job_icon(job_name)
fetch_s3_image("#{job_name.downcase}.png", 'job-icons')
end
def fetch_weapon_image(weapon, mainhand: false)
folder = mainhand ? 'weapon-main' : 'weapon-grid'
fetch_s3_image("#{weapon.granblue_id}.jpg", folder)
end
def fetch_user_picture(picture_identifier)
# Assuming user pictures are stored as PNG in a folder called 'user-pictures'
fetch_s3_image("#{picture_identifier}.png", 'profile')
end
def cleanup
@tempfiles.each do |tempfile|
tempfile.close
tempfile.unlink
end
@tempfiles.clear
end
private
def create_temp_file
temp_file = Tempfile.new(['image', '.jpg'])
temp_file.binmode
@tempfiles << temp_file
temp_file
end
def download_from_s3(key, temp_file)
response = @aws_service.s3_client.get_object(
bucket: @aws_service.bucket,
key: key
)
temp_file.write(response.body.read)
temp_file.rewind
end
def create_mini_magick_image(temp_file)
MiniMagick::Image.new(temp_file.path)
end
def handle_fetch_error(error, key)
Rails.logger.error "Error fetching image #{key}: #{error.message}"
Rails.logger.error error.backtrace.join("\n")
nil
end
end
end