@ilikexiaoni/pi-task-list 0.4.0 → 0.4.2

package/lib/task-list-persistence.ts

@@ -0,0 +1,116 @@
+import {
+  applyTaskAction,
+  cloneState,
+  cloneTaskCommand,
+  createEmptyState,
+  isPersistedTaskOperation,
+  isPersistedTaskState,
+  normalizeState,
+  type TaskOperation,
+  type TaskState,
+} from "./task-list-core.ts";
+
+export const TASK_LIST_TOOL_NAME = "task_list";
+export const TASK_LIST_STATE_ENTRY_TYPE = "task-list-state";
+
+interface RecordLike {
+  [key: string]: unknown;
+}
+
+interface BranchMessageEntry {
+  type?: unknown;
+  message?: unknown;
+  customType?: unknown;
+  data?: unknown;
+}
+
+function isRecord(value: unknown): value is RecordLike {
+  return typeof value === "object" && value !== null;
+}
+
+function snapshotFromValue(value: unknown): TaskState | undefined {
+  if (!isRecord(value)) return undefined;
+  if (isPersistedTaskState(value.state)) return normalizeState(value.state);
+  if (isPersistedTaskState(value)) return normalizeState(value);
+  return undefined;
+}
+
+function toolDetails(entry: unknown): unknown {
+  if (!isRecord(entry) || entry.type !== "message" || !isRecord(entry.message)) return undefined;
+  const message = entry.message;
+  if (message.role !== "toolResult" || message.toolName !== TASK_LIST_TOOL_NAME) return undefined;
+  return message.details;
+}
+
+function snapshotFromEntry(entry: BranchMessageEntry): TaskState | undefined {
+  if (entry.type === "custom" && entry.customType === TASK_LIST_STATE_ENTRY_TYPE) {
+    return snapshotFromValue(entry.data);
+  }
+  return snapshotFromValue(toolDetails(entry));
+}
+
+function operationFromValue(value: unknown): TaskOperation | undefined {
+  if (!isRecord(value)) return undefined;
+  const operation = value.operation;
+  if (!isPersistedTaskOperation(operation)) return undefined;
+  return {
+    version: 1,
+    command: cloneTaskCommand(operation.command),
+    at: operation.at.slice(0, 64),
+  };
+}
+
+function operationFromEntry(entry: BranchMessageEntry): TaskOperation | undefined {
+  return operationFromValue(toolDetails(entry));
+}
+
+/**
+ * Apply a compact task-list operation without allowing a malformed historical
+ * entry to prevent restoration of later valid session data.
+ */
+export function replayTaskOperation(state: TaskState, operation: TaskOperation): TaskState | undefined {
+  try {
+    return applyTaskAction(state, operation.command, operation.at).state;
+  } catch {
+    return undefined;
+  }
+}
+
+/**
+ * Restore the current branch from the most recent usable snapshot and replay
+ * only later mutations. Full snapshots from releases before operation-based
+ * persistence remain supported for backward compatibility.
+ */
+export function restoreTaskListState(branch: readonly unknown[]): TaskState {
+  let snapshot: TaskState | undefined;
+  let snapshotIndex = -1;
+
+  for (let index = branch.length - 1; index >= 0; index -= 1) {
+    const candidate = snapshotFromEntry(branch[index] as BranchMessageEntry);
+    if (!candidate) continue;
+    snapshot = candidate;
+    snapshotIndex = index;
+    break;
+  }
+
+  let state = snapshot ? cloneState(snapshot) : createEmptyState();
+  for (let index = snapshotIndex + 1; index < branch.length; index += 1) {
+    const operation = operationFromEntry(branch[index] as BranchMessageEntry);
+    if (!operation) continue;
+    const replayed = replayTaskOperation(state, operation);
+    if (replayed) state = replayed;
+  }
+
+  return normalizeState(state);
+}
+
+/**
+ * Update an already restored in-memory state from a task_list tool result.
+ * New releases use operations; older sessions expose a full `state` snapshot.
+ */
+export function applyTaskListToolDetails(current: TaskState, details: unknown): TaskState | undefined {
+  const snapshot = snapshotFromValue(details);
+  if (snapshot) return snapshot;
+  const operation = operationFromValue(details);
+  return operation ? replayTaskOperation(current, operation) : undefined;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ilikexiaoni/pi-task-list",
-  "version": "0.4.0",
+  "version": "0.4.2",
   "type": "module",
   "description": "A generic structured task list extension for Pi with dependencies, evidence, and visible progress.",
   "keywords": [