|
import base64
|
|
import datetime
|
|
import json
|
|
import urllib
|
|
|
|
import flask
|
|
import requests
|
|
import src.functions.config
|
|
|
|
redirectdownloadBP = flask.Blueprint("redirectdownload", __name__)
|
|
|
|
|
|
@redirectdownloadBP.route("/api/v1/redirectdownload/<name>")
|
|
async def redirectdownloadFunction(name):
|
|
id = flask.request.args.get("id")
|
|
itag = flask.request.args.get("itag")
|
|
|
|
config = src.functions.config.readConfig()
|
|
if config.get("kill_switch") == True:
|
|
return
|
|
|
|
if (
|
|
datetime.datetime.strptime(
|
|
config.get("token_expiry", datetime.datetime.utcnow()),
|
|
"%Y-%m-%d %H:%M:%S.%f",
|
|
)
|
|
<= datetime.datetime.utcnow()
|
|
):
|
|
config, drive = src.functions.credentials.refreshCredentials(config)
|
|
with open("config.json", "w+") as w:
|
|
json.dump(obj=config, fp=w, sort_keys=True, indent=4)
|
|
|
|
tmp_metadata = src.functions.metadata.jsonExtract(
|
|
src.functions.metadata.readMetadata(config), "id", id, False
|
|
)
|
|
if tmp_metadata:
|
|
name = tmp_metadata.get("name", name)
|
|
args = "?"
|
|
for arg in flask.request.args:
|
|
args += "%s=%s&" % (
|
|
arg,
|
|
urllib.parse.quote(flask.request.args.get(arg, "").encode("utf-8")),
|
|
)
|
|
session = {
|
|
"access_token": config.get("access_token"),
|
|
"client_id": config.get("client_id"),
|
|
"client_secret": config.get("client_secret"),
|
|
"refresh_token": config.get("refresh_token"),
|
|
"token_expiry": config.get("token_expiry"),
|
|
}
|
|
|
|
session["url"] = "https://www.googleapis.com/drive/v3/files/%s?alt=media" % (id)
|
|
if itag and itag != "" and config.get("transcoded") == True:
|
|
req = requests.get(
|
|
"https://drive.google.com/get_video_info?docid=%s" % (id),
|
|
headers={"Authorization": "Bearer %s" % (config.get("access_token"))},
|
|
)
|
|
parsed = urllib.parse.parse_qs(urllib.parse.unquote(req.text))
|
|
if parsed.get("status") == ["ok"]:
|
|
for stream in parsed["url"]:
|
|
if ("itag=%s" % (itag)) in stream:
|
|
url = stream
|
|
break
|
|
cookie_string = "; ".join(
|
|
[str(x) + "=" + str(y) for x, y in req.cookies.items()]
|
|
)
|
|
session["cookie"] = cookie_string
|
|
session["transcoded"] = config.get("transcoded")
|
|
session["url"] = url
|
|
|
|
sessionB64 = base64.b64encode(json.dumps(session).encode("ascii")).decode("ascii")
|
|
|
|
if config.get("cloudflare") not in ["local", "", None]:
|
|
return flask.redirect(
|
|
config.get("cloudflare")
|
|
+ "/api/v1/download/%s%ssession=%s&" % (name, args, sessionB64),
|
|
code=302,
|
|
)
|
|
elif config.get("cloudflare") == "local":
|
|
return flask.redirect(
|
|
"http://localhost:31146/api/v1/download/%s%ssession=%s&"
|
|
% (name, args, sessionB64),
|
|
code=302,
|
|
)
|
|
else:
|
|
return flask.redirect(
|
|
"/api/v1/download/%s%ssession=%s&"
|
|
% (urllib.parse.quote(name.encode("utf-8")), args, sessionB64),
|
|
code=302,
|
|
)
|
|
|
