darkfi/bin/app/gui/print_tree.py

from pydrk import SceneNodeType, PropertyType, CallArgType, Expr
from .api import api

def join(parent_path, child_name):
    if parent_path == "/":
        return f"/{child_name}"
    return f"{parent_path}/{child_name}"

def print_tree(node_path="/", depth=None):
    print(node_path)
    print_node_info(node_path, depth, indent=1)

def print_node_info(parent_path, depth, indent):
    if indent - 1 == depth:
        return

    ws = " "*4*indent
    for (child_name, child_id, child_type) in api.get_children(parent_path):
        match child_type:
            case SceneNodeType.ROOT:
                child_type = "root"
            case SceneNodeType.WINDOW:
                child_type = "window"
            case SceneNodeType.WINDOW_INPUT:
                child_type = "window_input"
            case SceneNodeType.KEYBOARD:
                child_type = "keyboard"
            case SceneNodeType.MOUSE:
                child_type = "mouse"
            case SceneNodeType.LAYER:
                child_type = "layer"
            case SceneNodeType.OBJECT:
                child_type = "object"
            case SceneNodeType.VECTOR_ART:
                child_type = "vector_art"
            case SceneNodeType.TEXT:
                child_type = "text"
            case SceneNodeType.TEXTURE:
                child_type = "texture"
            case SceneNodeType.FONTS:
                child_type = "fonts"
            case SceneNodeType.FONT:
                child_type = "font"
            case SceneNodeType.CHAT_VIEW:
                child_type = "chat_view"
            case SceneNodeType.BUTTON:
                child_type = "button"
            case SceneNodeType.SETTING:
                child_type = "setting"

        desc = f"{ws}{child_name}:{child_id}/"
        desc += " "*(50 - len(desc))
        desc += f"[{child_type}]"
        print(desc)

        if parent_path == "/":
            child_path = "/" + child_name
        else:
            child_path = parent_path + "/" + child_name

        print_node_info(child_path, depth, indent+1)

    for prop in api.get_properties(parent_path):
        prop_val = api.get_property_value(parent_path, prop.name)

        def fmt_str(pv):
            return pv if isinstance(pv, Expr) else f"\"{pv}\""

        def fmt_f32(pv):
            return pv if isinstance(pv, Expr) else f"{pv:.2f}"

        if prop.type == PropertyType.STR:
            prop_val = "[" + ", ".join(fmt_str(pv) for pv in prop_val) + "]"
        elif prop.type == PropertyType.FLOAT32:
            prop_val = "[" + ", ".join(fmt_f32(pv) for pv in prop_val) + "]"

        if len(prop_val) == 1:
            prop_val = prop_val[0]

        prop_val = f" = {prop_val}"

        prop_type = PropertyType.to_str(prop.type)

        print(f"{ws}{prop.name}: {prop_type}{prop_val}")
        if prop.depends:
            print(f"{ws}    depends: {prop.depends}")

    for sig in api.get_signals(parent_path):
        print(f"{ws}~{sig}")
        for slot_name, slot_id in api.get_slots(parent_path, sig):
            print(f"{ws}- '{slot_name}' ({slot_id})")

    for method_name in api.get_methods(parent_path):
        args, results = api.get_method(parent_path, method_name)

        args = [f"{name}: " + CallArgType.to_str(typ) for (name, _, typ) in args]
        results = [f"{name}: " + CallArgType.to_str(typ) for (name, _, typ) in results]

        method_str = f"{method_name}(" + ", ".join(args) + ") -> (" + ", ".join(results) + ")"
        print(f"{ws}{method_str}")