mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-09-20 14:45:56 +08:00
54 lines
1.9 KiB
Ruby
54 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
namespace :active_storage do
|
|
ID_PARTITION_LIMIT = 1_000_000_000
|
|
DIGEST = OpenSSL::Digest.const_get('SHA1').new
|
|
|
|
desc 'Copy all files from Paperclip to ActiveStorage, only same storage types'
|
|
task :migrate_files, [:before] => :environment do |_, _args|
|
|
if ENV['PAPERCLIP_STORAGE'] == 'filesystem' || ENV['ACTIVESTORAGE_SERVICE'] == 'local'
|
|
ActiveStorage::Blob.find_each do |blob|
|
|
src_path = Rails.root.join('public', 'system', blob.key)
|
|
|
|
next unless src_path.exist?
|
|
|
|
blob.transaction do
|
|
blob.key = ActiveStorage::Blob.generate_unique_secure_token
|
|
dst_path = ActiveStorage::Blob.service.path_for(blob.key)
|
|
|
|
puts "Moving #{src_path} to #{dst_path}"
|
|
FileUtils.mkdir_p(File.dirname(dst_path))
|
|
FileUtils.mv(src_path, dst_path)
|
|
blob.save!
|
|
end
|
|
end
|
|
puts 'Finished'
|
|
elsif ENV['PAPERCLIP_STORAGE'] == 's3' || ENV['ACTIVESTORAGE_SERVICE'] == 'amazon'
|
|
S3_BUCKET = Aws::S3::Resource.new.bucket(ENV['S3_BUCKET'])
|
|
|
|
ActiveStorage::Blob.find_each do |blob|
|
|
next unless blob.key.match?(%r{assets|experiments|temp_files|tiny_mce_assets|users|zip_exports\/})
|
|
|
|
src_path = ENV['S3_SUBFOLDER'] ? File.join(ENV['S3_SUBFOLDER'], blob.key) : blob.key
|
|
src_obj = S3_BUCKET.object(src_path)
|
|
|
|
next unless src_obj.exists?
|
|
|
|
blob.transaction do
|
|
blob.key = ActiveStorage::Blob.generate_unique_secure_token
|
|
dst_path = ENV['S3_SUBFOLDER'] ? File.join(ENV['S3_SUBFOLDER'], blob.key) : blob.key
|
|
|
|
puts "Moving #{src_path} to #{dst_path}"
|
|
|
|
src_obj.move_to(bucket: S3_BUCKET.name, key: dst_path)
|
|
blob.save!
|
|
end
|
|
rescue StandardError => e
|
|
puts 'Caught exception copying object ' + src_path + ':'
|
|
puts e.message
|
|
end
|
|
puts 'Finished'
|
|
end
|
|
end
|
|
end
|