2019-05-02 11:37:35 +02:00
|
|
|
class VirusScannerJob < ApplicationJob
|
2020-09-03 11:00:59 +02:00
|
|
|
# If by the time the job runs the blob has been deleted, ignore the error
|
2020-07-13 14:31:21 +02:00
|
|
|
discard_on ActiveRecord::RecordNotFound
|
2020-09-03 11:00:59 +02:00
|
|
|
# If the file is deleted during the scan, ignore the error
|
|
|
|
discard_on ActiveStorage::FileNotFoundError
|
2021-03-11 18:36:25 +01:00
|
|
|
# If for some reason the file appears invalid, retry for a while
|
2021-04-06 14:59:12 +02:00
|
|
|
retry_on(ActiveStorage::IntegrityError, attempts: 5, wait: 5.seconds) do |job, _error|
|
|
|
|
blob = job.arguments.first
|
|
|
|
|
|
|
|
metadata = {
|
|
|
|
virus_scan_result: ActiveStorage::VirusScanner::INTEGRITY_ERROR,
|
|
|
|
scanned_at: Time.zone.now
|
|
|
|
}
|
|
|
|
|
2021-04-06 18:00:54 +02:00
|
|
|
merge_and_update_metadata(blob, metadata)
|
2021-04-06 14:59:12 +02:00
|
|
|
end
|
2021-03-11 18:36:25 +01:00
|
|
|
|
2019-05-02 11:37:35 +02:00
|
|
|
def perform(blob)
|
2021-03-11 14:42:57 +01:00
|
|
|
if blob.virus_scanner.done? then return end
|
|
|
|
|
2019-05-02 11:37:35 +02:00
|
|
|
metadata = extract_metadata_via_virus_scanner(blob)
|
2022-11-09 12:33:20 +01:00
|
|
|
|
2021-04-06 18:00:54 +02:00
|
|
|
VirusScannerJob.merge_and_update_metadata(blob, metadata)
|
2019-05-02 11:37:35 +02:00
|
|
|
end
|
|
|
|
|
|
|
|
def extract_metadata_via_virus_scanner(blob)
|
|
|
|
ActiveStorage::VirusScanner.new(blob).metadata
|
|
|
|
end
|
2021-04-06 18:00:54 +02:00
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def self.merge_and_update_metadata(blob, metadata)
|
|
|
|
blob.update!(metadata: blob.metadata.merge(metadata))
|
|
|
|
end
|
2019-05-02 11:37:35 +02:00
|
|
|
end
|