mirror of https://github.com/tootsuite/mastodon
33 lines
1.2 KiB
Ruby
33 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Paperclip
|
|
class BlurhashTranscoder < Paperclip::Processor
|
|
def make
|
|
return @file unless options[:style] == :small || options[:blurhash]
|
|
|
|
width, height, data = blurhash_params
|
|
# Guard against segfaults if data has unexpected size
|
|
raise RangeError, "Invalid image data size (expected #{width * height * 3}, got #{data.size})" if data.size != width * height * 3 # TODO: should probably be another exception type
|
|
|
|
attachment.instance.blurhash = Blurhash.encode(width, height, data, **(options[:blurhash] || {}))
|
|
|
|
@file
|
|
rescue Vips::Error => e
|
|
raise Paperclip::Error, "Error while generating blurhash for #{@basename}: #{e}"
|
|
end
|
|
|
|
private
|
|
|
|
def blurhash_params
|
|
if Rails.configuration.x.use_vips
|
|
image = Vips::Image.thumbnail(@file.path, 100)
|
|
[image.width, image.height, image.colourspace(:srgb).extract_band(0, n: 3).to_a.flatten]
|
|
else
|
|
pixels = convert(':source -flatten -depth 8 -compress none RGB:-', source: File.expand_path(@file.path)).unpack('C*')
|
|
geometry = options.fetch(:file_geometry_parser).from_file(@file)
|
|
[geometry.width, geometry.height, pixels]
|
|
end
|
|
end
|
|
end
|
|
end
|