250 lines
7.2 KiB
Ruby
250 lines
7.2 KiB
Ruby
require 'resolv-replace'
|
|
|
|
class PixivApiClient
|
|
extend Memoist
|
|
|
|
API_VERSION = "1"
|
|
CLIENT_ID = "bYGKuGVw91e0NMfPGp44euvGt59s"
|
|
CLIENT_SECRET = "HP3RmkgAmEGro0gn1x9ioawQE8WMfvLXDz3ZqxpK"
|
|
|
|
# Tools to not include in the tags list. We don't tag digital media, so
|
|
# including these results in bad translated tags suggestions.
|
|
TOOLS_BLACKLIST = %w[
|
|
Photoshop Illustrator Fireworks Flash Painter PaintShopPro pixiv\ Sketch
|
|
CLIP\ STUDIO\ PAINT IllustStudio ComicStudio RETAS\ STUDIO SAI PhotoStudio
|
|
Pixia NekoPaint PictBear openCanvas ArtRage Expression Inkscape GIMP
|
|
CGillust COMICWORKS MS_Paint EDGE AzPainter AzPainter2 AzDrawing
|
|
PicturePublisher SketchBookPro Processing 4thPaint GraphicsGale mdiapp
|
|
Paintgraphic AfterEffects drawr CLIP\ PAINT\ Lab FireAlpaca Pixelmator
|
|
AzDrawing2 MediBang\ Paint Krita ibisPaint Procreate Live2D
|
|
Lightwave3D Shade Poser STRATA AnimationMaster XSI CARRARA CINEMA4D Maya
|
|
3dsMax Blender ZBrush Metasequoia Sunny3D Bryce Vue Hexagon\ King SketchUp
|
|
VistaPro Sculptris Comi\ Po! modo DAZ\ Studio 3D-Coat
|
|
]
|
|
|
|
class Error < Exception ; end
|
|
class BadIDError < Error ; end
|
|
|
|
class WorkResponse
|
|
attr_reader :json, :pages, :name, :moniker, :user_id, :page_count, :tags
|
|
attr_reader :artist_commentary_title, :artist_commentary_desc
|
|
|
|
def initialize(json)
|
|
@json = json
|
|
@name = json["user"]["name"]
|
|
@user_id = json["user"]["id"]
|
|
@moniker = json["user"]["account"]
|
|
@page_count = json["page_count"].to_i
|
|
@artist_commentary_title = json["title"].to_s
|
|
@artist_commentary_desc = json["caption"].to_s
|
|
@tags = json["tags"].reject {|x| x =~ /^http:/}
|
|
@tags += json["tools"] - TOOLS_BLACKLIST
|
|
|
|
if json["metadata"]
|
|
if json["metadata"]["zip_urls"]
|
|
@pages = json["metadata"]["zip_urls"]
|
|
elsif page_count > 1
|
|
@pages = json["metadata"]["pages"].map {|x| x["image_urls"]["large"]}
|
|
end
|
|
end
|
|
|
|
if @pages.nil? && json["image_urls"]
|
|
@pages = [json["image_urls"]["large"]]
|
|
end
|
|
end
|
|
end
|
|
|
|
class NovelResponse
|
|
extend Memoist
|
|
|
|
attr_reader :json
|
|
|
|
def initialize(json)
|
|
@json = json
|
|
end
|
|
|
|
def name
|
|
json["user"]["name"]
|
|
end
|
|
|
|
def user_id
|
|
json["user"]["id"]
|
|
end
|
|
|
|
def moniker
|
|
json["user"]["account"]
|
|
end
|
|
|
|
def page_count
|
|
json["page_count"].to_i
|
|
end
|
|
|
|
def artist_commentary_title
|
|
json["title"]
|
|
end
|
|
|
|
def artist_commentary_desc
|
|
json["caption"]
|
|
end
|
|
|
|
def tags
|
|
json["tags"]
|
|
end
|
|
|
|
def pages
|
|
# ex:
|
|
# https://i.pximg.net/c/150x150_80/novel-cover-master/img/2017/07/27/23/14/17/8465454_80685d10e6df4d7d53ad347ddc18a36b_master1200.jpg (6096b)
|
|
# =>
|
|
# https://i.pximg.net/novel-cover-original/img/2017/07/27/23/14/17/8465454_80685d10e6df4d7d53ad347ddc18a36b.jpg (532129b)
|
|
[find_original(json["image_urls"]["small"])]
|
|
end
|
|
memoize :pages
|
|
|
|
public
|
|
PXIMG = %r!\Ahttps?://i\.pximg\.net/c/\d+x\d+_\d+/novel-cover-master/img/(?<timestamp>\d+/\d+/\d+/\d+/\d+/\d+)/(?<filename>\d+_[a-f0-9]+)_master\d+\.(?<ext>jpg|jpeg|png|gif)!i
|
|
|
|
def find_original(x)
|
|
if x =~ PXIMG
|
|
return "https://i.pximg.net/novel-cover-original/img/#{$~[:timestamp]}/#{$~[:filename]}.#{$~[:ext]}"
|
|
end
|
|
|
|
return x
|
|
end
|
|
end
|
|
|
|
class FanboxResponse
|
|
attr_reader :json
|
|
|
|
def initialize(json)
|
|
@json = json
|
|
end
|
|
|
|
def name
|
|
json["body"]["user"]["name"]
|
|
end
|
|
|
|
def user_id
|
|
json["body"]["user"]["userId"]
|
|
end
|
|
|
|
def moniker
|
|
""
|
|
end
|
|
|
|
def page_count
|
|
json["body"]["body"]["images"].size
|
|
end
|
|
|
|
def artist_commentary_title
|
|
json["body"]["title"]
|
|
end
|
|
|
|
def artist_commentary_desc
|
|
json["body"]["body"]["text"]
|
|
end
|
|
|
|
def tags
|
|
[]
|
|
end
|
|
|
|
def pages
|
|
json["body"]["body"]["images"].map {|x| x["originalUrl"]}
|
|
end
|
|
end
|
|
|
|
def work(illust_id)
|
|
headers = Danbooru.config.http_headers.merge(
|
|
"Referer" => "http://www.pixiv.net",
|
|
"Content-Type" => "application/x-www-form-urlencoded",
|
|
"Authorization" => "Bearer #{access_token}"
|
|
)
|
|
params = {
|
|
"image_sizes" => "large",
|
|
"include_stats" => "true"
|
|
}
|
|
|
|
url = "https://public-api.secure.pixiv.net/v#{API_VERSION}/works/#{illust_id.to_i}.json"
|
|
resp = HTTParty.get(url, Danbooru.config.httparty_options.deep_merge(query: params, headers: headers))
|
|
body = resp.body.force_encoding("utf-8")
|
|
json = JSON.parse(body)
|
|
|
|
if resp.success?
|
|
WorkResponse.new(json["response"][0])
|
|
elsif json["status"] == "failure" && json.dig("errors", "system", "message") =~ /対象のイラストは見つかりませんでした。/
|
|
raise BadIDError.new("Pixiv ##{illust_id} not found: work was deleted, made private, or ID is invalid.")
|
|
else
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
rescue JSON::ParserError
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
|
|
def fanbox(fanbox_id)
|
|
url = "https://www.pixiv.net/ajax/fanbox/post?postId=#{fanbox_id.to_i}"
|
|
resp = agent.get(url)
|
|
json = JSON.parse(resp.body)
|
|
if resp.code == "200"
|
|
FanboxResponse.new(json)
|
|
elsif json["status"] == "failure"
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
rescue JSON::ParserError
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
|
|
def novel(novel_id)
|
|
headers = Danbooru.config.http_headers.merge(
|
|
"Referer" => "http://www.pixiv.net",
|
|
"Content-Type" => "application/x-www-form-urlencoded",
|
|
"Authorization" => "Bearer #{access_token}"
|
|
)
|
|
|
|
url = "https://public-api.secure.pixiv.net/v#{API_VERSION}/novels/#{novel_id.to_i}.json"
|
|
resp = HTTParty.get(url, Danbooru.config.httparty_options.deep_merge(headers: headers))
|
|
body = resp.body.force_encoding("utf-8")
|
|
json = JSON.parse(body)
|
|
|
|
if resp.success?
|
|
NovelResponse.new(json["response"][0])
|
|
elsif json["status"] == "failure" && json.dig("errors", "system", "message") =~ /対象のイラストは見つかりませんでした。/
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
rescue JSON::ParserError
|
|
raise Error.new("Pixiv API call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
|
|
def access_token
|
|
Cache.get("pixiv-papi-access-token", 3000) do
|
|
access_token = nil
|
|
headers = {
|
|
"Referer" => "http://www.pixiv.net"
|
|
}
|
|
params = {
|
|
"username" => Danbooru.config.pixiv_login,
|
|
"password" => Danbooru.config.pixiv_password,
|
|
"grant_type" => "password",
|
|
"client_id" => CLIENT_ID,
|
|
"client_secret" => CLIENT_SECRET
|
|
}
|
|
url = "https://oauth.secure.pixiv.net/auth/token"
|
|
|
|
resp = HTTParty.post(url, Danbooru.config.httparty_options.deep_merge(body: params, headers: headers))
|
|
body = resp.body.force_encoding("utf-8")
|
|
|
|
if resp.success?
|
|
json = JSON.parse(body)
|
|
access_token = json["response"]["access_token"]
|
|
else
|
|
raise Error.new("Pixiv API access token call failed (status=#{resp.code} body=#{body})")
|
|
end
|
|
|
|
access_token
|
|
end
|
|
end
|
|
|
|
def agent
|
|
PixivWebAgent.build
|
|
end
|
|
memoize :agent
|
|
end
|