Kumi
ef2ab5e519
Introduces MIT License for the project to clarify usage rights. Adds README with setup instructions and project details. Enhances app.py to allow configurable port via environment variable.
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from flask import Flask, render_template, request, Response, abort
|
|
from werkzeug.exceptions import InternalServerError
|
|
|
|
from urllib.request import urlopen
|
|
from urllib.error import HTTPError
|
|
from urllib.parse import unquote
|
|
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route("/")
|
|
def home():
|
|
domain = request.url_root
|
|
return render_template("index.html", domain=domain)
|
|
|
|
|
|
@app.route("/<path:url>")
|
|
def proxy(url):
|
|
cdnjs_url = f"https://cdnjs.cloudflare.com/{url}"
|
|
|
|
def generate():
|
|
# Subfunction to allow streaming the data instead of
|
|
# downloading all of it at once
|
|
try:
|
|
with urlopen(unquote(cdnjs_url)) as data:
|
|
while True:
|
|
chunk = data.read(1024 * 1024)
|
|
if not chunk:
|
|
break
|
|
yield chunk
|
|
except HTTPError as e:
|
|
abort(e.code)
|
|
|
|
try:
|
|
with urlopen(unquote(cdnjs_url)) as data:
|
|
content_type = data.headers["content-type"]
|
|
except HTTPError as e:
|
|
abort(e.code)
|
|
except KeyError:
|
|
raise InternalServerError()
|
|
|
|
headers = dict()
|
|
filename = url.split("/")[-1]
|
|
headers["Content-Disposition"] = f'attachment; filename="{filename}"'
|
|
|
|
return Response(generate(), content_type=content_type, headers=headers)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=os.environ.get("PORT", 5679))
|
|
|