246 lines
6.6 KiB
Ruby
246 lines
6.6 KiB
Ruby
require 'resolv-replace'
|
|
|
|
class PixivApiClient
|
|
extend Memoist
|
|
|
|
API_VERSION = "1"
|
|
CLIENT_ID = "bYGKuGVw91e0NMfPGp44euvGt59s"
|
|
CLIENT_SECRET = "HP3RmkgAmEGro0gn1x9ioawQE8WMfvLXDz3ZqxpK"
|
|
CLIENT_HASH_SALT = "28c1fdd170a5204386cb1313c7077b34f83e4aaf4aa829ce78c231e05b0bae2c"
|
|
|
|
class Error < StandardError; 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"]
|
|
|
|
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
|
|
if json["body"]["body"]
|
|
json["body"]["body"]["images"].map {|x| x["originalUrl"]}
|
|
else
|
|
[]
|
|
end
|
|
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"
|
|
response = Danbooru::Http.cache(1.minute).headers(headers).get(url, params: params)
|
|
json = response.parse
|
|
|
|
if response.code == 200
|
|
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=#{response.code} body=#{response.body})")
|
|
end
|
|
rescue JSON::ParserError
|
|
raise Error.new("Pixiv API call failed (status=#{response.code} body=#{response.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
|
|
|
|
client_time = Time.now.rfc3339
|
|
client_hash = Digest::MD5.hexdigest(client_time + CLIENT_HASH_SALT)
|
|
|
|
headers = {
|
|
"Referer": "http://www.pixiv.net",
|
|
"X-Client-Time": client_time,
|
|
"X-Client-Hash": client_hash
|
|
}
|
|
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
|