wscript

#!/usr/bin/env waf
import os
import time
# fixme: better to be a waf tool
import moo
from waflib.Task import Task
from waflib.TaskGen import taskgen_method, extension, declare_chain

APPNAME = 'moo'

def options(opt):
    opt.load('compiler_cxx')    

def configure(cfg):
    cfg.env.CXXFLAGS += ['-std=c++17', '-ggdb3',
                         '-Wall', '-Wpedantic', '-Werror']
    cfg.load('compiler_cxx')
    cfg.find_program('moo', var='MOO', mandatory=True)
    cfg.find_program('dot', var='DOT', mandatory=True)
    cfg.find_program('clang-format', var='CLANG_FORMAT', mandatory=False)
    p = dict(mandatory=True, args='--cflags --libs')
    cfg.check_cfg(package='libzmq', uselib_store='ZMQ', **p);


class codegen(Task): 
    color   = 'PINK'

    def run(self):
        model = self.inputs[0]
        templ = self.inputs[1]
        top = self.generator.bld.path.abspath()
        mpath = [os.path.join(top, "models")]
        data = moo.io.load(model.abspath(),mpath)
        moot = moo.io.load("moo.jsonnet", mpath, "templ")

        if hasattr(self, 'structpath'):
            data = moo.util.select_path(data, self.structpath)

        text = moo.template.render(templ.abspath(), dict(model=data, moo=moot))
        with open(self.outputs[0].abspath(), 'wb') as fp:
            fp.write(text.encode())

    def scan(self):
        deps = list()
        for maybe in self.inputs:
            extra = list()
            top = self.generator.bld.path.abspath()
            mpath = [os.path.join(top, "models")]
            tpath = [os.path.join(top, "templates")]
            if maybe.name.endswith('.jsonnet'):
                extra = moo.jsonnet.imports(maybe.abspath(),mpath)
            if maybe.name.endswith('.j2'):
                extra = moo.template.imports(maybe.abspath(),tpath)
            for one in extra:
                node = self.generator.bld.root.find_resource(one)
                deps.append(node)
        return (deps,time.time())


@extension('.jsonnet')
def add_jsonnet_deps(tgen, model):
    assert(model)
    tgt = tgen.bld.path.find_or_declare(tgen.target)
    assert(tgt)
    tmpl = tgen.template
    if isinstance(tmpl, str):
        tmpl = tgen.bld.path.find_resource(tgen.template)
    if not tmpl:
        raise ValueError(f"file not found: {tgen.template}")
    tsk = tgen.create_task('codegen', [model, tmpl], tgt)
    if hasattr(tgen, "structpath"):
        tsk.structpath=tgen.structpath

@extension('.dot')
def make_dotters(tgen, dot):
    for ext in ["png", "pdf", "svg"]:
        out = dot.change_ext('.'+ext)
        tsk = tgen.create_task('dotter', dot, [out])
        tsk.env.DOT_FMT=ext


class dotter(Task):
    run_str = "${DOT} -T${DOT_FMT} -o ${TGT} ${SRC}"


def build(bld):
    sd = [one.parent.abspath() for one in bld.path.ant_glob("**/wscript_build")]
    if (sd):
        bld.recurse(sd)