@playwo/opencode-cursor-oauth 0.0.0-dev.1b946f85e9b0 → 0.0.0-dev.2a59bf1639ea

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 (72) hide show
  1. package/README.md +19 -91
  2. package/dist/auth.js +1 -2
  3. package/dist/constants.d.ts +2 -0
  4. package/dist/constants.js +2 -0
  5. package/dist/cursor/bidi-session.d.ts +13 -0
  6. package/dist/cursor/bidi-session.js +149 -0
  7. package/dist/cursor/config.d.ts +4 -0
  8. package/dist/cursor/config.js +4 -0
  9. package/dist/cursor/connect-framing.d.ts +10 -0
  10. package/dist/cursor/connect-framing.js +80 -0
  11. package/dist/cursor/headers.d.ts +6 -0
  12. package/dist/cursor/headers.js +16 -0
  13. package/dist/cursor/index.d.ts +5 -0
  14. package/dist/cursor/index.js +5 -0
  15. package/dist/cursor/unary-rpc.d.ts +13 -0
  16. package/dist/cursor/unary-rpc.js +181 -0
  17. package/dist/index.d.ts +2 -14
  18. package/dist/index.js +2 -297
  19. package/dist/logger.js +7 -2
  20. package/dist/models.js +1 -23
  21. package/dist/openai/index.d.ts +3 -0
  22. package/dist/openai/index.js +3 -0
  23. package/dist/openai/messages.d.ts +39 -0
  24. package/dist/openai/messages.js +223 -0
  25. package/dist/openai/tools.d.ts +7 -0
  26. package/dist/openai/tools.js +58 -0
  27. package/dist/openai/types.d.ts +41 -0
  28. package/dist/openai/types.js +1 -0
  29. package/dist/plugin/cursor-auth-plugin.d.ts +3 -0
  30. package/dist/plugin/cursor-auth-plugin.js +140 -0
  31. package/dist/proto/agent_pb.js +637 -319
  32. package/dist/provider/index.d.ts +2 -0
  33. package/dist/provider/index.js +2 -0
  34. package/dist/provider/model-cost.d.ts +9 -0
  35. package/dist/provider/model-cost.js +206 -0
  36. package/dist/provider/models.d.ts +8 -0
  37. package/dist/provider/models.js +86 -0
  38. package/dist/proxy/bridge-non-streaming.d.ts +3 -0
  39. package/dist/proxy/bridge-non-streaming.js +119 -0
  40. package/dist/proxy/bridge-session.d.ts +5 -0
  41. package/dist/proxy/bridge-session.js +13 -0
  42. package/dist/proxy/bridge-streaming.d.ts +5 -0
  43. package/dist/proxy/bridge-streaming.js +311 -0
  44. package/dist/proxy/bridge.d.ts +3 -0
  45. package/dist/proxy/bridge.js +3 -0
  46. package/dist/proxy/chat-completion.d.ts +2 -0
  47. package/dist/proxy/chat-completion.js +113 -0
  48. package/dist/proxy/conversation-meta.d.ts +12 -0
  49. package/dist/proxy/conversation-meta.js +1 -0
  50. package/dist/proxy/conversation-state.d.ts +35 -0
  51. package/dist/proxy/conversation-state.js +95 -0
  52. package/dist/proxy/cursor-request.d.ts +5 -0
  53. package/dist/proxy/cursor-request.js +86 -0
  54. package/dist/proxy/index.d.ts +12 -0
  55. package/dist/proxy/index.js +12 -0
  56. package/dist/proxy/server.d.ts +6 -0
  57. package/dist/proxy/server.js +89 -0
  58. package/dist/proxy/sse.d.ts +5 -0
  59. package/dist/proxy/sse.js +5 -0
  60. package/dist/proxy/state-sync.d.ts +2 -0
  61. package/dist/proxy/state-sync.js +17 -0
  62. package/dist/proxy/stream-dispatch.d.ts +42 -0
  63. package/dist/proxy/stream-dispatch.js +491 -0
  64. package/dist/proxy/stream-state.d.ts +9 -0
  65. package/dist/proxy/stream-state.js +1 -0
  66. package/dist/proxy/title.d.ts +1 -0
  67. package/dist/proxy/title.js +103 -0
  68. package/dist/proxy/types.d.ts +27 -0
  69. package/dist/proxy/types.js +1 -0
  70. package/dist/proxy.d.ts +2 -20
  71. package/dist/proxy.js +2 -1385
  72. package/package.json +1 -1
@@ -0,0 +1,491 @@
1
+ import { create, toBinary } from "@bufbuild/protobuf";
2
+ import { AgentClientMessageSchema, ClientHeartbeatSchema, ConversationStateStructureSchema, BackgroundShellSpawnResultSchema, DeleteResultSchema, DeleteRejectedSchema, DiagnosticsResultSchema, ExecClientMessageSchema, FetchErrorSchema, FetchResultSchema, GetBlobResultSchema, GrepErrorSchema, GrepResultSchema, KvClientMessageSchema, LsRejectedSchema, LsResultSchema, McpResultSchema, ReadRejectedSchema, ReadResultSchema, RequestContextResultSchema, RequestContextSchema, RequestContextSuccessSchema, SetBlobResultSchema, ShellRejectedSchema, ShellResultSchema, WriteRejectedSchema, WriteResultSchema, WriteShellStdinErrorSchema, WriteShellStdinResultSchema, } from "../proto/agent_pb";
3
+ import { CONNECT_END_STREAM_FLAG } from "../cursor/config";
4
+ import { logPluginError, logPluginWarn } from "../logger";
5
+ import { decodeMcpArgsMap } from "../openai/tools";
6
+ export function parseConnectEndStream(data) {
7
+ try {
8
+ const payload = JSON.parse(new TextDecoder().decode(data));
9
+ const error = payload?.error;
10
+ if (error) {
11
+ const code = error.code ?? "unknown";
12
+ const message = error.message ?? "Unknown error";
13
+ return new Error(`Connect error ${code}: ${message}`);
14
+ }
15
+ return null;
16
+ }
17
+ catch {
18
+ return new Error("Failed to parse Connect end stream");
19
+ }
20
+ }
21
+ export function makeHeartbeatBytes() {
22
+ const heartbeat = create(AgentClientMessageSchema, {
23
+ message: {
24
+ case: "clientHeartbeat",
25
+ value: create(ClientHeartbeatSchema, {}),
26
+ },
27
+ });
28
+ return toBinary(AgentClientMessageSchema, heartbeat);
29
+ }
30
+ export function scheduleBridgeEnd(bridge) {
31
+ queueMicrotask(() => {
32
+ if (bridge.alive)
33
+ bridge.end();
34
+ });
35
+ }
36
+ /**
37
+ * Create a stateful parser for Connect protocol frames.
38
+ * Handles buffering partial data across chunks.
39
+ */
40
+ export function createConnectFrameParser(onMessage, onEndStream) {
41
+ let pending = Buffer.alloc(0);
42
+ return (incoming) => {
43
+ pending = Buffer.concat([pending, incoming]);
44
+ while (pending.length >= 5) {
45
+ const flags = pending[0];
46
+ const msgLen = pending.readUInt32BE(1);
47
+ if (pending.length < 5 + msgLen)
48
+ break;
49
+ const messageBytes = pending.subarray(5, 5 + msgLen);
50
+ pending = pending.subarray(5 + msgLen);
51
+ if (flags & CONNECT_END_STREAM_FLAG) {
52
+ onEndStream(messageBytes);
53
+ }
54
+ else {
55
+ onMessage(messageBytes);
56
+ }
57
+ }
58
+ };
59
+ }
60
+ const THINKING_TAG_NAMES = [
61
+ "think",
62
+ "thinking",
63
+ "reasoning",
64
+ "thought",
65
+ "think_intent",
66
+ ];
67
+ const MAX_THINKING_TAG_LEN = 16; // </think_intent> is 15 chars
68
+ /**
69
+ * Strip thinking tags from streamed text, routing tagged content to reasoning.
70
+ * Buffers partial tags across chunk boundaries.
71
+ */
72
+ export function createThinkingTagFilter() {
73
+ let buffer = "";
74
+ let inThinking = false;
75
+ return {
76
+ process(text) {
77
+ const input = buffer + text;
78
+ buffer = "";
79
+ let content = "";
80
+ let reasoning = "";
81
+ let lastIdx = 0;
82
+ const re = new RegExp(`<(/?)(?:${THINKING_TAG_NAMES.join("|")})\\s*>`, "gi");
83
+ let match;
84
+ while ((match = re.exec(input)) !== null) {
85
+ const before = input.slice(lastIdx, match.index);
86
+ if (inThinking)
87
+ reasoning += before;
88
+ else
89
+ content += before;
90
+ inThinking = match[1] !== "/";
91
+ lastIdx = re.lastIndex;
92
+ }
93
+ const rest = input.slice(lastIdx);
94
+ // Buffer a trailing '<' that could be the start of a thinking tag.
95
+ const ltPos = rest.lastIndexOf("<");
96
+ if (ltPos >= 0 &&
97
+ rest.length - ltPos < MAX_THINKING_TAG_LEN &&
98
+ /^<\/?[a-z_]*$/i.test(rest.slice(ltPos))) {
99
+ buffer = rest.slice(ltPos);
100
+ const before = rest.slice(0, ltPos);
101
+ if (inThinking)
102
+ reasoning += before;
103
+ else
104
+ content += before;
105
+ }
106
+ else {
107
+ if (inThinking)
108
+ reasoning += rest;
109
+ else
110
+ content += rest;
111
+ }
112
+ return { content, reasoning };
113
+ },
114
+ flush() {
115
+ const b = buffer;
116
+ buffer = "";
117
+ if (!b)
118
+ return { content: "", reasoning: "" };
119
+ return inThinking
120
+ ? { content: "", reasoning: b }
121
+ : { content: b, reasoning: "" };
122
+ },
123
+ };
124
+ }
125
+ export function computeUsage(state) {
126
+ const completion_tokens = state.outputTokens;
127
+ const total_tokens = state.totalTokens || completion_tokens;
128
+ const prompt_tokens = Math.max(0, total_tokens - completion_tokens);
129
+ return { prompt_tokens, completion_tokens, total_tokens };
130
+ }
131
+ export function processServerMessage(msg, blobStore, mcpTools, sendFrame, state, onText, onMcpExec, onCheckpoint, onTurnEnded, onUnsupportedMessage, onUnhandledExec) {
132
+ const msgCase = msg.message.case;
133
+ if (msgCase === "interactionUpdate") {
134
+ handleInteractionUpdate(msg.message.value, state, onText, onMcpExec, onTurnEnded, onUnsupportedMessage);
135
+ }
136
+ else if (msgCase === "kvServerMessage") {
137
+ handleKvMessage(msg.message.value, blobStore, sendFrame);
138
+ }
139
+ else if (msgCase === "execServerMessage") {
140
+ handleExecMessage(msg.message.value, mcpTools, sendFrame, onMcpExec, onUnhandledExec);
141
+ }
142
+ else if (msgCase === "execServerControlMessage") {
143
+ onUnsupportedMessage?.({
144
+ category: "execServerControl",
145
+ caseName: msg.message.value.message.case ?? "undefined",
146
+ });
147
+ }
148
+ else if (msgCase === "interactionQuery") {
149
+ onUnsupportedMessage?.({
150
+ category: "interactionQuery",
151
+ caseName: msg.message.value.query.case ?? "undefined",
152
+ });
153
+ }
154
+ else if (msgCase === "conversationCheckpointUpdate") {
155
+ const stateStructure = msg.message.value;
156
+ if (stateStructure.tokenDetails) {
157
+ state.totalTokens = stateStructure.tokenDetails.usedTokens;
158
+ }
159
+ if (onCheckpoint) {
160
+ onCheckpoint(toBinary(ConversationStateStructureSchema, stateStructure));
161
+ }
162
+ }
163
+ else {
164
+ onUnsupportedMessage?.({
165
+ category: "agentMessage",
166
+ caseName: msgCase ?? "undefined",
167
+ });
168
+ }
169
+ }
170
+ function handleInteractionUpdate(update, state, onText, onMcpExec, onTurnEnded, onUnsupportedMessage) {
171
+ const updateCase = update.message?.case;
172
+ if (updateCase === "textDelta") {
173
+ const delta = update.message.value.text || "";
174
+ if (delta)
175
+ onText(delta, false);
176
+ }
177
+ else if (updateCase === "thinkingDelta") {
178
+ const delta = update.message.value.text || "";
179
+ if (delta)
180
+ onText(delta, true);
181
+ }
182
+ else if (updateCase === "tokenDelta") {
183
+ state.outputTokens += update.message.value.tokens ?? 0;
184
+ }
185
+ else if (updateCase === "partialToolCall") {
186
+ const partial = update.message.value;
187
+ if (partial.callId && partial.argsTextDelta) {
188
+ state.interactionToolArgsText.set(partial.callId, partial.argsTextDelta);
189
+ }
190
+ }
191
+ else if (updateCase === "toolCallCompleted") {
192
+ const exec = decodeInteractionToolCall(update.message.value, state);
193
+ if (exec)
194
+ onMcpExec(exec);
195
+ else {
196
+ onUnsupportedMessage?.({
197
+ category: "toolCall",
198
+ caseName: update.message.value?.toolCall?.tool?.case ?? "undefined",
199
+ detail: "toolCallCompleted",
200
+ });
201
+ }
202
+ }
203
+ else if (updateCase === "turnEnded") {
204
+ onTurnEnded?.();
205
+ }
206
+ else if (updateCase === "toolCallStarted" ||
207
+ updateCase === "toolCallDelta" ||
208
+ updateCase === "thinkingCompleted" ||
209
+ updateCase === "userMessageAppended" ||
210
+ updateCase === "summary" ||
211
+ updateCase === "summaryStarted" ||
212
+ updateCase === "summaryCompleted" ||
213
+ updateCase === "heartbeat" ||
214
+ updateCase === "stepStarted" ||
215
+ updateCase === "stepCompleted") {
216
+ return;
217
+ }
218
+ else {
219
+ onUnsupportedMessage?.({
220
+ category: "interactionUpdate",
221
+ caseName: updateCase ?? "undefined",
222
+ });
223
+ }
224
+ // toolCallStarted, partialToolCall, toolCallDelta, toolCallCompleted
225
+ // are intentionally ignored. MCP tool calls flow through the exec
226
+ // message path (mcpArgs → mcpResult), not interaction updates.
227
+ }
228
+ function decodeInteractionToolCall(update, state) {
229
+ const callId = update.callId ?? "";
230
+ const toolCase = update.toolCall?.tool?.case;
231
+ if (toolCase !== "mcpToolCall")
232
+ return null;
233
+ const mcpArgs = update.toolCall?.tool?.value?.args;
234
+ if (!mcpArgs)
235
+ return null;
236
+ const toolCallId = mcpArgs.toolCallId || callId || crypto.randomUUID();
237
+ if (state.emittedToolCallIds.has(toolCallId))
238
+ return null;
239
+ const decodedMap = decodeMcpArgsMap(mcpArgs.args ?? {});
240
+ const partialArgsText = callId
241
+ ? state.interactionToolArgsText.get(callId)?.trim()
242
+ : undefined;
243
+ let decodedArgs = "{}";
244
+ if (Object.keys(decodedMap).length > 0) {
245
+ decodedArgs = JSON.stringify(decodedMap);
246
+ }
247
+ else if (partialArgsText) {
248
+ decodedArgs = partialArgsText;
249
+ }
250
+ state.emittedToolCallIds.add(toolCallId);
251
+ if (callId)
252
+ state.interactionToolArgsText.delete(callId);
253
+ return {
254
+ execId: callId || toolCallId,
255
+ execMsgId: 0,
256
+ toolCallId,
257
+ toolName: mcpArgs.toolName || mcpArgs.name || "unknown_mcp_tool",
258
+ decodedArgs,
259
+ };
260
+ }
261
+ /** Send a KV client response back to Cursor. */
262
+ function sendKvResponse(kvMsg, messageCase, value, sendFrame) {
263
+ const response = create(KvClientMessageSchema, {
264
+ id: kvMsg.id,
265
+ message: { case: messageCase, value: value },
266
+ });
267
+ const clientMsg = create(AgentClientMessageSchema, {
268
+ message: { case: "kvClientMessage", value: response },
269
+ });
270
+ sendFrame(toBinary(AgentClientMessageSchema, clientMsg));
271
+ }
272
+ function handleKvMessage(kvMsg, blobStore, sendFrame) {
273
+ const kvCase = kvMsg.message.case;
274
+ if (kvCase === "getBlobArgs") {
275
+ const blobId = kvMsg.message.value.blobId;
276
+ const blobIdKey = Buffer.from(blobId).toString("hex");
277
+ const blobData = blobStore.get(blobIdKey);
278
+ if (!blobData) {
279
+ logPluginWarn("Cursor requested missing blob", {
280
+ blobId: blobIdKey,
281
+ knownBlobCount: blobStore.size,
282
+ });
283
+ }
284
+ sendKvResponse(kvMsg, "getBlobResult", create(GetBlobResultSchema, blobData ? { blobData } : {}), sendFrame);
285
+ }
286
+ else if (kvCase === "setBlobArgs") {
287
+ const { blobId, blobData } = kvMsg.message.value;
288
+ blobStore.set(Buffer.from(blobId).toString("hex"), blobData);
289
+ sendKvResponse(kvMsg, "setBlobResult", create(SetBlobResultSchema, {}), sendFrame);
290
+ }
291
+ }
292
+ function handleExecMessage(execMsg, mcpTools, sendFrame, onMcpExec, onUnhandledExec) {
293
+ const execCase = execMsg.message.case;
294
+ if (execCase === "requestContextArgs") {
295
+ const requestContext = create(RequestContextSchema, {
296
+ rules: [],
297
+ repositoryInfo: [],
298
+ tools: mcpTools,
299
+ gitRepos: [],
300
+ projectLayouts: [],
301
+ mcpInstructions: [],
302
+ fileContents: {},
303
+ customSubagents: [],
304
+ });
305
+ const result = create(RequestContextResultSchema, {
306
+ result: {
307
+ case: "success",
308
+ value: create(RequestContextSuccessSchema, { requestContext }),
309
+ },
310
+ });
311
+ sendExecResult(execMsg, "requestContextResult", result, sendFrame);
312
+ return;
313
+ }
314
+ if (execCase === "mcpArgs") {
315
+ const mcpArgs = execMsg.message.value;
316
+ const decoded = decodeMcpArgsMap(mcpArgs.args ?? {});
317
+ onMcpExec({
318
+ execId: execMsg.execId,
319
+ execMsgId: execMsg.id,
320
+ toolCallId: mcpArgs.toolCallId || crypto.randomUUID(),
321
+ toolName: mcpArgs.toolName || mcpArgs.name,
322
+ decodedArgs: JSON.stringify(decoded),
323
+ });
324
+ return;
325
+ }
326
+ // --- Reject native Cursor tools ---
327
+ // The model tries these first. We must respond with rejection/error
328
+ // so it falls back to our MCP tools (registered via RequestContext).
329
+ const REJECT_REASON = "Tool not available in this environment. Use the MCP tools provided instead.";
330
+ if (execCase === "readArgs") {
331
+ const args = execMsg.message.value;
332
+ const result = create(ReadResultSchema, {
333
+ result: {
334
+ case: "rejected",
335
+ value: create(ReadRejectedSchema, {
336
+ path: args.path,
337
+ reason: REJECT_REASON,
338
+ }),
339
+ },
340
+ });
341
+ sendExecResult(execMsg, "readResult", result, sendFrame);
342
+ return;
343
+ }
344
+ if (execCase === "lsArgs") {
345
+ const args = execMsg.message.value;
346
+ const result = create(LsResultSchema, {
347
+ result: {
348
+ case: "rejected",
349
+ value: create(LsRejectedSchema, {
350
+ path: args.path,
351
+ reason: REJECT_REASON,
352
+ }),
353
+ },
354
+ });
355
+ sendExecResult(execMsg, "lsResult", result, sendFrame);
356
+ return;
357
+ }
358
+ if (execCase === "grepArgs") {
359
+ const result = create(GrepResultSchema, {
360
+ result: {
361
+ case: "error",
362
+ value: create(GrepErrorSchema, { error: REJECT_REASON }),
363
+ },
364
+ });
365
+ sendExecResult(execMsg, "grepResult", result, sendFrame);
366
+ return;
367
+ }
368
+ if (execCase === "writeArgs") {
369
+ const args = execMsg.message.value;
370
+ const result = create(WriteResultSchema, {
371
+ result: {
372
+ case: "rejected",
373
+ value: create(WriteRejectedSchema, {
374
+ path: args.path,
375
+ reason: REJECT_REASON,
376
+ }),
377
+ },
378
+ });
379
+ sendExecResult(execMsg, "writeResult", result, sendFrame);
380
+ return;
381
+ }
382
+ if (execCase === "deleteArgs") {
383
+ const args = execMsg.message.value;
384
+ const result = create(DeleteResultSchema, {
385
+ result: {
386
+ case: "rejected",
387
+ value: create(DeleteRejectedSchema, {
388
+ path: args.path,
389
+ reason: REJECT_REASON,
390
+ }),
391
+ },
392
+ });
393
+ sendExecResult(execMsg, "deleteResult", result, sendFrame);
394
+ return;
395
+ }
396
+ if (execCase === "shellArgs" || execCase === "shellStreamArgs") {
397
+ const args = execMsg.message.value;
398
+ const result = create(ShellResultSchema, {
399
+ result: {
400
+ case: "rejected",
401
+ value: create(ShellRejectedSchema, {
402
+ command: args.command ?? "",
403
+ workingDirectory: args.workingDirectory ?? "",
404
+ reason: REJECT_REASON,
405
+ isReadonly: false,
406
+ }),
407
+ },
408
+ });
409
+ sendExecResult(execMsg, "shellResult", result, sendFrame);
410
+ return;
411
+ }
412
+ if (execCase === "backgroundShellSpawnArgs") {
413
+ const args = execMsg.message.value;
414
+ const result = create(BackgroundShellSpawnResultSchema, {
415
+ result: {
416
+ case: "rejected",
417
+ value: create(ShellRejectedSchema, {
418
+ command: args.command ?? "",
419
+ workingDirectory: args.workingDirectory ?? "",
420
+ reason: REJECT_REASON,
421
+ isReadonly: false,
422
+ }),
423
+ },
424
+ });
425
+ sendExecResult(execMsg, "backgroundShellSpawnResult", result, sendFrame);
426
+ return;
427
+ }
428
+ if (execCase === "writeShellStdinArgs") {
429
+ const result = create(WriteShellStdinResultSchema, {
430
+ result: {
431
+ case: "error",
432
+ value: create(WriteShellStdinErrorSchema, { error: REJECT_REASON }),
433
+ },
434
+ });
435
+ sendExecResult(execMsg, "writeShellStdinResult", result, sendFrame);
436
+ return;
437
+ }
438
+ if (execCase === "fetchArgs") {
439
+ const args = execMsg.message.value;
440
+ const result = create(FetchResultSchema, {
441
+ result: {
442
+ case: "error",
443
+ value: create(FetchErrorSchema, {
444
+ url: args.url ?? "",
445
+ error: REJECT_REASON,
446
+ }),
447
+ },
448
+ });
449
+ sendExecResult(execMsg, "fetchResult", result, sendFrame);
450
+ return;
451
+ }
452
+ if (execCase === "diagnosticsArgs") {
453
+ const result = create(DiagnosticsResultSchema, {});
454
+ sendExecResult(execMsg, "diagnosticsResult", result, sendFrame);
455
+ return;
456
+ }
457
+ // MCP resource/screen/computer exec types
458
+ const miscCaseMap = {
459
+ listMcpResourcesExecArgs: "listMcpResourcesExecResult",
460
+ readMcpResourceExecArgs: "readMcpResourceExecResult",
461
+ recordScreenArgs: "recordScreenResult",
462
+ computerUseArgs: "computerUseResult",
463
+ };
464
+ const resultCase = miscCaseMap[execCase];
465
+ if (resultCase) {
466
+ sendExecResult(execMsg, resultCase, create(McpResultSchema, {}), sendFrame);
467
+ return;
468
+ }
469
+ logPluginError("Unhandled Cursor exec type", {
470
+ execCase: execCase ?? "undefined",
471
+ execId: execMsg.execId,
472
+ execMsgId: execMsg.id,
473
+ });
474
+ onUnhandledExec?.({
475
+ execCase: execCase ?? "undefined",
476
+ execId: execMsg.execId,
477
+ execMsgId: execMsg.id,
478
+ });
479
+ }
480
+ /** Send an exec client message back to Cursor. */
481
+ function sendExecResult(execMsg, messageCase, value, sendFrame) {
482
+ const execClientMessage = create(ExecClientMessageSchema, {
483
+ id: execMsg.id,
484
+ execId: execMsg.execId,
485
+ message: { case: messageCase, value: value },
486
+ });
487
+ const clientMessage = create(AgentClientMessageSchema, {
488
+ message: { case: "execClientMessage", value: execClientMessage },
489
+ });
490
+ sendFrame(toBinary(AgentClientMessageSchema, clientMessage));
491
+ }
@@ -0,0 +1,9 @@
1
+ import type { PendingExec } from "./types";
2
+ export interface StreamState {
3
+ toolCallIndex: number;
4
+ pendingExecs: PendingExec[];
5
+ outputTokens: number;
6
+ totalTokens: number;
7
+ interactionToolArgsText: Map<string, string>;
8
+ emittedToolCallIds: Set<string>;
9
+ }
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1 @@
1
+ export declare function handleTitleGenerationRequest(sourceText: string, accessToken: string, modelId: string, stream: boolean): Promise<Response>;
@@ -0,0 +1,103 @@
1
+ import { create, fromBinary, toBinary } from "@bufbuild/protobuf";
2
+ import { NameAgentRequestSchema, NameAgentResponseSchema, } from "../proto/agent_pb";
3
+ import { callCursorUnaryRpc, decodeConnectUnaryBody } from "../cursor";
4
+ import { SSE_HEADERS } from "./sse";
5
+ function deriveFallbackTitle(text) {
6
+ const cleaned = text
7
+ .replace(/<[^>]+>/g, " ")
8
+ .replace(/\[[^\]]+\]/g, " ")
9
+ .replace(/[^\p{L}\p{N}'’\-\s]+/gu, " ")
10
+ .replace(/\s+/g, " ")
11
+ .trim();
12
+ if (!cleaned)
13
+ return "";
14
+ const words = cleaned.split(" ").filter(Boolean).slice(0, 6);
15
+ return finalizeTitle(words.map(titleCaseWord).join(" "));
16
+ }
17
+ function titleCaseWord(word) {
18
+ if (!word)
19
+ return word;
20
+ return word[0].toUpperCase() + word.slice(1);
21
+ }
22
+ function finalizeTitle(value) {
23
+ return value
24
+ .replace(/^#{1,6}\s*/, "")
25
+ .replace(/[.!?,:;]+$/g, "")
26
+ .replace(/\s+/g, " ")
27
+ .trim()
28
+ .slice(0, 80)
29
+ .trim();
30
+ }
31
+ function createBufferedSSETextResponse(modelId, text, usage) {
32
+ const completionId = `chatcmpl-${crypto.randomUUID().replace(/-/g, "").slice(0, 28)}`;
33
+ const created = Math.floor(Date.now() / 1000);
34
+ const payload = [
35
+ {
36
+ id: completionId,
37
+ object: "chat.completion.chunk",
38
+ created,
39
+ model: modelId,
40
+ choices: [{ index: 0, delta: { content: text }, finish_reason: null }],
41
+ },
42
+ {
43
+ id: completionId,
44
+ object: "chat.completion.chunk",
45
+ created,
46
+ model: modelId,
47
+ choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
48
+ },
49
+ {
50
+ id: completionId,
51
+ object: "chat.completion.chunk",
52
+ created,
53
+ model: modelId,
54
+ choices: [],
55
+ usage,
56
+ },
57
+ ]
58
+ .map((chunk) => `data: ${JSON.stringify(chunk)}\n\n`)
59
+ .join("") + "data: [DONE]\n\n";
60
+ return new Response(payload, { headers: SSE_HEADERS });
61
+ }
62
+ export async function handleTitleGenerationRequest(sourceText, accessToken, modelId, stream) {
63
+ const requestBody = toBinary(NameAgentRequestSchema, create(NameAgentRequestSchema, {
64
+ userMessage: sourceText,
65
+ }));
66
+ const response = await callCursorUnaryRpc({
67
+ accessToken,
68
+ rpcPath: "/agent.v1.AgentService/NameAgent",
69
+ requestBody,
70
+ timeoutMs: 5_000,
71
+ });
72
+ if (response.timedOut) {
73
+ throw new Error("Cursor title generation timed out");
74
+ }
75
+ if (response.exitCode !== 0) {
76
+ throw new Error(`Cursor title generation failed with HTTP ${response.exitCode}`);
77
+ }
78
+ const payload = decodeConnectUnaryBody(response.body) ?? response.body;
79
+ const decoded = fromBinary(NameAgentResponseSchema, payload);
80
+ const title = finalizeTitle(decoded.name) ||
81
+ deriveFallbackTitle(sourceText) ||
82
+ "Untitled Session";
83
+ const usage = { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 };
84
+ if (stream) {
85
+ return createBufferedSSETextResponse(modelId, title, usage);
86
+ }
87
+ const completionId = `chatcmpl-${crypto.randomUUID().replace(/-/g, "").slice(0, 28)}`;
88
+ const created = Math.floor(Date.now() / 1000);
89
+ return new Response(JSON.stringify({
90
+ id: completionId,
91
+ object: "chat.completion",
92
+ created,
93
+ model: modelId,
94
+ choices: [
95
+ {
96
+ index: 0,
97
+ message: { role: "assistant", content: title },
98
+ finish_reason: "stop",
99
+ },
100
+ ],
101
+ usage,
102
+ }), { headers: { "Content-Type": "application/json" } });
103
+ }
@@ -0,0 +1,27 @@
1
+ import type { CursorSession } from "../cursor/bidi-session";
2
+ import type { ConversationRequestMetadata } from "./conversation-meta";
3
+ import type { McpToolDefinition } from "../proto/agent_pb";
4
+ export interface CursorRequestPayload {
5
+ requestBytes: Uint8Array;
6
+ blobStore: Map<string, Uint8Array>;
7
+ mcpTools: McpToolDefinition[];
8
+ }
9
+ /** A pending tool execution waiting for results from the caller. */
10
+ export interface PendingExec {
11
+ execId: string;
12
+ execMsgId: number;
13
+ toolCallId: string;
14
+ toolName: string;
15
+ /** Decoded arguments JSON string for SSE tool_calls emission. */
16
+ decodedArgs: string;
17
+ }
18
+ /** A live Cursor session kept alive across requests for tool result continuation. */
19
+ export interface ActiveBridge {
20
+ bridge: CursorSession;
21
+ heartbeatTimer: NodeJS.Timeout;
22
+ blobStore: Map<string, Uint8Array>;
23
+ mcpTools: McpToolDefinition[];
24
+ pendingExecs: PendingExec[];
25
+ modelId: string;
26
+ metadata: ConversationRequestMetadata;
27
+ }
@@ -0,0 +1 @@
1
+ export {};
package/dist/proxy.d.ts CHANGED
@@ -1,20 +1,2 @@
1
- interface CursorUnaryRpcOptions {
2
- accessToken: string;
3
- rpcPath: string;
4
- requestBody: Uint8Array;
5
- url?: string;
6
- timeoutMs?: number;
7
- transport?: "auto" | "fetch" | "http2";
8
- }
9
- export declare function callCursorUnaryRpc(options: CursorUnaryRpcOptions): Promise<{
10
- body: Uint8Array;
11
- exitCode: number;
12
- timedOut: boolean;
13
- }>;
14
- export declare function getProxyPort(): number | undefined;
15
- export declare function startProxy(getAccessToken: () => Promise<string>, models?: ReadonlyArray<{
16
- id: string;
17
- name: string;
18
- }>): Promise<number>;
19
- export declare function stopProxy(): void;
20
- export {};
1
+ export { getProxyPort, startProxy, stopProxy } from "./proxy/index";
2
+ export { callCursorUnaryRpc } from "./cursor";