@r1z4x/tezgah 1.3.0 → 1.5.0

package/hooks/tezgah_update.py

@@ -22,6 +22,7 @@ import sys
 import time
 
 import tezgah_paths as tp
+import tezgah_store
 
 REPO = os.environ.get("TEZGAH_REPO") or "r1z4x/tezgah"
 CHECK_EVERY = 24 * 3600
@@ -30,10 +31,6 @@ SEMVER = re.compile(r"^v?(\d+)\.(\d+)\.(\d+)$")
 ARROW = "\u2191"  # ↑: East Asian width Ambiguous, one cell, never an emoji
 
 
-def cache_path():
-    return os.path.join(tp.cache_dir(), "update.json")
-
-
 def endpoint():
     """Where the newest release is read, repointed by TEZGAH_UPDATE_URL (tests).
 
@@ -73,32 +70,15 @@ def disabled():
 
 
 def read_cache():
-    try:
-        with open(cache_path(), encoding="utf-8") as fh:
-            data = json.load(fh)
-        return data if isinstance(data, dict) else {}
-    except Exception:
-        return {}
+    """The `update_check` store's one document, `{checked, latest}`, or {}."""
+    data = tezgah_store.doc("update_check", "")
+    return data if isinstance(data, dict) else {}
 
 
 def write_cache(data):
-    """Write the cache whole, owner-only, through a rename, so a redraw reading it
+    """Write the cache whole, in one statement, so a redraw reading it
     mid-write sees the old answer or the new one and never half of either."""
-    path = cache_path()
-    tmp = "%s.%d.tmp" % (path, os.getpid())
-    try:
-        os.makedirs(os.path.dirname(path), mode=0o700, exist_ok=True)
-        fd = os.open(tmp, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, 0o600)
-        with os.fdopen(fd, "w", encoding="utf-8") as fh:
-            json.dump(data, fh)
-        os.replace(tmp, path)
-        return True
-    except OSError:
-        try:
-            os.remove(tmp)
-        except OSError:
-            pass
-        return False
+    return tezgah_store.put_doc("update_check", "", data)
 
 
 # Every 0.x ever published, retired by 1.0.0: npm served 0.17.0 through 0.32.0

package/hosts/codex/hook.py

@@ -158,6 +158,11 @@ def main():
     event = payload.get("hook_event_name") or "SessionStart"
     cwd = payload.get("cwd") or os.getcwd()
     session_id = payload.get("session_id")
+    # codex hands its hooks its own env, which carries no codex marker (only the
+    # shell tool gets CODEX_THREAD_ID), so the judge seam's session_cli() could
+    # not tell this is a codex session without it.
+    if session_id:
+        os.environ.setdefault("CODEX_THREAD_ID", str(session_id))
     safe(session_id, bind_session, session_id)  # the switch latch (tezgah_paths.off)
     if event == "SessionStart":
         safe(session_id, attest_session, "codex", session_id, cwd)

package/hosts/cursor/hook.py

@@ -52,7 +52,8 @@ try:
     from tezgah_integrity import (  # noqa: E402
         SUBAGENT_CHANNEL, bind_session, note, note_tool, ran_nothing, report_bytes,
         stop_reason)
-    from tezgah_paths import cache_dir, off  # noqa: E402
+    from tezgah_paths import off  # noqa: E402
+    import tezgah_store  # noqa: E402
     from tezgah_taste import write_note  # noqa: E402
     from tezgah_untrusted import marks  # noqa: E402
 except Exception as exc:
@@ -215,25 +216,21 @@ def first_time(session_id, tag):
     """True the first time a tag is seen for a session; fail-open when unwritable."""
     if not session_id:
         return True
-    mark = os.path.join(cache_dir(), "reinforced", slug(str(session_id)), tag)
-    if os.path.exists(mark):
-        return False
     try:
-        os.makedirs(os.path.dirname(mark), exist_ok=True)
-        open(mark, "w", encoding="utf-8").close()
-    except OSError:
-        pass
-    return True
+        return tezgah_store.mark("cursor_reinforced", "%s/%s" % (slug(str(session_id)), tag))
+    except tezgah_store.ERRORS:
+        return True
 
 
-def answer_path(session_id):
-    """Where the last assistant text of a conversation is kept.
+def answer_key(session_id):
+    """The key of the last assistant text of a conversation in the
+    `cursor_answer` store.
 
     Cursor's `stop` payload carries only `status`/`loop_count`, so the reply the
     Stop rule has to read arrives on the earlier `afterAgentResponse`, which
-    documents `{"text": "<assistant final text>"}`. One file per conversation,
-    overwritten."""
-    return os.path.join(cache_dir(), "answer", slug(str(session_id or "nosession")))
+    documents `{"text": "<assistant final text>"}`. One document per
+    conversation, overwritten."""
+    return slug(str(session_id or "nosession"))
 
 
 # A long reply is kept by its two ends rather than by one slice: the Stop rule's
@@ -255,29 +252,20 @@ def remember_answer(session_id, text):
     count of what was dropped, then `ANSWER_TAIL` characters of its tail - so
     both the opener the Stop rule reads and the closing marker it must not lose
     survive the round trip. A reply with no `conversation_id` is not stored; a
-    file that cannot be written is not fatal."""
+    store that cannot be written is not fatal."""
     if not session_id or not text:
         return
-    path = answer_path(session_id)
-    try:
-        os.makedirs(os.path.dirname(path), exist_ok=True)
-        text = str(text)
-        dropped = len(text) - ANSWER_HEAD - ANSWER_TAIL
-        if dropped > 0:
-            text = text[:ANSWER_HEAD] + ANSWER_CUT % dropped + text[-ANSWER_TAIL:]
-        with open(path, "w", encoding="utf-8") as fh:
-            fh.write(text)
-    except OSError:
-        pass
+    text = str(text)
+    dropped = len(text) - ANSWER_HEAD - ANSWER_TAIL
+    if dropped > 0:
+        text = text[:ANSWER_HEAD] + ANSWER_CUT % dropped + text[-ANSWER_TAIL:]
+    tezgah_store.put_doc("cursor_answer", answer_key(session_id), text)
 
 
 def last_answer(session_id):
     """The reply remembered for this conversation, or ""."""
-    try:
-        with open(answer_path(session_id), encoding="utf-8") as fh:
-            return fh.read()
-    except OSError:
-        return ""
+    text = tezgah_store.doc("cursor_answer", answer_key(session_id))
+    return text if isinstance(text, str) else ""
 
 
 def dispatch(payload):