discourse/lib/file_store/local_store.rb

153 lines
3.8 KiB
Ruby

# frozen_string_literal: true
require "file_store/base_store"
module FileStore
class LocalStore < BaseStore
def store_file(file, path)
copy_file(file, "#{public_dir}#{path}")
"#{Discourse.base_path}#{path}"
end
def remove_file(url, _)
return unless is_relative?(url)
source = "#{public_dir}#{url}"
return unless File.exist?(source)
destination = "#{public_dir}#{url.sub("/uploads/", "/uploads/tombstone/")}"
dir = Pathname.new(destination).dirname
FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
FileUtils.remove(destination) if File.exist?(destination)
FileUtils.move(source, destination, force: true)
FileUtils.touch(destination)
end
def has_been_uploaded?(url)
is_relative?(url) || is_local?(url)
end
def absolute_base_url
"#{Discourse.base_url_no_prefix}#{relative_base_url}"
end
def absolute_base_cdn_url
"#{Discourse.asset_host}#{relative_base_url}"
end
def relative_base_url
File.join(Discourse.base_path, upload_path)
end
def temporary_upload_path(filename)
FileStore::BaseStore.temporary_upload_path(filename, folder_prefix: relative_base_url)
end
def external?
false
end
def download_url(upload)
return unless upload
File.join(relative_base_url, upload.sha1)
end
def cdn_url(url)
UrlHelper.local_cdn_url(url)
end
def path_for(upload)
url = upload.try(:url)
"#{public_dir}#{upload.url}" if url && url[0] == "/" && url[1] != "/"
end
def purge_tombstone(grace_period)
if Dir.exist?(Discourse.store.tombstone_dir)
Discourse::Utils.execute_command(
"find",
tombstone_dir,
"-mtime",
"+#{grace_period}",
"-type",
"f",
"-delete",
)
end
end
def get_path_for(type, upload_id, sha, extension)
prefix_path(super(type, upload_id, sha, extension))
end
def copy_file(file, path)
dir = Pathname.new(path).dirname
FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
# move the file to the right location
# not using mv, cause permissions are no good on move
File.open(path, "wb") { |f| f.write(file.read) }
end
def is_relative?(url)
url.present? && url.start_with?(relative_base_url)
end
def is_local?(url)
return false if url.blank?
absolute_url = url.start_with?("//") ? SiteSetting.scheme + ":" + url : url
absolute_url.start_with?(absolute_base_url) || absolute_url.start_with?(absolute_base_cdn_url)
end
def public_dir
File.join(Rails.root, "public")
end
def tombstone_dir
"#{public_dir}#{relative_base_url.sub("/uploads/", "/uploads/tombstone/")}"
end
def list_missing_uploads(skip_optimized: false)
list_missing(Upload)
list_missing(OptimizedImage) unless skip_optimized
end
def copy_from(source_path)
FileUtils.mkdir_p(File.join(public_dir, upload_path))
Discourse::Utils.execute_command(
"rsync",
"-a",
"--safe-links",
"#{source_path}/",
"#{upload_path}/",
failure_message: "Failed to copy uploads.",
chdir: public_dir,
)
end
private
def list_missing(model)
count = 0
model.find_each do |upload|
# could be a remote image
next unless upload.url =~ %r{\A/[^/]}
path = "#{public_dir}#{upload.url}"
bad = true
begin
bad = false if File.size(path) != 0
rescue StandardError
# something is messed up
end
if bad
count += 1
puts path
end
end
puts "#{count} of #{model.count} #{model.name.underscore.pluralize} are missing" if count > 0
end
def prefix_path(path)
File.join("/", upload_path, path)
end
end
end