82 lines
1.6 KiB
Ruby
82 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module HasUrl
|
|
extend ActiveSupport::Concern
|
|
|
|
REMOTE_URL_SQL = "url LIKE '//%' OR url LIKE 'https://%' OR url LIKE 'http://%'"
|
|
|
|
included do
|
|
scope :remote, -> { where(REMOTE_URL_SQL) }
|
|
scope :local, -> { where.not(REMOTE_URL_SQL) }
|
|
end
|
|
|
|
class_methods do
|
|
def extract_url(url)
|
|
url.match(self::URL_REGEX)
|
|
end
|
|
|
|
def extract_sha1(path)
|
|
data = extract_url(path)
|
|
return if data.blank?
|
|
|
|
sha1 = data[2]
|
|
return if sha1&.length != Upload::SHA1_LENGTH
|
|
|
|
sha1
|
|
end
|
|
|
|
def get_from_url(url)
|
|
return if url.blank?
|
|
|
|
uri =
|
|
begin
|
|
URI(UrlHelper.unencode(url))
|
|
rescue URI::Error
|
|
end
|
|
|
|
return if uri&.path.blank?
|
|
data = extract_url(uri.path)
|
|
|
|
if data.blank?
|
|
result = nil
|
|
result ||= self.find_by(url: uri.path)
|
|
return result
|
|
end
|
|
|
|
result = nil
|
|
|
|
if self.name == "Upload"
|
|
sha1 = data[2]
|
|
result = self.find_by(sha1: sha1) if sha1&.length == Upload::SHA1_LENGTH
|
|
end
|
|
|
|
result || self.find_by("url LIKE ?", "%#{data[1]}")
|
|
end
|
|
|
|
def get_from_urls(upload_urls)
|
|
urls = []
|
|
sha1s = []
|
|
|
|
upload_urls.each do |url|
|
|
next if url.blank?
|
|
|
|
uri =
|
|
begin
|
|
URI(UrlHelper.unencode(url))
|
|
rescue URI::Error
|
|
end
|
|
|
|
next if uri&.path.blank?
|
|
urls << uri.path
|
|
|
|
if data = extract_url(uri.path).presence
|
|
urls << data[1]
|
|
sha1s << data[2] if self.name == "Upload"
|
|
end
|
|
end
|
|
|
|
self.where(url: urls).or(self.where(sha1: sha1s))
|
|
end
|
|
end
|
|
end
|