jaclang 0.7.25__py3-none-any.whl → 0.7.27__py3-none-any.whl

jaclang/plugin/default.py

@@ -11,7 +11,7 @@ from collections import OrderedDict
 from dataclasses import field
 from functools import wraps
 from logging import getLogger
-from typing import Any, Callable, Mapping, Optional, Sequence, Type, Union
+from typing import Any, Callable, Mapping, Optional, Sequence, Type, Union, cast
 from uuid import UUID
 
 from jaclang.compiler.constant import colors
@@ -41,6 +41,7 @@ from jaclang.runtimelib.constructs import (
 )
 from jaclang.runtimelib.importer import ImportPathSpec, JacImporter, PythonImporter
 from jaclang.runtimelib.machine import JacMachine, JacProgram
+from jaclang.runtimelib.memory import Shelf, ShelfStorage
 from jaclang.runtimelib.utils import collect_node_connections, traverse_graph
 
 
@@ -50,9 +51,30 @@ hookimpl = pluggy.HookimplMarker("jac")
 logger = getLogger(__name__)
 
 
+class JacCallableImplementation:
+    """Callable Implementations."""
+
+    @staticmethod
+    def get_object(id: str) -> Architype | None:
+        """Get object by id."""
+        if id == "root":
+            return Jac.get_context().root.architype
+        elif obj := Jac.get_context().mem.find_by_id(UUID(id)):
+            return obj.architype
+
+        return None
+
+
 class JacAccessValidationImpl:
     """Jac Access Validation Implementations."""
 
+    @staticmethod
+    @hookimpl
+    def elevate_root() -> None:
+        """Elevate context root to system_root."""
+        jctx = Jac.get_context()
+        jctx.root = jctx.system_root
+
     @staticmethod
     @hookimpl
     def allow_root(
@@ -379,7 +401,8 @@ class JacWalkerImpl:
         if walker.next:
             current_node = walker.next[-1].architype
             for i in warch._jac_entry_funcs_:
-                if not i.trigger:
+                trigger = i.get_funcparam_annotations(i.func)
+                if not trigger:
                     if i.func:
                         i.func(warch, current_node)
                     else:
@@ -387,7 +410,8 @@ class JacWalkerImpl:
         while len(walker.next):
             if current_node := walker.next.pop(0).architype:
                 for i in current_node._jac_entry_funcs_:
-                    if not i.trigger or isinstance(warch, i.trigger):
+                    trigger = i.get_funcparam_annotations(i.func)
+                    if not trigger or isinstance(warch, trigger):
                         if i.func:
                             i.func(current_node, warch)
                         else:
@@ -395,27 +419,30 @@ class JacWalkerImpl:
                     if walker.disengaged:
                         return warch
                 for i in warch._jac_entry_funcs_:
-                    if not i.trigger or isinstance(current_node, i.trigger):
-                        if i.func and i.trigger:
+                    trigger = i.get_funcparam_annotations(i.func)
+                    if not trigger or isinstance(current_node, trigger):
+                        if i.func and trigger:
                             i.func(warch, current_node)
-                        elif not i.trigger:
+                        elif not trigger:
                             continue
                         else:
                             raise ValueError(f"No function {i.name} to call.")
                     if walker.disengaged:
                         return warch
                 for i in warch._jac_exit_funcs_:
-                    if not i.trigger or isinstance(current_node, i.trigger):
-                        if i.func and i.trigger:
+                    trigger = i.get_funcparam_annotations(i.func)
+                    if not trigger or isinstance(current_node, trigger):
+                        if i.func and trigger:
                             i.func(warch, current_node)
-                        elif not i.trigger:
+                        elif not trigger:
                             continue
                         else:
                             raise ValueError(f"No function {i.name} to call.")
                     if walker.disengaged:
                         return warch
                 for i in current_node._jac_exit_funcs_:
-                    if not i.trigger or isinstance(warch, i.trigger):
+                    trigger = i.get_funcparam_annotations(i.func)
+                    if not trigger or isinstance(warch, trigger):
                         if i.func:
                             i.func(current_node, warch)
                         else:
@@ -423,7 +450,8 @@ class JacWalkerImpl:
                     if walker.disengaged:
                         return warch
         for i in warch._jac_exit_funcs_:
-            if not i.trigger:
+            trigger = i.get_funcparam_annotations(i.func)
+            if not trigger:
                 if i.func:
                     i.func(warch, current_node)
                 else:
@@ -562,14 +590,32 @@ class JacFeatureImpl(
 
     @staticmethod
     @hookimpl
-    def get_object(id: str) -> Architype | None:
-        """Get object by id."""
-        if id == "root":
-            return Jac.get_context().root.architype
-        elif obj := Jac.get_context().mem.find_by_id(UUID(id)):
-            return obj.architype
+    def reset_graph(root: Optional[Root] = None) -> int:
+        """Purge current or target graph."""
+        ctx = Jac.get_context()
+        mem = cast(ShelfStorage, ctx.mem)
+        ranchor = root.__jac__ if root else ctx.root
+
+        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
 
-        return None
+            if loaded_anchor := mem.find_by_id(anchor.id):
+                deleted_count += 1
+                Jac.destroy(loaded_anchor)
+
+        return deleted_count
+
+    @staticmethod
+    @hookimpl
+    def get_object_func() -> Callable[[str], Architype | None]:
+        """Get object by id func."""
+        return JacCallableImplementation.get_object
 
     @staticmethod
     @hookimpl

jaclang/plugin/feature.py

@@ -41,6 +41,11 @@ from jaclang.plugin.spec import (
 class JacAccessValidation:
     """Jac Access Validation Specs."""
 
+    @staticmethod
+    def elevate_root() -> None:
+        """Elevate context root to system_root."""
+        plugin_manager.hook.elevate_root()
+
     @staticmethod
     def allow_root(
         architype: Architype,
@@ -254,10 +259,20 @@ class JacFeature(
         """Get current execution context."""
         return plugin_manager.hook.get_context()
 
+    @staticmethod
+    def reset_graph(root: Optional[Root] = None) -> int:
+        """Purge current or target graph."""
+        return plugin_manager.hook.reset_graph(root=root)
+
     @staticmethod
     def get_object(id: str) -> Architype | None:
         """Get object given id."""
-        return plugin_manager.hook.get_object(id=id)
+        return plugin_manager.hook.get_object_func()(id=id)
+
+    @staticmethod
+    def get_object_func() -> Callable[[str], Architype | None]:
+        """Get object given id."""
+        return plugin_manager.hook.get_object_func()
 
     @staticmethod
     def object_ref(obj: Architype) -> str:

jaclang/plugin/spec.py

@@ -46,6 +46,12 @@ P = ParamSpec("P")
 class JacAccessValidationSpec:
     """Jac Access Validation Specs."""
 
+    @staticmethod
+    @hookspec(firstresult=True)
+    def elevate_root() -> None:
+        """Elevate context root to system_root."""
+        raise NotImplementedError
+
     @staticmethod
     @hookspec(firstresult=True)
     def allow_root(
@@ -246,8 +252,14 @@ class JacFeatureSpec(
 
     @staticmethod
     @hookspec(firstresult=True)
-    def get_object(id: str) -> Architype | None:
-        """Get object by id."""
+    def reset_graph(root: Optional[Root]) -> int:
+        """Purge current or target graph."""
+        raise NotImplementedError
+
+    @staticmethod
+    @hookspec(firstresult=True)
+    def get_object_func() -> Callable[[str], Architype | None]:
+        """Get object by id func."""
         raise NotImplementedError
 
     @staticmethod

jaclang/plugin/tests/fixtures/graph_purger.jac

@@ -0,0 +1,101 @@
+node A {
+    has id: int;
+}
+
+node B {
+    has id: int;
+}
+
+node C {
+    has id: int;
+}
+
+node D {
+    has id: int;
+}
+
+node E {
+    has id: int;
+}
+
+
+walker populate {
+    can setup1 with `root entry {
+        for i in range(2) {
+            here ++> A(id=i);
+        }
+        visit [-->];
+    }
+
+    can setup2 with A entry {
+        for i in range(2) {
+            here ++> B(id=i);
+        }
+        visit [-->];
+    }
+
+    can setup3 with B entry {
+        for i in range(2) {
+            here ++> C(id=i);
+        }
+        visit [-->];
+    }
+
+    can setup4 with C entry {
+        for i in range(2) {
+            here ++> D(id=i);
+        }
+        visit [-->];
+    }
+
+    can setup5 with D entry {
+        for i in range(2) {
+            here ++> E(id=i);
+        }
+        visit [-->];
+    }
+}
+
+walker traverse {
+    can enter1 with `root entry {
+        print(here);
+        visit [-->];
+    }
+
+    can enter2 with A entry {
+        print(here);
+        visit [-->];
+    }
+
+    can enter3 with B entry {
+        print(here);
+        visit [-->];
+    }
+
+    can enter4 with C entry {
+        print(here);
+        visit [-->];
+    }
+
+    can enter5 with D entry {
+        print(here);
+        visit [-->];
+    }
+
+    can enter6 with E entry {
+        print(here);
+        visit [-->];
+    }
+}
+
+walker purge {
+    can purge with `root entry {
+        print(Jac.reset_graph());
+    }
+}
+
+walker check {
+    can enter with `root entry {
+        print(len(Jac.get_context().mem.__shelf__.values()));
+    }
+}

jaclang/plugin/tests/fixtures/other_root_access.jac

@@ -23,6 +23,25 @@ walker update_node {
     }
 }
 
+walker update_target_node {
+    has val: int;
+    has node_id: str;
+
+    can enter with `root entry {
+        target_node = &(self.node_id);
+        target_node.val = self.val;
+    }
+}
+
+walker update_node_forced {
+    has val: int;
+
+    can enter2 with A entry {
+        Jac.elevate_root();
+        here.val = self.val;
+    }
+}
+
 walker create_node {
     has val: int;
 

jaclang/plugin/tests/fixtures/savable_object.jac

@@ -0,0 +1,84 @@
+enum Enum {
+    A = "a",
+    B = "b",
+    C = "c"
+}
+
+obj Child {
+    has val: int, arr: list[int], json: dict[str, int], enum_field: Enum;
+}
+
+obj Parent:Child: {
+    has child: Child;
+}
+
+obj SavableObject {
+    has val: int, arr: list[int], json: dict[str, int], parent: Parent, enum_field: Enum;
+}
+
+walker create_custom_object {
+    can enter1 with `root entry {
+        o = SavableObject(
+            val=0,
+            arr=[],
+            json={},
+            parent=Parent(
+                val=1,
+                arr=[1],
+                json={"a": 1},
+                child=Child(
+                    val=2,
+                    arr=[1, 2],
+                    json={"a": 1, "b": 2},
+                    enum_field = Enum.C
+                ),
+                enum_field = Enum.B
+            ),
+            enum_field = Enum.A
+        );
+        Jac.save(o);
+        print(jid(o));
+        print(o);
+    }
+}
+
+walker get_custom_object {
+    has object_id: str;
+
+    can enter1 with `root entry {
+        try {
+            print(&(self.object_id));
+        } except Exception as e {
+            print(None);
+        }
+    }
+}
+
+walker update_custom_object {
+    has object_id: str;
+
+    can enter1 with `root entry {
+        savable_object = &(self.object_id);
+        savable_object.parent.child.json["c"] = 3;
+        savable_object.parent.child.arr.append(3);
+        savable_object.parent.child.val = 3;
+        savable_object.parent.child.enum_field = Enum.A;
+        savable_object.parent.json["b"] = 2;
+        savable_object.parent.arr.append(2);
+        savable_object.parent.val = 2;
+        savable_object.parent.enum_field = Enum.C;
+        savable_object.json["a"] = 1;
+        savable_object.arr.append(1);
+        savable_object.val = 1;
+        savable_object.enum_field = Enum.B;
+        print(savable_object);
+    }
+}
+
+walker delete_custom_object {
+    has object_id: str;
+
+    can enter1 with `root entry {
+        Jac.destroy(&(self.object_id));
+    }
+}

jaclang/plugin/tests/test_jaseci.py

@@ -269,6 +269,75 @@ class TestJaseciPlugin(TestCase):
         )
         self._del_session(session)
 
+    def test_walker_purger(self) -> None:
+        """Test simple persistent object."""
+        session = self.fixture_abs_path("test_walker_purger.session")
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="populate",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="traverse",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="check",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="purge",
+            args=[],
+        )
+        output = self.capturedOutput.getvalue().strip()
+        self.assertEqual(
+            output,
+            (
+                "Root()\n"
+                "A(id=0)\nA(id=1)\n"
+                "B(id=0)\nB(id=1)\nB(id=0)\nB(id=1)\n"
+                "C(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\nC(id=0)\nC(id=1)\n"
+                "D(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\n"
+                "D(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\nD(id=0)\nD(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "E(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\nE(id=0)\nE(id=1)\n"
+                "125\n124"
+            ),
+        )
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="traverse",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="check",
+            args=[],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("graph_purger.jac"),
+            session=session,
+            entrypoint="purge",
+            args=[],
+        )
+        output = self.capturedOutput.getvalue().strip()
+        self.assertEqual(output, "Root()\n1\n0")
+
+        self._del_session(session)
+
     def trigger_access_validation_test(
         self, give_access_to_full_graph: bool, via_all: bool = False
     ) -> None:
@@ -316,6 +385,22 @@ class TestJaseciPlugin(TestCase):
             node=self.nodes[0],
         )
 
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_target_node",
+            args=[20, self.nodes[1]],
+            session=session,
+            root=self.roots[0],
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_target_node",
+            args=[10, self.nodes[0]],
+            session=session,
+            root=self.roots[1],
+        )
+
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
             entrypoint="check_node",
@@ -340,6 +425,55 @@ class TestJaseciPlugin(TestCase):
         self.assertEqual(archs[0], "A(val=2)")
         self.assertEqual(archs[1], "A(val=1)")
 
+        ##############################################
+        #        WITH READ ACCESS BUT ELEVATED       #
+        ##############################################
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node_forced",
+            args=[20],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node_forced",
+            args=[10],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        archs = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertTrue(len(archs) == 2)
+
+        # ---------- UPDATE SHOULD HAPPEN ---------- #
+
+        self.assertEqual(archs[0], "A(val=20)")
+        self.assertEqual(archs[1], "A(val=10)")
+
+        # ---------- DISALLOW READ ACCESS ---------- #
+
         self._output2buffer()
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
@@ -400,7 +534,7 @@ class TestJaseciPlugin(TestCase):
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
             entrypoint="update_node",
-            args=[20],
+            args=[200],
             root=self.roots[0],
             node=self.nodes[1],
             session=session,
@@ -408,7 +542,7 @@ class TestJaseciPlugin(TestCase):
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
             entrypoint="update_node",
-            args=[10],
+            args=[100],
             session=session,
             root=self.roots[1],
             node=self.nodes[0],
@@ -433,10 +567,12 @@ class TestJaseciPlugin(TestCase):
         archs = self.capturedOutput.getvalue().strip().split("\n")
         self.assertTrue(len(archs) == 2)
 
-        # --------- UPDATE SHOULD HAPPEN -------- #
+        # ---------- UPDATE SHOULD HAPPEN ---------- #
 
-        self.assertEqual(archs[0], "A(val=20)")
-        self.assertEqual(archs[1], "A(val=10)")
+        self.assertEqual(archs[0], "A(val=200)")
+        self.assertEqual(archs[1], "A(val=100)")
+
+        # ---------- DISALLOW WRITE ACCESS --------- #
 
         self._output2buffer()
         cli.enter(
@@ -474,7 +610,7 @@ class TestJaseciPlugin(TestCase):
         )
         self.assertFalse(self.capturedOutput.getvalue().strip())
 
-        # --------- ROOTS RESET OWN NODE -------- #
+        # ---------- ROOTS RESET OWN NODE ---------- #
 
         cli.enter(
             filename=self.fixture_abs_path("other_root_access.jac"),
@@ -608,3 +744,75 @@ class TestJaseciPlugin(TestCase):
         )
 
         self._del_session(session)
+
+    def test_savable_object(self) -> None:
+        """Test ObjectAnchor save."""
+        global session
+        session = self.fixture_abs_path("other_root_access.session")
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("savable_object.jac"),
+            entrypoint="create_custom_object",
+            args=[],
+            session=session,
+        )
+
+        prints = self.capturedOutput.getvalue().strip().split("\n")
+        id = prints[0]
+
+        self.assertEqual(
+            "SavableObject(val=0, arr=[], json={}, parent=Parent(val=1, arr=[1], json"
+            "={'a': 1}, enum_field=<Enum.B: 'b'>, child=Child(val=2, arr=[1, 2], json"
+            "={'a': 1, 'b': 2}, enum_field=<Enum.C: 'c'>)), enum_field=<Enum.A: 'a'>)",
+            prints[1],
+        )
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("savable_object.jac"),
+            entrypoint="get_custom_object",
+            args=[id],
+            session=session,
+        )
+        self.assertEqual(
+            "SavableObject(val=0, arr=[], json={}, parent=Parent(val=1, arr=[1], json"
+            "={'a': 1}, enum_field=<Enum.B: 'b'>, child=Child(val=2, arr=[1, 2], json"
+            "={'a': 1, 'b': 2}, enum_field=<Enum.C: 'c'>)), enum_field=<Enum.A: 'a'>)",
+            self.capturedOutput.getvalue().strip(),
+        )
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("savable_object.jac"),
+            entrypoint="update_custom_object",
+            args=[id],
+            session=session,
+        )
+
+        self.assertEqual(
+            "SavableObject(val=1, arr=[1], json={'a': 1}, parent=Parent(val=2, arr=[1, 2], json"
+            "={'a': 1, 'b': 2}, enum_field=<Enum.C: 'c'>, child=Child(val=3, arr=[1, 2, 3], json"
+            "={'a': 1, 'b': 2, 'c': 3}, enum_field=<Enum.A: 'a'>)), enum_field=<Enum.B: 'b'>)",
+            self.capturedOutput.getvalue().strip(),
+        )
+
+        self._output2buffer()
+
+        cli.enter(
+            filename=self.fixture_abs_path("savable_object.jac"),
+            entrypoint="delete_custom_object",
+            args=[id],
+            session=session,
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("savable_object.jac"),
+            entrypoint="get_custom_object",
+            args=[id],
+            session=session,
+        )
+        self.assertEqual("None", self.capturedOutput.getvalue().strip())