jaclang 0.7.33__py3-none-any.whl → 0.8.0__py3-none-any.whl

jaclang/runtimelib/machine.py

@@ -1,147 +1,1385 @@
-"""Jac Machine module."""
+"""Jac Language Features."""
 
 from __future__ import annotations
 
+import ast as ast3
+import asyncio
+import fnmatch
+import html
 import inspect
-import marshal
 import os
 import sys
 import tempfile
 import types
-from contextvars import ContextVar
-from typing import Optional, Union
-
-from jaclang.compiler.absyntree import Module
-from jaclang.compiler.compile import compile_jac
-from jaclang.compiler.constant import Constants as Con
-from jaclang.compiler.semtable import SemRegistry
-from jaclang.runtimelib.architype import (
-    Architype,
-    EdgeArchitype,
-    NodeArchitype,
-    WalkerArchitype,
+from collections import OrderedDict
+from concurrent.futures import Future, ThreadPoolExecutor
+from dataclasses import MISSING, dataclass, field
+from functools import partial, wraps
+from inspect import getfile
+from logging import getLogger
+from typing import (
+    Any,
+    Callable,
+    Mapping,
+    Optional,
+    ParamSpec,
+    Sequence,
+    TYPE_CHECKING,
+    Type,
+    TypeAlias,
+    TypeVar,
+    Union,
+    cast,
 )
-from jaclang.utils.log import logging
+from uuid import UUID
 
 
-logger = logging.getLogger(__name__)
+from jaclang.compiler import unitree as ast
+from jaclang.compiler.constant import Constants as Con, EdgeDir, colors
+from jaclang.compiler.passes.main.pyast_gen_pass import PyastGenPass
+from jaclang.compiler.program import JacProgram
+from jaclang.runtimelib.archetype import (
+    DataSpatialFunction,
+    GenericEdge as _GenericEdge,
+    Root as _Root,
+)
+from jaclang.runtimelib.constructs import (
+    AccessLevel,
+    Anchor,
+    Archetype,
+    EdgeAnchor,
+    EdgeArchetype,
+    GenericEdge,
+    JacTestCheck,
+    NodeAnchor,
+    NodeArchetype,
+    Root,
+    WalkerAnchor,
+    WalkerArchetype,
+)
+from jaclang.runtimelib.memory import Memory, Shelf, ShelfStorage
+from jaclang.runtimelib.utils import (
+    all_issubclass,
+    collect_node_connections,
+    traverse_graph,
+)
+from jaclang.utils import infer_language
 
+import pluggy
 
-JACMACHINE_CONTEXT = ContextVar["JacMachine | None"]("JacMachine")
 
+plugin_manager = pluggy.PluginManager("jac")
+hookspec = pluggy.HookspecMarker("jac")
+hookimpl = pluggy.HookimplMarker("jac")
+logger = getLogger(__name__)
 
-class JacMachine:
-    """JacMachine to handle the VM-related functionalities and loaded programs."""
+T = TypeVar("T")
+P = ParamSpec("P")
 
-    def __init__(self, base_path: str = "") -> None:
-        """Initialize the JacMachine object."""
-        self.loaded_modules: dict[str, types.ModuleType] = {}
-        if not base_path:
-            base_path = os.getcwd()
-        # Ensure the base_path is a list rather than a string
-        self.base_path = base_path
-        self.base_path_dir = (
-            os.path.dirname(base_path)
-            if not os.path.isdir(base_path)
-            else os.path.abspath(base_path)
+
+class JacAccessValidation:
+    """Jac Access Validation Specs."""
+
+    @staticmethod
+    def elevate_root() -> None:
+        """Elevate context root to system_root."""
+        jctx = JacMachineInterface.get_context()
+        jctx.root_state = jctx.system_root
+
+    @staticmethod
+    def allow_root(
+        archetype: Archetype,
+        root_id: UUID,
+        level: AccessLevel | int | str = AccessLevel.READ,
+    ) -> None:
+        """Allow all access from target root graph to current Archetype."""
+        level = AccessLevel.cast(level)
+        access = archetype.__jac__.access.roots
+
+        _root_id = str(root_id)
+        if level != access.anchors.get(_root_id, AccessLevel.NO_ACCESS):
+            access.anchors[_root_id] = level
+
+    @staticmethod
+    def disallow_root(
+        archetype: Archetype,
+        root_id: UUID,
+        level: AccessLevel | int | str = AccessLevel.READ,
+    ) -> None:
+        """Disallow all access from target root graph to current Archetype."""
+        level = AccessLevel.cast(level)
+        access = archetype.__jac__.access.roots
+
+        access.anchors.pop(str(root_id), None)
+
+    @staticmethod
+    def perm_grant(
+        archetype: Archetype, level: AccessLevel | int | str = AccessLevel.READ
+    ) -> None:
+        """Allow everyone to access current Archetype."""
+        anchor = archetype.__jac__
+        level = AccessLevel.cast(level)
+        if level != anchor.access.all:
+            anchor.access.all = level
+
+    @staticmethod
+    def perm_revoke(archetype: Archetype) -> None:
+        """Disallow others to access current Archetype."""
+        anchor = archetype.__jac__
+        if anchor.access.all > AccessLevel.NO_ACCESS:
+            anchor.access.all = AccessLevel.NO_ACCESS
+
+    @staticmethod
+    def check_read_access(to: Anchor) -> bool:
+        """Read Access Validation."""
+        if not (
+            access_level := JacMachineInterface.check_access_level(to)
+            > AccessLevel.NO_ACCESS
+        ):
+            logger.info(
+                f"Current root doesn't have read access to {to.__class__.__name__}[{to.id}]"
+            )
+        return access_level
+
+    @staticmethod
+    def check_connect_access(to: Anchor) -> bool:
+        """Write Access Validation."""
+        if not (
+            access_level := JacMachineInterface.check_access_level(to)
+            > AccessLevel.READ
+        ):
+            logger.info(
+                f"Current root doesn't have connect access to {to.__class__.__name__}[{to.id}]"
+            )
+        return access_level
+
+    @staticmethod
+    def check_write_access(to: Anchor) -> bool:
+        """Write Access Validation."""
+        if not (
+            access_level := JacMachineInterface.check_access_level(to)
+            > AccessLevel.CONNECT
+        ):
+            logger.info(
+                f"Current root doesn't have write access to {to.__class__.__name__}[{to.id}]"
+            )
+        return access_level
+
+    @staticmethod
+    def check_access_level(to: Anchor) -> AccessLevel:
+        """Access validation."""
+        if not to.persistent:
+            return AccessLevel.WRITE
+
+        jctx = JacMachineInterface.get_context()
+
+        jroot = jctx.root_state
+
+        # if current root is system_root
+        # if current root id is equal to target anchor's root id
+        # if current root is the target anchor
+        if jroot == jctx.system_root or jroot.id == to.root or jroot == to:
+            return AccessLevel.WRITE
+
+        access_level = AccessLevel.NO_ACCESS
+
+        # if target anchor have set access.all
+        if (to_access := to.access).all > AccessLevel.NO_ACCESS:
+            access_level = to_access.all
+
+        # if target anchor's root have set allowed roots
+        # if current root is allowed to the whole graph of target anchor's root
+        if to.root and isinstance(to_root := jctx.mem.find_one(to.root), Anchor):
+            if to_root.access.all > access_level:
+                access_level = to_root.access.all
+
+            if (level := to_root.access.roots.check(str(jroot.id))) is not None:
+                access_level = level
+
+        # if target anchor have set allowed roots
+        # if current root is allowed to target anchor
+        if (level := to_access.roots.check(str(jroot.id))) is not None:
+            access_level = level
+
+        return access_level
+
+
+class JacNode:
+    """Jac Node Operations."""
+
+    @staticmethod
+    def node_dot(node: NodeArchetype, dot_file: Optional[str] = None) -> str:
+        """Generate Dot file for visualizing nodes and edges."""
+        visited_nodes: set[NodeAnchor] = set()
+        connections: set[tuple[NodeArchetype, NodeArchetype, str]] = set()
+        unique_node_id_dict = {}
+
+        collect_node_connections(node.__jac__, visited_nodes, connections)
+        dot_content = 'digraph {\nnode [style="filled", shape="ellipse", fillcolor="invis", fontcolor="black"];\n'
+        for idx, i in enumerate([nodes_.archetype for nodes_ in visited_nodes]):
+            unique_node_id_dict[i] = (i.__class__.__name__, str(idx))
+            dot_content += f'{idx} [label="{i}"];\n'
+        dot_content += 'edge [color="gray", style="solid"];\n'
+
+        for pair in list(set(connections)):
+            dot_content += (
+                f"{unique_node_id_dict[pair[0]][1]} -> {unique_node_id_dict[pair[1]][1]}"
+                f' [label="{pair[2]}"];\n'
+            )
+        if dot_file:
+            with open(dot_file, "w") as f:
+                f.write(dot_content + "}")
+        return dot_content + "}"
+
+    @staticmethod
+    def get_edges(
+        node: NodeAnchor,
+        dir: EdgeDir,
+        filter: Callable[[EdgeArchetype], bool] | None,
+        target_obj: list[NodeArchetype] | None,
+    ) -> list[EdgeArchetype]:
+        """Get edges connected to this node."""
+        ret_edges: list[EdgeArchetype] = []
+        for anchor in node.edges:
+            if (
+                (source := anchor.source)
+                and (target := anchor.target)
+                and (not filter or filter(anchor.archetype))
+                and source.archetype
+                and target.archetype
+            ):
+                if (
+                    dir in [EdgeDir.OUT, EdgeDir.ANY]
+                    and node == source
+                    and (not target_obj or target.archetype in target_obj)
+                    and JacMachineInterface.check_read_access(target)
+                ):
+                    ret_edges.append(anchor.archetype)
+                if (
+                    dir in [EdgeDir.IN, EdgeDir.ANY]
+                    and node == target
+                    and (not target_obj or source.archetype in target_obj)
+                    and JacMachineInterface.check_read_access(source)
+                ):
+                    ret_edges.append(anchor.archetype)
+        return ret_edges
+
+    @staticmethod
+    def edges_to_nodes(
+        node: NodeAnchor,
+        dir: EdgeDir,
+        filter: Callable[[EdgeArchetype], bool] | None,
+        target_obj: list[NodeArchetype] | None,
+    ) -> list[NodeArchetype]:
+        """Get set of nodes connected to this node."""
+        ret_edges: list[NodeArchetype] = []
+        for anchor in node.edges:
+            if (
+                (source := anchor.source)
+                and (target := anchor.target)
+                and (not filter or filter(anchor.archetype))
+                and source.archetype
+                and target.archetype
+            ):
+                if (
+                    dir in [EdgeDir.OUT, EdgeDir.ANY]
+                    and node == source
+                    and (not target_obj or target.archetype in target_obj)
+                    and JacMachineInterface.check_read_access(target)
+                ):
+                    ret_edges.append(target.archetype)
+                if (
+                    dir in [EdgeDir.IN, EdgeDir.ANY]
+                    and node == target
+                    and (not target_obj or source.archetype in target_obj)
+                    and JacMachineInterface.check_read_access(source)
+                ):
+                    ret_edges.append(source.archetype)
+        return ret_edges
+
+    @staticmethod
+    def remove_edge(node: NodeAnchor, edge: EdgeAnchor) -> None:
+        """Remove reference without checking sync status."""
+        for idx, ed in enumerate(node.edges):
+            if ed.id == edge.id:
+                node.edges.pop(idx)
+                break
+
+
+class JacEdge:
+    """Jac Edge Operations."""
+
+    @staticmethod
+    def detach(edge: EdgeAnchor) -> None:
+        """Detach edge from nodes."""
+        JacMachineInterface.remove_edge(node=edge.source, edge=edge)
+        JacMachineInterface.remove_edge(node=edge.target, edge=edge)
+
+
+class JacWalker:
+    """Jac Edge Operations."""
+
+    @staticmethod
+    def visit(
+        walker: WalkerArchetype,
+        expr: (
+            list[NodeArchetype | EdgeArchetype]
+            | list[NodeArchetype]
+            | list[EdgeArchetype]
+            | NodeArchetype
+            | EdgeArchetype
+        ),
+    ) -> bool:  # noqa: ANN401
+        """Jac's visit stmt feature."""
+        if isinstance(walker, WalkerArchetype):
+            """Walker visits node."""
+            wanch = walker.__jac__
+            before_len = len(wanch.next)
+            for anchor in (
+                (i.__jac__ for i in expr) if isinstance(expr, list) else [expr.__jac__]
+            ):
+                if anchor not in wanch.ignores:
+                    if isinstance(anchor, NodeAnchor):
+                        wanch.next.append(anchor)
+                    elif isinstance(anchor, EdgeAnchor):
+                        if target := anchor.target:
+                            wanch.next.append(target)
+                        else:
+                            raise ValueError("Edge has no target.")
+            return len(wanch.next) > before_len
+        else:
+            raise TypeError("Invalid walker object")
+
+    @staticmethod
+    def ignore(
+        walker: WalkerArchetype,
+        expr: (
+            list[NodeArchetype | EdgeArchetype]
+            | list[NodeArchetype]
+            | list[EdgeArchetype]
+            | NodeArchetype
+            | EdgeArchetype
+        ),
+    ) -> bool:  # noqa: ANN401
+        """Jac's ignore stmt feature."""
+        if isinstance(walker, WalkerArchetype):
+            wanch = walker.__jac__
+            before_len = len(wanch.ignores)
+            for anchor in (
+                (i.__jac__ for i in expr) if isinstance(expr, list) else [expr.__jac__]
+            ):
+                if anchor not in wanch.ignores:
+                    if isinstance(anchor, NodeAnchor):
+                        wanch.ignores.append(anchor)
+                    elif isinstance(anchor, EdgeAnchor):
+                        if target := anchor.target:
+                            wanch.ignores.append(target)
+                        else:
+                            raise ValueError("Edge has no target.")
+            return len(wanch.ignores) > before_len
+        else:
+            raise TypeError("Invalid walker object")
+
+    @staticmethod
+    def spawn_call(walker: WalkerAnchor, node: NodeAnchor) -> WalkerArchetype:
+        """Jac's spawn operator feature."""
+        warch = walker.archetype
+        walker.path = []
+        walker.next = [node]
+        current_node = node.archetype
+
+        # walker entry
+        for i in warch._jac_entry_funcs_:
+            if not i.trigger:
+                i.func(warch, current_node)
+            if walker.disengaged:
+                return warch
+
+        while len(walker.next):
+            if current_node := walker.next.pop(0).archetype:
+                # walker entry with
+                for i in warch._jac_entry_funcs_:
+                    if (
+                        i.trigger
+                        and all_issubclass(i.trigger, NodeArchetype)
+                        and isinstance(current_node, i.trigger)
+                    ):
+                        i.func(warch, current_node)
+                    if walker.disengaged:
+                        return warch
+
+                # node entry
+                for i in current_node._jac_entry_funcs_:
+                    if not i.trigger:
+                        i.func(current_node, warch)
+                    if walker.disengaged:
+                        return warch
+
+                # node entry with
+                for i in current_node._jac_entry_funcs_:
+                    if (
+                        i.trigger
+                        and all_issubclass(i.trigger, WalkerArchetype)
+                        and isinstance(warch, i.trigger)
+                    ):
+                        i.func(current_node, warch)
+                    if walker.disengaged:
+                        return warch
+
+                # node exit with
+                for i in current_node._jac_exit_funcs_:
+                    if (
+                        i.trigger
+                        and all_issubclass(i.trigger, WalkerArchetype)
+                        and isinstance(warch, i.trigger)
+                    ):
+                        i.func(current_node, warch)
+                    if walker.disengaged:
+                        return warch
+
+                # node exit
+                for i in current_node._jac_exit_funcs_:
+                    if not i.trigger:
+                        i.func(current_node, warch)
+                    if walker.disengaged:
+                        return warch
+
+                # walker exit with
+                for i in warch._jac_exit_funcs_:
+                    if (
+                        i.trigger
+                        and all_issubclass(i.trigger, NodeArchetype)
+                        and isinstance(current_node, i.trigger)
+                    ):
+                        i.func(warch, current_node)
+                    if walker.disengaged:
+                        return warch
+        # walker exit
+        for i in warch._jac_exit_funcs_:
+            if not i.trigger:
+                i.func(warch, current_node)
+            if walker.disengaged:
+                return warch
+
+        walker.ignores = []
+        return warch
+
+    @staticmethod
+    def spawn(op1: Archetype, op2: Archetype) -> WalkerArchetype | asyncio.Future:
+        """Jac's spawn operator feature."""
+        if isinstance(op1, WalkerArchetype):
+            warch = op1
+            walker = op1.__jac__
+            if isinstance(op2, NodeArchetype):
+                node = op2.__jac__
+            elif isinstance(op2, EdgeArchetype):
+                node = op2.__jac__.target
+            else:
+                raise TypeError("Invalid target object")
+        elif isinstance(op2, WalkerArchetype):
+            warch = op2
+            walker = op2.__jac__
+            if isinstance(op1, NodeArchetype):
+                node = op1.__jac__
+            elif isinstance(op1, EdgeArchetype):
+                node = op1.__jac__.target
+            else:
+                raise TypeError("Invalid target object")
+        else:
+            raise TypeError("Invalid walker object")
+
+        if warch.__jac_async__:
+            machine = JacMachineInterface.py_get_jac_machine()
+            _event_loop = machine._event_loop
+            func = partial(JacMachineInterface.spawn_call, *(walker, node))
+            return asyncio.ensure_future(
+                _event_loop.run_in_executor(None, func), loop=_event_loop
+            )
+        else:
+            return JacMachineInterface.spawn_call(walker=walker, node=node)
+
+    @staticmethod
+    def disengage(walker: WalkerArchetype) -> bool:
+        """Jac's disengage stmt feature."""
+        walker.__jac__.disengaged = True
+        return True
+
+
+class JacClassReferences:
+    """Default Classes References."""
+
+    TYPE_CHECKING: bool = TYPE_CHECKING
+    EdgeDir: TypeAlias = EdgeDir
+    DSFunc: TypeAlias = DataSpatialFunction
+
+    Obj: TypeAlias = Archetype
+    Node: TypeAlias = NodeArchetype
+    Edge: TypeAlias = EdgeArchetype
+    Walker: TypeAlias = WalkerArchetype
+
+    Root: TypeAlias = _Root
+    GenericEdge: TypeAlias = _GenericEdge
+
+
+class JacBuiltin:
+    """Jac Builtins."""
+
+    @staticmethod
+    def dotgen(
+        node: NodeArchetype,
+        depth: int,
+        traverse: bool,
+        edge_type: Optional[list[str]],
+        bfs: bool,
+        edge_limit: int,
+        node_limit: int,
+        dot_file: Optional[str],
+    ) -> str:
+        """Generate Dot file for visualizing nodes and edges."""
+        edge_type = edge_type if edge_type else []
+        visited_nodes: list[NodeArchetype] = []
+        node_depths: dict[NodeArchetype, int] = {node: 0}
+        queue: list = [[node, 0]]
+        connections: list[tuple[NodeArchetype, NodeArchetype, EdgeArchetype]] = []
+
+        def dfs(node: NodeArchetype, cur_depth: int) -> None:
+            """Depth first search."""
+            if node not in visited_nodes:
+                visited_nodes.append(node)
+                traverse_graph(
+                    node,
+                    cur_depth,
+                    depth,
+                    edge_type,
+                    traverse,
+                    connections,
+                    node_depths,
+                    visited_nodes,
+                    queue,
+                    bfs,
+                    dfs,
+                    node_limit,
+                    edge_limit,
+                )
+
+        if bfs:
+            cur_depth = 0
+            while queue:
+                current_node, cur_depth = queue.pop(0)
+                if current_node not in visited_nodes:
+                    visited_nodes.append(current_node)
+                    traverse_graph(
+                        current_node,
+                        cur_depth,
+                        depth,
+                        edge_type,
+                        traverse,
+                        connections,
+                        node_depths,
+                        visited_nodes,
+                        queue,
+                        bfs,
+                        dfs,
+                        node_limit,
+                        edge_limit,
+                    )
+        else:
+            dfs(node, cur_depth=0)
+        dot_content = (
+            'digraph {\nnode [style="filled", shape="ellipse", '
+            'fillcolor="invis", fontcolor="black"];\n'
         )
-        self.jac_program: Optional[JacProgram] = None
+        for source, target, edge in connections:
+            edge_label = html.escape(str(edge.__jac__.archetype))
+            dot_content += (
+                f"{visited_nodes.index(source)} -> {visited_nodes.index(target)} "
+                f' [label="{edge_label if "GenericEdge" not in edge_label else ""}"];\n'
+            )
+        for node_ in visited_nodes:
+            color = (
+                colors[node_depths[node_]] if node_depths[node_] < 25 else colors[24]
+            )
+            dot_content += (
+                f'{visited_nodes.index(node_)} [label="{html.escape(str(node_.__jac__.archetype))}"'
+                f'fillcolor="{color}"];\n'
+            )
+        if dot_file:
+            with open(dot_file, "w") as f:
+                f.write(dot_content + "}")
+        return dot_content + "}"
 
-        JACMACHINE_CONTEXT.set(self)
 
-    def attach_program(self, jac_program: "JacProgram") -> None:
-        """Attach a JacProgram to the machine."""
-        self.jac_program = jac_program
+class JacCmd:
+    """Jac CLI command."""
+
+    @staticmethod
+    def create_cmd() -> None:
+        """Create Jac CLI cmds."""
+
+
+class JacBasics:
+    """Jac Feature."""
+
+    @staticmethod
+    def setup() -> None:
+        """Set Class References."""
+
+    @staticmethod
+    def get_context() -> JacMachine:
+        """Get current execution context."""
+        return JacMachineInterface.py_get_jac_machine()
+
+    @staticmethod
+    def reset_graph(root: Optional[Root] = None) -> int:
+        """Purge current or target graph."""
+        ctx = JacMachineInterface.get_context()
+        mem = cast(ShelfStorage, ctx.mem)
+        ranchor = root.__jac__ if root else ctx.root_state
+
+        deleted_count = 0
+        for anchor in (
+            anchors.values()
+            if isinstance(anchors := mem.__shelf__, Shelf)
+            else mem.__mem__.values()
+        ):
+            if anchor == ranchor or anchor.root != ranchor.id:
+                continue
+
+            if loaded_anchor := mem.find_by_id(anchor.id):
+                deleted_count += 1
+                JacMachineInterface.destroy([loaded_anchor])
+
+        return deleted_count
+
+    @staticmethod
+    def get_object(id: str) -> Archetype | None:
+        """Get object given id."""
+        if id == "root":
+            return JacMachineInterface.get_context().root_state.archetype
+        elif obj := JacMachineInterface.get_context().mem.find_by_id(UUID(id)):
+            return obj.archetype
 
-    def get_mod_bundle(self) -> Optional[Module]:
-        """Retrieve the mod_bundle from the attached JacProgram."""
-        if self.jac_program:
-            return self.jac_program.mod_bundle
         return None
 
-    def get_bytecode(
-        self,
-        module_name: str,
-        full_target: str,
-        caller_dir: str,
-        cachable: bool = True,
-        reload: bool = False,
-    ) -> Optional[types.CodeType]:
-        """Retrieve bytecode from the attached JacProgram."""
-        if self.jac_program:
-            return self.jac_program.get_bytecode(
-                module_name, full_target, caller_dir, cachable, reload=reload
+    @staticmethod
+    def object_ref(obj: Archetype) -> str:
+        """Get object reference id."""
+        return obj.__jac__.id.hex
+
+    @staticmethod
+    def make_archetype(cls: Type[Archetype]) -> Type[Archetype]:
+        """Create a obj archetype."""
+        entries: OrderedDict[str, JacMachineInterface.DSFunc] = OrderedDict(
+            (fn.name, fn) for fn in cls._jac_entry_funcs_
+        )
+        exits: OrderedDict[str, JacMachineInterface.DSFunc] = OrderedDict(
+            (fn.name, fn) for fn in cls._jac_exit_funcs_
+        )
+        for func in cls.__dict__.values():
+            if callable(func):
+                if hasattr(func, "__jac_entry"):
+                    entries[func.__name__] = JacMachineInterface.DSFunc(
+                        func.__name__, func
+                    )
+                if hasattr(func, "__jac_exit"):
+                    exits[func.__name__] = JacMachineInterface.DSFunc(
+                        func.__name__, func
+                    )
+
+        cls._jac_entry_funcs_ = [*entries.values()]
+        cls._jac_exit_funcs_ = [*exits.values()]
+
+        dataclass(eq=False)(cls)
+        return cls
+
+    @staticmethod
+    def impl_patch_filename(
+        file_loc: str,
+    ) -> Callable[[Callable[P, T]], Callable[P, T]]:
+        """Update impl file location."""
+
+        def decorator(func: Callable[P, T]) -> Callable[P, T]:
+            try:
+                code = func.__code__
+                new_code = types.CodeType(
+                    code.co_argcount,
+                    code.co_posonlyargcount,
+                    code.co_kwonlyargcount,
+                    code.co_nlocals,
+                    code.co_stacksize,
+                    code.co_flags,
+                    code.co_code,
+                    code.co_consts,
+                    code.co_names,
+                    code.co_varnames,
+                    file_loc,
+                    code.co_name,
+                    code.co_qualname,
+                    code.co_firstlineno,
+                    code.co_linetable,
+                    code.co_exceptiontable,
+                    code.co_freevars,
+                    code.co_cellvars,
+                )
+                func.__code__ = new_code
+            except AttributeError:
+                pass
+            return func
+
+        return decorator
+
+    @staticmethod
+    def py_get_jac_machine() -> JacMachine:
+        """Get jac machine from python context."""
+        machine = JacBasics.py_find_jac_machine()
+        if not machine:
+            raise RuntimeError("Jac machine not found in python context. ")
+        return machine
+
+    @staticmethod
+    def py_find_jac_machine() -> Optional[JacMachine]:
+        """Get jac machine from python context."""
+        machine = None
+        for i in inspect.stack():
+            machine = i.frame.f_globals.get("__jac_mach__") or i.frame.f_locals.get(
+                "__jac_mach__"
             )
-        return None
+            if machine:
+                break
+        return machine
+
+    @staticmethod
+    def py_jac_import(
+        target: str,
+        base_path: str,
+        absorb: bool = False,
+        mdl_alias: Optional[str] = None,
+        override_name: Optional[str] = None,
+        items: Optional[dict[str, Union[str, Optional[str]]]] = None,
+        reload_module: Optional[bool] = False,
+    ) -> tuple[types.ModuleType, ...]:
+        """Core Import Process."""
+        machine = JacBasics.py_find_jac_machine()
+        if not machine:
+            machine = JacMachine(base_path=base_path)
+        return JacMachineInterface.jac_import(
+            mach=machine,
+            target=target,
+            base_path=base_path,
+            absorb=absorb,
+            mdl_alias=mdl_alias,
+            override_name=override_name,
+            items=items,
+            reload_module=reload_module,
+        )
+
+    @staticmethod
+    def jac_import(
+        mach: JacMachine,
+        target: str,
+        base_path: str,
+        absorb: bool = False,
+        mdl_alias: Optional[str] = None,
+        override_name: Optional[str] = None,
+        items: Optional[dict[str, Union[str, Optional[str]]]] = None,
+        reload_module: Optional[bool] = False,
+    ) -> tuple[types.ModuleType, ...]:
+        """Core Import Process."""
+        from jaclang.runtimelib.importer import (
+            ImportPathSpec,
+            JacImporter,
+            PythonImporter,
+        )
 
-    def get_sem_ir(self, mod_sem_ir: SemRegistry | None) -> None:
-        """Update semtable on the attached JacProgram."""
-        if self.jac_program and mod_sem_ir:
-            if self.jac_program.sem_ir:
-                self.jac_program.sem_ir.registry.update(mod_sem_ir.registry)
+        lng = infer_language(target, base_path)
+
+        spec = ImportPathSpec(
+            target,
+            base_path,
+            absorb,
+            mdl_alias,
+            override_name,
+            lng,
+            items,
+        )
+
+        if not mach.jac_program:
+            JacMachineInterface.attach_program(mach, JacProgram())
+
+        if lng == "py":
+            import_result = PythonImporter(mach).run_import(spec)
+        else:
+            import_result = JacImporter(mach).run_import(spec, reload_module)
+
+        return (
+            (import_result.ret_mod,)
+            if absorb or not items
+            else tuple(import_result.ret_items)
+        )
+
+    @staticmethod
+    def jac_test(test_fun: Callable) -> Callable:
+        """Create a test."""
+        file_path = getfile(test_fun)
+        func_name = test_fun.__name__
+
+        def test_deco() -> None:
+            test_fun(JacTestCheck())
+
+        test_deco.__name__ = test_fun.__name__
+        JacTestCheck.add_test(file_path, func_name, test_deco)
+
+        return test_deco
+
+    @staticmethod
+    def run_test(
+        mach: JacMachine,
+        filepath: str,
+        func_name: Optional[str] = None,
+        filter: Optional[str] = None,
+        xit: bool = False,
+        maxfail: Optional[int] = None,
+        directory: Optional[str] = None,
+        verbose: bool = False,
+    ) -> int:
+        """Run the test suite in the specified .jac file."""
+        test_file = False
+        ret_count = 0
+        if filepath:
+            if filepath.endswith(".jac"):
+                base, mod_name = os.path.split(filepath)
+                base = base if base else "./"
+                mod_name = mod_name[:-4]
+                if mod_name.endswith(".test"):
+                    mod_name = mod_name[:-5]
+                JacTestCheck.reset()
+                JacMachineInterface.jac_import(
+                    mach=mach, target=mod_name, base_path=base
+                )
+                JacTestCheck.run_test(
+                    xit, maxfail, verbose, os.path.abspath(filepath), func_name
+                )
+                ret_count = JacTestCheck.failcount
             else:
-                self.jac_program.sem_ir = mod_sem_ir
+                print("Not a .jac file.")
+        else:
+            directory = directory if directory else os.getcwd()
+
+        if filter or directory:
+            current_dir = directory if directory else os.getcwd()
+            for root_dir, _, files in os.walk(current_dir, topdown=True):
+                files = (
+                    [file for file in files if fnmatch.fnmatch(file, filter)]
+                    if filter
+                    else files
+                )
+                files = [
+                    file
+                    for file in files
+                    if not file.endswith((".test.jac", ".impl.jac"))
+                ]
+                for file in files:
+                    if file.endswith(".jac"):
+                        test_file = True
+                        print(f"\n\n\t\t* Inside {root_dir}" + "/" + f"{file} *")
+                        JacTestCheck.reset()
+                        JacMachineInterface.jac_import(
+                            mach=mach, target=file[:-4], base_path=root_dir
+                        )
+                        JacTestCheck.run_test(
+                            xit, maxfail, verbose, os.path.abspath(file), func_name
+                        )
+
+                    if JacTestCheck.breaker and (xit or maxfail):
+                        break
+                if JacTestCheck.breaker and (xit or maxfail):
+                    break
+            JacTestCheck.breaker = False
+            ret_count += JacTestCheck.failcount
+            JacTestCheck.failcount = 0
+            print("No test files found.") if not test_file else None
+
+        return ret_count
+
+    @staticmethod
+    def field(factory: Callable[[], T] | None = None, init: bool = True) -> T:
+        """Jac's field handler."""
+        if factory:
+            return field(default_factory=factory)
+        return field(init=init)
+
+    @staticmethod
+    def report(expr: Any, custom: bool = False) -> None:  # noqa: ANN401
+        """Jac's report stmt feature."""
+        ctx = JacMachineInterface.get_context()
+        if custom:
+            ctx.custom = expr
+        else:
+            ctx.reports.append(expr)
+
+    @staticmethod
+    def refs(
+        sources: NodeArchetype | list[NodeArchetype],
+        targets: NodeArchetype | list[NodeArchetype] | None = None,
+        dir: EdgeDir = EdgeDir.OUT,
+        filter: Callable[[EdgeArchetype], bool] | None = None,
+        edges_only: bool = False,
+    ) -> list[NodeArchetype] | list[EdgeArchetype]:
+        """Jac's apply_dir stmt feature."""
+        if isinstance(sources, NodeArchetype):
+            sources = [sources]
+        targ_obj_set: Optional[list[NodeArchetype]] = (
+            [targets]
+            if isinstance(targets, NodeArchetype)
+            else targets if targets else None
+        )
+        if edges_only:
+            connected_edges: list[EdgeArchetype] = []
+            for node in sources:
+                edges = JacMachineInterface.get_edges(
+                    node.__jac__, dir, filter, target_obj=targ_obj_set
+                )
+                connected_edges.extend(
+                    edge for edge in edges if edge not in connected_edges
+                )
+            return connected_edges
+        else:
+            connected_nodes: list[NodeArchetype] = []
+            for node in sources:
+                nodes = JacMachineInterface.edges_to_nodes(
+                    node.__jac__, dir, filter, target_obj=targ_obj_set
+                )
+                connected_nodes.extend(
+                    node for node in nodes if node not in connected_nodes
+                )
+            return connected_nodes
+
+    @staticmethod
+    def filter(
+        items: list[Archetype],
+        func: Callable[[Archetype], bool],
+    ) -> list[Archetype]:
+        """Jac's filter archetype list."""
+        return [item for item in items if func(item)]
+
+    @staticmethod
+    def connect(
+        left: NodeArchetype | list[NodeArchetype],
+        right: NodeArchetype | list[NodeArchetype],
+        edge: Type[EdgeArchetype] | EdgeArchetype | None = None,
+        undir: bool = False,
+        conn_assign: tuple[tuple, tuple] | None = None,
+        edges_only: bool = False,
+    ) -> list[NodeArchetype] | list[EdgeArchetype]:
+        """Jac's connect operator feature."""
+        left = [left] if isinstance(left, NodeArchetype) else left
+        right = [right] if isinstance(right, NodeArchetype) else right
+        edges = []
+
+        for i in left:
+            _left = i.__jac__
+            if JacMachineInterface.check_connect_access(_left):
+                for j in right:
+                    _right = j.__jac__
+                    if JacMachineInterface.check_connect_access(_right):
+                        edges.append(
+                            JacMachineInterface.build_edge(
+                                is_undirected=undir,
+                                conn_type=edge,
+                                conn_assign=conn_assign,
+                            )(_left, _right)
+                        )
+        return right if not edges_only else edges
+
+    @staticmethod
+    def disconnect(
+        left: NodeArchetype | list[NodeArchetype],
+        right: NodeArchetype | list[NodeArchetype],
+        dir: EdgeDir = EdgeDir.OUT,
+        filter: Callable[[EdgeArchetype], bool] | None = None,
+    ) -> bool:
+        """Jac's disconnect operator feature."""
+        disconnect_occurred = False
+        left = [left] if isinstance(left, NodeArchetype) else left
+        right = [right] if isinstance(right, NodeArchetype) else right
+
+        for i in left:
+            node = i.__jac__
+            for anchor in set(node.edges):
+                if (
+                    (source := anchor.source)
+                    and (target := anchor.target)
+                    and (not filter or filter(anchor.archetype))
+                    and source.archetype
+                    and target.archetype
+                ):
+                    if (
+                        dir in [EdgeDir.OUT, EdgeDir.ANY]
+                        and node == source
+                        and target.archetype in right
+                        and JacMachineInterface.check_connect_access(target)
+                    ):
+                        (
+                            JacMachineInterface.destroy([anchor])
+                            if anchor.persistent
+                            else JacMachineInterface.detach(anchor)
+                        )
+                        disconnect_occurred = True
+                    if (
+                        dir in [EdgeDir.IN, EdgeDir.ANY]
+                        and node == target
+                        and source.archetype in right
+                        and JacMachineInterface.check_connect_access(source)
+                    ):
+                        (
+                            JacMachineInterface.destroy([anchor])
+                            if anchor.persistent
+                            else JacMachineInterface.detach(anchor)
+                        )
+                        disconnect_occurred = True
+
+        return disconnect_occurred
+
+    @staticmethod
+    def assign(target: list[T], attr_val: tuple[tuple[str], tuple[Any]]) -> list[T]:
+        """Jac's assign comprehension feature."""
+        for obj in target:
+            attrs, values = attr_val
+            for attr, value in zip(attrs, values):
+                setattr(obj, attr, value)
+        return target
+
+    @staticmethod
+    def root() -> Root:
+        """Jac's root getter."""
+        return JacMachineInterface.py_get_jac_machine().get_root()
+
+    @staticmethod
+    def build_edge(
+        is_undirected: bool,
+        conn_type: Optional[Type[EdgeArchetype] | EdgeArchetype],
+        conn_assign: Optional[tuple[tuple, tuple]],
+    ) -> Callable[[NodeAnchor, NodeAnchor], EdgeArchetype]:
+        """Jac's root getter."""
+        ct = conn_type if conn_type else GenericEdge
+
+        def builder(source: NodeAnchor, target: NodeAnchor) -> EdgeArchetype:
+            edge = ct() if isinstance(ct, type) else ct
+
+            eanch = edge.__jac__ = EdgeAnchor(
+                archetype=edge,
+                source=source,
+                target=target,
+                is_undirected=is_undirected,
+            )
+            source.edges.append(eanch)
+            target.edges.append(eanch)
+
+            if conn_assign:
+                for fld, val in zip(conn_assign[0], conn_assign[1]):
+                    if hasattr(edge, fld):
+                        setattr(edge, fld, val)
+                    else:
+                        raise ValueError(f"Invalid attribute: {fld}")
+            if source.persistent or target.persistent:
+                JacMachineInterface.save(eanch)
+            return edge
+
+        return builder
+
+    @staticmethod
+    def save(
+        obj: Archetype | Anchor,
+    ) -> None:
+        """Destroy object."""
+        anchor = obj.__jac__ if isinstance(obj, Archetype) else obj
+
+        jctx = JacMachineInterface.get_context()
+
+        anchor.persistent = True
+        anchor.root = jctx.root_state.id
+
+        jctx.mem.set(anchor.id, anchor)
+
+        match anchor:
+            case NodeAnchor():
+                for ed in anchor.edges:
+                    if ed.is_populated() and not ed.persistent:
+                        JacMachineInterface.save(ed)
+            case EdgeAnchor():
+                if (src := anchor.source) and src.is_populated() and not src.persistent:
+                    JacMachineInterface.save(src)
+                if (trg := anchor.target) and trg.is_populated() and not trg.persistent:
+                    JacMachineInterface.save(trg)
+            case _:
+                pass
+
+    @staticmethod
+    def destroy(objs: Archetype | Anchor | list[Archetype | Anchor]) -> None:
+        """Destroy multiple objects passed in a tuple or list."""
+        obj_list = objs if isinstance(objs, list) else [objs]
+        for obj in obj_list:
+            if not isinstance(obj, (Archetype, Anchor)):
+                return
+            anchor = obj.__jac__ if isinstance(obj, Archetype) else obj
+
+            if JacMachineInterface.check_write_access(anchor):
+                match anchor:
+                    case NodeAnchor():
+                        for edge in anchor.edges[:]:
+                            JacMachineInterface.destroy([edge])
+                    case EdgeAnchor():
+                        JacMachineInterface.detach(anchor)
+                    case _:
+                        pass
+
+                JacMachineInterface.get_context().mem.remove(anchor.id)
+
+    @staticmethod
+    def entry(func: Callable) -> Callable:
+        """Mark a method as jac entry with this decorator."""
+        setattr(func, "__jac_entry", None)  # noqa:B010
+        return func
+
+    @staticmethod
+    def exit(func: Callable) -> Callable:
+        """Mark a method as jac exit with this decorator."""
+        setattr(func, "__jac_exit", None)  # noqa:B010
+        return func
+
+    @staticmethod
+    def with_llm(
+        file_loc: str,
+        model: Any,  # noqa: ANN401
+        model_params: dict[str, Any],
+        scope: str,
+        incl_info: list[tuple[str, str]],
+        excl_info: list[tuple[str, str]],
+        inputs: list[tuple[str, str, str, Any]],
+        outputs: tuple,
+        action: str,
+        _globals: dict,
+        _locals: Mapping,
+    ) -> Any:  # noqa: ANN401
+        """Jac's with_llm feature."""
+        raise ImportError(
+            "mtllm is not installed. Please install it with `pip install mtllm` and run `jac clean`."
+        )
+
+    @staticmethod
+    def gen_llm_body(_pass: PyastGenPass, node: ast.Ability) -> list[ast3.AST]:
+        """Generate the by LLM body."""
+        _pass.log_warning(
+            "MT-LLM is not installed. Please install it with `pip install mtllm`."
+        )
+        return [
+            _pass.sync(
+                ast3.Raise(
+                    _pass.sync(
+                        ast3.Call(
+                            func=_pass.sync(
+                                ast3.Name(id="ImportError", ctx=ast3.Load())
+                            ),
+                            args=[
+                                _pass.sync(
+                                    ast3.Constant(
+                                        value="mtllm is not installed. Please install it with `pip install mtllm` and run `jac clean`."  # noqa: E501
+                                    )
+                                )
+                            ],
+                            keywords=[],
+                        )
+                    )
+                )
+            )
+        ]
+
+    @staticmethod
+    def by_llm_call(
+        _pass: PyastGenPass,
+        model: ast3.AST,
+        model_params: dict[str, ast.Expr],
+        scope: ast3.AST,
+        inputs: Sequence[Optional[ast3.AST]],
+        outputs: Sequence[Optional[ast3.AST]] | ast3.Call,
+        action: Optional[ast3.AST],
+        include_info: list[tuple[str, ast3.AST]],
+        exclude_info: list[tuple[str, ast3.AST]],
+    ) -> ast3.Call:
+        """Return the LLM Call, e.g. _JacFeature.with_llm()."""
+        _pass.log_warning(
+            "MT-LLM is not installed. Please install it with `pip install mtllm`."
+        )
+        return ast3.Call(
+            func=_pass.sync(
+                ast3.Attribute(
+                    value=_pass.sync(ast3.Name(id="_Jac", ctx=ast3.Load())),
+                    attr="with_llm",
+                    ctx=ast3.Load(),
+                )
+            ),
+            args=[],
+            keywords=[
+                _pass.sync(
+                    ast3.keyword(
+                        arg="file_loc",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="model",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="model_params",
+                        value=_pass.sync(ast3.Dict(keys=[], values=[])),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="scope",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="incl_info",
+                        value=_pass.sync(ast3.List(elts=[], ctx=ast3.Load())),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="excl_info",
+                        value=_pass.sync(ast3.List(elts=[], ctx=ast3.Load())),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="inputs",
+                        value=_pass.sync(ast3.List(elts=[], ctx=ast3.Load())),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="outputs",
+                        value=_pass.sync(ast3.List(elts=[], ctx=ast3.Load())),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="action",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="_globals",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+                _pass.sync(
+                    ast3.keyword(
+                        arg="_locals",
+                        value=_pass.sync(ast3.Constant(value="None")),
+                    )
+                ),
+            ],
+        )
+
+    @staticmethod
+    def get_by_llm_call_args(_pass: PyastGenPass, node: ast.FuncCall) -> dict:
+        """Get the by LLM call args."""
+        return {
+            "model": None,
+            "model_params": {},
+            "scope": None,
+            "inputs": [],
+            "outputs": [],
+            "action": None,
+            "include_info": [],
+            "exclude_info": [],
+        }
+
 
-    def load_module(self, module_name: str, module: types.ModuleType) -> None:
+class JacUtils:
+    """Jac Machine Utilities."""
+
+    @staticmethod
+    def attach_program(mach: JacMachine, jac_program: JacProgram) -> None:
+        """Attach a JacProgram to the machine."""
+        mach.jac_program = jac_program
+
+    @staticmethod
+    def load_module(
+        mach: JacMachine, module_name: str, module: types.ModuleType
+    ) -> None:
         """Load a module into the machine."""
-        self.loaded_modules[module_name] = module
-        sys.modules[module_name] = module
+        mach.loaded_modules[module_name] = module
+        sys.modules[module_name] = module  # TODO: May want to nuke this one day
 
-    def list_modules(self) -> list[str]:
+    @staticmethod
+    def list_modules(mach: JacMachine) -> list[str]:
         """List all loaded modules."""
-        return list(self.loaded_modules.keys())
+        return list(mach.loaded_modules.keys())
 
-    def list_walkers(self, module_name: str) -> list[str]:
+    @staticmethod
+    def list_walkers(mach: JacMachine, module_name: str) -> list[str]:
         """List all walkers in a specific module."""
-        module = self.loaded_modules.get(module_name)
+        module = mach.loaded_modules.get(module_name)
         if module:
             walkers = []
             for name, obj in inspect.getmembers(module):
-                if isinstance(obj, type) and issubclass(obj, WalkerArchitype):
+                if isinstance(obj, type) and issubclass(obj, WalkerArchetype):
                     walkers.append(name)
             return walkers
         return []
 
-    def list_nodes(self, module_name: str) -> list[str]:
+    @staticmethod
+    def list_nodes(mach: JacMachine, module_name: str) -> list[str]:
         """List all nodes in a specific module."""
-        module = self.loaded_modules.get(module_name)
+        module = mach.loaded_modules.get(module_name)
         if module:
             nodes = []
             for name, obj in inspect.getmembers(module):
-                if isinstance(obj, type) and issubclass(obj, NodeArchitype):
+                if isinstance(obj, type) and issubclass(obj, NodeArchetype):
                     nodes.append(name)
             return nodes
         return []
 
-    def list_edges(self, module_name: str) -> list[str]:
+    @staticmethod
+    def list_edges(mach: JacMachine, module_name: str) -> list[str]:
         """List all edges in a specific module."""
-        module = self.loaded_modules.get(module_name)
+        module = mach.loaded_modules.get(module_name)
         if module:
             nodes = []
             for name, obj in inspect.getmembers(module):
-                if isinstance(obj, type) and issubclass(obj, EdgeArchitype):
+                if isinstance(obj, type) and issubclass(obj, EdgeArchetype):
                     nodes.append(name)
             return nodes
         return []
 
-    def create_architype_from_source(
-        self,
+    @staticmethod
+    def create_archetype_from_source(
+        mach: JacMachine,
         source_code: str,
         module_name: Optional[str] = None,
         base_path: Optional[str] = None,
         cachable: bool = False,
         keep_temporary_files: bool = False,
     ) -> Optional[types.ModuleType]:
-        """Dynamically creates architypes (nodes, walkers, etc.) from Jac source code."""
+        """Dynamically creates archetypes (nodes, walkers, etc.) from Jac source code."""
         from jaclang.runtimelib.importer import JacImporter, ImportPathSpec
 
         if not base_path:
-            base_path = self.base_path or os.getcwd()
+            base_path = mach.base_path or os.getcwd()
 
         if base_path and not os.path.exists(base_path):
             os.makedirs(base_path)
         if not module_name:
-            module_name = f"_dynamic_module_{len(self.loaded_modules)}"
+            module_name = f"_dynamic_module_{len(mach.loaded_modules)}"
         with tempfile.NamedTemporaryFile(
             mode="w",
             suffix=".jac",
@@ -153,7 +1391,7 @@ class JacMachine:
             tmp_file.write(source_code)
 
         try:
-            importer = JacImporter(self)
+            importer = JacImporter(mach)
             tmp_file_basename = os.path.basename(tmp_file_path)
             tmp_module_name, _ = os.path.splitext(tmp_file_basename)
 
@@ -161,7 +1399,6 @@ class JacMachine:
                 target=tmp_module_name,
                 base_path=base_path,
                 absorb=False,
-                cachable=cachable,
                 mdl_alias=None,
                 override_name=module_name,
                 lng="jac",
@@ -171,7 +1408,7 @@ class JacMachine:
             import_result = importer.run_import(spec, reload=False)
             module = import_result.ret_mod
 
-            self.loaded_modules[module_name] = module
+            mach.loaded_modules[module_name] = module
             return module
         except Exception as e:
             logger.error(f"Error importing dynamic module '{module_name}': {e}")
@@ -180,21 +1417,23 @@ class JacMachine:
             if not keep_temporary_files and os.path.exists(tmp_file_path):
                 os.remove(tmp_file_path)
 
+    @staticmethod
     def update_walker(
-        self, module_name: str, items: Optional[dict[str, Union[str, Optional[str]]]]
+        mach: JacMachine,
+        module_name: str,
+        items: Optional[dict[str, Union[str, Optional[str]]]],
     ) -> tuple[types.ModuleType, ...]:
         """Reimport the module."""
         from .importer import JacImporter, ImportPathSpec
 
-        if module_name in self.loaded_modules:
+        if module_name in mach.loaded_modules:
             try:
-                old_module = self.loaded_modules[module_name]
-                importer = JacImporter(self)
+                old_module = mach.loaded_modules[module_name]
+                importer = JacImporter(mach)
                 spec = ImportPathSpec(
                     target=module_name,
-                    base_path=self.base_path,
+                    base_path=mach.base_path,
                     absorb=False,
-                    cachable=True,
                     mdl_alias=None,
                     override_name=None,
                     lng="jac",
@@ -220,15 +1459,16 @@ class JacMachine:
             logger.warning(f"Module {module_name} not found in loaded modules.")
         return ()
 
+    @staticmethod
     def spawn_node(
-        self,
+        mach: JacMachine,
         node_name: str,
         attributes: Optional[dict] = None,
         module_name: str = "__main__",
-    ) -> NodeArchitype:
+    ) -> NodeArchetype:
         """Spawn a node instance of the given node_name with attributes."""
-        node_class = self.get_architype(module_name, node_name)
-        if isinstance(node_class, type) and issubclass(node_class, NodeArchitype):
+        node_class = JacMachineInterface.get_archetype(mach, module_name, node_name)
+        if isinstance(node_class, type) and issubclass(node_class, NodeArchetype):
             if attributes is None:
                 attributes = {}
             node_instance = node_class(**attributes)
@@ -236,15 +1476,16 @@ class JacMachine:
         else:
             raise ValueError(f"Node {node_name} not found.")
 
+    @staticmethod
     def spawn_walker(
-        self,
+        mach: JacMachine,
         walker_name: str,
         attributes: Optional[dict] = None,
         module_name: str = "__main__",
-    ) -> WalkerArchitype:
+    ) -> WalkerArchetype:
         """Spawn a walker instance of the given walker_name."""
-        walker_class = self.get_architype(module_name, walker_name)
-        if isinstance(walker_class, type) and issubclass(walker_class, WalkerArchitype):
+        walker_class = JacMachineInterface.get_archetype(mach, module_name, walker_name)
+        if isinstance(walker_class, type) and issubclass(walker_class, WalkerArchetype):
             if attributes is None:
                 attributes = {}
             walker_instance = walker_class(**attributes)
@@ -252,67 +1493,233 @@ class JacMachine:
         else:
             raise ValueError(f"Walker {walker_name} not found.")
 
-    def get_architype(
-        self, module_name: str, architype_name: str
-    ) -> Optional[Architype]:
-        """Retrieve an architype class from a module."""
-        module = self.loaded_modules.get(module_name)
+    @staticmethod
+    def get_archetype(
+        mach: JacMachine, module_name: str, archetype_name: str
+    ) -> Optional[Archetype]:
+        """Retrieve an archetype class from a module."""
+        module = mach.loaded_modules.get(module_name)
         if module:
-            return getattr(module, architype_name, None)
+            return getattr(module, archetype_name, None)
         return None
 
     @staticmethod
-    def get(base_path: str = "") -> "JacMachine":
-        """Get current jac machine."""
-        if (jac_machine := JACMACHINE_CONTEXT.get(None)) is None:
-            jac_machine = JacMachine(base_path)
-        return jac_machine
+    def await_obj(obj: Any) -> Any:  # noqa: ANN401
+        """Await an object if it is a coroutine or async or future function."""
+        machine = JacMachineInterface.py_get_jac_machine()
+        _event_loop = machine._event_loop
+        return _event_loop.run_until_complete(obj)
 
     @staticmethod
-    def detach() -> None:
-        """Detach current jac machine."""
-        JACMACHINE_CONTEXT.set(None)
+    def thread_run(func: Callable, *args: object) -> Future:  # noqa: ANN401
+        """Run a function in a thread."""
+        machine = JacMachine.py_get_jac_machine()
+        _executor = machine.pool
+        return _executor.submit(func, *args)
 
+    @staticmethod
+    def thread_wait(future: Any) -> None:  # noqa: ANN401
+        """Wait for a thread to finish."""
+        return future.result()
+
+
+class JacMachineInterface(
+    JacClassReferences,
+    JacAccessValidation,
+    JacNode,
+    JacEdge,
+    JacWalker,
+    JacBuiltin,
+    JacCmd,
+    JacBasics,
+    JacUtils,
+):
+    """Jac Feature."""
 
-class JacProgram:
-    """Class to hold the mod_bundle bytecode and sem_ir for Jac modules."""
+
+class JacMachine(JacMachineInterface):
+    """Jac Machine State."""
 
     def __init__(
         self,
-        mod_bundle: Optional[Module],
-        bytecode: Optional[dict[str, bytes]],
-        sem_ir: Optional[SemRegistry],
+        base_path: str = "",
+        session: Optional[str] = None,
+        root: Optional[str] = None,
+        interp_mode: bool = False,
     ) -> None:
-        """Initialize the JacProgram object."""
-        self.mod_bundle = mod_bundle
-        self.bytecode = bytecode or {}
-        self.sem_ir = sem_ir if sem_ir else SemRegistry()
+        """Initialize JacMachine."""
+        self.loaded_modules: dict[str, types.ModuleType] = {}
+        if not base_path:
+            base_path = os.getcwd()
+        # Ensure the base_path is a list rather than a string
+        self.base_path = base_path
+        self.base_path_dir = (
+            os.path.dirname(base_path)
+            if not os.path.isdir(base_path)
+            else os.path.abspath(base_path)
+        )
+        self.jac_program: JacProgram = JacProgram()
+        self.interp_mode = interp_mode
+        self.pool = ThreadPoolExecutor()
+        self._event_loop = asyncio.new_event_loop()
+        self.mem: Memory = ShelfStorage(session)
+        self.reports: list[Any] = []
+        sr_arch = Root()
+        sr_anch = sr_arch.__jac__
+        sr_anch.id = UUID(Con.SUPER_ROOT_UUID)
+        sr_anch.persistent = False
+        self.system_root = sr_anch
+        self.custom: Any = MISSING
+        if not isinstance(
+            system_root := self.mem.find_by_id(UUID(Con.SUPER_ROOT_UUID)), NodeAnchor
+        ):
+            system_root = cast(NodeAnchor, Root().__jac__)  # type: ignore[attr-defined]
+            system_root.id = UUID(Con.SUPER_ROOT_UUID)
+            self.mem.set(system_root.id, system_root)
 
-    def get_bytecode(
+        self.system_root = system_root
+
+        self.entry_node = self.root_state = self.init_anchor(root, self.system_root)
+
+    def init_anchor(
         self,
-        module_name: str,
-        full_target: str,
-        caller_dir: str,
-        cachable: bool = True,
-        reload: bool = False,
-    ) -> Optional[types.CodeType]:
-        """Get the bytecode for a specific module."""
-        if self.mod_bundle and isinstance(self.mod_bundle, Module):
-            codeobj = self.mod_bundle.mod_deps[full_target].gen.py_bytecode
-            return marshal.loads(codeobj) if isinstance(codeobj, bytes) else None
-        gen_dir = os.path.join(caller_dir, Con.JAC_GEN_DIR)
-        pyc_file_path = os.path.join(gen_dir, module_name + ".jbc")
-        if cachable and os.path.exists(pyc_file_path) and not reload:
-            with open(pyc_file_path, "rb") as f:
-                return marshal.load(f)
-
-        result = compile_jac(full_target, cache_result=cachable)
-        if result.errors_had:
-            for alrt in result.errors_had:
-                # We're not logging here, it already gets logged as the errors were added to the errors_had list.
-                # Regardless of the logging, this needs to be sent to the end user, so we'll printing it to stderr.
-                logger.error(alrt.pretty_print())
-        if result.ir.gen.py_bytecode is not None:
-            return marshal.loads(result.ir.gen.py_bytecode)
-        else:
-            return None
+        anchor_id: str | None,
+        default: NodeAnchor,
+    ) -> NodeAnchor:
+        """Load initial anchors."""
+        if anchor_id:
+            if isinstance(anchor := self.mem.find_by_id(UUID(anchor_id)), NodeAnchor):
+                return anchor
+            raise ValueError(f"Invalid anchor id {anchor_id} !")
+        return default
+
+    def set_entry_node(self, entry_node: str | None) -> None:
+        """Override entry."""
+        self.entry_node = self.init_anchor(entry_node, self.root_state)
+
+    def close(self) -> None:
+        """Close current ExecutionContext."""
+        call_jac_func_with_machine(mach=self, func=self.mem.close)
+
+    def get_root(self) -> Root:
+        """Get current root."""
+        return cast(Root, self.root_state.archetype)
+
+    def global_system_root(self) -> NodeAnchor:
+        """Get global system root."""
+        return self.system_root
+
+
+def call_jac_func_with_machine(  # TODO: remove this
+    mach: JacMachine, func: Callable, *args: Any  # noqa: ANN401
+) -> Any:  # noqa: ANN401
+    """Call Jac function with machine context in local."""
+    __jac_mach__ = mach  # noqa: F841
+    return func(*args)
+
+
+def generate_plugin_helpers(
+    plugin_class: Type[Any],
+) -> tuple[Type[Any], Type[Any], Type[Any]]:
+    """Generate three helper classes based on a plugin class.
+
+    - Spec class: contains @hookspec placeholder methods.
+    - Impl class: contains original plugin methods decorated with @hookimpl.
+    - Proxy class: contains methods that call plugin_manager.hook.<method>.
+
+    Returns:
+        Tuple of (SpecClass, ImplClass, ProxyClass).
+    """
+    # Markers for spec and impl
+    spec_methods = {}
+    impl_methods = {}
+    proxy_methods = {}
+
+    for name, method in inspect.getmembers(plugin_class, predicate=inspect.isfunction):
+        if name.startswith("_"):
+            continue
+
+        sig = inspect.signature(method)
+        sig_nodef = sig.replace(
+            parameters=[
+                p.replace(default=inspect.Parameter.empty)
+                for p in sig.parameters.values()
+            ]
+        )
+        doc = method.__doc__ or ""
+
+        # --- Spec placeholder ---
+        def make_spec(
+            name: str, sig_nodef: inspect.Signature, doc: str, method: Callable
+        ) -> Callable:
+            """Create a placeholder method for the spec class."""
+
+            @wraps(method)
+            def placeholder(*args: object, **kwargs: object) -> None:
+                pass
+
+            placeholder.__name__ = name
+            placeholder.__doc__ = doc
+            placeholder.__signature__ = sig_nodef  # type: ignore
+            return placeholder
+
+        spec_methods[name] = hookspec(firstresult=True)(
+            make_spec(name, sig_nodef, doc, method)
+        )
+
+        # --- Impl class: original methods with @hookimpl ---
+        wrapped_impl = wraps(method)(method)
+        wrapped_impl.__signature__ = sig_nodef  # type: ignore
+        impl_methods[name] = hookimpl(wrapped_impl)
+
+        # --- Proxy class: call through plugin_manager.hook ---
+        # Gather class variables and annotations from entire MRO (excluding built-ins)
+        class_vars: dict[str, Any] = {}
+        annotations: dict[str, Any] = {}
+        for base in reversed(plugin_class.__mro__):
+            if base is object:
+                continue
+            # collect annotations first so bases are overridden by subclasses
+            base_ann = getattr(base, "__annotations__", {})
+            annotations.update(base_ann)
+            for key, value in base.__dict__.items():
+                # skip private/special, methods, and descriptors
+                if key.startswith("__"):
+                    continue
+                if callable(value) and not isinstance(value, type):
+                    continue
+                class_vars[key] = value
+
+        def make_proxy(name: str, sig: inspect.Signature) -> Callable:
+            """Create a proxy method for the proxy class."""
+
+            def proxy(*args: object, **kwargs: object) -> object:
+                # bind positionals to parameter names
+                bound = sig.bind_partial(*args, **kwargs)  # noqa
+                bound.apply_defaults()
+                # grab the HookCaller
+                hookcaller = getattr(plugin_manager.hook, name)  # noqa
+                # call with named args only
+                return hookcaller(**bound.arguments)
+
+            proxy.__name__ = name
+            proxy.__signature__ = sig  # type: ignore
+            return proxy
+
+        proxy_methods[name] = make_proxy(name, sig)
+
+    # Construct classes
+    spec_cls = type(f"{plugin_class.__name__}Spec", (object,), spec_methods)
+    impl_cls = type(f"{plugin_class.__name__}Impl", (object,), impl_methods)
+    proxy_namespace = {}
+    proxy_namespace.update(class_vars)
+    if annotations:
+        proxy_namespace["__annotations__"] = annotations
+    proxy_namespace.update(proxy_methods)
+    proxy_cls = type(f"{plugin_class.__name__}", (object,), proxy_namespace)
+
+    return spec_cls, impl_cls, proxy_cls
+
+
+JacMachineSpec, JacMachineImpl, JacMachineInterface = generate_plugin_helpers(JacMachineInterface)  # type: ignore[misc]
+plugin_manager.add_hookspecs(JacMachineSpec)