1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
import { asArrayOf, asNumber, asString, isRecord } from "./util/guards";
export type BrowserState = {
readonly namespace: string;
readonly ownedTargetIds?: ReadonlyArray<string>;
readonly harnessWindowTargetId?: string;
readonly harnessWindowId?: number;
};
export const defaultState = (namespace = "default"): BrowserState => ({
namespace,
ownedTargetIds: [],
});
// Pi session JSONL is append-only and every custom entry shows up in the branch navigator, so a
// snapshot is only worth appending when it differs from the one the branch already holds (#26).
// Tab order is not a difference: ownership is a set.
const stateKey = (state: BrowserState): string =>
JSON.stringify({
namespace: state.namespace,
ownedTargetIds: [...(state.ownedTargetIds ?? [])].sort(),
harnessWindowTargetId: state.harnessWindowTargetId ?? null,
harnessWindowId: state.harnessWindowId ?? null,
});
export type StatePersister = {
// Appends the snapshot unless it equals the last one persisted or restored. Returns whether it wrote.
readonly persist: (state: BrowserState) => boolean;
// Records a snapshot that already lives in the session (restored from the branch) without writing it.
readonly markPersisted: (state: BrowserState) => void;
};
export const createStatePersister = (pi: ExtensionAPI): StatePersister => {
let last: string | undefined;
return {
persist: (state) => {
const key = stateKey(state);
if (key === last) return false;
pi.appendEntry<BrowserState>("browser-harness-state", state);
last = key;
return true;
},
markPersisted: (state) => {
last = stateKey(state);
},
};
};
const asPersistedState = (v: unknown): Partial<BrowserState> | undefined => {
if (!isRecord(v)) return undefined;
const namespace = asString(v["namespace"]);
const ownedTargetIds = asArrayOf(v["ownedTargetIds"], asString);
const harnessWindowTargetId = asString(v["harnessWindowTargetId"]);
const harnessWindowId = asNumber(v["harnessWindowId"]);
return {
...(namespace !== undefined ? { namespace } : {}),
...(ownedTargetIds !== undefined ? { ownedTargetIds } : {}),
...(harnessWindowTargetId !== undefined ? { harnessWindowTargetId } : {}),
...(harnessWindowId !== undefined ? { harnessWindowId } : {}),
};
};
export const restoreState = (ctx: ExtensionContext, currentNamespace?: string): BrowserState => {
const branchEntries = ctx.sessionManager.getBranch();
const fallback = defaultState(currentNamespace);
for (let i = branchEntries.length - 1; i >= 0; i--) {
const entry = branchEntries[i];
if (entry?.type === "custom" && entry.customType === "browser-harness-state") {
const data = asPersistedState(entry.data);
if (data) {
return {
...fallback,
...data,
namespace: currentNamespace ?? data.namespace ?? fallback.namespace,
};
}
}
}
return fallback;
};
|
