cutekit/osdk/model.py

import os
from enum import Enum
from typing import Any

from osdk.jexpr import Json
from osdk.logger import Logger


logger = Logger("model")

Props = dict[str, Any]


class Type(Enum):
    UNKNOWN = "unknown"
    PROJECT = "project"
    TARGET = "target"
    LIB = "lib"
    EXE = "exe"


class Manifest:
    id: str = ""
    type: Type = Type.UNKNOWN
    path: str = ""

    def __init__(self, json: Json = None, path: str = "", strict: bool = True, **kwargs: Any):
        if json is not None:
            if not "id" in json:
                raise ValueError("Missing id")

            self.id = json["id"]

            if not "type" in json and strict:
                raise ValueError("Missing type")

            self.type = Type(json["type"])

            self.path = path
        elif strict:
            raise ValueError("Missing json")

        for key in kwargs:
            setattr(self, key, kwargs[key])

    def toJson(self) -> Json:
        return {
            "id": self.id,
            "type": self.type.value,
            "path": self.path
        }

    def __str__(self):
        return f"Manifest(id={self.id}, type={self.type}, path={self.path})"

    def __repr__(self):
        return f"Manifest({id})"

    def dirname(self) -> str:
        return os.path.dirname(self.path)


class Extern:
    git: str = ""
    tag: str = ""

    def __init__(self, json: Json = None, strict: bool = True, **kwargs: Any):
        if json is not None:
            if not "git" in json and strict:
                raise ValueError("Missing git")

            self.git = json["git"]

            if not "tag" in json and strict:
                raise ValueError("Missing tag")

            self.tag = json["tag"]
        elif strict:
            raise ValueError("Missing json")

        for key in kwargs:
            setattr(self, key, kwargs[key])

    def toJson(self) -> Json:
        return {
            "git": self.git,
            "tag": self.tag
        }

    def __str__(self):
        return f"Extern(git={self.git}, tag={self.tag})"

    def __repr__(self):
        return f"Extern({self.git})"


class ProjectManifest(Manifest):
    description: str = ""
    extern: dict[str, Extern] = {}

    def __init__(self, json: Json = None, path: str = "", strict: bool = True, **kwargs: Any):
        if json is not None:
            if not "description" in json and strict:
                raise ValueError("Missing description")

            self.description = json["description"]

            self.extern = {k: Extern(v)
                           for k, v in json.get("extern", {}).items()}
        elif strict:
            raise ValueError("Missing json")

        super().__init__(json, path, strict, **kwargs)

    def toJson(self) -> Json:
        return {
            **super().toJson(),
            "description": self.description,
            "extern": {k: v.toJson() for k, v in self.extern.items()}
        }

    def __str__(self):
        return f"ProjectManifest(id={self.id}, type={self.type}, path={self.path}, description={self.description}, extern={self.extern})"

    def __repr__(self):
        return f"ProjectManifest({self.id})"


class Tool:
    cmd: str = ""
    args: list[str] = []
    files: list[str] = []

    def __init__(self, json: Json = None, strict: bool = True, **kwargs: Any):
        if json is not None:
            if not "cmd" in json and strict:
                raise ValueError("Missing cmd")

            self.cmd = json.get("cmd", self.cmd)

            if not "args" in json and strict:
                raise ValueError("Missing args")

            self.args = json.get("args", [])

            self.files = json.get("files", [])
        elif strict:
            raise ValueError("Missing json")

        for key in kwargs:
            setattr(self, key, kwargs[key])

    def toJson(self) -> Json:
        return {
            "cmd": self.cmd,
            "args": self.args,
            "files": self.files
        }

    def __str__(self):
        return f"Tool(cmd={self.cmd}, args={self.args}, files={self.files})"

    def __repr__(self):
        return f"Tool({self.cmd})"


Tools = dict[str, Tool]


class TargetManifest(Manifest):
    props: Props
    tools: Tools
    routing: dict[str, str]

    def __init__(self, json: Json = None, path: str = "", strict: bool = True, **kwargs: Any):
        if json is not None:
            if not "props" in json and strict:
                raise ValueError("Missing props")

            self.props = json["props"]

            if not "tools" in json and strict:
                raise ValueError("Missing tools")

            self.tools = {k: Tool(v) for k, v in json["tools"].items()}

            self.routing = json.get("routing", {})

        super().__init__(json, path, strict, **kwargs)

    def toJson(self) -> Json:
        return {
            **super().toJson(),
            "props": self.props,
            "tools": {k: v.toJson() for k, v in self.tools.items()},
            "routing": self.routing
        }

    def __repr__(self):
        return f"TargetManifest({self.id})"

    def route(self, componentSpec: str):
        return self.routing[componentSpec] if componentSpec in self.routing else componentSpec

    def cdefs(self) -> list[str]:
        defines: list[str] = []

        for key in self.props:
            macroname = key.lower().replace("-", "_")
            prop = self.props[key]
            macrovalue = str(prop).lower().replace(" ", "_").replace("-", "_")
            if isinstance(prop, bool):
                if prop:
                    defines += [f"-D__osdk_{macroname}__"]
            else:
                defines += [f"-D__osdk_{macroname}_{macrovalue}__"]

        return defines


class ComponentManifest(Manifest):
    decription: str = "(No description)"
    props: Props = {}
    tools: Tools = {}
    enableIf: dict[str, list[Any]] = {}
    requires: list[str] = []
    provides: list[str] = []
    subdirs: list[str] = []

    def __init__(self, json: Json = None, path: str = "", strict: bool = True, **kwargs: Any):
        if json is not None:
            self.decription = json.get("description", self.decription)
            self.props = json.get("props", self.props)
            self.tools = {k: Tool(v, strict=False)
                          for k, v in json.get("tools", {}).items()}
            self.enableIf = json.get("enableIf", self.enableIf)
            self.requires = json.get("requires", self.requires)
            self.provides = json.get("provides", self.provides)
            self.subdirs = list(map(lambda x: os.path.join(os.path.dirname(
                path), x), json.get("subdirs", [""])))

        super().__init__(json, path, strict, **kwargs)

    def toJson(self) -> Json:
        return {
            **super().toJson(),
            "description": self.decription,
            "props": self.props,
            "tools": {k: v.toJson() for k, v in self.tools.items()},
            "enableIf": self.enableIf,
            "requires": self.requires,
            "provides": self.provides,
            "subdirs": self.subdirs
        }

    def __repr__(self):
        return f"ComponentManifest({self.id})"

    def isEnabled(self, target: TargetManifest) -> tuple[bool, str]:
        for k, v in self.enableIf.items():
            if not k in target.props:
                logger.log(
                    f"Component {self.id} disabled by missing {k} in target")
                return False, f"Missing props '{k}' in target"

            if not target.props[k] in v:
                vStrs = [f"'{str(x)}'" for x in v]
                logger.log(
                    f"Component {self.id} disabled by {k}={target.props[k]} not in {v}")
                return False, f"Props missmatch for '{k}': Got '{target.props[k]}' but expected {', '.join(vStrs)}"

        return True, ""