jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug with copy-pasted urls in uploads/new erroring out

Browse Source
This commit is contained in:
Albert Yi
2018-06-18 10:10:30 -07:00
parent a9c4210bb5
commit cc24b0a54f
3 changed files with 55 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
app/logical/upload_service.rb
View File

@@ -213,6 +213,39 @@ class UploadService
tags << "animated_png" if is_animated_png?(upload, file)
tags.join(" ")
end
def self.download_from_source(source, referer_url: nil)
download = Downloads::File.new(source, referer_url: referer_url)
file = download.download!
context = {
downloaded_source: download.downloaded_source,
source: download.source
}
if download.data[:is_ugoira]
context[:ugoira] = {
frame_data: download.data[:ugoira_frame_data],
content_type: download.data[:ugoira_content_type]
}
end
yield(context)
return file
end
def self.download_for_upload(source, upload)
file = download_from_source(source, referer_url: upload.referer_url) do |context|
upload.downloaded_source = context[:downloaded_source]
upload.source = context[:source]
if context[:ugoira]
upload.context = { ugoira: context[:ugoira] }
end
end
return file
end
end
class Preprocessor
@@ -281,14 +314,7 @@ class UploadService
begin
if source.present?
file = download_from_source(source, referer_url: upload.referer_url) do |context|
upload.downloaded_source = context[:downloaded_source]
upload.source = context[:source]
if context[:ugoira]
upload.context = { ugoira: context[:ugoira] }
end
end
file = Utils.download_for_upload(source, upload)
elsif params[:file].present?
file = params[:file]
end
@@ -314,26 +340,6 @@ class UploadService
pred.save
return pred
end
def download_from_source(source, referer_url: nil)
download = Downloads::File.new(source, referer_url: referer_url)
file = download.download!
context = {
downloaded_source: download.downloaded_source,
source: download.source
}
if download.data[:is_ugoira]
context[:ugoira] = {
frame_data: download.data[:ugoira_frame_data],
content_type: download.data[:ugoira_content_type]
}
end
yield(context)
return file
end
end
class Replacer
@@ -506,10 +512,12 @@ class UploadService
@upload.update(status: "processing")
if @upload.file.nil? && source.present?
@upload.file = Utils.download_for_upload(source, @upload)
end
if @upload.file.present?
Utils.process_file(upload, @upload.file)
else
# sources will be handled in preprocessing now
end
@upload.save!