import flask import flask_bootstrap import redis import json import os import jinja2 import uuid import string import random import ipaddress import hashlib import time version = os.getenv("this_version", "master") static_url_path = "/" + version + "/static" app = flask.Flask(__name__, static_url_path=static_url_path) flask_bootstrap.Bootstrap(app) db = redis.StrictRedis(host='redis', port=6379, db=0) def render_flavor(flavor, template, data): return flask.render_template( os.path.join(flavor, template), **data ) @app.add_template_global def secret(length=16): charset = string.ascii_uppercase + string.digits return ''.join( random.SystemRandom().choice(charset) for _ in range(length) ) #Original copied from https://github.com/andrewlkho/ulagen def random_ipv6_subnet(): eui64 = uuid.getnode() >> 24 << 48 | 0xfffe000000 | uuid.getnode() & 0xffffff eui64_canon = "-".join([format(eui64, "02X")[i:i+2] for i in range(0, 18, 2)]) h = hashlib.sha1() h.update((eui64_canon + str(time.time() - time.mktime((1900, 1, 1, 0, 0, 0, 0, 1, -1)))).encode('utf-8')) globalid = h.hexdigest()[0:10] prefix = ":".join(("fd" + globalid[0:2], globalid[2:6], globalid[6:10])) return prefix def build_app(path): app.jinja_env.trim_blocks = True app.jinja_env.lstrip_blocks = True @app.context_processor def app_context(): return dict( versions=os.getenv("VERSIONS","master").split(','), stable_version = os.getenv("stable_version", "master") ) prefix_bp = flask.Blueprint(version.replace(".", "_"), __name__) prefix_bp.jinja_loader = jinja2.ChoiceLoader([ jinja2.FileSystemLoader(os.path.join(path, "templates")), jinja2.FileSystemLoader(os.path.join(path, "flavors")) ]) root_bp = flask.Blueprint("root", __name__) root_bp.jinja_loader = jinja2.ChoiceLoader([ jinja2.FileSystemLoader(os.path.join(path, "templates")), jinja2.FileSystemLoader(os.path.join(path, "flavors")) ]) @prefix_bp.context_processor @root_bp.context_processor def bp_context(version=version): return dict(version=version) @prefix_bp.route("/") @root_bp.route("/") def wizard(): return flask.render_template( 'wizard.html', flavor="compose", steps=sorted(os.listdir(os.path.join(path, "templates", "steps", "compose"))), subnet6=random_ipv6_subnet() ) @prefix_bp.route("/submit", methods=["POST"]) @root_bp.route("/submit", methods=["POST"]) def submit(): data = flask.request.form.copy() data['uid'] = str(uuid.uuid4()) try: data['dns'] = str(ipaddress.IPv4Network(data['subnet'], strict=False)[-2]) except ValueError as err: return "Error while generating files: " + str(err) db.set(data['uid'], json.dumps(data)) return flask.redirect(flask.url_for('.setup', uid=data['uid'])) @prefix_bp.route("/setup/", methods=["GET"]) @root_bp.route("/setup/", methods=["GET"]) def setup(uid): data = json.loads(db.get(uid)) flavor = data.get("flavor", "compose") rendered = render_flavor(flavor, "setup.html", data) return flask.render_template("setup.html", contents=rendered) @prefix_bp.route("/file//", methods=["GET"]) @root_bp.route("/file//", methods=["GET"]) def file(uid, filepath): data = json.loads(db.get(uid)) flavor = data.get("flavor", "compose") return flask.Response( render_flavor(flavor, filepath, data), mimetype="application/text" ) app.register_blueprint(prefix_bp, url_prefix="/{}".format(version)) app.register_blueprint(root_bp) if __name__ == "__main__": build_app("/tmp/mailutest") app.run(debug=True)