jaclang 0.7.16__py3-none-any.whl → 0.7.18__py3-none-any.whl

jaclang/cli/cli.py

@@ -2,14 +2,12 @@
 
 import ast as ast3
 import importlib
-import inspect
 import marshal
 import os
 import pickle
 import shutil
 import types
 from typing import Optional
-from uuid import UUID
 
 import jaclang.compiler.absyntree as ast
 from jaclang import jac_import
@@ -22,8 +20,9 @@ from jaclang.compiler.passes.tool.schedules import format_pass
 from jaclang.plugin.builtin import dotgen
 from jaclang.plugin.feature import JacCmd as Cmd
 from jaclang.plugin.feature import JacFeature as Jac
-from jaclang.runtimelib.constructs import Architype
-from jaclang.runtimelib.machine import JacProgram
+from jaclang.runtimelib.constructs import WalkerArchitype
+from jaclang.runtimelib.context import ExecutionContext
+from jaclang.runtimelib.machine import JacMachine, JacProgram
 from jaclang.utils.helpers import debugger as db
 from jaclang.utils.lang_tools import AstTool
 
@@ -68,12 +67,7 @@ def format(path: str, outfile: str = "", debug: bool = False) -> None:
 
 @cmd_registry.register
 def run(
-    filename: str,
-    session: str = "",
-    main: bool = True,
-    cache: bool = True,
-    walker: str = "",
-    node: str = "",
+    filename: str, session: str = "", main: bool = True, cache: bool = True
 ) -> None:
     """Run the specified .jac file."""
     # if no session specified, check if it was defined when starting the command shell
@@ -90,78 +84,89 @@ def run(
     base, mod = os.path.split(filename)
     base = base if base else "./"
     mod = mod[:-4]
-    Jac.context().init_memory(base_path=base, session=session)
+
+    jctx = ExecutionContext.create(session=session)
+
     if filename.endswith(".jac"):
-        ret_module = jac_import(
+        jac_import(
             target=mod,
             base_path=base,
             cachable=cache,
             override_name="__main__" if main else None,
         )
-        if ret_module is None:
-            loaded_mod = None
-        else:
-            (loaded_mod,) = ret_module
     elif filename.endswith(".jir"):
         with open(filename, "rb") as f:
-            ir = pickle.load(f)
-            jac_program = JacProgram(mod_bundle=ir, bytecode=None)
-            Jac.context().jac_machine.attach_program(jac_program)
-            ret_module = jac_import(
+            JacMachine(base).attach_program(
+                JacProgram(mod_bundle=pickle.load(f), bytecode=None)
+            )
+            jac_import(
                 target=mod,
                 base_path=base,
                 cachable=cache,
                 override_name="__main__" if main else None,
             )
-            if ret_module is None:
-                loaded_mod = None
-            else:
-                (loaded_mod,) = ret_module
     else:
-        print("Not a .jac file.")
-        return
-
-    if not node or node == "root":
-        entrypoint: Architype = Jac.get_root()
-    else:
-        obj = Jac.context().get_obj(UUID(node))
-        if obj is None:
-            print(f"Entrypoint {node} not found.")
-            return
-        entrypoint = obj
-
-    # TODO: handle no override name
-    if walker:
-        walker_module = dict(inspect.getmembers(loaded_mod)).get(walker)
-        if walker_module:
-            Jac.spawn_call(entrypoint, walker_module())
-        else:
-            print(f"Walker {walker} not found.")
+        jctx.close()
+        JacMachine.detach()
+        raise ValueError("Not a valid file!\nOnly supports `.jac` and `.jir`")
 
-    Jac.reset_context()
+    jctx.close()
+    JacMachine.detach()
 
 
 @cmd_registry.register
-def get_object(id: str, session: str = "") -> dict:
+def get_object(
+    filename: str, id: str, session: str = "", main: bool = True, cache: bool = True
+) -> dict:
     """Get the object with the specified id."""
     if session == "":
-        session = cmd_registry.args.session if "session" in cmd_registry.args else ""
+        session = (
+            cmd_registry.args.session
+            if hasattr(cmd_registry, "args")
+            and hasattr(cmd_registry.args, "session")
+            and cmd_registry.args.session
+            else ""
+        )
 
-    Jac.context().init_memory(session=session)
+    base, mod = os.path.split(filename)
+    base = base if base else "./"
+    mod = mod[:-4]
 
-    if id == "root":
-        id_uuid = UUID(int=0)
-    else:
-        id_uuid = UUID(id)
+    jctx = ExecutionContext.create(session=session)
 
-    obj = Jac.context().get_obj(id_uuid)
-    if obj is None:
-        print(f"Object with id {id} not found.")
-        Jac.reset_context()
-        return {}
+    if filename.endswith(".jac"):
+        jac_import(
+            target=mod,
+            base_path=base,
+            cachable=cache,
+            override_name="__main__" if main else None,
+        )
+    elif filename.endswith(".jir"):
+        with open(filename, "rb") as f:
+            JacMachine(base).attach_program(
+                JacProgram(mod_bundle=pickle.load(f), bytecode=None)
+            )
+            jac_import(
+                target=mod,
+                base_path=base,
+                cachable=cache,
+                override_name="__main__" if main else None,
+            )
+    else:
+        jctx.close()
+        JacMachine.detach()
+        raise ValueError("Not a valid file!\nOnly supports `.jac` and `.jir`")
+
+    data = {}
+    obj = Jac.get_object(id)
+    if obj:
+        data = obj.__jac__.__getstate__()
     else:
-        Jac.reset_context()
-        return obj.__getstate__()
+        print(f"Object with id {id} not found.")
+
+    jctx.close()
+    JacMachine.detach()
+    return data
 
 
 @cmd_registry.register
@@ -211,25 +216,78 @@ def lsp() -> None:
 
 
 @cmd_registry.register
-def enter(filename: str, entrypoint: str, args: list) -> None:
-    """Run the specified entrypoint function in the given .jac file.
+def enter(
+    filename: str,
+    entrypoint: str,
+    args: list,
+    session: str = "",
+    main: bool = True,
+    cache: bool = True,
+    root: str = "",
+    node: str = "",
+) -> None:
+    """
+    Run the specified entrypoint function in the given .jac file.
 
     :param filename: The path to the .jac file.
     :param entrypoint: The name of the entrypoint function.
     :param args: Arguments to pass to the entrypoint function.
+    :param session: shelve.Shelf file path.
+    :param root: root executor.
+    :param node: starting node.
     """
+    if session == "":
+        session = (
+            cmd_registry.args.session
+            if hasattr(cmd_registry, "args")
+            and hasattr(cmd_registry.args, "session")
+            and cmd_registry.args.session
+            else ""
+        )
+
+    base, mod = os.path.split(filename)
+    base = base if base else "./"
+    mod = mod[:-4]
+
+    jctx = ExecutionContext.create(session=session, root=root)
+
     if filename.endswith(".jac"):
-        base, mod_name = os.path.split(filename)
-        base = base if base else "./"
-        mod_name = mod_name[:-4]
-        (mod,) = jac_import(target=mod_name, base_path=base)
-        if not mod:
+        ret_module = jac_import(
+            target=mod,
+            base_path=base,
+            cachable=cache,
+            override_name="__main__" if main else None,
+        )
+    elif filename.endswith(".jir"):
+        with open(filename, "rb") as f:
+            JacMachine(base).attach_program(
+                JacProgram(mod_bundle=pickle.load(f), bytecode=None)
+            )
+            ret_module = jac_import(
+                target=mod,
+                base_path=base,
+                cachable=cache,
+                override_name="__main__" if main else None,
+            )
+    else:
+        jctx.close()
+        JacMachine.detach()
+        raise ValueError("Not a valid file!\nOnly supports `.jac` and `.jir`")
+
+    if ret_module:
+        (loaded_mod,) = ret_module
+        if not loaded_mod:
             print("Errors occurred while importing the module.")
-            return
         else:
-            getattr(mod, entrypoint)(*args)
-    else:
-        print("Not a .jac file.")
+            architype = getattr(loaded_mod, entrypoint)(*args)
+
+            jctx.set_entry_node(node)
+
+            if isinstance(architype, WalkerArchitype) and jctx.validate_access():
+                Jac.spawn_call(jctx.entry_node.architype, architype)
+
+    jctx.close()
+    JacMachine.detach()
 
 
 @cmd_registry.register
@@ -252,6 +310,8 @@ def test(
 
     jac test => jac test -d .
     """
+    jctx = ExecutionContext.create()
+
     failcount = Jac.run_test(
         filepath=filepath,
         filter=filter,
@@ -260,6 +320,9 @@ def test(
         directory=directory,
         verbose=verbose,
     )
+
+    jctx.close()
+
     if failcount:
         raise SystemExit(f"Tests failed: {failcount}")
 
@@ -361,13 +424,13 @@ def dot(
     base, mod = os.path.split(filename)
     base = base if base else "./"
     mod = mod[:-4]
-    Jac.context().init_memory(base_path=base, session=session)
+
+    jctx = ExecutionContext.create(session=session)
+
     if filename.endswith(".jac"):
-        jac_import(
-            target=mod,
-            base_path=base,
-        )
-        module = importlib.import_module(mod)
+        jac_machine = JacMachine(base)
+        jac_import(target=mod, base_path=base)
+        module = jac_machine.loaded_modules.get(mod)
         globals().update(vars(module))
         try:
             node = globals().get(initial, eval(initial)) if initial else None
@@ -385,7 +448,7 @@ def dot(
             import traceback
 
             traceback.print_exc()
-            Jac.reset_context()
+            jctx.close()
             return
         file_name = saveto if saveto else f"{mod}.dot"
         with open(file_name, "w") as file:
@@ -394,7 +457,7 @@ def dot(
     else:
         print("Not a .jac file.")
 
-    Jac.reset_context()
+    jctx.close()
 
 
 @cmd_registry.register

jaclang/compiler/absyntree.py

@@ -630,8 +630,11 @@ class Module(AstDocNode):
         self.impl_mod: list[Module] = []
         self.test_mod: list[Module] = []
         self.mod_deps: dict[str, Module] = {}
+        self.py_mod_dep_map: dict[str, str] = {}
+        self.py_raise_map: dict[str, str] = {}
         self.registry = registry
         self.terminals: list[Token] = terminals
+        self.is_raised_from_py: bool = False
         AstNode.__init__(self, kid=kid)
         AstDocNode.__init__(self, doc=doc)
 
@@ -956,6 +959,7 @@ class ModulePath(AstSymbolNode):
         self.level = level
         self.alias = alias
         self.sub_module: Optional[Module] = None
+        self.abs_path: Optional[str] = None
 
         name_spec = alias if alias else path[0] if path else None
 
@@ -977,7 +981,7 @@ class ModulePath(AstSymbolNode):
         )
 
     @property
-    def path_str(self) -> str:
+    def dot_path_str(self) -> str:
         """Get path string."""
         return ("." * self.level) + ".".join(
             [p.value for p in self.path] if self.path else [self.name_spec.sym_name]
@@ -985,7 +989,7 @@ class ModulePath(AstSymbolNode):
 
     def resolve_relative_path(self, target_item: Optional[str] = None) -> str:
         """Convert an import target string into a relative file path."""
-        target = self.path_str
+        target = self.dot_path_str
         if target_item:
             target += f".{target_item}"
         base_path = os.path.dirname(self.loc.mod_path)
@@ -1048,6 +1052,7 @@ class ModuleItem(AstSymbolNode):
             name_spec=alias if alias else name,
             sym_category=SymbolType.MOD_VAR,
         )
+        self.abs_path: Optional[str] = None
 
     @property
     def from_parent(self) -> Import:
@@ -4313,11 +4318,11 @@ class Ellipsis(Literal):
 class EmptyToken(Token):
     """EmptyToken node type for Jac Ast."""
 
-    def __init__(self) -> None:
+    def __init__(self, file_path: str = "") -> None:
         """Initialize empty token."""
         super().__init__(
             name="EmptyToken",
-            file_path="",
+            file_path=file_path,
             value="",
             line=0,
             end_line=0,

jaclang/compiler/constant.py

@@ -94,18 +94,18 @@ class Constants(StrEnum):
     HERE = "_jac_here_"
     JAC_FEATURE = "_Jac"
     ROOT = f"{JAC_FEATURE}.get_root()"
-    EDGES_TO_NODE = "_jac_.edges_to_nodes"
-    EDGE_REF = "_jac_.edge_ref"
-    CONNECT_NODE = "_jac_.connect_node"
-    DISCONNECT_NODE = "_jac_.disconnect_node"
-    WALKER_VISIT = "_jac_.visit_node"
-    WALKER_IGNORE = "_jac_.ignore_node"
-    DISENGAGE = "_jac_.disengage_now"
+    EDGES_TO_NODE = "__jac__.edges_to_nodes"
+    EDGE_REF = "__jac__.edge_ref"
+    CONNECT_NODE = "__jac__.connect_node"
+    DISCONNECT_NODE = "__jac__.disconnect_node"
+    WALKER_VISIT = "__jac__.visit_node"
+    WALKER_IGNORE = "__jac__.ignore_node"
+    DISENGAGE = "__jac__.disengage_now"
     OBJECT_CLASS = "_jac_Object_"
     NODE_CLASS = "_jac_Node_"
     EDGE_CLASS = "_jac_Edge_"
     WALKER_CLASS = "_jac_Walker_"
-    WITH_DIR = "_jac_.apply_dir"
+    WITH_DIR = "__jac__.apply_dir"
     EDGE_DIR = "_jac_Edge_Dir"
     ON_ENTRY = "_jac_ds_.on_entry"
     ON_EXIT = "_jac_ds_.on_exit"

jaclang/compiler/parser.py

@@ -163,7 +163,11 @@ class JacParser(Pass):
                 body=body,
                 is_imported=False,
                 terminals=self.terminals,
-                kid=kid if len(kid) else [ast.EmptyToken()],
+                kid=(
+                    kid
+                    if len(kid)
+                    else [ast.EmptyToken(file_path=self.parse_ref.mod_path)]
+                ),
             )
             return self.nu(mod)
 
@@ -1003,7 +1007,11 @@ class JacParser(Pass):
                         semstr=semstr,
                         params=params,
                         return_type=return_spec,
-                        kid=kid if len(kid) else [ast.EmptyToken()],
+                        kid=(
+                            kid
+                            if len(kid)
+                            else [ast.EmptyToken(file_path=self.parse_ref.mod_path)]
+                        ),
                     )
                 )
             else:

jaclang/compiler/passes/main/__init__.py

@@ -1,8 +1,8 @@
 """Collection of passes for Jac IR."""
 
 from .sub_node_tab_pass import SubNodeTabPass
-from .import_pass import JacImportPass, PyImportPass  # noqa: I100
 from .sym_tab_build_pass import SymTabBuildPass  # noqa: I100
+from .import_pass import JacImportPass, PyImportPass  # noqa: I100
 from .def_impl_match_pass import DeclImplMatchPass  # noqa: I100
 from .def_use_pass import DefUsePass  # noqa: I100
 from .pyout_pass import PyOutPass  # noqa: I100