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

pawoo: rewrite /small/ images to /original/

Browse Source
This commit is contained in:
evazion
2017-06-11 00:00:56 -05:00
parent 14196f36d8
commit 655e53fe65
2 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/logical/downloads/rewrite_strategies/base.rb
View File

@@ -5,6 +5,8 @@
module Downloads module Downloads
module RewriteStrategies module RewriteStrategies
class Base class Base
attr_reader :url
def initialize(url = nil) def initialize(url = nil)
@url = url @url = url
end end

16
app/logical/downloads/rewrite_strategies/pawoo.rb
View File

@@ -1,17 +1,13 @@
module Downloads module Downloads
module RewriteStrategies module RewriteStrategies
class Pawoo < Base class Pawoo < Base
attr_accessor :url, :source
def initialize(url)
@url = url
end
def rewrite(url, headers, data = {}) def rewrite(url, headers, data = {})
if PawooApiClient::Status.is_match?(url) if Sources::Strategies::Pawoo.url_match?(url)
client = PawooApiClient.new source = Sources::Strategies::Pawoo.new(url)
response = client.get_status(url) source.get
url = response.image_url url = source.image_url
elsif url =~ %r!\Ahttps?://img\.pawoo\.net/media_attachments/files/(\d+/\d+/\d+)/small/([a-z0-9]+\.\w+)\z!i
url = "https://img.pawoo.net/media_attachments/files/#{$1}/original/#{$2}"
end end
return [url, headers, data] return [url, headers, data]