larigira/larigira/rpc.py

import gc
import logging
from copy import deepcopy

from flask import (Blueprint, Flask, abort, current_app, jsonify, redirect,
                   render_template, request)
from flask_babel import Babel
from flask_bootstrap import Bootstrap
from flask_wtf.csrf import CSRFProtect

from cachelib import SimpleCache
from greenlet import greenlet

from .config import get_conf
from .dbadmin import db

rpc = Blueprint("rpc", __name__, url_prefix=get_conf()["ROUTE_PREFIX"] + "/api")
viewui = Blueprint(
    "view",
    __name__,
    url_prefix=get_conf()["ROUTE_PREFIX"] + "/view",
    template_folder="templates",
)


def send_to_parent(kind, *args):
    """similar to the behaviour of a ParentedLet"""
    if not hasattr(current_app, "queue"):
        logging.debug("no parent queue; aborting send")
        return
    msg = {
        "emitter": current_app._get_current_object(),
        "class": current_app._get_current_object().__class__.__name__,
        "kind": kind,
        "args": args,
    }
    current_app.queue.put(msg)


@rpc.route("/")
def rpc_index():
    rules = list(current_app.url_map.iter_rules())
    return jsonify({"rules": [r.rule for r in rules if r.rule.startswith("/api")]})


@rpc.route("/refresh")
def rpc_refresh():
    send_to_parent("refresh")
    return jsonify(dict(status="ok"))


@rpc.route("/audiospec", methods=["GET"])
def get_audiospec():
    return jsonify(current_app.larigira.controller.player.continous_audiospec)


@rpc.route("/audiospec", methods=["PUT"])
def change_audiospec():
    player = current_app.larigira.controller.player
    if request.json is None:
        abort(400, "Must send application/json data")
    if "spec" not in request.json or type(request.json["spec"]) is not dict:
        abort(400, "Object must have a key 'spec' whose value is an object")
    player.continous_audiospec = request.json["spec"]
    if "kind" not in request.json["spec"]:
        abort(400, "invalid audiospec")
    return jsonify(player.continous_audiospec)


@rpc.route("/audiospec", methods=["DELETE"])
def reset_audiospec():
    player = current_app.larigira.controller.player
    player.continous_audiospec = None
    return jsonify(player.continous_audiospec)


@rpc.route("/eventsenabled/toggle", methods=["POST"])
def toggle_events_enabled():
    status = current_app.larigira.controller.player.events_enabled
    current_app.larigira.controller.player.events_enabled = not status
    return jsonify(dict(events_enabled=not status))


@rpc.route("/eventsenabled", methods=["GET"])
def get_events_enabled():
    status = current_app.larigira.controller.player.events_enabled
    return jsonify(dict(events_enabled=status))


@rpc.route("/eventsenabled", methods=["PUT"])
def set_events_enabled():
    player = current_app.larigira.controller.player
    if request.json is None:
        abort(400, "Must send application/json data")
    if type(request.json) is not bool:
        abort(400, "Content must be a JSON boolean")
    player.events_enabled = request.json
    return jsonify(dict(events_enabled=request.json))


def get_scheduled_audiogen():
    larigira = current_app.larigira
    running = larigira.controller.monitor.running
    events = {t: {} for t in running.keys()}
    for timespec_eid in events:
        orig_info = running[timespec_eid]
        info = events[timespec_eid]
        info["running_time"] = orig_info["running_time"].isoformat()
        info["audiospecs"] = orig_info["audiospecs"]
        info["timespec"] = orig_info["timespec"]
        info["timespec"]["actions"] = {
            aid: spec
            for aid, spec in zip(info["timespec"]["actions"], info["audiospecs"])
        }
        info["greenlet"] = hex(id(orig_info["greenlet"]))
    return events


@viewui.route("/status/running")
def ui_wip():
    audiogens = get_scheduled_audiogen()
    return render_template(
        "running.html",
        audiogens=sorted(audiogens.items(), key=lambda x: x[1]["running_time"]),
    )


@rpc.route("/debug/running")
def rpc_wip():
    def treeify(flat):
        roots = [obid for obid in flat if flat[obid]["parent"] not in flat]
        tree = deepcopy(flat)
        for obid in tree:
            tree[obid]["children"] = {}

        to_remove = []
        for obid in tree:
            if obid in roots:
                continue
            if obid not in tree:
                current_app.logger.warning("How strange, {} not in tree".format(obid))
                continue
            tree[tree[obid]["parent"]]["children"][obid] = tree[obid]
            to_remove.append(obid)

        for obid in to_remove:
            del tree[obid]
        return tree

    greenlets = {}
    for ob in filter(lambda obj: isinstance(obj, greenlet), gc.get_objects()):
        objrepr = {"repr": repr(ob), "class": ob.__class__.__name__}
        if hasattr(ob, "parent_greenlet") and ob.parent_greenlet is not None:
            objrepr["parent"] = hex(id(ob.parent_greenlet))
        else:
            objrepr["parent"] = hex(id(ob.parent)) if ob.parent is not None else None
        if hasattr(ob, "doc"):
            objrepr["doc"] = ob.doc.split("\n")[0]
        elif ob.__doc__:
            objrepr["doc"] = ob.__doc__.split("\n")[0]

        greenlets[hex(id(ob))] = objrepr

    # TODO: make it a tree

    return jsonify(
        dict(
            greenlets=greenlets,
            greenlets_tree=treeify(greenlets),
            audiogens=get_scheduled_audiogen(),
        )
    )


def babel_get_locale():
    try:
        if request.accept_languages:
            return request.accept_languages[0][0]
    finally:
        return None


def create_app(queue, larigira):
    app = Flask("larigira", static_url_path=get_conf()["ROUTE_PREFIX"] + "/static")
    app.config.update(get_conf())
    Bootstrap(app)
    babel = Babel(app)
    babel.init_app(app, locale_selector=babel_get_locale)
    csrf = CSRFProtect(app)
    csrf.exempt(db)
    app.register_blueprint(rpc)
    app.register_blueprint(viewui)
    app.register_blueprint(db)
    app.route(get_conf()["ROUTE_PREFIX"] + "/")(
        lambda: redirect(get_conf()["ROUTE_PREFIX"] + get_conf()["HOME_URL"])
    )
    app.queue = queue
    app.larigira = larigira
    app.cache = SimpleCache()
    return app