service

#!/usr/bin/env python3

import _jsonnet as j
from aiohttp import web
import logging
import os
import json
import yaml

logger = logging.getLogger("service")
logger.setLevel(logging.INFO)

class Generator:

    def __init__(self, config, base="./templates/"):
        self.jsonnet_base = base
        self.config = config

    def process(self, config):

        res = j.evaluate_snippet("config", config, import_callback=self.load)
        return json.loads(res)

    def load(self, dir, filename):

        logger.info("Request jsonnet: %s %s", dir, filename)

        if filename == "config.json" and dir == "":
            print("config.json...")
            path = os.path.join(".", dir, filename)
            return str(path), self.config

        try:
            if dir:
                path = os.path.join(".", dir, filename)
            else:
                path = os.path.join(self.jsonnet_base, filename)

            logger.debug("Try: %s", path)
            with open(path, "rb") as f:
                logger.debug("Loaded: %s", path)
                return str(path), f.read()

        except:
            path = os.path.join(self.jsonnet_base, filename)
            logger.debug("Try: %s", path)
            with open(path, "rb") as f:
                logger.debug("Loaded: %s", path)
                return str(path), f.read()

class Api:
    def __init__(self, patterns="", config=None):

        if config:
            self.config = config
        else:
            self.config = {}

        self.patterns = patterns

        self.port = int(self.config.get("port", "8081"))
        self.app = web.Application(middlewares=[])

        self.app.add_routes([web.post("/api/generate", self.generate)])
        self.app.add_routes([web.get("/{tail:.*}", self.everything)])

    async def everything(self, request):

        if ".." in request.path:
            return web.HTTPNotFound()

        if request.path == "/":
            with open("dist/index.html", "r") as f:
                return web.Response(
                    text=f.read(), content_type="text/html"
                )

        print(request.path)

        if request.path == "/api/patterns":
            return web.Response(
                text=self.patterns, content_type="application/json"
            )

        if request.path.endswith(".css"):
            with open("dist" + request.path, "r") as f:
                data = f.read()
            return web.Response(
                text=data, content_type="text/css"
            )

        if request.path.endswith(".js"):
            with open("dist" + request.path, "r") as f:
                data = f.read()
            return web.Response(
                text=data, content_type="text/javascript"
            )

        if request.path.endswith(".html"):
            with open("dist" + request.path, "r") as f:
                data = f.read()
            return web.Response(
                text=data, content_type="text/html"
            )

        return web.HTTPNotFound()

    async def generate(self, request):

        print("Generate...")
        config = await request.text()

        print(config)

        config = config.encode("utf-8")

        gen = Generator(config)

        with open("./templates/config-loader.jsonnet", "r") as f:
            wrapper = f.read()
        processed = gen.process(wrapper)

        return web.Response(
            text=yaml.dump(processed), content_type = "text/plain"
        )

    def run(self):

        web.run_app(self.app, port=self.port)

print("Create all-patterns from templates...")
with open("./templates/all-patterns.jsonnet", "r") as f:
    cfg = f.read()
gen = Generator("")
patterns = gen.process(cfg)
print("Done.")

a = Api(patterns=json.dumps(patterns, indent=4))
a.run()