103 lines
2.4 KiB
Ruby
103 lines
2.4 KiB
Ruby
|
require 'uri'
|
||
|
|
require 'faraday'
|
||
|
|
require 'yaml'
|
||
|
|
|
||
|
|
class Localizer
|
||
|
|
|
||
|
|
IMAGE_FIELDS = ["media_artist_url"]
|
||
|
|
|
||
|
|
attr_reader :post_directory, :image_directory
|
||
|
|
|
||
|
|
def initialize(post_directory, image_directory)
|
||
|
|
@post_directory = post_directory
|
||
|
|
@image_directory = image_directory
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
def localize_images
|
||
|
|
Dir.foreach(@post_directory) do |file|
|
||
|
|
next if !(file.end_with?('md') || file.end_with?('.markdown'))
|
||
|
|
transform_file("#{@post_directory}#{File::SEPARATOR}#{file}")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
|
||
|
|
def local_image_filename(url)
|
||
|
|
uri = URI(url)
|
||
|
|
"#{image_directory}#{File::SEPARATOR}#{File.basename(uri.path)}"
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
# downloads the file locally if it hasn't already been downloaded
|
||
|
|
def ensure_local_image(url, local_file)
|
||
|
|
return local_file if File.exist?(local_file)
|
||
|
|
|
||
|
|
connection = Faraday.new do |faraday|
|
||
|
|
faraday.adapter(Faraday.default_adapter)
|
||
|
|
end
|
||
|
|
response = download_file_helper(connection, url)
|
||
|
|
File.open(local_file, 'wb') { |file| file << response.body }
|
||
|
|
|
||
|
|
local_file
|
||
|
|
end
|
||
|
|
|
||
|
|
# helper to make sure that we can follow redirects, which can be important for
|
||
|
|
# many of the cloud file sharing services out there.
|
||
|
|
def download_file_helper(connection, url)
|
||
|
|
response = connection.get do |req|
|
||
|
|
req.url(url)
|
||
|
|
end
|
||
|
|
|
||
|
|
case response.status
|
||
|
|
when 302
|
||
|
|
download_file_helper(connection, response['Location'])
|
||
|
|
when 200
|
||
|
|
response
|
||
|
|
else
|
||
|
|
raise "Error downloading file: #{response.status}"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
def transform_file(file)
|
||
|
|
puts "Transforming #{file}"
|
||
|
|
contents = File.read(file, encoding: 'UTF-8')
|
||
|
|
|
||
|
|
# Step 1: get all the images that must be downloaded
|
||
|
|
yaml = YAML.load(contents)
|
||
|
|
|
||
|
|
yaml['talks'].each do |talk|
|
||
|
|
IMAGE_FIELDS.each do |image_field|
|
||
|
|
|
||
|
|
image_url = talk[image_field]
|
||
|
|
|
||
|
|
next unless !image_url.nil? && (image_url.start_with?('http') || image_url.start_with?('https'))
|
||
|
|
|
||
|
|
local_image = local_image_filename(image_url)
|
||
|
|
|
||
|
|
begin
|
||
|
|
ensure_local_image(image_url, local_image)
|
||
|
|
rescue
|
||
|
|
puts "Error downloading locally for image: #{image_url}"
|
||
|
|
next
|
||
|
|
end
|
||
|
|
|
||
|
|
# talk[image_field] = local_image
|
||
|
|
contents = contents.gsub(image_url,"/"+local_image)
|
||
|
|
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# Step 3: replace the file itself
|
||
|
|
File.unlink(file)
|
||
|
|
File.open(file, "w", encoding: "UTF-8") { |file| file.puts contents }
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
localizer = Localizer.new('_posts', 'images')
|
||
|
|
localizer.localize_images
|