pi-x-ide 0.1.0

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 (55) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +138 -0
  3. package/README.zh.md +138 -0
  4. package/dist/src/pi/commands.d.ts +12 -0
  5. package/dist/src/pi/commands.js +96 -0
  6. package/dist/src/pi/commands.js.map +1 -0
  7. package/dist/src/pi/connection.d.ts +27 -0
  8. package/dist/src/pi/connection.js +136 -0
  9. package/dist/src/pi/connection.js.map +1 -0
  10. package/dist/src/pi/context.d.ts +8 -0
  11. package/dist/src/pi/context.js +66 -0
  12. package/dist/src/pi/context.js.map +1 -0
  13. package/dist/src/pi/discovery.d.ts +11 -0
  14. package/dist/src/pi/discovery.js +79 -0
  15. package/dist/src/pi/discovery.js.map +1 -0
  16. package/dist/src/pi/index.d.ts +4 -0
  17. package/dist/src/pi/index.js +182 -0
  18. package/dist/src/pi/index.js.map +1 -0
  19. package/dist/src/pi/state.d.ts +24 -0
  20. package/dist/src/pi/state.js +12 -0
  21. package/dist/src/pi/state.js.map +1 -0
  22. package/dist/src/pi/ui.d.ts +6 -0
  23. package/dist/src/pi/ui.js +84 -0
  24. package/dist/src/pi/ui.js.map +1 -0
  25. package/dist/src/shared/format.d.ts +23 -0
  26. package/dist/src/shared/format.js +84 -0
  27. package/dist/src/shared/format.js.map +1 -0
  28. package/dist/src/shared/paths.d.ts +5 -0
  29. package/dist/src/shared/paths.js +50 -0
  30. package/dist/src/shared/paths.js.map +1 -0
  31. package/dist/src/shared/protocol.d.ts +92 -0
  32. package/dist/src/shared/protocol.js +8 -0
  33. package/dist/src/shared/protocol.js.map +1 -0
  34. package/dist/src/shared/schema.d.ts +9 -0
  35. package/dist/src/shared/schema.js +94 -0
  36. package/dist/src/shared/schema.js.map +1 -0
  37. package/dist/src/shared/ws.d.ts +2 -0
  38. package/dist/src/shared/ws.js +12 -0
  39. package/dist/src/shared/ws.js.map +1 -0
  40. package/dist/test/shared.test.d.ts +1 -0
  41. package/dist/test/shared.test.js +101 -0
  42. package/dist/test/shared.test.js.map +1 -0
  43. package/package.json +55 -0
  44. package/src/pi/commands.ts +122 -0
  45. package/src/pi/connection.ts +155 -0
  46. package/src/pi/context.ts +82 -0
  47. package/src/pi/discovery.ts +88 -0
  48. package/src/pi/index.ts +190 -0
  49. package/src/pi/state.ts +29 -0
  50. package/src/pi/ui.ts +85 -0
  51. package/src/shared/format.ts +95 -0
  52. package/src/shared/paths.ts +47 -0
  53. package/src/shared/protocol.ts +107 -0
  54. package/src/shared/schema.ts +113 -0
  55. package/src/shared/ws.ts +8 -0
@@ -0,0 +1,136 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.IdeConnection = void 0;
7
+ const ws_1 = __importDefault(require("ws"));
8
+ const protocol_1 = require("../shared/protocol");
9
+ const schema_1 = require("../shared/schema");
10
+ const ws_2 = require("../shared/ws");
11
+ class IdeConnection {
12
+ candidate;
13
+ cwd;
14
+ callbacks;
15
+ socket;
16
+ nextId = 1;
17
+ closedByUser = false;
18
+ constructor(candidate, cwd, callbacks) {
19
+ this.candidate = candidate;
20
+ this.cwd = cwd;
21
+ this.callbacks = callbacks;
22
+ }
23
+ get isOpen() {
24
+ return this.socket?.readyState === ws_1.default.OPEN;
25
+ }
26
+ async connect() {
27
+ this.closedByUser = false;
28
+ const { lock } = this.candidate;
29
+ const socket = new ws_1.default(`ws://${lock.host}:${lock.port}`, {
30
+ headers: {
31
+ [protocol_1.AUTH_HEADER]: lock.authToken,
32
+ },
33
+ });
34
+ this.socket = socket;
35
+ socket.on("message", (raw) => this.handleMessage((0, ws_2.decodeRawData)(raw)));
36
+ socket.on("close", (_code, reason) => {
37
+ if (this.socket === socket)
38
+ this.socket = undefined;
39
+ this.callbacks.onDisconnected?.(reason.toString("utf8") || (this.closedByUser ? "closed" : "disconnected"));
40
+ });
41
+ socket.on("error", (error) => this.callbacks.onError?.(error));
42
+ await new Promise((resolve, reject) => {
43
+ const onOpen = () => {
44
+ cleanup();
45
+ this.sendInitialize();
46
+ resolve();
47
+ };
48
+ const onError = (error) => {
49
+ cleanup();
50
+ reject(error);
51
+ };
52
+ const cleanup = () => {
53
+ socket.off("open", onOpen);
54
+ socket.off("error", onError);
55
+ };
56
+ socket.once("open", onOpen);
57
+ socket.once("error", onError);
58
+ });
59
+ }
60
+ disconnect() {
61
+ this.closedByUser = true;
62
+ this.socket?.close();
63
+ this.socket = undefined;
64
+ }
65
+ sendInitialize() {
66
+ this.socket?.send(JSON.stringify({
67
+ jsonrpc: "2.0",
68
+ id: this.nextId++,
69
+ method: "initialize",
70
+ params: {
71
+ protocolVersion: protocol_1.PROTOCOL_VERSION,
72
+ client: { name: "pi-x-ide", version: "0.1.0" },
73
+ cwd: this.cwd,
74
+ },
75
+ }));
76
+ }
77
+ handleMessage(text) {
78
+ let parsed;
79
+ try {
80
+ parsed = JSON.parse(text);
81
+ }
82
+ catch {
83
+ return;
84
+ }
85
+ if (isRpcResponse(parsed)) {
86
+ const server = getServerInfo(parsed);
87
+ if (server)
88
+ this.callbacks.onConnected?.(server);
89
+ return;
90
+ }
91
+ if (!isNotification(parsed))
92
+ return;
93
+ if (parsed.method === "selection_changed" && (0, schema_1.isSelectionChangedParams)(parsed.params)) {
94
+ this.callbacks.onSelectionChanged?.(withReceivedAt(parsed.params));
95
+ }
96
+ else if (parsed.method === "selection_cleared" && (0, schema_1.isSelectionClearedParams)(parsed.params)) {
97
+ this.callbacks.onSelectionCleared?.(withReceivedAt(parsed.params));
98
+ }
99
+ else if (parsed.method === "at_mentioned" && (0, schema_1.isAtMentionedParams)(parsed.params)) {
100
+ const params = withReceivedAt(parsed.params);
101
+ this.callbacks.onAtMentioned?.(params);
102
+ }
103
+ }
104
+ }
105
+ exports.IdeConnection = IdeConnection;
106
+ function withReceivedAt(params) {
107
+ return { ...params, receivedAt: params.receivedAt ?? Date.now() };
108
+ }
109
+ function isNotification(value) {
110
+ return (typeof value === "object" &&
111
+ value !== null &&
112
+ value.jsonrpc === "2.0" &&
113
+ typeof value.method === "string");
114
+ }
115
+ function isRpcResponse(value) {
116
+ return (typeof value === "object" && value !== null && value.jsonrpc === "2.0" && "id" in value);
117
+ }
118
+ function getServerInfo(response) {
119
+ const result = response.result;
120
+ if (!result || typeof result !== "object")
121
+ return undefined;
122
+ const server = result.server;
123
+ if (!server || typeof server !== "object")
124
+ return undefined;
125
+ const name = server.name;
126
+ if (typeof name !== "string")
127
+ return undefined;
128
+ const version = server.version;
129
+ const ide = server.ide;
130
+ return {
131
+ name,
132
+ version: typeof version === "string" ? version : undefined,
133
+ ide: typeof ide === "string" ? ide : undefined,
134
+ };
135
+ }
136
+ //# sourceMappingURL=connection.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../src/pi/connection.ts"],"names":[],"mappings":";;;;;;AAAA,4CAA2B;AAC3B,iDAS4B;AAC5B,6CAA2G;AAC3G,qCAA6C;AAW7C,MAAa,aAAa;IAMb;IACQ;IACA;IAPX,MAAM,CAAa;IACnB,MAAM,GAAG,CAAC,CAAC;IACX,YAAY,GAAG,KAAK,CAAC;IAE7B,YACW,SAA4B,EACpB,GAAW,EACX,SAAiC;QAFzC,cAAS,GAAT,SAAS,CAAmB;QACpB,QAAG,GAAH,GAAG,CAAQ;QACX,cAAS,GAAT,SAAS,CAAwB;IACjD,CAAC;IAEJ,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,MAAM,EAAE,UAAU,KAAK,YAAS,CAAC,IAAI,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,MAAM,MAAM,GAAG,IAAI,YAAS,CAAC,QAAQ,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE;YAC7D,OAAO,EAAE;gBACP,CAAC,sBAAW,CAAC,EAAE,IAAI,CAAC,SAAS;aAC9B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,IAAA,kBAAa,EAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YACpD,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;QAC9G,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAE/D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,MAAM,GAAG,GAAG,EAAE;gBAClB,OAAO,EAAE,CAAC;gBACV,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC;YACF,MAAM,OAAO,GAAG,CAAC,KAAY,EAAE,EAAE;gBAC/B,OAAO,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC;YACF,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC3B,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/B,CAAC,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU;QACR,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1B,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,MAAM,EAAE,IAAI,CACf,IAAI,CAAC,SAAS,CAAC;YACb,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE;YACjB,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE;gBACN,eAAe,EAAE,2BAAgB;gBACjC,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE;gBAC9C,GAAG,EAAE,IAAI,CAAC,GAAG;aACd;SACF,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YACrC,IAAI,MAAM;gBAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;YAAE,OAAO;QACpC,IAAI,MAAM,CAAC,MAAM,KAAK,mBAAmB,IAAI,IAAA,iCAAwB,EAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACrF,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACrE,CAAC;aAAM,IAAI,MAAM,CAAC,MAAM,KAAK,mBAAmB,IAAI,IAAA,iCAAwB,EAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5F,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACrE,CAAC;aAAM,IAAI,MAAM,CAAC,MAAM,KAAK,cAAc,IAAI,IAAA,4BAAmB,EAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAClF,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;CACF;AAhGD,sCAgGC;AAED,SAAS,cAAc,CAA4D,MAAS;IAC1F,OAAO,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACb,KAA+B,CAAC,OAAO,KAAK,KAAK;QAClD,OAAQ,KAA8B,CAAC,MAAM,KAAK,QAAQ,CAC3D,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAK,KAA+B,CAAC,OAAO,KAAK,KAAK,IAAI,IAAI,IAAI,KAAK,CACnH,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,QAAyB;IAC9C,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC/B,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC5D,MAAM,MAAM,GAAI,MAA+B,CAAC,MAAM,CAAC;IACvD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC5D,MAAM,IAAI,GAAI,MAA6B,CAAC,IAAI,CAAC;IACjD,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC/C,MAAM,OAAO,GAAI,MAAgC,CAAC,OAAO,CAAC;IAC1D,MAAM,GAAG,GAAI,MAA4B,CAAC,GAAG,CAAC;IAC9C,OAAO;QACL,IAAI;QACJ,OAAO,EAAE,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QAC1D,GAAG,EAAE,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;KAC/C,CAAC;AACJ,CAAC"}
@@ -0,0 +1,8 @@
1
+ import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent" with {
2
+ "resolution-mode": "import"
3
+ };
4
+ import type { EditorSelectionSnapshot } from "../shared/protocol";
5
+ import type { PiIdeRuntime } from "./state";
6
+ export declare function registerContextHandlers(pi: ExtensionAPI, runtime: PiIdeRuntime): void;
7
+ export declare function setLatestSelection(runtime: PiIdeRuntime, snapshot: EditorSelectionSnapshot, ctx?: ExtensionContext): void;
8
+ export declare function clearLatestSelection(runtime: PiIdeRuntime, ctx?: ExtensionContext): void;
@@ -0,0 +1,66 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.registerContextHandlers = registerContextHandlers;
4
+ exports.setLatestSelection = setLatestSelection;
5
+ exports.clearLatestSelection = clearLatestSelection;
6
+ const format_1 = require("../shared/format");
7
+ const ui_1 = require("./ui");
8
+ const CONTEXT_MARKER = "pi-x-ide/editor-context";
9
+ function registerContextHandlers(pi, runtime) {
10
+ pi.on("before_agent_start", (_event, ctx) => {
11
+ runtime.ctx = ctx;
12
+ if (!runtime.enabled)
13
+ return;
14
+ if (!runtime.latestSelection)
15
+ return;
16
+ if (runtime.attachState !== "pending")
17
+ return;
18
+ runtime.turnSelection = runtime.latestSelection;
19
+ });
20
+ // Merge the active editor context into the submitted user prompt rather than
21
+ // adding a separate extension message.
22
+ pi.on("message_end", (event, ctx) => {
23
+ runtime.ctx = ctx;
24
+ if (!runtime.enabled || !runtime.turnSelection)
25
+ return;
26
+ if (event.message.role !== "user")
27
+ return;
28
+ if (messageContainsMarker(event.message))
29
+ return;
30
+ const text = `${(0, format_1.formatEditorContext)(runtime.turnSelection, { cwd: ctx.cwd })}\n<!-- ${CONTEXT_MARKER} -->\n`;
31
+ const message = mergeIntoUserMessage(event.message, text);
32
+ runtime.attachState = "sent";
33
+ runtime.turnSelection = undefined;
34
+ (0, ui_1.updateIdeUi)(runtime, ctx);
35
+ return { message };
36
+ });
37
+ }
38
+ function setLatestSelection(runtime, snapshot, ctx) {
39
+ const key = (0, format_1.snapshotKey)(snapshot);
40
+ runtime.latestSelection = snapshot;
41
+ if (runtime.latestSelectionKey !== key) {
42
+ runtime.latestSelectionKey = key;
43
+ runtime.attachState = "pending";
44
+ }
45
+ (0, ui_1.updateIdeUi)(runtime, ctx);
46
+ }
47
+ function clearLatestSelection(runtime, ctx) {
48
+ runtime.latestSelection = undefined;
49
+ runtime.latestSelectionKey = undefined;
50
+ runtime.turnSelection = undefined;
51
+ runtime.attachState = "idle";
52
+ (0, ui_1.updateIdeUi)(runtime, ctx);
53
+ }
54
+ function mergeIntoUserMessage(message, text) {
55
+ return {
56
+ ...message,
57
+ content: [{ type: "text", text }, ...normalizeUserContent(message.content)],
58
+ };
59
+ }
60
+ function normalizeUserContent(content) {
61
+ return typeof content === "string" ? [{ type: "text", text: content }] : content;
62
+ }
63
+ function messageContainsMarker(message) {
64
+ return JSON.stringify(message).includes(CONTEXT_MARKER);
65
+ }
66
+ //# sourceMappingURL=context.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"context.js","sourceRoot":"","sources":["../../../src/pi/context.ts"],"names":[],"mappings":";;AAUA,0DAyBC;AAED,gDAYC;AAED,oDAMC;AAtDD,6CAAoE;AAGpE,6BAAmC;AAEnC,MAAM,cAAc,GAAG,yBAAyB,CAAC;AAEjD,SAAgB,uBAAuB,CAAC,EAAgB,EAAE,OAAqB;IAC7E,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QAC1C,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;QAClB,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO;QAC7B,IAAI,CAAC,OAAO,CAAC,eAAe;YAAE,OAAO;QACrC,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS;YAAE,OAAO;QAE9C,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,eAAe,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAC7E,uCAAuC;IACvC,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAClC,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;QAClB,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa;YAAE,OAAO;QACvD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO;QAC1C,IAAI,qBAAqB,CAAC,KAAK,CAAC,OAAO,CAAC;YAAE,OAAO;QAEjD,MAAM,IAAI,GAAG,GAAG,IAAA,4BAAmB,EAAC,OAAO,CAAC,aAAa,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,UAAU,cAAc,QAAQ,CAAC;QAC7G,MAAM,OAAO,GAAG,oBAAoB,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC1D,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC;QAC7B,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;QAClC,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC1B,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,kBAAkB,CAChC,OAAqB,EACrB,QAAiC,EACjC,GAAsB;IAEtB,MAAM,GAAG,GAAG,IAAA,oBAAW,EAAC,QAAQ,CAAC,CAAC;IAClC,OAAO,CAAC,eAAe,GAAG,QAAQ,CAAC;IACnC,IAAI,OAAO,CAAC,kBAAkB,KAAK,GAAG,EAAE,CAAC;QACvC,OAAO,CAAC,kBAAkB,GAAG,GAAG,CAAC;QACjC,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAClC,CAAC;IACD,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;AAC5B,CAAC;AAED,SAAgB,oBAAoB,CAAC,OAAqB,EAAE,GAAsB;IAChF,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;IACpC,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;IACvC,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;IAClC,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC;IAC7B,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;AAC5B,CAAC;AAWD,SAAS,oBAAoB,CAAiC,OAAU,EAAE,IAAY;IACpF,OAAO;QACL,GAAG,OAAO;QACV,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;KAC5E,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAwC;IACpE,OAAO,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AACnF,CAAC;AAED,SAAS,qBAAqB,CAAC,OAA6B;IAC1D,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;AAC1D,CAAC"}
@@ -0,0 +1,11 @@
1
+ import { type LockFileCandidate } from "../shared/protocol";
2
+ export interface DiscoverOptions {
3
+ cwd: string;
4
+ lockDir?: string;
5
+ now?: number;
6
+ maxAgeMs?: number;
7
+ checkPid?: boolean;
8
+ }
9
+ export declare function sortCandidates(candidates: LockFileCandidate[]): LockFileCandidate[];
10
+ export declare function discoverIdeCandidates(options: DiscoverOptions): Promise<LockFileCandidate[]>;
11
+ export declare function resolveBestIdeCandidate(options: DiscoverOptions): Promise<LockFileCandidate | undefined>;
@@ -0,0 +1,79 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.sortCandidates = sortCandidates;
4
+ exports.discoverIdeCandidates = discoverIdeCandidates;
5
+ exports.resolveBestIdeCandidate = resolveBestIdeCandidate;
6
+ const promises_1 = require("node:fs/promises");
7
+ const node_path_1 = require("node:path");
8
+ const protocol_1 = require("../shared/protocol");
9
+ const paths_1 = require("../shared/paths");
10
+ const schema_1 = require("../shared/schema");
11
+ function isProcessAlive(pid) {
12
+ try {
13
+ process.kill(pid, 0);
14
+ return true;
15
+ }
16
+ catch {
17
+ return false;
18
+ }
19
+ }
20
+ function bestWorkspaceMatch(lock, cwd) {
21
+ let best;
22
+ for (const workspaceFolder of lock.workspaceFolders) {
23
+ const matchLength = (0, paths_1.relationshipMatchLength)(workspaceFolder, cwd);
24
+ if (matchLength <= 0)
25
+ continue;
26
+ if (!best || matchLength > best.matchLength) {
27
+ best = { matchLength, workspaceFolder };
28
+ }
29
+ }
30
+ return best;
31
+ }
32
+ function sortCandidates(candidates) {
33
+ return [...candidates].sort((a, b) => b.matchLength - a.matchLength || b.mtimeMs - a.mtimeMs || a.path.localeCompare(b.path));
34
+ }
35
+ async function discoverIdeCandidates(options) {
36
+ const lockDir = options.lockDir ?? (0, paths_1.resolveLockDir)();
37
+ const now = options.now ?? Date.now();
38
+ const maxAgeMs = options.maxAgeMs ?? 24 * 60 * 60 * 1000;
39
+ const checkPid = options.checkPid ?? true;
40
+ let entries;
41
+ try {
42
+ entries = await (0, promises_1.readdir)(lockDir);
43
+ }
44
+ catch {
45
+ return [];
46
+ }
47
+ const candidates = [];
48
+ for (const entry of entries) {
49
+ if (!entry.endsWith(protocol_1.LOCK_FILE_EXTENSION))
50
+ continue;
51
+ const path = (0, node_path_1.join)(lockDir, entry);
52
+ let content;
53
+ let mtimeMs;
54
+ try {
55
+ const [fileContent, fileStat] = await Promise.all([(0, promises_1.readFile)(path, "utf8"), (0, promises_1.stat)(path)]);
56
+ content = fileContent;
57
+ mtimeMs = fileStat.mtimeMs;
58
+ }
59
+ catch {
60
+ continue;
61
+ }
62
+ if (now - mtimeMs > maxAgeMs)
63
+ continue;
64
+ const lock = (0, schema_1.parseLockFileContent)(content);
65
+ if (!lock)
66
+ continue;
67
+ if (checkPid && typeof lock.pid === "number" && !isProcessAlive(lock.pid))
68
+ continue;
69
+ const match = bestWorkspaceMatch(lock, options.cwd);
70
+ if (!match)
71
+ continue;
72
+ candidates.push({ path, lock, mtimeMs, ...match });
73
+ }
74
+ return sortCandidates(candidates);
75
+ }
76
+ async function resolveBestIdeCandidate(options) {
77
+ return (await discoverIdeCandidates(options))[0];
78
+ }
79
+ //# sourceMappingURL=discovery.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"discovery.js","sourceRoot":"","sources":["../../../src/pi/discovery.ts"],"names":[],"mappings":";;AAsCA,wCAIC;AAED,sDAuCC;AAED,0DAEC;AAvFD,+CAA2D;AAC3D,yCAAiC;AACjC,iDAAmG;AACnG,2CAA0E;AAC1E,6CAAwD;AAUxD,SAAS,cAAc,CAAC,GAAW;IACjC,IAAI,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CACzB,IAAiB,EACjB,GAAW;IAEX,IAAI,IAAkE,CAAC;IACvE,KAAK,MAAM,eAAe,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,WAAW,GAAG,IAAA,+BAAuB,EAAC,eAAe,EAAE,GAAG,CAAC,CAAC;QAClE,IAAI,WAAW,IAAI,CAAC;YAAE,SAAS;QAC/B,IAAI,CAAC,IAAI,IAAI,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YAC5C,IAAI,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC;QAC1C,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAgB,cAAc,CAAC,UAA+B;IAC5D,OAAO,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CACzB,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CACjG,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,qBAAqB,CAAC,OAAwB;IAClE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAA,sBAAc,GAAE,CAAC;IACpD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IACzD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAE1C,IAAI,OAAiB,CAAC;IACtB,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,IAAA,kBAAO,EAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,UAAU,GAAwB,EAAE,CAAC;IAC3C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,8BAAmB,CAAC;YAAE,SAAS;QACnD,MAAM,IAAI,GAAG,IAAA,gBAAI,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAClC,IAAI,OAAe,CAAC;QACpB,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAA,mBAAQ,EAAC,IAAI,EAAE,MAAM,CAAC,EAAE,IAAA,eAAI,EAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxF,OAAO,GAAG,WAAW,CAAC;YACtB,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;QAED,IAAI,GAAG,GAAG,OAAO,GAAG,QAAQ;YAAE,SAAS;QACvC,MAAM,IAAI,GAAG,IAAA,6BAAoB,EAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,IAAI,QAAQ,IAAI,OAAO,IAAI,CAAC,GAAG,KAAK,QAAQ,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;YAAE,SAAS;QAEpF,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK;YAAE,SAAS;QAErB,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,OAAO,cAAc,CAAC,UAAU,CAAC,CAAC;AACpC,CAAC;AAEM,KAAK,UAAU,uBAAuB,CAAC,OAAwB;IACpE,OAAO,CAAC,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACnD,CAAC"}
@@ -0,0 +1,4 @@
1
+ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent" with {
2
+ "resolution-mode": "import"
3
+ };
4
+ export default function (pi: ExtensionAPI): void;
@@ -0,0 +1,182 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.default = default_1;
4
+ const format_1 = require("../shared/format");
5
+ const discovery_1 = require("./discovery");
6
+ const connection_1 = require("./connection");
7
+ const commands_1 = require("./commands");
8
+ const context_1 = require("./context");
9
+ const state_1 = require("./state");
10
+ const ui_1 = require("./ui");
11
+ const RECONNECT_DELAY_MS = 2_000;
12
+ function default_1(pi) {
13
+ const runtime = (0, state_1.createRuntime)();
14
+ (0, context_1.registerContextHandlers)(pi, runtime);
15
+ (0, commands_1.registerIdeCommand)(pi, runtime, {
16
+ refreshCandidates: (ctx) => refreshCandidates(runtime, ctx),
17
+ connectAuto: (ctx) => connectAuto(runtime, ctx),
18
+ connectCandidate: (candidate, ctx) => connectCandidate(runtime, candidate, ctx),
19
+ disconnect: (ctx, disabled) => disconnect(runtime, ctx, disabled),
20
+ });
21
+ pi.on("session_start", async (_event, ctx) => {
22
+ runtime.ctx = ctx;
23
+ runtime.cwd = ctx.cwd;
24
+ if (!runtime.enabled) {
25
+ runtime.connectionStatus = "disabled";
26
+ (0, ui_1.updateIdeUi)(runtime, ctx);
27
+ return;
28
+ }
29
+ await connectAuto(runtime, ctx);
30
+ });
31
+ pi.on("session_shutdown", (_event, ctx) => {
32
+ runtime.ctx = ctx;
33
+ if (runtime.reconnectTimer)
34
+ clearTimeout(runtime.reconnectTimer);
35
+ runtime.reconnectTimer = undefined;
36
+ runtime.connection?.disconnect();
37
+ runtime.connection = undefined;
38
+ (0, ui_1.clearIdeUi)(runtime, ctx);
39
+ });
40
+ }
41
+ async function refreshCandidates(runtime, ctx) {
42
+ runtime.ctx = ctx;
43
+ runtime.cwd = ctx.cwd;
44
+ runtime.candidates = await (0, discovery_1.discoverIdeCandidates)({ cwd: ctx.cwd });
45
+ return runtime.candidates;
46
+ }
47
+ async function connectAuto(runtime, ctx) {
48
+ runtime.enabled = true;
49
+ const candidates = await refreshCandidates(runtime, ctx);
50
+ const candidate = candidates[0];
51
+ if (!candidate) {
52
+ runtime.connectionStatus = "disconnected";
53
+ runtime.connectionMessage = "No matching IDE lock files found.";
54
+ runtime.currentCandidate = undefined;
55
+ runtime.connectedServer = undefined;
56
+ (0, ui_1.updateIdeUi)(runtime, ctx);
57
+ return;
58
+ }
59
+ await connectCandidate(runtime, candidate, ctx);
60
+ }
61
+ async function connectCandidate(runtime, candidate, ctx) {
62
+ runtime.ctx = ctx;
63
+ runtime.cwd = ctx.cwd;
64
+ runtime.enabled = true;
65
+ if (runtime.reconnectTimer)
66
+ clearTimeout(runtime.reconnectTimer);
67
+ runtime.reconnectTimer = undefined;
68
+ const previous = runtime.connection;
69
+ runtime.connection = undefined;
70
+ previous?.disconnect();
71
+ runtime.currentCandidate = candidate;
72
+ runtime.connectedServer = undefined;
73
+ runtime.connectionStatus = "connecting";
74
+ runtime.connectionMessage = `Connecting to ${candidate.lock.name} at ${candidate.lock.host}:${candidate.lock.port}`;
75
+ (0, ui_1.updateIdeUi)(runtime, ctx);
76
+ const connection = new connection_1.IdeConnection(candidate, ctx.cwd, {
77
+ onConnected: (server) => {
78
+ if (runtime.connection !== connection)
79
+ return;
80
+ runtime.connectedServer = server;
81
+ runtime.connectionStatus = "connected";
82
+ runtime.connectionMessage = undefined;
83
+ (0, ui_1.updateIdeUi)(runtime);
84
+ },
85
+ onDisconnected: (reason) => {
86
+ if (runtime.connection !== connection)
87
+ return;
88
+ runtime.connection = undefined;
89
+ runtime.connectedServer = undefined;
90
+ runtime.connectionStatus = runtime.enabled ? "disconnected" : "disabled";
91
+ runtime.connectionMessage = reason;
92
+ (0, ui_1.updateIdeUi)(runtime);
93
+ if (runtime.enabled)
94
+ scheduleReconnect(runtime);
95
+ },
96
+ onSelectionChanged: (snapshot) => {
97
+ if (runtime.connection !== connection)
98
+ return;
99
+ (0, context_1.setLatestSelection)(runtime, snapshot);
100
+ },
101
+ onSelectionCleared: () => {
102
+ if (runtime.connection !== connection)
103
+ return;
104
+ (0, context_1.clearLatestSelection)(runtime);
105
+ },
106
+ onAtMentioned: (params) => {
107
+ if (runtime.connection !== connection)
108
+ return;
109
+ handleAtMentioned(runtime, params);
110
+ },
111
+ onError: (error) => {
112
+ if (runtime.connection !== connection)
113
+ return;
114
+ runtime.connectionStatus = "error";
115
+ runtime.connectionMessage = error.message;
116
+ (0, ui_1.updateIdeUi)(runtime);
117
+ },
118
+ });
119
+ runtime.connection = connection;
120
+ try {
121
+ await connection.connect();
122
+ if (runtime.connection === connection && runtime.connectionStatus === "connecting") {
123
+ runtime.connectionStatus = "connected";
124
+ runtime.connectionMessage = undefined;
125
+ (0, ui_1.updateIdeUi)(runtime, ctx);
126
+ }
127
+ }
128
+ catch (error) {
129
+ if (runtime.connection === connection) {
130
+ runtime.connection = undefined;
131
+ runtime.connectionStatus = "error";
132
+ runtime.connectionMessage = error instanceof Error ? error.message : String(error);
133
+ (0, ui_1.updateIdeUi)(runtime, ctx);
134
+ scheduleReconnect(runtime);
135
+ }
136
+ }
137
+ }
138
+ function disconnect(runtime, ctx, disabled = false) {
139
+ runtime.ctx = ctx;
140
+ if (runtime.reconnectTimer)
141
+ clearTimeout(runtime.reconnectTimer);
142
+ runtime.reconnectTimer = undefined;
143
+ const connection = runtime.connection;
144
+ runtime.connection = undefined;
145
+ connection?.disconnect();
146
+ runtime.enabled = !disabled;
147
+ runtime.connectedServer = undefined;
148
+ runtime.connectionStatus = disabled ? "disabled" : "disconnected";
149
+ runtime.connectionMessage = disabled ? "IDE integration disabled." : "Disconnected.";
150
+ if (disabled) {
151
+ runtime.latestSelection = undefined;
152
+ runtime.latestSelectionKey = undefined;
153
+ runtime.turnSelection = undefined;
154
+ runtime.attachState = "idle";
155
+ }
156
+ (0, ui_1.updateIdeUi)(runtime, ctx);
157
+ }
158
+ function scheduleReconnect(runtime) {
159
+ if (runtime.reconnectTimer || !runtime.enabled)
160
+ return;
161
+ runtime.reconnectTimer = setTimeout(() => {
162
+ runtime.reconnectTimer = undefined;
163
+ const ctx = runtime.ctx;
164
+ if (!ctx || !runtime.enabled)
165
+ return;
166
+ connectAuto(runtime, ctx).catch((error) => {
167
+ runtime.connectionStatus = "error";
168
+ runtime.connectionMessage = error instanceof Error ? error.message : String(error);
169
+ (0, ui_1.updateIdeUi)(runtime);
170
+ });
171
+ }, RECONNECT_DELAY_MS);
172
+ }
173
+ function handleAtMentioned(runtime, params) {
174
+ (0, context_1.setLatestSelection)(runtime, params);
175
+ const ctx = runtime.ctx;
176
+ if (!ctx?.hasUI)
177
+ return;
178
+ const text = params.rangeText || (0, format_1.formatRangeMention)(params, { cwd: ctx.cwd });
179
+ ctx.ui.pasteToEditor(text);
180
+ ctx.ui.notify(`Attached ${text}`, "info");
181
+ }
182
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/pi/index.ts"],"names":[],"mappings":";;AAcA,4BA8BC;AAzCD,6CAAsD;AAEtD,2CAAoD;AACpD,6CAA6C;AAC7C,yCAAgD;AAChD,uCAA8F;AAC9F,mCAA2D;AAC3D,6BAA+C;AAE/C,MAAM,kBAAkB,GAAG,KAAK,CAAC;AAEjC,mBAAyB,EAAgB;IACvC,MAAM,OAAO,GAAG,IAAA,qBAAa,GAAE,CAAC;IAEhC,IAAA,iCAAuB,EAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACrC,IAAA,6BAAkB,EAAC,EAAE,EAAE,OAAO,EAAE;QAC9B,iBAAiB,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE,GAAG,CAAC;QAC3D,WAAW,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC;QAC/C,gBAAgB,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC;QAC/E,UAAU,EAAE,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC;KAClE,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC3C,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;QAClB,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,CAAC,gBAAgB,GAAG,UAAU,CAAC;YACtC,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,OAAO;QACT,CAAC;QACD,MAAM,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;QACxC,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;QAClB,IAAI,OAAO,CAAC,cAAc;YAAE,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACjE,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;QACnC,OAAO,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC;QACjC,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;QAC/B,IAAA,eAAU,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC9B,OAAqB,EACrB,GAA+C;IAE/C,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;IAClB,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IACtB,OAAO,CAAC,UAAU,GAAG,MAAM,IAAA,iCAAqB,EAAC,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;IACnE,OAAO,OAAO,CAAC,UAAU,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,OAAqB,EAAE,GAA+C;IAC/F,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;IACvB,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACzD,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,CAAC,gBAAgB,GAAG,cAAc,CAAC;QAC1C,OAAO,CAAC,iBAAiB,GAAG,mCAAmC,CAAC;QAChE,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;QACrC,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;QACpC,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC1B,OAAO;IACT,CAAC;IACD,MAAM,gBAAgB,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;AAClD,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,OAAqB,EACrB,SAA4B,EAC5B,GAA+C;IAE/C,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;IAClB,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IACtB,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;IACvB,IAAI,OAAO,CAAC,cAAc;QAAE,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACjE,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAEnC,MAAM,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACpC,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAC/B,QAAQ,EAAE,UAAU,EAAE,CAAC;IAEvB,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACrC,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;IACpC,OAAO,CAAC,gBAAgB,GAAG,YAAY,CAAC;IACxC,OAAO,CAAC,iBAAiB,GAAG,iBAAiB,SAAS,CAAC,IAAI,CAAC,IAAI,OAAO,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IACpH,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAE1B,MAAM,UAAU,GAAG,IAAI,0BAAa,CAAC,SAAS,EAAE,GAAG,CAAC,GAAG,EAAE;QACvD,WAAW,EAAE,CAAC,MAAM,EAAE,EAAE;YACtB,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC;YACjC,OAAO,CAAC,gBAAgB,GAAG,WAAW,CAAC;YACvC,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACtC,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;QACvB,CAAC;QACD,cAAc,EAAE,CAAC,MAAM,EAAE,EAAE;YACzB,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YAC/B,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;YACpC,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC;YACzE,OAAO,CAAC,iBAAiB,GAAG,MAAM,CAAC;YACnC,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;YACrB,IAAI,OAAO,CAAC,OAAO;gBAAE,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,kBAAkB,EAAE,CAAC,QAAQ,EAAE,EAAE;YAC/B,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,IAAA,4BAAkB,EAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACxC,CAAC;QACD,kBAAkB,EAAE,GAAG,EAAE;YACvB,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,IAAA,8BAAoB,EAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,aAAa,EAAE,CAAC,MAAM,EAAE,EAAE;YACxB,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,iBAAiB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;gBAAE,OAAO;YAC9C,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACnC,OAAO,CAAC,iBAAiB,GAAG,KAAK,CAAC,OAAO,CAAC;YAC1C,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;QACvB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,CAAC,UAAU,GAAG,UAAU,CAAC;IAChC,IAAI,CAAC;QACH,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC;QAC3B,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,IAAI,OAAO,CAAC,gBAAgB,KAAK,YAAY,EAAE,CAAC;YACnF,OAAO,CAAC,gBAAgB,GAAG,WAAW,CAAC;YACvC,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACtC,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;YACtC,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;YAC/B,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACnC,OAAO,CAAC,iBAAiB,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnF,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC1B,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,OAAqB,EAAE,GAA+C,EAAE,QAAQ,GAAG,KAAK;IAC1G,OAAO,CAAC,GAAG,GAAG,GAAG,CAAC;IAClB,IAAI,OAAO,CAAC,cAAc;QAAE,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACjE,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IACnC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACtC,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAC/B,UAAU,EAAE,UAAU,EAAE,CAAC;IACzB,OAAO,CAAC,OAAO,GAAG,CAAC,QAAQ,CAAC;IAC5B,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;IACpC,OAAO,CAAC,gBAAgB,GAAG,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC;IAClE,OAAO,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,eAAe,CAAC;IACrF,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;QACpC,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;QACvC,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;QAClC,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC;IAC/B,CAAC;IACD,IAAA,gBAAW,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAqB;IAC9C,IAAI,OAAO,CAAC,cAAc,IAAI,CAAC,OAAO,CAAC,OAAO;QAAE,OAAO;IACvD,OAAO,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;QACvC,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;QACnC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACxB,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO;QACrC,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACjD,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC;YACnC,OAAO,CAAC,iBAAiB,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnF,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,kBAAkB,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAqB,EAAE,MAAyB;IACzE,IAAA,4BAAkB,EAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACpC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IACxB,IAAI,CAAC,GAAG,EAAE,KAAK;QAAE,OAAO;IACxB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,IAAI,IAAA,2BAAkB,EAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;IAC9E,GAAG,CAAC,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC3B,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;AAC5C,CAAC"}
@@ -0,0 +1,24 @@
1
+ import type { ExtensionContext } from "@earendil-works/pi-coding-agent" with { "resolution-mode": "import" };
2
+ import type { AttachState, EditorSelectionSnapshot, LockFileCandidate } from "../shared/protocol";
3
+ import type { IdeConnection } from "./connection";
4
+ export interface PiIdeRuntime {
5
+ ctx?: ExtensionContext;
6
+ cwd?: string;
7
+ enabled: boolean;
8
+ connection?: IdeConnection;
9
+ currentCandidate?: LockFileCandidate;
10
+ candidates: LockFileCandidate[];
11
+ connectedServer?: {
12
+ name: string;
13
+ version?: string;
14
+ ide?: string;
15
+ };
16
+ connectionStatus: "idle" | "connecting" | "connected" | "disconnected" | "error" | "disabled";
17
+ connectionMessage?: string;
18
+ latestSelection?: EditorSelectionSnapshot;
19
+ latestSelectionKey?: string;
20
+ attachState: AttachState;
21
+ turnSelection?: EditorSelectionSnapshot;
22
+ reconnectTimer?: NodeJS.Timeout;
23
+ }
24
+ export declare function createRuntime(): PiIdeRuntime;
@@ -0,0 +1,12 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.createRuntime = createRuntime;
4
+ function createRuntime() {
5
+ return {
6
+ enabled: true,
7
+ candidates: [],
8
+ connectionStatus: "idle",
9
+ attachState: "idle",
10
+ };
11
+ }
12
+ //# sourceMappingURL=state.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"state.js","sourceRoot":"","sources":["../../../src/pi/state.ts"],"names":[],"mappings":";;AAqBA,sCAOC;AAPD,SAAgB,aAAa;IAC3B,OAAO;QACL,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,EAAE;QACd,gBAAgB,EAAE,MAAM;QACxB,WAAW,EAAE,MAAM;KACpB,CAAC;AACJ,CAAC"}
@@ -0,0 +1,6 @@
1
+ import type { ExtensionContext } from "@earendil-works/pi-coding-agent" with { "resolution-mode": "import" };
2
+ import type { PiIdeRuntime } from "./state";
3
+ export declare function updateIdeUi(runtime: PiIdeRuntime, ctx?: ExtensionContext | undefined): void;
4
+ export declare function clearIdeUi(runtime: PiIdeRuntime, ctx?: ExtensionContext | undefined): void;
5
+ export declare function buildStatusLine(runtime: PiIdeRuntime, cwd?: string): string;
6
+ export declare function buildWidget(runtime: PiIdeRuntime, cwd?: string): string[] | undefined;
@@ -0,0 +1,84 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.updateIdeUi = updateIdeUi;
4
+ exports.clearIdeUi = clearIdeUi;
5
+ exports.buildStatusLine = buildStatusLine;
6
+ exports.buildWidget = buildWidget;
7
+ const format_1 = require("../shared/format");
8
+ const paths_1 = require("../shared/paths");
9
+ function updateIdeUi(runtime, ctx = runtime.ctx) {
10
+ if (!ctx?.hasUI)
11
+ return;
12
+ ctx.ui.setWidget("pi-x-ide", (_tui, theme) => ({
13
+ render(width) {
14
+ const status = buildStatusLine(runtime, ctx.cwd);
15
+ const text = truncatePlainStatus(status, width);
16
+ const pad = " ".repeat(Math.max(0, width - text.length));
17
+ const color = isPendingEditorContext(runtime) ? "success" : "dim";
18
+ return [pad + theme.fg(color, text)];
19
+ },
20
+ invalidate() { },
21
+ }), { placement: "aboveEditor" });
22
+ }
23
+ function clearIdeUi(runtime, ctx = runtime.ctx) {
24
+ if (!ctx?.hasUI)
25
+ return;
26
+ ctx.ui.setWidget("pi-x-ide", undefined);
27
+ }
28
+ function truncatePlainStatus(text, width) {
29
+ if (width <= 0)
30
+ return "";
31
+ if (text.length <= width)
32
+ return text;
33
+ if (width <= 3)
34
+ return ".".repeat(width);
35
+ return `${text.slice(0, width - 3)}...`;
36
+ }
37
+ function isPendingEditorContext(runtime) {
38
+ return !!runtime.latestSelection && runtime.attachState === "pending";
39
+ }
40
+ function buildStatusLine(runtime, cwd) {
41
+ if (!runtime.enabled || runtime.connectionStatus === "disabled")
42
+ return "IDE: off";
43
+ if (runtime.connectionStatus === "connecting")
44
+ return "IDE: connecting";
45
+ if (runtime.connectionStatus === "error")
46
+ return `IDE: error${runtime.connectionMessage ? ` ${runtime.connectionMessage}` : ""}`;
47
+ if (runtime.connectionStatus !== "connected")
48
+ return "IDE: disconnected";
49
+ const ide = runtime.connectedServer?.ide ?? runtime.currentCandidate?.lock.ide ?? "ide";
50
+ const selection = runtime.latestSelection;
51
+ if (!selection)
52
+ return `IDE: ${ide} ✓`;
53
+ const rel = (0, paths_1.toRelativeDisplayPath)(selection.filePath, selection.workspaceFolder, cwd);
54
+ const range = (0, format_1.describeRanges)(selection.ranges);
55
+ return `IDE: ${ide} ✓ ${rel}${range === "open file" ? "" : range} ${runtime.attachState}`;
56
+ }
57
+ function buildWidget(runtime, cwd) {
58
+ if (!runtime.enabled || runtime.connectionStatus === "disabled")
59
+ return undefined;
60
+ if (runtime.connectionStatus !== "connected" &&
61
+ runtime.connectionStatus !== "connecting" &&
62
+ runtime.connectionStatus !== "error") {
63
+ return undefined;
64
+ }
65
+ const lines = [];
66
+ const ide = runtime.connectedServer?.name ?? runtime.currentCandidate?.lock.name ?? "IDE";
67
+ lines.push(`IDE: ${ide} (${runtime.connectionStatus})`);
68
+ if (runtime.currentCandidate) {
69
+ lines.push(`Workspace: ${runtime.currentCandidate.workspaceFolder}`);
70
+ }
71
+ if (runtime.latestSelection) {
72
+ const selection = runtime.latestSelection;
73
+ lines.push(`File: ${(0, paths_1.toRelativeDisplayPath)(selection.filePath, selection.workspaceFolder, cwd)}`);
74
+ lines.push(`Range: ${(0, format_1.describeRanges)(selection.ranges)}`);
75
+ lines.push(`Attach: ${runtime.attachState}`);
76
+ if (selection.receivedAt)
77
+ lines.push(`Updated: ${new Date(selection.receivedAt).toLocaleTimeString()}`);
78
+ }
79
+ else if (runtime.connectionMessage) {
80
+ lines.push(runtime.connectionMessage);
81
+ }
82
+ return lines;
83
+ }
84
+ //# sourceMappingURL=ui.js.map