mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-23 03:02:50 +01:00
474 lines
16 KiB
Python
474 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2017-2023 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Utility classes to setup OAuth and link accounts to gallery-dl"""
|
|
|
|
from .common import Extractor, Message
|
|
from .. import text, oauth, util, config, exception
|
|
from ..output import stdout_write
|
|
from ..cache import cache, memcache
|
|
import urllib.parse
|
|
import binascii
|
|
import hashlib
|
|
|
|
REDIRECT_URI_LOCALHOST = "http://localhost:6414/"
|
|
REDIRECT_URI_HTTPS = "https://mikf.github.io/gallery-dl/oauth-redirect.html"
|
|
|
|
|
|
class OAuthBase(Extractor):
|
|
"""Base class for OAuth Helpers"""
|
|
category = "oauth"
|
|
redirect_uri = REDIRECT_URI_LOCALHOST
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.client = None
|
|
|
|
def _init(self):
|
|
self.cache = config.get(("extractor", self.category), "cache", True)
|
|
if self.cache and cache is memcache:
|
|
self.log.warning("cache file is not writeable")
|
|
self.cache = False
|
|
|
|
def oauth_config(self, key, default=None):
|
|
value = config.interpolate(("extractor", self.subcategory), key)
|
|
return value if value is not None else default
|
|
|
|
def recv(self):
|
|
"""Open local HTTP server and recv callback parameters"""
|
|
import socket
|
|
stdout_write("Waiting for response. (Cancel with Ctrl+c)\n")
|
|
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
server.bind((self.config("host", "localhost"),
|
|
self.config("port", 6414)))
|
|
server.listen(1)
|
|
|
|
# workaround for ctrl+c not working during server.accept on Windows
|
|
if util.WINDOWS:
|
|
server.settimeout(1.0)
|
|
while True:
|
|
try:
|
|
self.client = server.accept()[0]
|
|
break
|
|
except socket.timeout:
|
|
pass
|
|
server.close()
|
|
|
|
data = self.client.recv(1024).decode()
|
|
path = data.split(" ", 2)[1]
|
|
return text.parse_query(path.partition("?")[2])
|
|
|
|
def send(self, msg):
|
|
"""Send 'msg' to the socket opened in 'recv()'"""
|
|
stdout_write(msg)
|
|
self.client.send(b"HTTP/1.1 200 OK\r\n\r\n" + msg.encode())
|
|
self.client.close()
|
|
|
|
def open(self, url, params, recv=None):
|
|
"""Open 'url' in browser amd return response parameters"""
|
|
url += "?" + urllib.parse.urlencode(params)
|
|
|
|
browser = self.config("browser", True)
|
|
if browser:
|
|
try:
|
|
import webbrowser
|
|
browser = webbrowser.get()
|
|
except Exception:
|
|
browser = None
|
|
|
|
if browser and browser.open(url):
|
|
name = getattr(browser, "name", "Browser")
|
|
self.log.info("Opening URL in %s:", name.capitalize())
|
|
else:
|
|
self.log.info("Please open this URL in your browser:")
|
|
|
|
stdout_write("\n{}\n\n".format(url))
|
|
return (recv or self.recv)()
|
|
|
|
def error(self, msg):
|
|
return self.send(
|
|
"Remote server reported an error:\n\n{}\n".format(msg))
|
|
|
|
def _oauth1_authorization_flow(
|
|
self, default_key, default_secret,
|
|
request_token_url, authorize_url, access_token_url):
|
|
"""Perform the OAuth 1.0a authorization flow"""
|
|
|
|
api_key = self.oauth_config("api-key") or default_key
|
|
api_secret = self.oauth_config("api-secret") or default_secret
|
|
self.session = oauth.OAuth1Session(api_key, api_secret)
|
|
|
|
self.log.info("Using %s %s API key (%s)",
|
|
"default" if api_key == default_key else "custom",
|
|
self.subcategory, api_key)
|
|
|
|
# get a request token
|
|
params = {"oauth_callback": self.redirect_uri}
|
|
data = self.session.get(request_token_url, params=params).text
|
|
|
|
data = text.parse_query(data)
|
|
self.session.auth.token_secret = data["oauth_token_secret"]
|
|
|
|
# get the user's authorization
|
|
params = {"oauth_token": data["oauth_token"], "perms": "read"}
|
|
data = self.open(authorize_url, params)
|
|
|
|
# exchange the request token for an access token
|
|
data = self.session.get(access_token_url, params=data).text
|
|
data = text.parse_query(data)
|
|
token = data["oauth_token"]
|
|
token_secret = data["oauth_token_secret"]
|
|
|
|
# write to cache
|
|
if self.cache:
|
|
key = (self.subcategory, self.session.auth.consumer_key)
|
|
oauth._token_cache.update(key, (token, token_secret))
|
|
self.log.info("Writing tokens to cache")
|
|
|
|
# display tokens
|
|
self.send(self._generate_message(
|
|
("access-token", "access-token-secret"),
|
|
(token, token_secret),
|
|
))
|
|
|
|
def _oauth2_authorization_code_grant(
|
|
self, client_id, client_secret, default_id, default_secret,
|
|
auth_url, token_url, scope="read", duration="permanent",
|
|
key="refresh_token", auth=True, cache=None, instance=None):
|
|
"""Perform an OAuth2 authorization code grant"""
|
|
|
|
client_id = str(client_id) if client_id else default_id
|
|
client_secret = client_secret or default_secret
|
|
|
|
self.log.info("Using %s %s client ID (%s)",
|
|
"default" if client_id == default_id else "custom",
|
|
instance or self.subcategory, client_id)
|
|
|
|
state = "gallery-dl_{}_{}".format(
|
|
self.subcategory,
|
|
oauth.nonce(8),
|
|
)
|
|
|
|
auth_params = {
|
|
"client_id" : client_id,
|
|
"response_type": "code",
|
|
"state" : state,
|
|
"redirect_uri" : self.redirect_uri,
|
|
"duration" : duration,
|
|
"scope" : scope,
|
|
}
|
|
|
|
# receive an authorization code
|
|
params = self.open(auth_url, auth_params)
|
|
|
|
# check authorization response
|
|
if state != params.get("state"):
|
|
self.send("'state' mismatch: expected {}, got {}.\n".format(
|
|
state, params.get("state")))
|
|
return
|
|
if "error" in params:
|
|
return self.error(params)
|
|
|
|
# exchange authorization code for a token
|
|
data = {
|
|
"grant_type" : "authorization_code",
|
|
"code" : params["code"],
|
|
"redirect_uri": self.redirect_uri,
|
|
}
|
|
|
|
if auth:
|
|
auth = util.HTTPBasicAuth(client_id, client_secret)
|
|
else:
|
|
auth = None
|
|
data["client_id"] = client_id
|
|
data["client_secret"] = client_secret
|
|
|
|
data = self.session.post(token_url, data=data, auth=auth).json()
|
|
|
|
# check token response
|
|
if "error" in data:
|
|
return self.error(data)
|
|
|
|
token = data[key]
|
|
token_name = key.replace("_", "-")
|
|
|
|
# write to cache
|
|
if self.cache and cache:
|
|
cache.update(instance or ("#" + str(client_id)), token)
|
|
self.log.info("Writing '%s' to cache", token_name)
|
|
|
|
# display token
|
|
self.send(self._generate_message(
|
|
(token_name,), (token,),
|
|
))
|
|
|
|
def _generate_message(self, names, values):
|
|
_vh, _va, _is, _it = (
|
|
("This value has", "this value", "is", "it")
|
|
if len(names) == 1 else
|
|
("These values have", "these values", "are", "them")
|
|
)
|
|
|
|
msg = "\nYour {} {}\n\n{}\n\n".format(
|
|
" and ".join("'" + n + "'" for n in names),
|
|
_is,
|
|
"\n".join(values),
|
|
)
|
|
|
|
opt = self.oauth_config(names[0])
|
|
if self.cache and (opt is None or opt == "cache"):
|
|
msg += _vh + " been cached and will automatically be used.\n"
|
|
else:
|
|
msg += "Put " + _va + " into your configuration file as \n"
|
|
msg += " and\n".join(
|
|
"'extractor." + self.subcategory + "." + n + "'"
|
|
for n in names
|
|
)
|
|
if self.cache:
|
|
msg += (
|
|
"\nor set\n'extractor.{}.{}' to \"cache\""
|
|
.format(self.subcategory, names[0])
|
|
)
|
|
msg += "\nto use {}.\n".format(_it)
|
|
|
|
return msg
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# OAuth 1.0a
|
|
|
|
class OAuthFlickr(OAuthBase):
|
|
subcategory = "flickr"
|
|
pattern = "oauth:flickr$"
|
|
example = "oauth:flickr"
|
|
redirect_uri = REDIRECT_URI_HTTPS
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import flickr
|
|
|
|
self._oauth1_authorization_flow(
|
|
flickr.FlickrAPI.API_KEY,
|
|
flickr.FlickrAPI.API_SECRET,
|
|
"https://www.flickr.com/services/oauth/request_token",
|
|
"https://www.flickr.com/services/oauth/authorize",
|
|
"https://www.flickr.com/services/oauth/access_token",
|
|
)
|
|
|
|
|
|
class OAuthSmugmug(OAuthBase):
|
|
subcategory = "smugmug"
|
|
pattern = "oauth:smugmug$"
|
|
example = "oauth:smugmug"
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import smugmug
|
|
|
|
self._oauth1_authorization_flow(
|
|
smugmug.SmugmugAPI.API_KEY,
|
|
smugmug.SmugmugAPI.API_SECRET,
|
|
"https://api.smugmug.com/services/oauth/1.0a/getRequestToken",
|
|
"https://api.smugmug.com/services/oauth/1.0a/authorize",
|
|
"https://api.smugmug.com/services/oauth/1.0a/getAccessToken",
|
|
)
|
|
|
|
|
|
class OAuthTumblr(OAuthBase):
|
|
subcategory = "tumblr"
|
|
pattern = "oauth:tumblr$"
|
|
example = "oauth:tumblr"
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import tumblr
|
|
|
|
self._oauth1_authorization_flow(
|
|
tumblr.TumblrAPI.API_KEY,
|
|
tumblr.TumblrAPI.API_SECRET,
|
|
"https://www.tumblr.com/oauth/request_token",
|
|
"https://www.tumblr.com/oauth/authorize",
|
|
"https://www.tumblr.com/oauth/access_token",
|
|
)
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# OAuth 2.0
|
|
|
|
class OAuthDeviantart(OAuthBase):
|
|
subcategory = "deviantart"
|
|
pattern = "oauth:deviantart$"
|
|
example = "oauth:deviantart"
|
|
redirect_uri = REDIRECT_URI_HTTPS
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import deviantart
|
|
|
|
self._oauth2_authorization_code_grant(
|
|
self.oauth_config("client-id"),
|
|
self.oauth_config("client-secret"),
|
|
deviantart.DeviantartOAuthAPI.CLIENT_ID,
|
|
deviantart.DeviantartOAuthAPI.CLIENT_SECRET,
|
|
"https://www.deviantart.com/oauth2/authorize",
|
|
"https://www.deviantart.com/oauth2/token",
|
|
scope="browse user.manage",
|
|
cache=deviantart._refresh_token_cache,
|
|
)
|
|
|
|
|
|
class OAuthReddit(OAuthBase):
|
|
subcategory = "reddit"
|
|
pattern = "oauth:reddit$"
|
|
example = "oauth:reddit"
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import reddit
|
|
|
|
self.session.headers["User-Agent"] = reddit.RedditAPI.USER_AGENT
|
|
self._oauth2_authorization_code_grant(
|
|
self.oauth_config("client-id"),
|
|
"",
|
|
reddit.RedditAPI.CLIENT_ID,
|
|
"",
|
|
"https://www.reddit.com/api/v1/authorize",
|
|
"https://www.reddit.com/api/v1/access_token",
|
|
scope="read history",
|
|
cache=reddit._refresh_token_cache,
|
|
)
|
|
|
|
|
|
class OAuthMastodon(OAuthBase):
|
|
subcategory = "mastodon"
|
|
pattern = "oauth:mastodon:(?:https?://)?([^/?#]+)"
|
|
example = "oauth:mastodon:mastodon.social"
|
|
|
|
def __init__(self, match):
|
|
OAuthBase.__init__(self, match)
|
|
self.instance = match.group(1)
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import mastodon
|
|
|
|
for _, root, application in mastodon.MastodonExtractor.instances:
|
|
if self.instance == root.partition("://")[2]:
|
|
break
|
|
else:
|
|
application = self._register(self.instance)
|
|
|
|
self._oauth2_authorization_code_grant(
|
|
application["client-id"],
|
|
application["client-secret"],
|
|
application["client-id"],
|
|
application["client-secret"],
|
|
"https://{}/oauth/authorize".format(self.instance),
|
|
"https://{}/oauth/token".format(self.instance),
|
|
instance=self.instance,
|
|
key="access_token",
|
|
cache=mastodon._access_token_cache,
|
|
)
|
|
|
|
@cache(maxage=36500*86400, keyarg=1)
|
|
def _register(self, instance):
|
|
self.log.info("Registering application for '%s'", instance)
|
|
|
|
url = "https://{}/api/v1/apps".format(instance)
|
|
data = {
|
|
"client_name": "gdl:" + oauth.nonce(8),
|
|
"redirect_uris": self.redirect_uri,
|
|
"scopes": "read",
|
|
}
|
|
data = self.session.post(url, data=data).json()
|
|
|
|
if "client_id" not in data or "client_secret" not in data:
|
|
raise exception.StopExtraction(
|
|
"Failed to register new application: '%s'", data)
|
|
|
|
data["client-id"] = data.pop("client_id")
|
|
data["client-secret"] = data.pop("client_secret")
|
|
|
|
self.log.info("client-id:\n%s", data["client-id"])
|
|
self.log.info("client-secret:\n%s", data["client-secret"])
|
|
|
|
return data
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
|
|
class OAuthPixiv(OAuthBase):
|
|
subcategory = "pixiv"
|
|
pattern = "oauth:pixiv$"
|
|
example = "oauth:pixiv"
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
from . import pixiv
|
|
|
|
code_verifier = util.generate_token(32)
|
|
digest = hashlib.sha256(code_verifier.encode()).digest()
|
|
code_challenge = binascii.b2a_base64(
|
|
digest)[:-2].decode().replace("+", "-").replace("/", "_")
|
|
|
|
url = "https://app-api.pixiv.net/web/v1/login"
|
|
params = {
|
|
"code_challenge": code_challenge,
|
|
"code_challenge_method": "S256",
|
|
"client": "pixiv-android",
|
|
}
|
|
code = self.open(url, params, self._input)
|
|
|
|
url = "https://oauth.secure.pixiv.net/auth/token"
|
|
headers = {
|
|
"User-Agent": "PixivAndroidApp/5.0.234 (Android 11; Pixel 5)",
|
|
}
|
|
data = {
|
|
"client_id" : self.oauth_config(
|
|
"client-id" , pixiv.PixivAppAPI.CLIENT_ID),
|
|
"client_secret" : self.oauth_config(
|
|
"client-secret", pixiv.PixivAppAPI.CLIENT_SECRET),
|
|
"code" : code,
|
|
"code_verifier" : code_verifier,
|
|
"grant_type" : "authorization_code",
|
|
"include_policy": "true",
|
|
"redirect_uri" : "https://app-api.pixiv.net"
|
|
"/web/v1/users/auth/pixiv/callback",
|
|
}
|
|
data = self.session.post(url, headers=headers, data=data).json()
|
|
|
|
if "error" in data:
|
|
stdout_write("\n{}\n".format(data))
|
|
if data["error"] in ("invalid_request", "invalid_grant"):
|
|
stdout_write("'code' expired, try again\n\n")
|
|
return
|
|
|
|
token = data["refresh_token"]
|
|
if self.cache:
|
|
username = self.oauth_config("username")
|
|
pixiv._refresh_token_cache.update(username, token)
|
|
self.log.info("Writing 'refresh-token' to cache")
|
|
|
|
stdout_write(self._generate_message(("refresh-token",), (token,)))
|
|
|
|
def _input(self):
|
|
stdout_write("""\
|
|
1) Open your browser's Developer Tools (F12) and switch to the Network tab
|
|
2) Login
|
|
3) Select the last network monitor entry ('callback?state=...')
|
|
4) Copy its 'code' query parameter, paste it below, and press Enter
|
|
|
|
- This 'code' will expire 30 seconds after logging in.
|
|
- Copy-pasting more than just the 'code' value will work as well,
|
|
like the entire URL or several query parameters.
|
|
|
|
""")
|
|
code = input("code: ")
|
|
return code.rpartition("=")[2].strip()
|