snorklerjoe/CubeServer

"""Flask blueprint managing API client (team) configuration stuff"""

import logging
from datetime import datetime
from subprocess import call
from os import environ, mkdir, chdir
import os.path
from shutil import rmtree
from random import randint, random

from cubeserver_app import settings

from flask import Blueprint, render_template, session, make_response

bp = Blueprint("config", __name__, url_prefix="/setup", template_folder="templates")


def get_api_cert():
    cert = environ.get("API_CERT")
    if not cert:
        cert_filename = environ.get("API_CERT_FILENAME", "/etc/ssl/api_cert/server.pem")
        if os.path.exists(cert_filename):
            cert = open(cert_filename, "r").read()

    return cert


@bp.route("/")
def index():
    """Renders the main page for this blueprint"""
    return render_template("index.html.jinja2", rand=random())


@bp.route("/client_config.py")
def py_file():
    """Renders a configuration python file"""
    team_name = session["team_name"] if "team_name" in session else None
    team_secret = session["team_secret"] if "team_secret" in session else None
    logging.info(f"Rendering client_config.py for {team_name}")
    return render_template(
        "client_config.py.jinja2",
        timestamp=datetime.now().isoformat(),
        team_name=team_name,
        team_secret=team_secret,
        ssid=environ["AP_SSID"],
        api_host=environ["API_HOST"],
        port=environ["API_PORT"],
        server_cert=get_api_cert(),
    )


@bp.route("/CircuitPython_lib.zip")
def package_circuitpython_lib():
    """Packages and downloads the CircuitPython ZIP library"""
    logging.info(f"Packing CircuitPython ZIP")
    working_dir = f"/tmp/lib_pack_{str(randint(100,999))}"
    mkdir(working_dir)
    chdir(working_dir)
    output_path = f"{working_dir}/download.zip"
    client_config_path = f"{working_dir}/client_config.py"
    with open(client_config_path, "w") as fh:
        fh.write(py_file())
    call(
        [
            "/code/package_lib.sh",
            settings.API_WRAPPER_GIT_URL_CIRCUITPYTHON,
            client_config_path,
            output_path,
        ]
    )
    # TODO: avoid RAM overhead of loading file into memory?
    output_raw = bytes()
    with open(output_path, "rb") as fh:
        output_raw = fh.read()
    # Clean up:
    rmtree(working_dir)
    # Formulate response:
    response = make_response(output_raw)
    response.headers.set("Content-Type", "application/zip")
    response.headers.set(
        "Content-Disposition",
        "attachment",
        filename=settings.API_WRAPPER_ZIP_FILENAME_CIRCUITPYTHON,
    )
    return response


# This is more important than you think. Don't remove it. :)
@bp.route("/api_cert.pem")
def api_cert():
    """Downloads the pem file for the cert of the api
    for server verification purposes"""
    logging.info("Downloading api_cert.pem")
    api_cert = get_api_cert() or "undefined"
    response = make_response(api_cert)
    response.headers.set("Content-Type", "application/x-pem-file")
    response.headers.set(
        "Content-Disposition",
        "attachment",
        filename=settings.API_WRAPPER_PEM_FILENAME,
    )
    return response