2013-10-14 20:27:41 +08:00
|
|
|
module Jobs
|
|
|
|
class CleanUpUploads < Jobs::Scheduled
|
2014-02-06 07:14:41 +08:00
|
|
|
every 1.hour
|
2013-10-14 20:27:41 +08:00
|
|
|
|
|
|
|
def execute(args)
|
2013-10-16 16:55:42 +08:00
|
|
|
return unless SiteSetting.clean_up_uploads?
|
2013-10-14 20:27:41 +08:00
|
|
|
|
2016-08-02 00:35:57 +08:00
|
|
|
ignore_urls = []
|
|
|
|
ignore_urls |= UserProfile.uniq.where("profile_background IS NOT NULL AND profile_background != ''").pluck(:profile_background)
|
|
|
|
ignore_urls |= UserProfile.uniq.where("card_background IS NOT NULL AND card_background != ''").pluck(:card_background)
|
|
|
|
ignore_urls |= Category.uniq.where("logo_url IS NOT NULL AND logo_url != ''").pluck(:logo_url)
|
|
|
|
ignore_urls |= Category.uniq.where("background_url IS NOT NULL AND background_url != ''").pluck(:background_url)
|
2016-07-01 15:22:30 +08:00
|
|
|
|
2016-09-09 04:58:07 +08:00
|
|
|
# Any URLs in site settings are fair game
|
2016-10-25 16:43:57 +08:00
|
|
|
ignore_urls |= [
|
|
|
|
SiteSetting.logo_url,
|
|
|
|
SiteSetting.logo_small_url,
|
|
|
|
SiteSetting.favicon_url,
|
|
|
|
SiteSetting.apple_touch_icon_url
|
|
|
|
]
|
2016-09-09 04:58:07 +08:00
|
|
|
|
2016-08-02 00:35:57 +08:00
|
|
|
ids = []
|
|
|
|
ids |= PostUpload.uniq.pluck(:upload_id)
|
|
|
|
ids |= User.uniq.where("uploaded_avatar_id IS NOT NULL").pluck(:uploaded_avatar_id)
|
|
|
|
ids |= UserAvatar.uniq.where("gravatar_upload_id IS NOT NULL").pluck(:gravatar_upload_id)
|
2016-07-01 15:22:30 +08:00
|
|
|
|
2013-11-28 05:01:41 +08:00
|
|
|
grace_period = [SiteSetting.clean_orphan_uploads_grace_period_hours, 1].max
|
2013-10-14 20:27:41 +08:00
|
|
|
|
2016-08-02 00:35:57 +08:00
|
|
|
result = Upload.where("retain_hours IS NULL OR created_at < current_timestamp - interval '1 hour' * retain_hours")
|
|
|
|
result = result.where("created_at < ?", grace_period.hour.ago)
|
|
|
|
result = result.where("id NOT IN (?)", ids) if !ids.empty?
|
|
|
|
result = result.where("url NOT IN (?)", ignore_urls) if !ignore_urls.empty?
|
2013-10-14 20:27:41 +08:00
|
|
|
|
2016-08-02 00:35:57 +08:00
|
|
|
result.find_each do |upload|
|
|
|
|
next if QueuedPost.where("raw LIKE '%#{upload.sha1}%'").exists?
|
|
|
|
next if Draft.where("data LIKE '%#{upload.sha1}%'").exists?
|
|
|
|
upload.destroy
|
2016-07-01 15:22:30 +08:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2013-10-14 20:27:41 +08:00
|
|
|
end
|