10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
# File 'lib/upload_fixer.rb', line 10
def self.fix_extension_on_upload(upload)
is_external = Discourse.store.external?
previous_url = upload.url.dup
source =
if is_external
"https:#{previous_url}"
else
Discourse.store.path_for(upload)
end
correct_extension = FastImage.type(source).to_s.downcase
current_extension = upload.extension.to_s.downcase
if correct_extension.present?
correct_extension = "jpg" if correct_extension == "jpeg"
current_extension = "jpg" if current_extension == "jpeg"
if correct_extension != current_extension
new_filename = change_extension(upload.original_filename, correct_extension)
new_url = change_extension(previous_url, correct_extension)
if is_external
new_url = "/#{new_url}"
source = Discourse.store.get_path_for_upload(upload)
destination = change_extension(source, correct_extension)
Discourse.store.copy_file(previous_url, source, destination)
upload.update!(
original_filename: new_filename,
url: new_url,
extension: correct_extension,
)
DbHelper.remap(previous_url, upload.url)
Discourse.store.remove_file(previous_url, source)
else
destination = change_extension(source, correct_extension)
FileUtils.copy(source, destination)
upload.update!(
original_filename: new_filename,
url: new_url,
extension: correct_extension,
)
DbHelper.remap(previous_url, upload.url)
tombstone_path = source.sub("/uploads/", "/uploads/tombstone/")
FileUtils.mkdir_p(File.dirname(tombstone_path))
FileUtils.move(source, tombstone_path)
end
end
end
rescue => e
STDERR.puts "Skipping upload: failed to correct extension on upload id: #{upload.id} #{current_extension} => #{correct_extension}"
STDERR.puts e
end
|