@opencode-trace/plugin 0.0.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +124 -0
- package/dist/integration.test.d.ts +2 -0
- package/dist/integration.test.d.ts.map +1 -0
- package/dist/integration.test.js +76 -0
- package/dist/integration.test.js.map +1 -0
- package/dist/plugin-instance.d.ts +29 -0
- package/dist/plugin-instance.d.ts.map +1 -0
- package/dist/plugin-instance.js +219 -0
- package/dist/plugin-instance.js.map +1 -0
- package/dist/plugin-instance.test.d.ts +2 -0
- package/dist/plugin-instance.test.d.ts.map +1 -0
- package/dist/plugin-instance.test.js +135 -0
- package/dist/plugin-instance.test.js.map +1 -0
- package/dist/redact.d.ts +2 -0
- package/dist/redact.d.ts.map +1 -0
- package/dist/redact.js +40 -0
- package/dist/redact.js.map +1 -0
- package/dist/redact.test.d.ts +2 -0
- package/dist/redact.test.d.ts.map +1 -0
- package/dist/redact.test.js +77 -0
- package/dist/redact.test.js.map +1 -0
- package/dist/state-queue.d.ts +14 -0
- package/dist/state-queue.d.ts.map +1 -0
- package/dist/state-queue.js +44 -0
- package/dist/state-queue.js.map +1 -0
- package/dist/state-queue.test.d.ts +2 -0
- package/dist/state-queue.test.d.ts.map +1 -0
- package/dist/state-queue.test.js +99 -0
- package/dist/state-queue.test.js.map +1 -0
- package/dist/trace.d.ts +32 -0
- package/dist/trace.d.ts.map +1 -0
- package/dist/trace.js +100 -0
- package/dist/trace.js.map +1 -0
- package/dist/trace.test.d.ts +2 -0
- package/dist/trace.test.d.ts.map +1 -0
- package/dist/trace.test.js +359 -0
- package/dist/trace.test.js.map +1 -0
- package/dist/write-queue.d.ts +14 -0
- package/dist/write-queue.d.ts.map +1 -0
- package/dist/write-queue.js +62 -0
- package/dist/write-queue.js.map +1 -0
- package/dist/write-queue.test.d.ts +2 -0
- package/dist/write-queue.test.d.ts.map +1 -0
- package/dist/write-queue.test.js +92 -0
- package/dist/write-queue.test.js.map +1 -0
- package/package.json +48 -0
package/dist/redact.js
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
const SENSITIVE_HEADERS = [
|
|
2
|
+
"authorization",
|
|
3
|
+
"api-key",
|
|
4
|
+
"x-api-key",
|
|
5
|
+
"apikey",
|
|
6
|
+
"x-apikey",
|
|
7
|
+
"token",
|
|
8
|
+
"x-token",
|
|
9
|
+
"access-token",
|
|
10
|
+
"x-access-token",
|
|
11
|
+
"secret",
|
|
12
|
+
"x-secret",
|
|
13
|
+
"cookie",
|
|
14
|
+
];
|
|
15
|
+
function isSensitiveHeader(key) {
|
|
16
|
+
return SENSITIVE_HEADERS.includes(key.toLowerCase());
|
|
17
|
+
}
|
|
18
|
+
export function redactHeaders(headers) {
|
|
19
|
+
const disabled = process.env.OPENCODE_TRACE_REDACT === "false";
|
|
20
|
+
if (disabled) {
|
|
21
|
+
return headers;
|
|
22
|
+
}
|
|
23
|
+
const result = {};
|
|
24
|
+
for (const [key, value] of Object.entries(headers)) {
|
|
25
|
+
if (isSensitiveHeader(key)) {
|
|
26
|
+
const lowerValue = value.toLowerCase();
|
|
27
|
+
if (lowerValue.startsWith("bearer ")) {
|
|
28
|
+
result[key] = "Bearer [REDACTED]";
|
|
29
|
+
}
|
|
30
|
+
else {
|
|
31
|
+
result[key] = "[REDACTED]";
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
else {
|
|
35
|
+
result[key] = value;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
return result;
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=redact.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redact.js","sourceRoot":"","sources":["../src/redact.ts"],"names":[],"mappings":"AAAA,MAAM,iBAAiB,GAAG;IACxB,eAAe;IACf,SAAS;IACT,WAAW;IACX,QAAQ;IACR,UAAU;IACV,OAAO;IACP,SAAS;IACT,cAAc;IACd,gBAAgB;IAChB,QAAQ;IACR,UAAU;IACV,QAAQ;CACT,CAAC;AAEF,SAAS,iBAAiB,CAAC,GAAW;IACpC,OAAO,iBAAiB,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,OAA+B;IAE/B,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,OAAO,CAAC;IAC/D,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,IAAI,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,MAAM,CAAC,GAAG,CAAC,GAAG,mBAAmB,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;YAC7B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redact.test.d.ts","sourceRoot":"","sources":["../src/redact.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
import { describe, test, expect, beforeEach, afterEach, vi } from "vitest";
|
|
2
|
+
describe("redactHeaders", () => {
|
|
3
|
+
const originalEnv = process.env.OPENCODE_TRACE_REDACT;
|
|
4
|
+
beforeEach(() => {
|
|
5
|
+
vi.stubEnv("OPENCODE_TRACE_REDACT", "true");
|
|
6
|
+
});
|
|
7
|
+
afterEach(() => {
|
|
8
|
+
vi.unstubAllEnvs();
|
|
9
|
+
if (originalEnv !== undefined) {
|
|
10
|
+
process.env.OPENCODE_TRACE_REDACT = originalEnv;
|
|
11
|
+
}
|
|
12
|
+
else {
|
|
13
|
+
delete process.env.OPENCODE_TRACE_REDACT;
|
|
14
|
+
}
|
|
15
|
+
});
|
|
16
|
+
test("redacts authorization header with Bearer token", async () => {
|
|
17
|
+
const { redactHeaders } = await import("./redact.js");
|
|
18
|
+
const headers = {
|
|
19
|
+
authorization: "Bearer sk-proj-abc123def456ghi789",
|
|
20
|
+
"content-type": "application/json",
|
|
21
|
+
};
|
|
22
|
+
const result = redactHeaders(headers);
|
|
23
|
+
expect(result.authorization).toBe("Bearer [REDACTED]");
|
|
24
|
+
expect(result["content-type"]).toBe("application/json");
|
|
25
|
+
});
|
|
26
|
+
test("redacts authorization header without Bearer prefix", async () => {
|
|
27
|
+
const { redactHeaders } = await import("./redact.js");
|
|
28
|
+
const headers = {
|
|
29
|
+
authorization: "sk-proj-abc123def456ghi789",
|
|
30
|
+
};
|
|
31
|
+
const result = redactHeaders(headers);
|
|
32
|
+
expect(result.authorization).toBe("[REDACTED]");
|
|
33
|
+
});
|
|
34
|
+
test("redacts api-key header", async () => {
|
|
35
|
+
const { redactHeaders } = await import("./redact.js");
|
|
36
|
+
const headers = {
|
|
37
|
+
"api-key": "my-secret-key-123",
|
|
38
|
+
"x-api-key": "another-secret-key",
|
|
39
|
+
};
|
|
40
|
+
const result = redactHeaders(headers);
|
|
41
|
+
expect(result["api-key"]).toBe("[REDACTED]");
|
|
42
|
+
expect(result["x-api-key"]).toBe("[REDACTED]");
|
|
43
|
+
});
|
|
44
|
+
test("does not redact when OPENCODE_TRACE_REDACT is false", async () => {
|
|
45
|
+
vi.stubEnv("OPENCODE_TRACE_REDACT", "false");
|
|
46
|
+
const { redactHeaders } = await import("./redact.js");
|
|
47
|
+
const headers = {
|
|
48
|
+
authorization: "Bearer sk-proj-abc123",
|
|
49
|
+
};
|
|
50
|
+
const result = redactHeaders(headers);
|
|
51
|
+
expect(result.authorization).toBe("Bearer sk-proj-abc123");
|
|
52
|
+
});
|
|
53
|
+
test("redacts by default when OPENCODE_TRACE_REDACT is not set", async () => {
|
|
54
|
+
vi.stubEnv("OPENCODE_TRACE_REDACT", undefined);
|
|
55
|
+
delete process.env.OPENCODE_TRACE_REDACT;
|
|
56
|
+
const { redactHeaders } = await import("./redact.js");
|
|
57
|
+
const headers = {
|
|
58
|
+
authorization: "Bearer sk-proj-abc123",
|
|
59
|
+
};
|
|
60
|
+
const result = redactHeaders(headers);
|
|
61
|
+
expect(result.authorization).toBe("Bearer [REDACTED]");
|
|
62
|
+
});
|
|
63
|
+
test("preserves non-sensitive headers", async () => {
|
|
64
|
+
vi.stubEnv("OPENCODE_TRACE_REDACT", "true");
|
|
65
|
+
const { redactHeaders } = await import("./redact.js");
|
|
66
|
+
const headers = {
|
|
67
|
+
"content-type": "application/json",
|
|
68
|
+
accept: "text/html",
|
|
69
|
+
"user-agent": "Mozilla/5.0",
|
|
70
|
+
};
|
|
71
|
+
const result = redactHeaders(headers);
|
|
72
|
+
expect(result["content-type"]).toBe("application/json");
|
|
73
|
+
expect(result.accept).toBe("text/html");
|
|
74
|
+
expect(result["user-agent"]).toBe("Mozilla/5.0");
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
//# sourceMappingURL=redact.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redact.test.js","sourceRoot":"","sources":["../src/redact.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE3E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAEtD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,WAAW,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;QAC3C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,aAAa,EAAE,mCAAmC;YAClD,cAAc,EAAE,kBAAkB;SACnC,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,aAAa,EAAE,4BAA4B;SAC5C,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,SAAS,EAAE,mBAAmB;YAC9B,WAAW,EAAE,oBAAoB;SAClC,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACrE,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;QAC7C,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,aAAa,EAAE,uBAAuB;SACvC,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QAC1E,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,SAAS,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;QACzC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,aAAa,EAAE,uBAAuB;SACvC,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QACjD,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG;YACd,cAAc,EAAE,kBAAkB;YAClC,MAAM,EAAE,WAAW;YACnB,YAAY,EAAE,aAAa;SAC5B,CAAC;QACF,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { StateManager } from "@opencode-trace/core/state";
|
|
2
|
+
import type { TraceRecord } from "./trace.js";
|
|
3
|
+
export declare class AsyncStateQueue {
|
|
4
|
+
private queue;
|
|
5
|
+
private stateManager;
|
|
6
|
+
private writing;
|
|
7
|
+
private batchSize;
|
|
8
|
+
constructor(batchSize?: number);
|
|
9
|
+
setStateManager(manager: StateManager): void;
|
|
10
|
+
enqueue(session: string, seq: number, record: TraceRecord): void;
|
|
11
|
+
private processQueue;
|
|
12
|
+
flush(): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=state-queue.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-queue.d.ts","sourceRoot":"","sources":["../src/state-queue.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAE/D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C,qBAAa,eAAe;IAC1B,OAAO,CAAC,KAAK,CAAoE;IACjF,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,SAAS,CAAS;gBAEd,SAAS,GAAE,MAAW;IAIlC,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAI5C,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,GAAG,IAAI;YAOlD,YAAY;IAoBpB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAK7B"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { logger } from "@opencode-trace/core";
|
|
2
|
+
export class AsyncStateQueue {
|
|
3
|
+
queue = [];
|
|
4
|
+
stateManager = null;
|
|
5
|
+
writing = false;
|
|
6
|
+
batchSize;
|
|
7
|
+
constructor(batchSize = 10) {
|
|
8
|
+
this.batchSize = batchSize;
|
|
9
|
+
}
|
|
10
|
+
setStateManager(manager) {
|
|
11
|
+
this.stateManager = manager;
|
|
12
|
+
}
|
|
13
|
+
enqueue(session, seq, record) {
|
|
14
|
+
this.queue.push({ session, seq, record });
|
|
15
|
+
if (!this.writing && this.stateManager) {
|
|
16
|
+
this.processQueue();
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
async processQueue() {
|
|
20
|
+
this.writing = true;
|
|
21
|
+
while (this.queue.length > 0) {
|
|
22
|
+
const batch = this.queue.splice(0, this.batchSize);
|
|
23
|
+
for (const { session, seq, record } of batch) {
|
|
24
|
+
try {
|
|
25
|
+
// NOTE: StateManager.writeRecord is currently sync (Task 3 will convert to async)
|
|
26
|
+
await this.stateManager.writeRecord(session, seq, record);
|
|
27
|
+
}
|
|
28
|
+
catch (err) {
|
|
29
|
+
logger.error("SQLite update failed", { error: String(err) });
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
this.writing = false;
|
|
34
|
+
if (this.queue.length > 0 && !this.writing && this.stateManager) {
|
|
35
|
+
this.processQueue();
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
async flush() {
|
|
39
|
+
while (this.writing || this.queue.length > 0) {
|
|
40
|
+
await new Promise(resolve => setTimeout(resolve, 10));
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=state-queue.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-queue.js","sourceRoot":"","sources":["../src/state-queue.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAG9C,MAAM,OAAO,eAAe;IAClB,KAAK,GAAiE,EAAE,CAAC;IACzE,YAAY,GAAwB,IAAI,CAAC;IACzC,OAAO,GAAY,KAAK,CAAC;IACzB,SAAS,CAAS;IAE1B,YAAY,YAAoB,EAAE;QAChC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,eAAe,CAAC,OAAqB;QACnC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;IAC9B,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,GAAW,EAAE,MAAmB;QACvD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvC,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YACnD,KAAK,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;gBAC7C,IAAI,CAAC;oBACH,kFAAkF;oBAClF,MAAM,IAAI,CAAC,YAAa,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;gBAC7D,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QAErB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAChE,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-queue.test.d.ts","sourceRoot":"","sources":["../src/state-queue.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
import { describe, test, expect, beforeEach, afterEach, vi } from "vitest";
|
|
2
|
+
import { AsyncStateQueue } from "./state-queue.js";
|
|
3
|
+
import { StateManager } from "@opencode-trace/core/state";
|
|
4
|
+
import { logger } from "@opencode-trace/core";
|
|
5
|
+
import { mkdtempSync, rmSync } from "node:fs";
|
|
6
|
+
import { tmpdir } from "node:os";
|
|
7
|
+
import { join } from "node:path";
|
|
8
|
+
describe("AsyncStateQueue", () => {
|
|
9
|
+
let tempDir;
|
|
10
|
+
let queue;
|
|
11
|
+
let stateManager;
|
|
12
|
+
beforeEach(async () => {
|
|
13
|
+
tempDir = mkdtempSync(join(tmpdir(), "state-queue-test-"));
|
|
14
|
+
queue = new AsyncStateQueue();
|
|
15
|
+
stateManager = new StateManager(tempDir);
|
|
16
|
+
await stateManager.init();
|
|
17
|
+
queue.setStateManager(stateManager);
|
|
18
|
+
});
|
|
19
|
+
afterEach(() => {
|
|
20
|
+
rmSync(tempDir, { recursive: true, force: true });
|
|
21
|
+
});
|
|
22
|
+
test("enqueue calls StateManager.writeRecord asynchronously", async () => {
|
|
23
|
+
const sessionId = "test-session";
|
|
24
|
+
stateManager.startSession(sessionId);
|
|
25
|
+
const record = {
|
|
26
|
+
id: 1,
|
|
27
|
+
purpose: "test",
|
|
28
|
+
requestAt: "2026-05-07T00:00:00Z",
|
|
29
|
+
responseAt: "2026-05-07T00:00:01Z",
|
|
30
|
+
request: { method: "GET", url: "https://example.com", headers: {}, body: null },
|
|
31
|
+
response: { status: 200, statusText: "OK", headers: {}, body: null },
|
|
32
|
+
error: null,
|
|
33
|
+
};
|
|
34
|
+
queue.enqueue(sessionId, 1, record);
|
|
35
|
+
await queue.flush();
|
|
36
|
+
const session = stateManager.getSession(sessionId);
|
|
37
|
+
expect(session?.requestCount).toBe(1);
|
|
38
|
+
});
|
|
39
|
+
test("enqueue processes items in batches", async () => {
|
|
40
|
+
const sessionId = "batch-test";
|
|
41
|
+
stateManager.startSession(sessionId);
|
|
42
|
+
const records = Array.from({ length: 25 }, (_, i) => ({
|
|
43
|
+
id: i + 1,
|
|
44
|
+
purpose: `test-${i}`,
|
|
45
|
+
requestAt: new Date().toISOString(),
|
|
46
|
+
responseAt: new Date().toISOString(),
|
|
47
|
+
request: { method: "GET", url: `https://example.com/${i}`, headers: {}, body: null },
|
|
48
|
+
response: { status: 200, statusText: "OK", headers: {}, body: null },
|
|
49
|
+
error: null,
|
|
50
|
+
}));
|
|
51
|
+
for (let i = 0; i < 25; i++) {
|
|
52
|
+
queue.enqueue(sessionId, i + 1, records[i]);
|
|
53
|
+
}
|
|
54
|
+
await queue.flush();
|
|
55
|
+
const session = stateManager.getSession(sessionId);
|
|
56
|
+
expect(session?.requestCount).toBe(25);
|
|
57
|
+
});
|
|
58
|
+
test("enqueue handles errors gracefully", async () => {
|
|
59
|
+
const sessionId = "error-test";
|
|
60
|
+
stateManager.startSession(sessionId);
|
|
61
|
+
const loggerErrorSpy = vi.spyOn(logger, 'error').mockImplementation(() => logger);
|
|
62
|
+
const record = {
|
|
63
|
+
id: 1,
|
|
64
|
+
purpose: "test",
|
|
65
|
+
requestAt: new Date().toISOString(),
|
|
66
|
+
responseAt: new Date().toISOString(),
|
|
67
|
+
request: { method: "GET", url: "https://example.com", headers: {}, body: null },
|
|
68
|
+
response: { status: 200, statusText: "OK", headers: {}, body: null },
|
|
69
|
+
error: null,
|
|
70
|
+
};
|
|
71
|
+
vi.spyOn(stateManager, 'writeRecord').mockImplementationOnce(() => {
|
|
72
|
+
throw new Error("Test error");
|
|
73
|
+
});
|
|
74
|
+
queue.enqueue(sessionId, 1, record);
|
|
75
|
+
await queue.flush();
|
|
76
|
+
expect(loggerErrorSpy).toHaveBeenCalledWith("SQLite update failed", { error: "Error: Test error" });
|
|
77
|
+
loggerErrorSpy.mockRestore();
|
|
78
|
+
vi.restoreAllMocks();
|
|
79
|
+
});
|
|
80
|
+
test("flush waits for queue to drain", async () => {
|
|
81
|
+
const sessionId = "flush-test";
|
|
82
|
+
stateManager.startSession(sessionId);
|
|
83
|
+
for (let i = 0; i < 5; i++) {
|
|
84
|
+
queue.enqueue(sessionId, i + 1, {
|
|
85
|
+
id: i + 1,
|
|
86
|
+
purpose: `test-${i}`,
|
|
87
|
+
requestAt: new Date().toISOString(),
|
|
88
|
+
responseAt: new Date().toISOString(),
|
|
89
|
+
request: { method: "GET", url: "https://example.com", headers: {}, body: null },
|
|
90
|
+
response: { status: 200, statusText: "OK", headers: {}, body: null },
|
|
91
|
+
error: null,
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
await queue.flush();
|
|
95
|
+
const session = stateManager.getSession(sessionId);
|
|
96
|
+
expect(session?.requestCount).toBe(5);
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
//# sourceMappingURL=state-queue.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-queue.test.js","sourceRoot":"","sources":["../src/state-queue.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC3E,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,OAAe,CAAC;IACpB,IAAI,KAAsB,CAAC;IAC3B,IAAI,YAA0B,CAAC;IAE/B,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QAC3D,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9B,YAAY,GAAG,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,SAAS,GAAG,cAAc,CAAC;QACjC,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,CAAC;YACL,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,sBAAsB;YACjC,UAAU,EAAE,sBAAsB;YAClC,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,qBAAqB,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YAC/E,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACpE,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAEpC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,SAAS,GAAG,YAAY,CAAC;QAC/B,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACpD,EAAE,EAAE,CAAC,GAAG,CAAC;YACT,OAAO,EAAE,QAAQ,CAAC,EAAE;YACpB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,uBAAuB,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACpF,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACpE,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC,CAAC;QAEJ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,SAAS,GAAG,YAAY,CAAC;QAC/B,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,cAAc,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC;QAElF,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,CAAC;YACL,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,qBAAqB,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YAC/E,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;YACpE,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC,sBAAsB,CAAC,GAAG,EAAE;YAChE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAEpC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,sBAAsB,EACtB,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAC/B,CAAC;QAEF,cAAc,CAAC,WAAW,EAAE,CAAC;QAC7B,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,SAAS,GAAG,YAAY,CAAC;QAC/B,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,EAAE;gBAC9B,EAAE,EAAE,CAAC,GAAG,CAAC;gBACT,OAAO,EAAE,QAAQ,CAAC,EAAE;gBACpB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACpC,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,qBAAqB,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;gBAC/E,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;gBACpE,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/trace.d.ts
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import type { PluginModule } from "@opencode-ai/plugin";
|
|
2
|
+
export interface TraceRequest {
|
|
3
|
+
method: string;
|
|
4
|
+
url: string;
|
|
5
|
+
headers: Record<string, string>;
|
|
6
|
+
body: unknown;
|
|
7
|
+
}
|
|
8
|
+
export interface TraceResponse {
|
|
9
|
+
status: number;
|
|
10
|
+
statusText: string;
|
|
11
|
+
headers: Record<string, string>;
|
|
12
|
+
body: unknown;
|
|
13
|
+
}
|
|
14
|
+
export interface TraceRecord {
|
|
15
|
+
id: number;
|
|
16
|
+
purpose: string;
|
|
17
|
+
requestAt: string;
|
|
18
|
+
responseAt: string;
|
|
19
|
+
request: TraceRequest;
|
|
20
|
+
response: TraceResponse | null;
|
|
21
|
+
error: {
|
|
22
|
+
message: string;
|
|
23
|
+
stack?: string;
|
|
24
|
+
} | null;
|
|
25
|
+
requestSentAt?: number;
|
|
26
|
+
firstTokenAt?: number;
|
|
27
|
+
lastTokenAt?: number;
|
|
28
|
+
}
|
|
29
|
+
export declare function _resetForTesting(): void;
|
|
30
|
+
declare const entrypoint: PluginModule;
|
|
31
|
+
export default entrypoint;
|
|
32
|
+
//# sourceMappingURL=trace.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAU,YAAY,EAAsB,MAAM,qBAAqB,CAAC;AAMpF,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,EAAE,OAAO,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,EAAE,OAAO,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,YAAY,CAAC;IACtB,QAAQ,EAAE,aAAa,GAAG,IAAI,CAAC;IAC/B,KAAK,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAClD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAID,wBAAgB,gBAAgB,IAAI,IAAI,CAKvC;AAoGD,QAAA,MAAM,UAAU,EAAE,YAGjB,CAAC;AAEF,eAAe,UAAU,CAAC"}
|
package/dist/trace.js
ADDED
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { join } from "node:path";
|
|
2
|
+
import { logger } from "@opencode-trace/core";
|
|
3
|
+
import { TracePlugin } from "./plugin-instance.js";
|
|
4
|
+
let testPlugin = null;
|
|
5
|
+
export function _resetForTesting() {
|
|
6
|
+
if (testPlugin) {
|
|
7
|
+
testPlugin.uninstallInterceptor();
|
|
8
|
+
testPlugin = null;
|
|
9
|
+
}
|
|
10
|
+
}
|
|
11
|
+
const plugin = async (input) => {
|
|
12
|
+
const traceDir = join(input.directory, ".opencode-trace");
|
|
13
|
+
const instance = new TracePlugin(traceDir);
|
|
14
|
+
instance.installInterceptor();
|
|
15
|
+
await instance.initStateManager();
|
|
16
|
+
if (!instance.getStateManager()) {
|
|
17
|
+
logger.error("StateManager initialization failed");
|
|
18
|
+
}
|
|
19
|
+
testPlugin = instance;
|
|
20
|
+
const hooks = {
|
|
21
|
+
event: async ({ event }) => {
|
|
22
|
+
if (!instance.getStateManager())
|
|
23
|
+
return;
|
|
24
|
+
if (event.type === "session.created" || event.type === "session.updated") {
|
|
25
|
+
const session = event.properties.info;
|
|
26
|
+
const existing = instance.getStateManager().getSession(session.id);
|
|
27
|
+
if (!existing) {
|
|
28
|
+
instance.getStateManager().startSession(session.id);
|
|
29
|
+
}
|
|
30
|
+
instance.getStateManager().updateSessionMetadata(session.id, {
|
|
31
|
+
title: session.title,
|
|
32
|
+
parentID: session.parentID,
|
|
33
|
+
folderPath: session.directory,
|
|
34
|
+
});
|
|
35
|
+
if (session.parentID) {
|
|
36
|
+
instance.getStateManager().addSubSession(session.parentID, session.id);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
},
|
|
40
|
+
"tool.execute.after": async (input, output) => {
|
|
41
|
+
if (!instance.getStateManager())
|
|
42
|
+
return;
|
|
43
|
+
if (input.tool === "task") {
|
|
44
|
+
const metadata = output.metadata;
|
|
45
|
+
if (metadata && typeof metadata.session_id === "string") {
|
|
46
|
+
instance.getStateManager().addSubSession(input.sessionID, metadata.session_id);
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
},
|
|
50
|
+
tool: {
|
|
51
|
+
trace_enable: {
|
|
52
|
+
description: "Enable trace recording for the current session. When global trace is disabled, only enabled sessions will be recorded.",
|
|
53
|
+
args: {},
|
|
54
|
+
execute: async (_, context) => {
|
|
55
|
+
if (!instance.getStateManager()) {
|
|
56
|
+
return "StateManager not initialized";
|
|
57
|
+
}
|
|
58
|
+
instance.getStateManager().setSessionEnabled(context.sessionID, true);
|
|
59
|
+
return `Trace enabled for session ${context.sessionID}`;
|
|
60
|
+
},
|
|
61
|
+
},
|
|
62
|
+
trace_disable: {
|
|
63
|
+
description: "Disable trace recording for the current session. This session will not be recorded even if global trace is enabled.",
|
|
64
|
+
args: {},
|
|
65
|
+
execute: async (_, context) => {
|
|
66
|
+
if (!instance.getStateManager()) {
|
|
67
|
+
return "StateManager not initialized";
|
|
68
|
+
}
|
|
69
|
+
instance.getStateManager().setSessionEnabled(context.sessionID, false);
|
|
70
|
+
return `Trace disabled for session ${context.sessionID}`;
|
|
71
|
+
},
|
|
72
|
+
},
|
|
73
|
+
trace_status: {
|
|
74
|
+
description: "Check the trace recording status for the current session. Shows both global and session-level status.",
|
|
75
|
+
args: {},
|
|
76
|
+
execute: async (_, context) => {
|
|
77
|
+
if (!instance.getStateManager()) {
|
|
78
|
+
return "StateManager not initialized";
|
|
79
|
+
}
|
|
80
|
+
const globalEnabled = instance.getStateManager().getGlobalState("global_trace_enabled") === "true";
|
|
81
|
+
const sessionEnabled = instance.getStateManager().getSessionEnabled(context.sessionID);
|
|
82
|
+
const willRecord = instance.getStateManager().isTraceEnabled(context.sessionID);
|
|
83
|
+
return JSON.stringify({
|
|
84
|
+
globalEnabled,
|
|
85
|
+
sessionEnabled,
|
|
86
|
+
willRecord,
|
|
87
|
+
sessionId: context.sessionID,
|
|
88
|
+
}, null, 2);
|
|
89
|
+
},
|
|
90
|
+
},
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
return hooks;
|
|
94
|
+
};
|
|
95
|
+
const entrypoint = {
|
|
96
|
+
id: "ljw1004.opencode-trace",
|
|
97
|
+
server: plugin,
|
|
98
|
+
};
|
|
99
|
+
export default entrypoint;
|
|
100
|
+
//# sourceMappingURL=trace.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.js","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AA6BnD,IAAI,UAAU,GAAuB,IAAI,CAAC;AAE1C,MAAM,UAAU,gBAAgB;IAC9B,IAAI,UAAU,EAAE,CAAC;QACf,UAAU,CAAC,oBAAoB,EAAE,CAAC;QAClC,UAAU,GAAG,IAAI,CAAC;IACpB,CAAC;AACH,CAAC;AAED,MAAM,MAAM,GAAW,KAAK,EAAE,KAAkB,EAAE,EAAE;IAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;IAC1D,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE3C,QAAQ,CAAC,kBAAkB,EAAE,CAAC;IAC9B,MAAM,QAAQ,CAAC,gBAAgB,EAAE,CAAC;IAElC,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACrD,CAAC;IAED,UAAU,GAAG,QAAQ,CAAC;IAEtB,MAAM,KAAK,GAAU;QACnB,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAoB,EAAE,EAAE;YAC3C,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE;gBAAE,OAAO;YAExC,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACzE,MAAM,OAAO,GAAI,KAAK,CAAC,UAAgC,CAAC,IAAI,CAAC;gBAE7D,MAAM,QAAQ,GAAG,QAAQ,CAAC,eAAe,EAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,CAAC,eAAe,EAAG,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACvD,CAAC;gBAED,QAAQ,CAAC,eAAe,EAAG,CAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,EAAE;oBAC5D,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,UAAU,EAAE,OAAO,CAAC,SAAS;iBAC9B,CAAC,CAAC;gBAEH,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACrB,QAAQ,CAAC,eAAe,EAAG,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC;QACH,CAAC;QAED,oBAAoB,EAAE,KAAK,EACzB,KAAyE,EACzE,MAA4D,EAC5D,EAAE;YACF,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE;gBAAE,OAAO;YAExC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,MAAM,CAAC,QAA+C,CAAC;gBACxE,IAAI,QAAQ,IAAI,OAAO,QAAQ,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;oBACxD,QAAQ,CAAC,eAAe,EAAG,CAAC,aAAa,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,EAAE;YACJ,YAAY,EAAE;gBACZ,WAAW,EAAE,wHAAwH;gBACrI,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;oBAC5B,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,CAAC;wBAChC,OAAO,8BAA8B,CAAC;oBACxC,CAAC;oBACD,QAAQ,CAAC,eAAe,EAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;oBACvE,OAAO,6BAA6B,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC1D,CAAC;aACF;YACD,aAAa,EAAE;gBACb,WAAW,EAAE,qHAAqH;gBAClI,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;oBAC5B,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,CAAC;wBAChC,OAAO,8BAA8B,CAAC;oBACxC,CAAC;oBACD,QAAQ,CAAC,eAAe,EAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;oBACxE,OAAO,8BAA8B,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3D,CAAC;aACF;YACD,YAAY,EAAE;gBACZ,WAAW,EAAE,uGAAuG;gBACpH,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;oBAC5B,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,CAAC;wBAChC,OAAO,8BAA8B,CAAC;oBACxC,CAAC;oBACD,MAAM,aAAa,GAAG,QAAQ,CAAC,eAAe,EAAG,CAAC,cAAc,CAAC,sBAAsB,CAAC,KAAK,MAAM,CAAC;oBACpG,MAAM,cAAc,GAAG,QAAQ,CAAC,eAAe,EAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBACxF,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,EAAG,CAAC,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBACjF,OAAO,IAAI,CAAC,SAAS,CAAC;wBACpB,aAAa;wBACb,cAAc;wBACd,UAAU;wBACV,SAAS,EAAE,OAAO,CAAC,SAAS;qBAC7B,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBACd,CAAC;aACF;SACF;KACF,CAAC;IAEF,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,UAAU,GAAiB;IAC/B,EAAE,EAAE,wBAAwB;IAC5B,MAAM,EAAE,MAAM;CACf,CAAC;AAEF,eAAe,UAAU,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.test.d.ts","sourceRoot":"","sources":["../src/trace.test.ts"],"names":[],"mappings":""}
|