statepatch 0.2.0__tar.gz → 0.2.1__tar.gz

{statepatch-0.2.0 → statepatch-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: statepatch
-Version: 0.2.0
+Version: 0.2.1
 Summary: Immer-style ops over a plain JSON tree: observables that emit every mutation as an op
 Project-URL: Repository, https://github.com/assistant-ui/harness-sdk
 License-Expression: MIT

{statepatch-0.2.0 → statepatch-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "statepatch"
-version = "0.2.0"
+version = "0.2.1"
 description = "Immer-style ops over a plain JSON tree: observables that emit every mutation as an op"
 readme = "README.md"
 license = "MIT"

{statepatch-0.2.0 → statepatch-0.2.1}/src/statepatch/_wire.py

@@ -34,15 +34,16 @@ def _snapshot(node: Any) -> Any:
 
 
 def plain(value: Any) -> Any:
-    """Resolve a proxy or observable to a plain snapshot; pass anything else through."""
+    """Resolve a proxy or observable to its live stored node (mutating it emits no ops); pass anything else through."""
     from ._core import Observable, Proxy
 
-    if isinstance(value, Proxy):
-        _owner, _rel, node = value._target()
-        return _snapshot(node)
     if isinstance(value, Observable):
-        return _snapshot(value._state)
-    return value
+        value = Proxy(value, ())
+    if not isinstance(value, Proxy):
+        return value
+    owner, rel, node = value._target()
+    mounted = any(path[: len(rel)] == rel for _child, path, _unsub in owner._mounts.values())
+    return _snapshot(node) if mounted else node
 
 
 def _adopt(value: Any, base: Path) -> tuple[Any, list[tuple[Path, "Observable"]]]:

{statepatch-0.2.0 → statepatch-0.2.1}/tests/test_observable.py

@@ -4,7 +4,7 @@ mounted observables forward their op streams at the mount path."""
 from datetime import datetime, timezone
 
 import pytest
-from statepatch import Op, append, observable
+from statepatch import Observable, Op, append, observable, plain
 
 
 def collect(obs):
@@ -180,6 +180,26 @@ def test_snapshot_ops_carry_plain_values_for_nested_mounts():
     assert ops[-1] == Op("replace", ("wrap", "inner", "n"), 7)
 
 
+def test_plain_returns_the_live_node():
+    state = observable({"runs": [{"id": "r1"}]})
+    ops = collect(state)
+    assert plain(state) is state._state
+    assert plain(state["runs"]) is state._state["runs"]
+    assert plain(state["runs"][0]["id"]) == "r1"
+    plain(state["runs"]).append({"id": "r2"})
+    assert state.value == {"runs": [{"id": "r1"}, {"id": "r2"}]}
+    assert ops == []
+
+
+def test_plain_unwraps_mounts_into_a_snapshot():
+    child = observable({"n": 0})
+    parent = observable({"wrap": {"inner": child}, "other": [1]})
+    assert plain(parent["wrap"]) == {"inner": {"n": 0}}
+    assert not isinstance(plain(parent["wrap"])["inner"], Observable)
+    assert plain(parent["wrap"]["inner"]) is child._state
+    assert plain(parent["other"]) is parent._state["other"]
+
+
 def test_container_helpers():
     state = observable({"d": {"a": 1}, "l": [1, 2, 3]})
     ops = collect(state)