atom-lite 0.4.1 → 0.5.1

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.
Files changed (88) hide show
  1. package/CHANGELOG.md +11 -0
  2. package/README.md +1 -1
  3. package/dist/agent/loop.d.ts +9 -0
  4. package/dist/agent/loop.js +233 -47
  5. package/dist/agent/loop.js.map +1 -1
  6. package/dist/agent/model-turn.d.ts +37 -0
  7. package/dist/agent/model-turn.js +53 -0
  8. package/dist/agent/model-turn.js.map +1 -0
  9. package/dist/agent/turn-recorder.d.ts +33 -0
  10. package/dist/agent/turn-recorder.js +74 -0
  11. package/dist/agent/turn-recorder.js.map +1 -0
  12. package/dist/agent/types.d.ts +4 -0
  13. package/dist/cli/app.js +373 -33
  14. package/dist/cli/app.js.map +1 -1
  15. package/dist/cli/commands.d.ts +3 -0
  16. package/dist/cli/commands.js +20 -5
  17. package/dist/cli/commands.js.map +1 -1
  18. package/dist/cli/main.js +2 -2
  19. package/dist/cli/main.js.map +1 -1
  20. package/dist/cli/mouse.js +4 -1
  21. package/dist/cli/mouse.js.map +1 -1
  22. package/dist/cli/stream-text.js +1 -0
  23. package/dist/cli/stream-text.js.map +1 -1
  24. package/dist/cli/theme.d.ts +1 -1
  25. package/dist/cli/theme.js +2 -0
  26. package/dist/cli/theme.js.map +1 -1
  27. package/dist/cli/transcript.d.ts +15 -1
  28. package/dist/cli/transcript.js +40 -1
  29. package/dist/cli/transcript.js.map +1 -1
  30. package/dist/cli/trust-prompt.d.ts +24 -0
  31. package/dist/cli/trust-prompt.js +89 -0
  32. package/dist/cli/trust-prompt.js.map +1 -0
  33. package/dist/config/defaults.d.ts +2 -0
  34. package/dist/config/defaults.js +2 -0
  35. package/dist/config/defaults.js.map +1 -1
  36. package/dist/config/load.js +4 -0
  37. package/dist/config/load.js.map +1 -1
  38. package/dist/config/paths.d.ts +2 -0
  39. package/dist/config/paths.js +12 -1
  40. package/dist/config/paths.js.map +1 -1
  41. package/dist/config/schema.d.ts +4 -0
  42. package/dist/config/schema.js +2 -0
  43. package/dist/config/schema.js.map +1 -1
  44. package/dist/context/manager.d.ts +17 -3
  45. package/dist/context/manager.js +129 -41
  46. package/dist/context/manager.js.map +1 -1
  47. package/dist/extensions/api.d.ts +51 -1
  48. package/dist/extensions/api.js +124 -6
  49. package/dist/extensions/api.js.map +1 -1
  50. package/dist/extensions/discovery.d.ts +16 -0
  51. package/dist/extensions/discovery.js +191 -108
  52. package/dist/extensions/discovery.js.map +1 -1
  53. package/dist/extensions/loader.d.ts +42 -1
  54. package/dist/extensions/loader.js +272 -32
  55. package/dist/extensions/loader.js.map +1 -1
  56. package/dist/extensions/session-manager.d.ts +44 -0
  57. package/dist/extensions/session-manager.js +137 -0
  58. package/dist/extensions/session-manager.js.map +1 -0
  59. package/dist/extensions/tool-adapter.d.ts +12 -0
  60. package/dist/extensions/tool-adapter.js +157 -0
  61. package/dist/extensions/tool-adapter.js.map +1 -0
  62. package/dist/extensions/trust.d.ts +2 -0
  63. package/dist/extensions/trust.js +32 -17
  64. package/dist/extensions/trust.js.map +1 -1
  65. package/dist/extensions/types.d.ts +45 -3
  66. package/dist/extensions/ui.d.ts +21 -0
  67. package/dist/extensions/ui.js +75 -0
  68. package/dist/extensions/ui.js.map +1 -0
  69. package/dist/index.d.ts +1 -0
  70. package/dist/index.js +1 -0
  71. package/dist/index.js.map +1 -1
  72. package/dist/mcp/stdio-transport.js +3 -20
  73. package/dist/mcp/stdio-transport.js.map +1 -1
  74. package/dist/runtime/scheduler.d.ts +4 -3
  75. package/dist/runtime/scheduler.js +7 -105
  76. package/dist/runtime/scheduler.js.map +1 -1
  77. package/dist/runtime/tool-executor.d.ts +18 -0
  78. package/dist/runtime/tool-executor.js +126 -0
  79. package/dist/runtime/tool-executor.js.map +1 -0
  80. package/dist/security/process.d.ts +10 -0
  81. package/dist/security/process.js +36 -0
  82. package/dist/security/process.js.map +1 -1
  83. package/dist/sessions/store.js +2 -3
  84. package/dist/sessions/store.js.map +1 -1
  85. package/dist/tools/bash.js +6 -3
  86. package/dist/tools/bash.js.map +1 -1
  87. package/dist/tools/types.d.ts +6 -0
  88. package/package.json +3 -1
@@ -1,8 +1,8 @@
1
1
  import { join } from "node:path";
2
- import { homedir } from "node:os";
3
2
  import { loadProjectInstructions, statSignature } from "./instructions.js";
4
3
  import { DefaultSkillRegistry, skillCacheSignature } from "../skills/registry.js";
5
4
  import { estimateMessageTokens } from "./budget.js";
5
+ import { userConfigDir } from "../config/paths.js";
6
6
  const DEFAULT_MAX_INPUT = 128000;
7
7
  export class ContextManager {
8
8
  opts;
@@ -10,25 +10,74 @@ export class ContextManager {
10
10
  skillCache;
11
11
  modelContextWindow;
12
12
  compactedThisBuild = false;
13
+ forceCompact = false;
14
+ buildCount = 0;
15
+ sessionMemory = null;
16
+ sessionMemoryCount = 0;
17
+ lastCompactionSummary = null;
18
+ lastCompactionCoveredCount = 0;
19
+ compactEveryNTurns = 0;
13
20
  constructor(opts) {
14
21
  this.opts = opts;
22
+ this.compactEveryNTurns = opts.compactEveryNTurns ?? 0;
15
23
  }
16
- /** Update the provider/model used for compaction after a switch. */
17
24
  setProvider(provider, model) {
18
25
  this.opts.provider = provider;
19
26
  if (model)
20
27
  this.opts.model = model;
21
28
  }
22
- /** Use the selected model's real window unless the user pinned a budget. */
23
29
  setModelContextWindow(window) {
24
30
  this.modelContextWindow = window;
25
31
  }
26
- /** True once if the most recent build compacted history. */
27
32
  consumeCompaction() {
28
33
  const compacted = this.compactedThisBuild;
29
34
  this.compactedThisBuild = false;
30
35
  return compacted;
31
36
  }
37
+ consumeCompactionSummary() {
38
+ const summary = this.lastCompactionSummary;
39
+ this.lastCompactionSummary = null;
40
+ return summary;
41
+ }
42
+ consumeCompactionCoveredCount() {
43
+ const count = this.lastCompactionCoveredCount;
44
+ this.lastCompactionCoveredCount = 0;
45
+ return count;
46
+ }
47
+ getLastCompactionCoveredCount() {
48
+ return this.lastCompactionCoveredCount;
49
+ }
50
+ setSessionMemory(summary, coveredCount) {
51
+ this.sessionMemory = summary;
52
+ if (summary === null) {
53
+ this.sessionMemoryCount = 0;
54
+ }
55
+ else if (typeof coveredCount === "number" && coveredCount >= 0) {
56
+ this.sessionMemoryCount = coveredCount;
57
+ }
58
+ }
59
+ getSessionMemory() {
60
+ return this.sessionMemory;
61
+ }
62
+ setForceCompact(force) {
63
+ this.forceCompact = force;
64
+ }
65
+ getCompactEveryNTurns() {
66
+ return this.compactEveryNTurns;
67
+ }
68
+ async compactHistory(_workspace, history, focus) {
69
+ const keep = Math.min(6, history.length);
70
+ const older = history.slice(0, history.length - keep);
71
+ if (older.length === 0)
72
+ return null;
73
+ const summary = await this.summarize(older, focus);
74
+ const coveredCount = history.length - keep;
75
+ this.sessionMemory = summary;
76
+ this.sessionMemoryCount = coveredCount;
77
+ this.lastCompactionSummary = summary;
78
+ this.lastCompactionCoveredCount = coveredCount;
79
+ return summary;
80
+ }
32
81
  budget() {
33
82
  const max = this.opts.maxInputTokens ?? this.modelContextWindow ?? DEFAULT_MAX_INPUT;
34
83
  return Math.max(1000, max - this.opts.reserveOutputTokens);
@@ -36,9 +85,45 @@ export class ContextManager {
36
85
  threshold() {
37
86
  return Math.floor(this.budget() * (this.opts.compactAtPercent / 100));
38
87
  }
88
+ async summarize(older, focus) {
89
+ let summaryText = "";
90
+ try {
91
+ const focusLine = focus
92
+ ? `\nFocus the summary on: ${focus}`
93
+ : "";
94
+ const summaryReq = {
95
+ model: this.opts.model,
96
+ messages: [
97
+ {
98
+ role: "system",
99
+ content: "Summarize the following conversation history compactly, preserving changed files, decisions, failing tests, and unfinished work. Output only the summary." +
100
+ focusLine,
101
+ },
102
+ {
103
+ role: "user",
104
+ content: [{ type: "text", text: JSON.stringify(older) }],
105
+ },
106
+ ],
107
+ tools: [],
108
+ };
109
+ const resp = await this.opts.provider.generate(summaryReq);
110
+ summaryText =
111
+ resp.message.content
112
+ ?.map((p) => (p.type === "text" ? p.text : ""))
113
+ .join("") ?? "";
114
+ }
115
+ catch {
116
+ summaryText = "";
117
+ }
118
+ if (!summaryText) {
119
+ summaryText = "[Earlier conversation truncated to fit context budget.]";
120
+ }
121
+ return summaryText;
122
+ }
39
123
  async build(workspace, history) {
40
124
  this.compactedThisBuild = false;
41
- const userPath = this.opts.userInstructionsPath ?? join(homedir(), ".config", "atom-lite", "AGENTS.md");
125
+ this.lastCompactionSummary = null;
126
+ const userPath = this.opts.userInstructionsPath ?? join(userConfigDir(), "AGENTS.md");
42
127
  const instructionSignature = `${this.opts.maxProjectInstructionBytes}|` +
43
128
  statSignature([userPath, join(workspace, "AGENTS.md")]);
44
129
  let instructions;
@@ -63,56 +148,59 @@ export class ContextManager {
63
148
  const systemPrompt = skillIndex
64
149
  ? `${this.opts.systemPrompt}\n\n${skillIndex}`
65
150
  : this.opts.systemPrompt;
151
+ this.buildCount++;
152
+ const periodic = this.compactEveryNTurns > 0 &&
153
+ this.buildCount % this.compactEveryNTurns === 0;
154
+ // Restore persisted session memory by injecting it and trimming the
155
+ // replayed tail, so resumed long sessions stay compact.
156
+ let workingHistory = history;
157
+ if (this.sessionMemory) {
158
+ const start = Math.min(this.sessionMemoryCount, history.length);
159
+ workingHistory = history.slice(start);
160
+ if (workingHistory.length === 0) {
161
+ workingHistory = history.slice(-Math.min(6, history.length));
162
+ }
163
+ }
66
164
  const messages = [{ role: "system", content: systemPrompt }];
67
165
  for (const instr of instructions) {
68
166
  messages.push({ role: "system", content: instr });
69
167
  }
70
- messages.push(...history);
71
- const total = messages.reduce((s, m) => s + estimateMessageTokens(m), 0);
72
- if (total <= this.threshold())
73
- return messages;
74
- // Compact: keep last 6 messages verbatim, summarize the rest.
75
- const keep = Math.min(6, history.length);
76
- const recent = history.slice(-keep);
77
- const older = history.slice(0, history.length - keep);
78
- if (older.length === 0)
79
- return messages;
80
- let summaryText = "";
81
- try {
82
- const summaryReq = {
83
- model: this.opts.model,
84
- messages: [
168
+ if (this.sessionMemory) {
169
+ messages.push({
170
+ role: "assistant",
171
+ content: [
85
172
  {
86
- role: "system",
87
- content: "Summarize the following conversation history compactly, preserving changed files, decisions, failing tests, and unfinished work. Output only the summary.",
88
- },
89
- {
90
- role: "user",
91
- content: [{ type: "text", text: JSON.stringify(older) }],
173
+ type: "text",
174
+ text: `Context summary from previous compaction:\n${this.sessionMemory}`,
92
175
  },
93
176
  ],
94
- tools: [],
95
- };
96
- const resp = await this.opts.provider.generate(summaryReq);
97
- summaryText =
98
- resp.message.content
99
- ?.map((p) => (p.type === "text" ? p.text : ""))
100
- .join("") ?? "";
177
+ });
101
178
  }
102
- catch {
103
- summaryText = "";
104
- }
105
- if (!summaryText) {
106
- // Fallback: truncate old tool output with marker.
107
- summaryText = "[Earlier conversation truncated to fit context budget.]";
179
+ messages.push(...workingHistory);
180
+ const total = messages.reduce((s, m) => s + estimateMessageTokens(m), 0);
181
+ const overThreshold = total > this.threshold() && workingHistory.length > 6;
182
+ if (!this.forceCompact && !periodic && !overThreshold) {
183
+ return messages;
108
184
  }
185
+ this.forceCompact = false;
186
+ const keep = Math.min(6, workingHistory.length);
187
+ const recent = workingHistory.slice(-keep);
188
+ const older = workingHistory.slice(0, workingHistory.length - keep);
189
+ if (older.length === 0)
190
+ return messages;
191
+ const summary = await this.summarize(older);
192
+ const coveredCount = this.sessionMemoryCount + older.length;
193
+ this.lastCompactionSummary = summary;
194
+ this.sessionMemory = summary;
195
+ this.sessionMemoryCount = coveredCount;
196
+ this.lastCompactionCoveredCount = coveredCount;
109
197
  this.compactedThisBuild = true;
110
198
  const compacted = [
111
- { role: "system", content: this.opts.systemPrompt },
199
+ { role: "system", content: systemPrompt },
112
200
  ...instructions.map((i) => ({ role: "system", content: i })),
113
201
  {
114
202
  role: "assistant",
115
- content: [{ type: "text", text: `Context summary: ${summaryText}` }],
203
+ content: [{ type: "text", text: `Context summary: ${summary}` }],
116
204
  },
117
205
  ...recent,
118
206
  ];
@@ -1 +1 @@
1
- {"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/context/manager.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAClF,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAapD,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAEjC,MAAM,OAAO,cAAc;IACjB,IAAI,CAAwB;IAC5B,gBAAgB,CAAyC;IACzD,UAAU,CAA2D;IACrE,kBAAkB,CAAqB;IACvC,kBAAkB,GAAG,KAAK,CAAC;IAEnC,YAAY,IAA2B;QACrC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,oEAAoE;IACpE,WAAW,CAAC,QAAyB,EAAE,KAAc;QACnD,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC9B,IAAI,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrC,CAAC;IAED,4EAA4E;IAC5E,qBAAqB,CAAC,MAA0B;QAC9C,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC;IACnC,CAAC;IAED,4DAA4D;IAC5D,iBAAiB;QACf,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC;QAC1C,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;QAChC,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM;QACZ,MAAM,GAAG,GACP,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,kBAAkB,IAAI,iBAAiB,CAAC;QAC3E,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC7D,CAAC;IAEO,SAAS;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,GAAG,GAAG,CAAC,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB,EAAE,OAAkB;QAC/C,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;QACxG,MAAM,oBAAoB,GACxB,GAAG,IAAI,CAAC,IAAI,CAAC,0BAA0B,GAAG;YAC1C,aAAa,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;QAC1D,IAAI,YAAsB,CAAC;QAC3B,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,KAAK,oBAAoB,EAAE,CAAC;YAC9D,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,uBAAuB,CACpC,SAAS,EACT,IAAI,CAAC,IAAI,CAAC,0BAA0B,EACpC,QAAQ,CACT,CAAC;YACF,IAAI,CAAC,gBAAgB,GAAG,EAAE,SAAS,EAAE,oBAAoB,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;QAClF,CAAC;QAED,MAAM,cAAc,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,UAAkB,CAAC;QACvB,IACE,IAAI,CAAC,UAAU,EAAE,SAAS,KAAK,SAAS;YACxC,IAAI,CAAC,UAAU,CAAC,SAAS,KAAK,cAAc,EAC5C,CAAC;YACD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAAC;YACrD,UAAU,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;QAChF,CAAC;QACD,MAAM,YAAY,GAAG,UAAU;YAC7B,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,OAAO,UAAU,EAAE;YAC9C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;QAC3B,MAAM,QAAQ,GAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QACxE,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;YACjC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;QAE1B,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,qBAAqB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACzE,IAAI,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO,QAAQ,CAAC;QAE/C,8DAA8D;QAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;QACtD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAExC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK;gBACtB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,QAAiB;wBACvB,OAAO,EACL,2JAA2J;qBAC9J;oBACD;wBACE,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;qBAClE;iBACF;gBACD,KAAK,EAAE,EAAE;aACV,CAAC;YACF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,UAAiB,CAAC,CAAC;YAClE,WAAW;gBACT,IAAI,CAAC,OAAO,CAAC,OAAO;oBAClB,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBACnD,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,GAAG,EAAE,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,kDAAkD;YAClD,WAAW,GAAG,yDAAyD,CAAC;QAC1E,CAAC;QAED,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAC/B,MAAM,SAAS,GAAc;YAC3B,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACnD,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YACrE;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,WAAW,EAAE,EAAE,CAAC;aACrE;YACD,GAAG,MAAM;SACV,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}
1
+ {"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/context/manager.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAClF,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAcnD,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAEjC,MAAM,OAAO,cAAc;IACjB,IAAI,CAAwB;IAC5B,gBAAgB,CAAyC;IACzD,UAAU,CAA2D;IACrE,kBAAkB,CAAqB;IACvC,kBAAkB,GAAG,KAAK,CAAC;IAC3B,YAAY,GAAG,KAAK,CAAC;IACrB,UAAU,GAAG,CAAC,CAAC;IACf,aAAa,GAAkB,IAAI,CAAC;IACpC,kBAAkB,GAAG,CAAC,CAAC;IACvB,qBAAqB,GAAkB,IAAI,CAAC;IAC5C,0BAA0B,GAAG,CAAC,CAAC;IAC/B,kBAAkB,GAAG,CAAC,CAAC;IAE/B,YAAY,IAA2B;QACrC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,IAAI,CAAC,CAAC;IACzD,CAAC;IAED,WAAW,CAAC,QAAyB,EAAE,KAAc;QACnD,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC9B,IAAI,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrC,CAAC;IAED,qBAAqB,CAAC,MAA0B;QAC9C,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC;IACnC,CAAC;IAED,iBAAiB;QACf,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC;QAC1C,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;QAChC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,wBAAwB;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CAAC;QAC3C,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAClC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,6BAA6B;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,0BAA0B,CAAC;QAC9C,IAAI,CAAC,0BAA0B,GAAG,CAAC,CAAC;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,6BAA6B;QAC3B,OAAO,IAAI,CAAC,0BAA0B,CAAC;IACzC,CAAC;IAED,gBAAgB,CAAC,OAAsB,EAAE,YAAqB;QAC5D,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACrB,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;QAC9B,CAAC;aAAM,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,IAAI,CAAC,EAAE,CAAC;YACjE,IAAI,CAAC,kBAAkB,GAAG,YAAY,CAAC;QACzC,CAAC;IACH,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,KAAc;QAC5B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED,qBAAqB;QACnB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,UAAkB,EAClB,OAAkB,EAClB,KAAc;QAEd,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;QACtD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACnD,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;QAC3C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,IAAI,CAAC,kBAAkB,GAAG,YAAY,CAAC;QACvC,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC;QACrC,IAAI,CAAC,0BAA0B,GAAG,YAAY,CAAC;QAC/C,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,MAAM;QACZ,MAAM,GAAG,GACP,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,kBAAkB,IAAI,iBAAiB,CAAC;QAC3E,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC7D,CAAC;IAEO,SAAS;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,GAAG,GAAG,CAAC,CAAC,CAAC;IACxE,CAAC;IAEO,KAAK,CAAC,SAAS,CACrB,KAAgB,EAChB,KAAc;QAEd,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,KAAK;gBACrB,CAAC,CAAC,2BAA2B,KAAK,EAAE;gBACpC,CAAC,CAAC,EAAE,CAAC;YACP,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK;gBACtB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,QAAiB;wBACvB,OAAO,EACL,2JAA2J;4BAC3J,SAAS;qBACZ;oBACD;wBACE,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;qBAClE;iBACF;gBACD,KAAK,EAAE,EAAE;aACV,CAAC;YACF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,UAAiB,CAAC,CAAC;YAClE,WAAW;gBACT,IAAI,CAAC,OAAO,CAAC,OAAO;oBAClB,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBACnD,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,GAAG,EAAE,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,WAAW,GAAG,yDAAyD,CAAC;QAC1E,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB,EAAE,OAAkB;QAC/C,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC;QAChC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE,WAAW,CAAC,CAAC;QACtF,MAAM,oBAAoB,GACxB,GAAG,IAAI,CAAC,IAAI,CAAC,0BAA0B,GAAG;YAC1C,aAAa,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;QAC1D,IAAI,YAAsB,CAAC;QAC3B,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,KAAK,oBAAoB,EAAE,CAAC;YAC9D,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,uBAAuB,CACpC,SAAS,EACT,IAAI,CAAC,IAAI,CAAC,0BAA0B,EACpC,QAAQ,CACT,CAAC;YACF,IAAI,CAAC,gBAAgB,GAAG,EAAE,SAAS,EAAE,oBAAoB,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;QAClF,CAAC;QAED,MAAM,cAAc,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,UAAkB,CAAC;QACvB,IACE,IAAI,CAAC,UAAU,EAAE,SAAS,KAAK,SAAS;YACxC,IAAI,CAAC,UAAU,CAAC,SAAS,KAAK,cAAc,EAC5C,CAAC;YACD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,oBAAoB,CAAC,SAAS,CAAC,CAAC;YACrD,UAAU,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;QAChF,CAAC;QACD,MAAM,YAAY,GAAG,UAAU;YAC7B,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,OAAO,UAAU,EAAE;YAC9C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;QAE3B,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,QAAQ,GACZ,IAAI,CAAC,kBAAkB,GAAG,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,kBAAkB,KAAK,CAAC,CAAC;QAElD,oEAAoE;QACpE,wDAAwD;QACxD,IAAI,cAAc,GAAG,OAAO,CAAC;QAC7B,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YAChE,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChC,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QACxE,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;YACjC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,8CAA8C,IAAI,CAAC,aAAa,EAAE;qBACzE;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,qBAAqB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACzE,MAAM,aAAa,GACjB,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,QAAQ,IAAI,CAAC,aAAa,EAAE,CAAC;YACtD,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAE1B,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;QACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAExC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5D,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC;QACrC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,IAAI,CAAC,kBAAkB,GAAG,YAAY,CAAC;QACvC,IAAI,CAAC,0BAA0B,GAAG,YAAY,CAAC;QAC/C,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAE/B,MAAM,SAAS,GAAc;YAC3B,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;YACzC,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YACrE;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,OAAO,EAAE,EAAE,CAAC;aACjE;YACD,GAAG,MAAM;SACV,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}
@@ -1,9 +1,12 @@
1
- import type { AtomEvent, ExecOptions, ExecResult, ExtensionCommand, ExtensionContext, ExtensionModel, ExtensionProvider, ExtensionTool, ExtensionWidget, ExtensionEventHandler } from "./types.js";
1
+ import type { AtomEvent, ExecOptions, ExecResult, ExtensionCommand, ExtensionContext, ExtensionModel, ExtensionProvider, ExtensionTool, ExtensionWidget, ExtensionUIContext, ExtensionEventHandler } from "./types.js";
2
2
  export interface ExtensionAPI {
3
3
  id: string;
4
+ ui: ExtensionUIContext;
4
5
  registerTool(tool: ExtensionTool): void;
5
6
  unregisterTool(name: string): void;
6
7
  setActiveTools(names: string[]): void;
8
+ getActiveTools(): string[];
9
+ getAllTools(): Array<Record<string, unknown>>;
7
10
  registerCommand(command: ExtensionCommand): void;
8
11
  unregisterCommand(name: string): void;
9
12
  on(event: AtomEvent, handler: ExtensionEventHandler): () => void;
@@ -19,6 +22,32 @@ export interface ExtensionAPI {
19
22
  notify(message: string, level?: "info" | "warning" | "error"): void;
20
23
  registerWidget(location: ExtensionWidget["location"], widget: ExtensionWidget): void;
21
24
  unregisterWidget(id: string): void;
25
+ registerToolRenderer(resolver: any): void;
26
+ registerMessageRenderer(customType: string, renderer: any): void;
27
+ registerEntryRenderer(customType: string, renderer: any): void;
28
+ registerMarkdownTransformer(transformer: any): void;
29
+ registerShortcut(shortcut: string, options: {
30
+ description?: string;
31
+ handler: (ctx: any) => void | Promise<void>;
32
+ }): void;
33
+ registerFlag(name: string, options: any): void;
34
+ getFlag(name: string): boolean | string | undefined;
35
+ addAutocompleteProvider(factory: any): void;
36
+ setWidget(key: string, content: string[] | ((tui: any, theme: any) => any) | undefined, options?: {
37
+ placement?: "aboveEditor" | "belowEditor";
38
+ }): void;
39
+ setFooter(factory: ((tui: any, theme: any, footerData: any) => any) | undefined): void;
40
+ setHeader(factory: ((tui: any, theme: any) => any) | undefined): void;
41
+ setTitle(title: string): void;
42
+ setWorkingMessage(message?: string): void;
43
+ setWorkingVisible(visible: boolean): void;
44
+ setWorkingIndicator(options?: {
45
+ frames?: string[];
46
+ intervalMs?: number;
47
+ }): void;
48
+ setHiddenThinkingLabel(label?: string): void;
49
+ getToolsExpanded(): boolean;
50
+ setToolsExpanded(expanded: boolean): void;
22
51
  select(title: string, options: string[]): Promise<string | undefined>;
23
52
  confirm(title: string, message: string): Promise<boolean>;
24
53
  input(title: string, placeholder?: string): Promise<string | undefined>;
@@ -26,6 +55,7 @@ export interface ExtensionAPI {
26
55
  getSession(): {
27
56
  id: string;
28
57
  };
58
+ getSessionManager(): any;
29
59
  isProjectTrusted(): boolean;
30
60
  registerProvider(provider: ExtensionProvider): void;
31
61
  unregisterProvider(name: string): void;
@@ -38,6 +68,26 @@ export interface ExtensionRegistries {
38
68
  commands: Map<string, ExtensionCommand>;
39
69
  handlers: Map<AtomEvent, Set<ExtensionEventHandler>>;
40
70
  widgets: Map<string, ExtensionWidget>;
71
+ uiWidgets: Map<string, any>;
72
+ toolRenderers: any[];
73
+ messageRenderers: Map<string, any>;
74
+ entryRenderers: Map<string, any>;
75
+ markdownTransformers: any[];
76
+ shortcuts: Map<string, any>;
77
+ flags: Map<string, any>;
78
+ flagValues: Map<string, any>;
79
+ autocompleteProviders: any[];
80
+ footerFactory?: any;
81
+ headerFactory?: any;
82
+ title?: string;
83
+ workingMessage?: string;
84
+ workingVisible: boolean;
85
+ workingIndicator?: {
86
+ frames?: string[];
87
+ intervalMs?: number;
88
+ };
89
+ hiddenThinkingLabel?: string;
90
+ toolsExpanded: boolean;
41
91
  providers: Map<string, ExtensionProvider>;
42
92
  models: ExtensionModel[];
43
93
  entries: Array<{
@@ -1,9 +1,21 @@
1
+ import { createDefaultExtensionUI } from "./ui.js";
1
2
  export function createExtensionRegistries() {
2
3
  return {
3
4
  tools: new Map(),
4
5
  commands: new Map(),
5
6
  handlers: new Map(),
6
7
  widgets: new Map(),
8
+ uiWidgets: new Map(),
9
+ toolRenderers: [],
10
+ messageRenderers: new Map(),
11
+ entryRenderers: new Map(),
12
+ markdownTransformers: [],
13
+ shortcuts: new Map(),
14
+ flags: new Map(),
15
+ flagValues: new Map(),
16
+ autocompleteProviders: [],
17
+ workingVisible: true,
18
+ toolsExpanded: false,
7
19
  providers: new Map(),
8
20
  models: [],
9
21
  entries: [],
@@ -12,8 +24,13 @@ export function createExtensionRegistries() {
12
24
  };
13
25
  }
14
26
  export function createExtensionAPI(id, ctx, registries, sessionId) {
27
+ if (typeof ctx.hasUI !== "boolean")
28
+ ctx.hasUI = false;
29
+ if (!ctx.ui)
30
+ ctx.ui = createDefaultExtensionUI(registries, ctx);
15
31
  return {
16
32
  id,
33
+ ui: ctx.ui,
17
34
  registerTool: (tool) => {
18
35
  registries.tools.set(tool.name, tool);
19
36
  },
@@ -23,8 +40,27 @@ export function createExtensionAPI(id, ctx, registries, sessionId) {
23
40
  setActiveTools: (names) => {
24
41
  registries.activeTools = new Set(names);
25
42
  },
43
+ getActiveTools: () => {
44
+ if (registries.activeTools)
45
+ return [...registries.activeTools];
46
+ return [...registries.tools.keys()];
47
+ },
48
+ getAllTools: () => {
49
+ return [...registries.tools.values()].map((tool) => ({
50
+ name: tool.name,
51
+ label: tool.label,
52
+ description: tool.description,
53
+ parameters: tool.parameters ?? {},
54
+ exposure: tool.exposure ?? "direct",
55
+ annotations: tool.annotations,
56
+ }));
57
+ },
26
58
  registerCommand: (command) => {
27
- registries.commands.set(command.name, command);
59
+ const normalized = { ...command };
60
+ if (!normalized.run && typeof normalized.handler === "function") {
61
+ normalized.run = normalized.handler;
62
+ }
63
+ registries.commands.set(normalized.name, normalized);
28
64
  },
29
65
  unregisterCommand: (name) => {
30
66
  registries.commands.delete(name);
@@ -40,11 +76,22 @@ export function createExtensionAPI(id, ctx, registries, sessionId) {
40
76
  registries.handlers.get(event)?.delete(handler);
41
77
  },
42
78
  appendEntry: (type, data) => {
79
+ if (ctx.sessionManager?.appendEntry) {
80
+ return ctx.sessionManager.appendEntry(type, data);
81
+ }
43
82
  registries.entries.push({ type, data });
44
83
  },
45
84
  getSessionEntries: (filter) => {
85
+ if (ctx.sessionManager?.getEntries) {
86
+ return ctx.sessionManager
87
+ .getEntries()
88
+ .filter((entry) => filter?.type
89
+ ? entry?.customType === filter.type || entry?.type === filter.type
90
+ : true);
91
+ }
46
92
  return registries.entries.filter((entry) => filter?.type ? entry.type === filter.type : true);
47
93
  },
94
+ getSessionManager: () => ctx.sessionManager,
48
95
  setStatus: (part, value) => {
49
96
  registries.status.set(part, value);
50
97
  },
@@ -58,14 +105,85 @@ export function createExtensionAPI(id, ctx, registries, sessionId) {
58
105
  unregisterWidget: (widgetId) => {
59
106
  registries.widgets.delete(widgetId);
60
107
  },
61
- select: async () => undefined,
62
- confirm: async () => false,
63
- input: async () => undefined,
108
+ registerToolRenderer: (resolver) => {
109
+ registries.toolRenderers.push(resolver);
110
+ },
111
+ registerMessageRenderer: (customType, renderer) => {
112
+ registries.messageRenderers.set(customType, renderer);
113
+ },
114
+ registerEntryRenderer: (customType, renderer) => {
115
+ registries.entryRenderers.set(customType, renderer);
116
+ },
117
+ registerMarkdownTransformer: (transformer) => {
118
+ if (registries.markdownTransformers.length > 0) {
119
+ registries.markdownTransformers[registries.markdownTransformers.length - 1] = transformer;
120
+ }
121
+ else {
122
+ registries.markdownTransformers.push(transformer);
123
+ }
124
+ },
125
+ registerShortcut: (shortcut, options) => {
126
+ registries.shortcuts.set(shortcut, options);
127
+ },
128
+ registerFlag: (name, options) => {
129
+ registries.flags.set(name, options);
130
+ if (options?.default !== undefined)
131
+ registries.flagValues.set(name, options.default);
132
+ },
133
+ getFlag: (name) => registries.flagValues.get(name),
134
+ addAutocompleteProvider: (factory) => {
135
+ registries.autocompleteProviders.push(factory);
136
+ },
137
+ setWidget: (key, content, options) => ctx.ui.setWidget(key, content, options),
138
+ setFooter: (factory) => ctx.ui.setFooter(factory),
139
+ setHeader: (factory) => ctx.ui.setHeader(factory),
140
+ setTitle: (title) => ctx.ui.setTitle(title),
141
+ setWorkingMessage: (message) => ctx.ui.setWorkingMessage(message),
142
+ setWorkingVisible: (visible) => ctx.ui.setWorkingVisible(visible),
143
+ setWorkingIndicator: (options) => ctx.ui.setWorkingIndicator(options),
144
+ setHiddenThinkingLabel: (label) => ctx.ui.setHiddenThinkingLabel(label),
145
+ getToolsExpanded: () => ctx.ui.getToolsExpanded(),
146
+ setToolsExpanded: (expanded) => ctx.ui.setToolsExpanded(expanded),
147
+ select: () => {
148
+ throw new Error("extension runtime not initialized");
149
+ },
150
+ confirm: () => {
151
+ throw new Error("extension runtime not initialized");
152
+ },
153
+ input: () => {
154
+ throw new Error("extension runtime not initialized");
155
+ },
64
156
  getWorkspace: () => ctx.workspace,
65
157
  getSession: () => ({ id: sessionId }),
66
158
  isProjectTrusted: () => ctx.trusted,
67
- registerProvider: (provider) => {
68
- registries.providers.set(provider.name, provider);
159
+ registerProvider: (providerOrName, config) => {
160
+ const name = typeof providerOrName === "string"
161
+ ? providerOrName
162
+ : providerOrName?.name ?? providerOrName?.id;
163
+ const cfg = typeof providerOrName === "string"
164
+ ? (config ?? {})
165
+ : providerOrName?.config ?? providerOrName ?? {};
166
+ if (!name) {
167
+ // eslint-disable-next-line no-console
168
+ console.error("[extension] registerProvider requires a provider name");
169
+ return;
170
+ }
171
+ const unsupported = [];
172
+ if (cfg?.oauth)
173
+ unsupported.push("oauth");
174
+ if (cfg?.streamSimple)
175
+ unsupported.push("streamSimple");
176
+ if (cfg?.classifiers)
177
+ unsupported.push("classifiers");
178
+ if (cfg?.refreshModels)
179
+ unsupported.push("refreshModels");
180
+ if (cfg?.images)
181
+ unsupported.push("images");
182
+ if (unsupported.length > 0) {
183
+ // eslint-disable-next-line no-console
184
+ console.error(`[extension] provider "${name}" uses reduced-compatibility features: ${unsupported.join(", ")}`);
185
+ }
186
+ registries.providers.set(name, { name, config: { ...cfg } });
69
187
  },
70
188
  unregisterProvider: (name) => {
71
189
  registries.providers.delete(name);
@@ -1 +1 @@
1
- {"version":3,"file":"api.js","sourceRoot":"","sources":["../../src/extensions/api.ts"],"names":[],"mappings":"AAqDA,MAAM,UAAU,yBAAyB;IACvC,OAAO;QACL,KAAK,EAAE,IAAI,GAAG,EAAE;QAChB,QAAQ,EAAE,IAAI,GAAG,EAAE;QACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;QACnB,OAAO,EAAE,IAAI,GAAG,EAAE;QAClB,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,IAAI;QACjB,MAAM,EAAE,IAAI,GAAG,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,EAAU,EACV,GAAqB,EACrB,UAA+B,EAC/B,SAAiB;IAEjB,OAAO;QACL,EAAE;QACF,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;YACrB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACxC,CAAC;QACD,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE;YACvB,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE;YACxB,UAAU,CAAC,WAAW,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,eAAe,EAAE,CAAC,OAAO,EAAE,EAAE;YAC3B,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC;QACD,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1B,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QACD,EAAE,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACrB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAC5C,CAAC;YACD,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7C,OAAO,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/D,CAAC;QACD,GAAG,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACtB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,WAAW,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;YAC1B,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;QACD,iBAAiB,EAAE,CAAC,MAAM,EAAE,EAAE;YAC5B,OAAO,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CACzC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CACjD,CAAC;QACJ,CAAC;QACD,SAAS,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACzB,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC;QACD,MAAM,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACzB,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE;YACpC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QACD,gBAAgB,EAAE,CAAC,QAAQ,EAAE,EAAE;YAC7B,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;QACD,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;QAC7B,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,KAAK;QAC1B,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS;QAC5B,YAAY,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,SAAS;QACjC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;QACrC,gBAAgB,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO;QACnC,gBAAgB,EAAE,CAAC,QAAQ,EAAE,EAAE;YAC7B,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACpD,CAAC;QACD,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QACD,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE;YACvB,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,GAAG,EAAE,EAAE,OAAO,GAAG,EAAE,EAAE,EAAE;YAC/C,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;YAC5D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,EAAE,IAAI,EAAE;oBACzC,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,SAAS;oBACjC,QAAQ,EAAE,MAAM;oBAChB,OAAO,EAAE,OAAO,CAAC,SAAS;iBAC3B,CAAC,CAAC;gBACH,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,KAIf,CAAC;gBACF,OAAO;oBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;oBAC5B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;oBAC5B,QAAQ,EAAE,OAAO,CAAC,MAAM,IAAI,CAAC;iBAC9B,CAAC;YACJ,CAAC;QACH,CAAC;QACD,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE;YACf,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,OAAO,EAAE,CAAC,CAAC;QAChD,CAAC;KACF,CAAC;AACJ,CAAC"}
1
+ {"version":3,"file":"api.js","sourceRoot":"","sources":["../../src/extensions/api.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AAqFnD,MAAM,UAAU,yBAAyB;IACvC,OAAO;QACL,KAAK,EAAE,IAAI,GAAG,EAAE;QAChB,QAAQ,EAAE,IAAI,GAAG,EAAE;QACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;QACnB,OAAO,EAAE,IAAI,GAAG,EAAE;QAClB,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,aAAa,EAAE,EAAE;QACjB,gBAAgB,EAAE,IAAI,GAAG,EAAE;QAC3B,cAAc,EAAE,IAAI,GAAG,EAAE;QACzB,oBAAoB,EAAE,EAAE;QACxB,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,KAAK,EAAE,IAAI,GAAG,EAAE;QAChB,UAAU,EAAE,IAAI,GAAG,EAAE;QACrB,qBAAqB,EAAE,EAAE;QACzB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,KAAK;QACpB,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,MAAM,EAAE,EAAE;QACV,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,IAAI;QACjB,MAAM,EAAE,IAAI,GAAG,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,EAAU,EACV,GAAqB,EACrB,UAA+B,EAC/B,SAAiB;IAEjB,IAAI,OAAO,GAAG,CAAC,KAAK,KAAK,SAAS;QAAE,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC;IACtD,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,GAAG,CAAC,EAAE,GAAG,wBAAwB,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;IAChE,OAAO;QACL,EAAE;QACF,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;YACrB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACxC,CAAC;QACD,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE;YACvB,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QACD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE;YACxB,UAAU,CAAC,WAAW,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,cAAc,EAAE,GAAG,EAAE;YACnB,IAAI,UAAU,CAAC,WAAW;gBAAE,OAAO,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;YAC/D,OAAO,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACtC,CAAC;QACD,WAAW,EAAE,GAAG,EAAE;YAChB,OAAO,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACnD,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;gBACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,QAAQ;gBACnC,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC,CAAC;QACN,CAAC;QACD,eAAe,EAAE,CAAC,OAAO,EAAE,EAAE;YAC3B,MAAM,UAAU,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,OAAO,UAAU,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBAChE,UAAU,CAAC,GAAG,GAAG,UAAU,CAAC,OAAO,CAAC;YACtC,CAAC;YACD,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;QACD,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1B,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QACD,EAAE,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACrB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAC5C,CAAC;YACD,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7C,OAAO,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/D,CAAC;QACD,GAAG,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACtB,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,WAAW,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;YAC1B,IAAI,GAAG,CAAC,cAAc,EAAE,WAAW,EAAE,CAAC;gBACpC,OAAO,GAAG,CAAC,cAAc,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpD,CAAC;YACD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;QACD,iBAAiB,EAAE,CAAC,MAAM,EAAE,EAAE;YAC5B,IAAI,GAAG,CAAC,cAAc,EAAE,UAAU,EAAE,CAAC;gBACnC,OAAO,GAAG,CAAC,cAAc;qBACtB,UAAU,EAAE;qBACZ,MAAM,CAAC,CAAC,KAAU,EAAE,EAAE,CACrB,MAAM,EAAE,IAAI;oBACV,CAAC,CAAC,KAAK,EAAE,UAAU,KAAK,MAAM,CAAC,IAAI,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,CAAC,IAAI;oBAClE,CAAC,CAAC,IAAI,CACT,CAAC;YACN,CAAC;YACD,OAAO,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CACzC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CACjD,CAAC;QACJ,CAAC;QACD,iBAAiB,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,cAAc;QAC3C,SAAS,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACzB,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC;QACD,MAAM,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACzB,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE;YACpC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QACD,gBAAgB,EAAE,CAAC,QAAQ,EAAE,EAAE;YAC7B,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;QACD,oBAAoB,EAAE,CAAC,QAAQ,EAAE,EAAE;YACjC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,uBAAuB,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE;YAChD,UAAU,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACxD,CAAC;QACD,qBAAqB,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE;YAC9C,UAAU,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACtD,CAAC;QACD,2BAA2B,EAAE,CAAC,WAAW,EAAE,EAAE;YAC3C,IAAI,UAAU,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/C,UAAU,CAAC,oBAAoB,CAAC,UAAU,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC;YAC5F,CAAC;iBAAM,CAAC;gBACN,UAAU,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;QACD,gBAAgB,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE;YACtC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QACD,YAAY,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE;YAC9B,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACpC,IAAI,OAAO,EAAE,OAAO,KAAK,SAAS;gBAAE,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACvF,CAAC;QACD,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC;QAClD,uBAAuB,EAAE,CAAC,OAAO,EAAE,EAAE;YACnC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,CAAC;QACD,SAAS,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC;QAC7E,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC;QACjD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC;QACjD,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC3C,iBAAiB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QACjE,iBAAiB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QACjE,mBAAmB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,mBAAmB,CAAC,OAAO,CAAC;QACrE,sBAAsB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,sBAAsB,CAAC,KAAK,CAAC;QACvE,gBAAgB,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,gBAAgB,EAAE;QACjD,gBAAgB,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC;QACjE,MAAM,EAAE,GAAG,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,EAAE,GAAG,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,KAAK,EAAE,GAAG,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,YAAY,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,SAAS;QACjC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;QACrC,gBAAgB,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO;QACnC,gBAAgB,EAAE,CAAC,cAAmB,EAAE,MAAY,EAAE,EAAE;YACtD,MAAM,IAAI,GACR,OAAO,cAAc,KAAK,QAAQ;gBAChC,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,cAAc,EAAE,IAAI,IAAI,cAAc,EAAE,EAAE,CAAC;YACjD,MAAM,GAAG,GACP,OAAO,cAAc,KAAK,QAAQ;gBAChC,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,CAAC;gBAChB,CAAC,CAAC,cAAc,EAAE,MAAM,IAAI,cAAc,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,sCAAsC;gBACtC,OAAO,CAAC,KAAK,CAAC,uDAAuD,CAAC,CAAC;gBACvE,OAAO;YACT,CAAC;YACD,MAAM,WAAW,GAAa,EAAE,CAAC;YACjC,IAAI,GAAG,EAAE,KAAK;gBAAE,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,GAAG,EAAE,YAAY;gBAAE,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACxD,IAAI,GAAG,EAAE,WAAW;gBAAE,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtD,IAAI,GAAG,EAAE,aAAa;gBAAE,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1D,IAAI,GAAG,EAAE,MAAM;gBAAE,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,sCAAsC;gBACtC,OAAO,CAAC,KAAK,CACX,yBAAyB,IAAI,0CAA0C,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAChG,CAAC;YACJ,CAAC;YACD,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QACD,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE;YACvB,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,GAAG,EAAE,EAAE,OAAO,GAAG,EAAE,EAAE,EAAE;YAC/C,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;YAC5D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,EAAE,IAAI,EAAE;oBACzC,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,SAAS;oBACjC,QAAQ,EAAE,MAAM;oBAChB,OAAO,EAAE,OAAO,CAAC,SAAS;iBAC3B,CAAC,CAAC;gBACH,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,KAIf,CAAC;gBACF,OAAO;oBACL,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;oBAC5B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;oBAC5B,QAAQ,EAAE,OAAO,CAAC,MAAM,IAAI,CAAC;iBAC9B,CAAC;YACJ,CAAC;QACH,CAAC;QACD,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE;YACf,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,OAAO,EAAE,CAAC,CAAC;QAChD,CAAC;KACF,CAAC;AACJ,CAAC"}
@@ -7,20 +7,36 @@ export interface ExtensionManifest {
7
7
  engines?: {
8
8
  "atom-lite"?: string;
9
9
  };
10
+ /** Which manifest field declared the extension entrypoints. */
11
+ kind?: "pi" | "atom";
10
12
  }
11
13
  export interface DiscoveredExtension {
12
14
  id: string;
13
15
  source: ExtensionSource;
16
+ /** Human-readable origin for diagnostics (for example "pi global"). */
17
+ origin?: string;
14
18
  rootPath: string;
15
19
  entryPath: string;
16
20
  manifest?: ExtensionManifest;
17
21
  errors: string[];
22
+ /** Project extensions are trusted only after an explicit project decision. */
23
+ trusted: boolean;
24
+ /** Set when the extension was discovered but must not be loaded. */
25
+ skipReason?: string;
18
26
  }
19
27
  export interface DiscoverExtensionsOptions {
20
28
  cliPaths?: string[];
29
+ /** Extra global extension directory (Atom-native). */
21
30
  userDir?: string;
31
+ /** Extra project extension directory (Atom-native). */
22
32
  projectDir?: string;
23
33
  packageRoots?: string[];
34
+ /** Workspace root. Enables project and settings discovery and trust checks. */
35
+ cwd?: string;
36
+ /** Pi agent directory (`~/.pi/agent` by default). */
37
+ agentDir?: string;
38
+ trustedProjects?: string[];
39
+ deniedProjects?: string[];
24
40
  }
25
41
  export declare function discoverExtensions(options?: DiscoverExtensionsOptions): DiscoveredExtension[];
26
42
  export declare function formatExtensionDiscovery(extensions: DiscoveredExtension[]): string;