tiennm99/ccs
1
0 Fork 0
mirror of https://github.com/tiennm99/ccs.git synced 2026-10-03 05:19:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(cursor): fix test isolation and daemon exit handling

Browse Source 
- Convert PID_FILE constant to getPidFilePath() function to respect CCS_HOME changes at runtime
- Add proc.on('exit') handler to clear interval on silent process crashes
- Ensures test isolation by computing paths dynamically
This commit is contained in:
Tam Nhu Tran committed 2026-02-11 19:06:59 +07:00
1 parent aaa31c6427
commit fe97d720d4
8 files changed
+208 -288
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/cursor/cursor-daemon.ts
+26 -7
View File
@@ -25,7 +25,13 @@ function getCursorDir(): string {
return path.join(getCcsDir(), 'cursor'); return path.join(getCcsDir(), 'cursor');
} }
const PID_FILE = path.join(getCursorDir(), 'daemon.pid'); /**
* Get PID file path.
* Computed at runtime to respect CCS_HOME changes (e.g., in tests).
*/
function getPidFilePath(): string {
return path.join(getCursorDir(), 'daemon.pid');
}
/** /**
* Check if cursor daemon is running on the specified port. * Check if cursor daemon is running on the specified port.
@@ -77,9 +83,10 @@ export async function getDaemonStatus(port: number): Promise<CursorDaemonStatus>
* Read PID from file. * Read PID from file.
*/ */
function getPidFromFile(): number | null { function getPidFromFile(): number | null {
const pidFile = getPidFilePath();
try { try {
if (fs.existsSync(PID_FILE)) { if (fs.existsSync(pidFile)) {
const content = fs.readFileSync(PID_FILE, 'utf8').trim(); const content = fs.readFileSync(pidFile, 'utf8').trim();
const pid = parseInt(content, 10); const pid = parseInt(content, 10);
return isNaN(pid) ? null : pid; return isNaN(pid) ? null : pid;
} }
@@ -93,12 +100,13 @@ function getPidFromFile(): number | null {
* Write PID to file. * Write PID to file.
*/ */
function writePidToFile(pid: number): void { function writePidToFile(pid: number): void {
const pidFile = getPidFilePath();
try { try {
const dir = path.dirname(PID_FILE); const dir = path.dirname(pidFile);
if (!fs.existsSync(dir)) { if (!fs.existsSync(dir)) {
fs.mkdirSync(dir, { recursive: true, mode: 0o700 }); fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
} }
fs.writeFileSync(PID_FILE, pid.toString(), { mode: 0o600 }); fs.writeFileSync(pidFile, pid.toString(), { mode: 0o600 });
} catch { } catch {
// Ignore errors // Ignore errors
} }
@@ -108,9 +116,10 @@ function writePidToFile(pid: number): void {
* Remove PID file. * Remove PID file.
*/ */
function removePidFile(): void { function removePidFile(): void {
const pidFile = getPidFilePath();
try { try {
if (fs.existsSync(PID_FILE)) { if (fs.existsSync(pidFile)) {
fs.unlinkSync(PID_FILE); fs.unlinkSync(pidFile);
} }
} catch { } catch {
// Ignore errors // Ignore errors
@@ -197,6 +206,16 @@ export async function startDaemon(
error: `Failed to start daemon: ${err.message}`, error: `Failed to start daemon: ${err.message}`,
}); });
}); });
proc.on('exit', (code) => {
if (code !== 0 && code !== null) {
clearInterval(checkInterval);
resolve({
success: false,
error: `Daemon process exited with code ${code}`,
});
}
});
} catch (err) { } catch (err) {
resolve({ resolve({
success: false, success: false,
src/cursor/cursor-executor.ts
+102 -117
View File
@@ -3,12 +3,12 @@
* Handles HTTP/2 requests to Cursor API with protobuf encoding/decoding * Handles HTTP/2 requests to Cursor API with protobuf encoding/decoding
*/ */
import * as crypto from "crypto"; import * as crypto from 'crypto';
import * as zlib from "zlib"; import * as zlib from 'zlib';
import type { IncomingHttpHeaders } from "http"; import type { IncomingHttpHeaders } from 'http';
import { generateCursorBody, extractTextFromResponse } from "./cursor-protobuf.js"; import { generateCursorBody, extractTextFromResponse } from './cursor-protobuf.js';
import { buildCursorRequest } from "./cursor-translator.js"; import { buildCursorRequest } from './cursor-translator.js';
import type { CursorMessage, CursorTool } from "./cursor-protobuf-schema.js"; import type { CursorMessage, CursorTool } from './cursor-protobuf-schema.js';
/** Compression flags for response parsing */ /** Compression flags for response parsing */
const COMPRESS_FLAG = { const COMPRESS_FLAG = {
@@ -59,10 +59,10 @@ interface Http2Response {
/** Detect cloud environment */ /** Detect cloud environment */
function isCloudEnv(): boolean { function isCloudEnv(): boolean {
if (typeof caches !== "undefined" && typeof caches === "object") return true; if (typeof caches !== 'undefined' && typeof caches === 'object') return true;
try { try {
// Check for EdgeRuntime without causing compilation error // Check for EdgeRuntime without causing compilation error
if (typeof (globalThis as { EdgeRuntime?: string }).EdgeRuntime !== "undefined") return true; if (typeof (globalThis as { EdgeRuntime?: string }).EdgeRuntime !== 'undefined') return true;
} catch { } catch {
// Continue // Continue
} }
@@ -70,12 +70,12 @@ function isCloudEnv(): boolean {
} }
/** Lazy import http2 */ /** Lazy import http2 */
let http2Module: typeof import("http2") | null = null; let http2Module: typeof import('http2') | null = null;
async function getHttp2() { async function getHttp2() {
if (http2Module) return http2Module; if (http2Module) return http2Module;
if (!isCloudEnv()) { if (!isCloudEnv()) {
try { try {
http2Module = await import("http2"); http2Module = await import('http2');
return http2Module; return http2Module;
} catch { } catch {
return null; return null;
@@ -91,7 +91,7 @@ function decompressPayload(payload: Buffer, flags: number): Buffer {
// Check if payload is JSON error // Check if payload is JSON error
if (payload.length > 10 && payload[0] === 0x7b && payload[1] === 0x22) { if (payload.length > 10 && payload[0] === 0x7b && payload[1] === 0x22) {
try { try {
const text = payload.toString("utf-8"); const text = payload.toString('utf-8');
if (text.startsWith('{"error"')) { if (text.startsWith('{"error"')) {
return payload; return payload;
} }
@@ -128,28 +128,28 @@ function createErrorResponse(jsonError: {
jsonError?.error?.details?.[0]?.debug?.details?.title || jsonError?.error?.details?.[0]?.debug?.details?.title ||
jsonError?.error?.details?.[0]?.debug?.details?.detail || jsonError?.error?.details?.[0]?.debug?.details?.detail ||
jsonError?.error?.message || jsonError?.error?.message ||
"API Error"; 'API Error';
const isRateLimit = jsonError?.error?.code === "resource_exhausted"; const isRateLimit = jsonError?.error?.code === 'resource_exhausted';
return new Response( return new Response(
JSON.stringify({ JSON.stringify({
error: { error: {
message: errorMsg, message: errorMsg,
type: isRateLimit ? "rate_limit_error" : "api_error", type: isRateLimit ? 'rate_limit_error' : 'api_error',
code: jsonError?.error?.details?.[0]?.debug?.error || "unknown", code: jsonError?.error?.details?.[0]?.debug?.error || 'unknown',
}, },
}), }),
{ {
status: isRateLimit ? 429 : 400, status: isRateLimit ? 429 : 400,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
} }
); );
} }
export class CursorExecutor { export class CursorExecutor {
private readonly baseUrl = "https://api2.cursor.sh"; private readonly baseUrl = 'https://api2.cursor.sh';
private readonly chatPath = "/aiserver.v1.AiService/StreamChat"; private readonly chatPath = '/aiserver.v1.AiService/StreamChat';
buildUrl(): string { buildUrl(): string {
return `${this.baseUrl}${this.chatPath}`; return `${this.baseUrl}${this.chatPath}`;
@@ -175,8 +175,8 @@ export class CursorExecutor {
t = byteArray[i]; t = byteArray[i];
} }
const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_"; const alphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_';
let encoded = ""; let encoded = '';
for (let i = 0; i < byteArray.length; i += 3) { for (let i = 0; i < byteArray.length; i += 3) {
const a = byteArray[i]; const a = byteArray[i];
@@ -203,48 +203,41 @@ export class CursorExecutor {
const ghostMode = credentials.providerSpecificData?.ghostMode !== false; const ghostMode = credentials.providerSpecificData?.ghostMode !== false;
if (!machineId) { if (!machineId) {
throw new Error("Machine ID is required for Cursor API"); throw new Error('Machine ID is required for Cursor API');
} }
const cleanToken = accessToken.includes("::") const cleanToken = accessToken.includes('::') ? accessToken.split('::')[1] : accessToken;
? accessToken.split("::")[1]
: accessToken;
return { return {
authorization: `Bearer ${cleanToken}`, authorization: `Bearer ${cleanToken}`,
"connect-accept-encoding": "gzip", 'connect-accept-encoding': 'gzip',
"connect-protocol-version": "1", 'connect-protocol-version': '1',
"content-type": "application/connect+proto", 'content-type': 'application/connect+proto',
"user-agent": "connect-es/1.6.1", 'user-agent': 'connect-es/1.6.1',
"x-amzn-trace-id": `Root=${crypto.randomUUID()}`, 'x-amzn-trace-id': `Root=${crypto.randomUUID()}`,
"x-client-key": crypto.createHash("sha256").update(cleanToken).digest("hex"), 'x-client-key': crypto.createHash('sha256').update(cleanToken).digest('hex'),
"x-cursor-checksum": this.generateChecksum(machineId), 'x-cursor-checksum': this.generateChecksum(machineId),
"x-cursor-client-version": "2.3.41", 'x-cursor-client-version': '2.3.41',
"x-cursor-client-type": "ide", 'x-cursor-client-type': 'ide',
"x-cursor-client-os": 'x-cursor-client-os':
process.platform === "win32" process.platform === 'win32'
? "windows" ? 'windows'
: process.platform === "darwin" : process.platform === 'darwin'
? "macos" ? 'macos'
: "linux", : 'linux',
"x-cursor-client-arch": process.arch === "arm64" ? "aarch64" : "x64", 'x-cursor-client-arch': process.arch === 'arm64' ? 'aarch64' : 'x64',
"x-cursor-client-device-type": "desktop", 'x-cursor-client-device-type': 'desktop',
"x-cursor-config-version": crypto.randomUUID(), 'x-cursor-config-version': crypto.randomUUID(),
"x-cursor-timezone": 'x-cursor-timezone': Intl.DateTimeFormat().resolvedOptions().timeZone || 'UTC',
Intl.DateTimeFormat().resolvedOptions().timeZone || "UTC", 'x-ghost-mode': ghostMode ? 'true' : 'false',
"x-ghost-mode": ghostMode ? "true" : "false", 'x-request-id': crypto.randomUUID(),
"x-request-id": crypto.randomUUID(), 'x-session-id': crypto.createHash('sha256').update(cleanToken).digest('hex').substring(0, 36),
"x-session-id": crypto
.createHash("sha256")
.update(cleanToken)
.digest("hex")
.substring(0, 36),
}; };
} }
transformRequest( transformRequest(
model: string, model: string,
body: ExecutorParams["body"], body: ExecutorParams['body'],
stream: boolean, stream: boolean,
credentials: CursorCredentials credentials: CursorCredentials
): Uint8Array { ): Uint8Array {
@@ -262,7 +255,7 @@ export class CursorExecutor {
signal?: AbortSignal signal?: AbortSignal
): Promise<Http2Response> { ): Promise<Http2Response> {
const response = await fetch(url, { const response = await fetch(url, {
method: "POST", method: 'POST',
headers, headers,
body, body,
signal, signal,
@@ -288,7 +281,7 @@ export class CursorExecutor {
): Promise<Http2Response> { ): Promise<Http2Response> {
const http2 = await getHttp2(); const http2 = await getHttp2();
if (!http2) { if (!http2) {
throw new Error("http2 module not available"); throw new Error('http2 module not available');
} }
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
@@ -297,40 +290,40 @@ export class CursorExecutor {
const chunks: Buffer[] = []; const chunks: Buffer[] = [];
let responseHeaders: IncomingHttpHeaders = {}; let responseHeaders: IncomingHttpHeaders = {};
client.on("error", reject); client.on('error', reject);
const req = client.request({ const req = client.request({
":method": "POST", ':method': 'POST',
":path": urlObj.pathname, ':path': urlObj.pathname,
":authority": urlObj.host, ':authority': urlObj.host,
":scheme": "https", ':scheme': 'https',
...headers, ...headers,
}); });
req.on("response", (hdrs) => { req.on('response', (hdrs) => {
responseHeaders = hdrs; responseHeaders = hdrs;
}); });
req.on("data", (chunk: Buffer) => { req.on('data', (chunk: Buffer) => {
chunks.push(chunk); chunks.push(chunk);
}); });
req.on("end", () => { req.on('end', () => {
client.close(); client.close();
resolve({ resolve({
status: Number(responseHeaders[":status"]), status: Number(responseHeaders[':status']),
headers: responseHeaders, headers: responseHeaders,
body: Buffer.concat(chunks), body: Buffer.concat(chunks),
}); });
}); });
req.on("error", (err) => { req.on('error', (err) => {
client.close(); client.close();
reject(err); reject(err);
}); });
if (signal) { if (signal) {
signal.addEventListener("abort", () => { signal.addEventListener('abort', () => {
req.close(); req.close();
client.close(); client.close();
reject(new Error("Request aborted")); reject(new Error('Request aborted'));
}); });
} }
@@ -343,7 +336,7 @@ export class CursorExecutor {
response: Response; response: Response;
url: string; url: string;
headers: Record<string, string>; headers: Record<string, string>;
transformedBody: ExecutorParams["body"]; transformedBody: ExecutorParams['body'];
}> { }> {
const { model, body, stream, credentials, signal } = params; const { model, body, stream, credentials, signal } = params;
const url = this.buildUrl(); const url = this.buildUrl();
@@ -357,18 +350,18 @@ export class CursorExecutor {
: await this.makeFetchRequest(url, headers, transformedBody, signal); : await this.makeFetchRequest(url, headers, transformedBody, signal);
if (response.status !== 200) { if (response.status !== 200) {
const errorText = response.body?.toString() || "Unknown error"; const errorText = response.body?.toString() || 'Unknown error';
const errorResponse = new Response( const errorResponse = new Response(
JSON.stringify({ JSON.stringify({
error: { error: {
message: `[${response.status}]: ${errorText}`, message: `[${response.status}]: ${errorText}`,
type: "invalid_request_error", type: 'invalid_request_error',
code: "", code: '',
}, },
}), }),
{ {
status: response.status, status: response.status,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
} }
); );
return { response: errorResponse, url, headers, transformedBody: body }; return { response: errorResponse, url, headers, transformedBody: body };
@@ -385,29 +378,25 @@ export class CursorExecutor {
JSON.stringify({ JSON.stringify({
error: { error: {
message: (error as Error).message, message: (error as Error).message,
type: "connection_error", type: 'connection_error',
code: "", code: '',
}, },
}), }),
{ {
status: 500, status: 500,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
} }
); );
return { response: errorResponse, url, headers, transformedBody: body }; return { response: errorResponse, url, headers, transformedBody: body };
} }
} }
transformProtobufToJSON( transformProtobufToJSON(buffer: Buffer, model: string, body: ExecutorParams['body']): Response {
buffer: Buffer,
model: string,
body: ExecutorParams["body"]
): Response {
const responseId = `chatcmpl-cursor-${Date.now()}`; const responseId = `chatcmpl-cursor-${Date.now()}`;
const created = Math.floor(Date.now() / 1000); const created = Math.floor(Date.now() / 1000);
let offset = 0; let offset = 0;
let totalContent = ""; let totalContent = '';
const toolCalls: Array<{ const toolCalls: Array<{
id: string; id: string;
type: string; type: string;
@@ -438,8 +427,8 @@ export class CursorExecutor {
payload = decompressPayload(payload, flags); payload = decompressPayload(payload, flags);
try { try {
const text = payload.toString("utf-8"); const text = payload.toString('utf-8');
if (text.startsWith("{") && text.includes('"error"')) { if (text.startsWith('{') && text.includes('"error"')) {
return createErrorResponse(JSON.parse(text)); return createErrorResponse(JSON.parse(text));
} }
} catch { } catch {
@@ -453,13 +442,13 @@ export class CursorExecutor {
JSON.stringify({ JSON.stringify({
error: { error: {
message: result.error, message: result.error,
type: "rate_limit_error", type: 'rate_limit_error',
code: "rate_limited", code: 'rate_limited',
}, },
}), }),
{ {
status: 429, status: 429,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
} }
); );
} }
@@ -518,7 +507,7 @@ export class CursorExecutor {
function: { name: string; arguments: string }; function: { name: string; arguments: string };
}>; }>;
} = { } = {
role: "assistant", role: 'assistant',
content: totalContent || null, content: totalContent || null,
}; };
@@ -528,14 +517,14 @@ export class CursorExecutor {
const completion = { const completion = {
id: responseId, id: responseId,
object: "chat.completion", object: 'chat.completion',
created, created,
model, model,
choices: [ choices: [
{ {
index: 0, index: 0,
message, message,
finish_reason: toolCalls.length > 0 ? "tool_calls" : "stop", finish_reason: toolCalls.length > 0 ? 'tool_calls' : 'stop',
}, },
], ],
usage: { usage: {
@@ -547,21 +536,17 @@ export class CursorExecutor {
return new Response(JSON.stringify(completion), { return new Response(JSON.stringify(completion), {
status: 200, status: 200,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
}); });
} }
transformProtobufToSSE( transformProtobufToSSE(buffer: Buffer, model: string, body: ExecutorParams['body']): Response {
buffer: Buffer,
model: string,
body: ExecutorParams["body"]
): Response {
const responseId = `chatcmpl-cursor-${Date.now()}`; const responseId = `chatcmpl-cursor-${Date.now()}`;
const created = Math.floor(Date.now() / 1000); const created = Math.floor(Date.now() / 1000);
const chunks: string[] = []; const chunks: string[] = [];
let offset = 0; let offset = 0;
let totalContent = ""; let totalContent = '';
const toolCalls: Array<{ const toolCalls: Array<{
id: string; id: string;
type: string; type: string;
@@ -593,8 +578,8 @@ export class CursorExecutor {
payload = decompressPayload(payload, flags); payload = decompressPayload(payload, flags);
try { try {
const text = payload.toString("utf-8"); const text = payload.toString('utf-8');
if (text.startsWith("{") && text.includes('"error"')) { if (text.startsWith('{') && text.includes('"error"')) {
return createErrorResponse(JSON.parse(text)); return createErrorResponse(JSON.parse(text));
} }
} catch { } catch {
@@ -608,13 +593,13 @@ export class CursorExecutor {
JSON.stringify({ JSON.stringify({
error: { error: {
message: result.error, message: result.error,
type: "rate_limit_error", type: 'rate_limit_error',
code: "rate_limited", code: 'rate_limited',
}, },
}), }),
{ {
status: 429, status: 429,
headers: { "Content-Type": "application/json" }, headers: { 'Content-Type': 'application/json' },
} }
); );
} }
@@ -626,13 +611,13 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
{ {
index: 0, index: 0,
delta: { role: "assistant", content: "" }, delta: { role: 'assistant', content: '' },
finish_reason: null, finish_reason: null,
}, },
], ],
@@ -649,7 +634,7 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
@@ -660,7 +645,7 @@ export class CursorExecutor {
{ {
index: existing.index, index: existing.index,
id: tc.id, id: tc.id,
type: "function", type: 'function',
function: { function: {
name: tc.function.name, name: tc.function.name,
arguments: tc.function.arguments, arguments: tc.function.arguments,
@@ -682,7 +667,7 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
@@ -693,7 +678,7 @@ export class CursorExecutor {
{ {
index: toolCallIndex, index: toolCallIndex,
id: tc.id, id: tc.id,
type: "function", type: 'function',
function: { function: {
name: tc.function.name, name: tc.function.name,
arguments: tc.function.arguments, arguments: tc.function.arguments,
@@ -714,7 +699,7 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
@@ -722,7 +707,7 @@ export class CursorExecutor {
index: 0, index: 0,
delta: delta:
chunks.length === 0 && toolCalls.length === 0 chunks.length === 0 && toolCalls.length === 0
? { role: "assistant", content: result.text } ? { role: 'assistant', content: result.text }
: { content: result.text }, : { content: result.text },
finish_reason: null, finish_reason: null,
}, },
@@ -736,13 +721,13 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
{ {
index: 0, index: 0,
delta: { role: "assistant", content: "" }, delta: { role: 'assistant', content: '' },
finish_reason: null, finish_reason: null,
}, },
], ],
@@ -753,14 +738,14 @@ export class CursorExecutor {
chunks.push( chunks.push(
`data: ${JSON.stringify({ `data: ${JSON.stringify({
id: responseId, id: responseId,
object: "chat.completion.chunk", object: 'chat.completion.chunk',
created, created,
model, model,
choices: [ choices: [
{ {
index: 0, index: 0,
delta: {}, delta: {},
finish_reason: toolCalls.length > 0 ? "tool_calls" : "stop", finish_reason: toolCalls.length > 0 ? 'tool_calls' : 'stop',
}, },
], ],
usage: { usage: {
@@ -770,14 +755,14 @@ export class CursorExecutor {
}, },
})}\n\n` })}\n\n`
); );
chunks.push("data: [DONE]\n\n"); chunks.push('data: [DONE]\n\n');
return new Response(chunks.join(""), { return new Response(chunks.join(''), {
status: 200, status: 200,
headers: { headers: {
"Content-Type": "text/event-stream", 'Content-Type': 'text/event-stream',
"Cache-Control": "no-cache", 'Cache-Control': 'no-cache',
Connection: "keep-alive", Connection: 'keep-alive',
}, },
}); });
} }
src/cursor/cursor-protobuf-decoder.ts
+23 -49
View File
@@ -3,21 +3,14 @@
* Implements ConnectRPC protobuf wire format decoding * Implements ConnectRPC protobuf wire format decoding
*/ */
import * as zlib from "zlib"; import * as zlib from 'zlib';
import { import { WIRE_TYPE, FIELD, type WireType } from './cursor-protobuf-schema.js';
WIRE_TYPE,
FIELD,
type WireType,
} from "./cursor-protobuf-schema.js";
/** /**
* Decode a varint from buffer * Decode a varint from buffer
* Returns [value, newOffset] * Returns [value, newOffset]
*/ */
export function decodeVarint( export function decodeVarint(buffer: Uint8Array, offset: number): [number, number] {
buffer: Uint8Array,
offset: number
): [number, number] {
let result = 0; let result = 0;
let shift = 0; let shift = 0;
let pos = offset; let pos = offset;
@@ -77,10 +70,7 @@ export function decodeField(
export function decodeMessage( export function decodeMessage(
data: Uint8Array data: Uint8Array
): Map<number, Array<{ wireType: WireType; value: Uint8Array | number }>> { ): Map<number, Array<{ wireType: WireType; value: Uint8Array | number }>> {
const fields = new Map< const fields = new Map<number, Array<{ wireType: WireType; value: Uint8Array | number }>>();
number,
Array<{ wireType: WireType; value: Uint8Array | number }>
>();
let pos = 0; let pos = 0;
while (pos < data.length) { while (pos < data.length) {
@@ -110,8 +100,7 @@ export function parseConnectRPCFrame(buffer: Buffer): {
if (buffer.length < 5) return null; if (buffer.length < 5) return null;
const flags = buffer[0]; const flags = buffer[0];
const length = const length = (buffer[1] << 24) | (buffer[2] << 16) | (buffer[3] << 8) | buffer[4];
(buffer[1] << 24) | (buffer[2] << 16) | (buffer[3] << 8) | buffer[4];
if (buffer.length < 5 + length) return null; if (buffer.length < 5 + length) return null;
@@ -144,24 +133,20 @@ function extractToolCall(toolCallData: Uint8Array): {
isLast: boolean; isLast: boolean;
} | null { } | null {
const toolCall = decodeMessage(toolCallData); const toolCall = decodeMessage(toolCallData);
let toolCallId = ""; let toolCallId = '';
let toolName = ""; let toolName = '';
let rawArgs = ""; let rawArgs = '';
let isLast = false; let isLast = false;
// Extract tool call ID // Extract tool call ID
if (toolCall.has(FIELD.TOOL_ID)) { if (toolCall.has(FIELD.TOOL_ID)) {
const fullId = new TextDecoder().decode( const fullId = new TextDecoder().decode(toolCall.get(FIELD.TOOL_ID)![0].value as Uint8Array);
toolCall.get(FIELD.TOOL_ID)![0].value as Uint8Array toolCallId = fullId.split('\n')[0]; // Take first line
);
toolCallId = fullId.split("\n")[0]; // Take first line
} }
// Extract tool name // Extract tool name
if (toolCall.has(FIELD.TOOL_NAME)) { if (toolCall.has(FIELD.TOOL_NAME)) {
toolName = new TextDecoder().decode( toolName = new TextDecoder().decode(toolCall.get(FIELD.TOOL_NAME)![0].value as Uint8Array);
toolCall.get(FIELD.TOOL_NAME)![0].value as Uint8Array
);
} }
// Extract is_last flag // Extract is_last flag
@@ -172,14 +157,10 @@ function extractToolCall(toolCallData: Uint8Array): {
// Extract MCP params - nested real tool info // Extract MCP params - nested real tool info
if (toolCall.has(FIELD.TOOL_MCP_PARAMS)) { if (toolCall.has(FIELD.TOOL_MCP_PARAMS)) {
try { try {
const mcpParams = decodeMessage( const mcpParams = decodeMessage(toolCall.get(FIELD.TOOL_MCP_PARAMS)![0].value as Uint8Array);
toolCall.get(FIELD.TOOL_MCP_PARAMS)![0].value as Uint8Array
);
if (mcpParams.has(FIELD.MCP_TOOLS_LIST)) { if (mcpParams.has(FIELD.MCP_TOOLS_LIST)) {
const tool = decodeMessage( const tool = decodeMessage(mcpParams.get(FIELD.MCP_TOOLS_LIST)![0].value as Uint8Array);
mcpParams.get(FIELD.MCP_TOOLS_LIST)![0].value as Uint8Array
);
if (tool.has(FIELD.MCP_NESTED_NAME)) { if (tool.has(FIELD.MCP_NESTED_NAME)) {
toolName = new TextDecoder().decode( toolName = new TextDecoder().decode(
@@ -200,18 +181,16 @@ function extractToolCall(toolCallData: Uint8Array): {
// Fallback to raw_args // Fallback to raw_args
if (!rawArgs && toolCall.has(FIELD.TOOL_RAW_ARGS)) { if (!rawArgs && toolCall.has(FIELD.TOOL_RAW_ARGS)) {
rawArgs = new TextDecoder().decode( rawArgs = new TextDecoder().decode(toolCall.get(FIELD.TOOL_RAW_ARGS)![0].value as Uint8Array);
toolCall.get(FIELD.TOOL_RAW_ARGS)![0].value as Uint8Array
);
} }
if (toolCallId && toolName) { if (toolCallId && toolName) {
return { return {
id: toolCallId, id: toolCallId,
type: "function", type: 'function',
function: { function: {
name: toolName, name: toolName,
arguments: rawArgs || "{}", arguments: rawArgs || '{}',
}, },
isLast, isLast,
}; };
@@ -223,26 +202,23 @@ function extractToolCall(toolCallData: Uint8Array): {
/** /**
* Extract text and thinking from response data * Extract text and thinking from response data
*/ */
function extractTextAndThinking( function extractTextAndThinking(responseData: Uint8Array): {
responseData: Uint8Array text: string | null;
): { text: string | null; thinking: string | null } { thinking: string | null;
} {
const nested = decodeMessage(responseData); const nested = decodeMessage(responseData);
let text: string | null = null; let text: string | null = null;
let thinking: string | null = null; let thinking: string | null = null;
// Extract text // Extract text
if (nested.has(FIELD.RESPONSE_TEXT)) { if (nested.has(FIELD.RESPONSE_TEXT)) {
text = new TextDecoder().decode( text = new TextDecoder().decode(nested.get(FIELD.RESPONSE_TEXT)![0].value as Uint8Array);
nested.get(FIELD.RESPONSE_TEXT)![0].value as Uint8Array
);
} }
// Extract thinking // Extract thinking
if (nested.has(FIELD.THINKING)) { if (nested.has(FIELD.THINKING)) {
try { try {
const thinkingMsg = decodeMessage( const thinkingMsg = decodeMessage(nested.get(FIELD.THINKING)![0].value as Uint8Array);
nested.get(FIELD.THINKING)![0].value as Uint8Array
);
if (thinkingMsg.has(FIELD.THINKING_TEXT)) { if (thinkingMsg.has(FIELD.THINKING_TEXT)) {
thinking = new TextDecoder().decode( thinking = new TextDecoder().decode(
thinkingMsg.get(FIELD.THINKING_TEXT)![0].value as Uint8Array thinkingMsg.get(FIELD.THINKING_TEXT)![0].value as Uint8Array
@@ -275,9 +251,7 @@ export function extractTextFromResponse(payload: Uint8Array): {
// Field 1: ClientSideToolV2Call // Field 1: ClientSideToolV2Call
if (fields.has(FIELD.TOOL_CALL)) { if (fields.has(FIELD.TOOL_CALL)) {
const toolCall = extractToolCall( const toolCall = extractToolCall(fields.get(FIELD.TOOL_CALL)![0].value as Uint8Array);
fields.get(FIELD.TOOL_CALL)![0].value as Uint8Array
);
if (toolCall) { if (toolCall) {
return { text: null, error: null, toolCall, thinking: null }; return { text: null, error: null, toolCall, thinking: null };
} }
src/cursor/cursor-protobuf-encoder.ts
+24 -55
View File
@@ -3,8 +3,8 @@
* Implements ConnectRPC protobuf wire format encoding * Implements ConnectRPC protobuf wire format encoding
*/ */
import { randomUUID } from "crypto"; import { randomUUID } from 'crypto';
import * as zlib from "zlib"; import * as zlib from 'zlib';
import { import {
WIRE_TYPE, WIRE_TYPE,
ROLE, ROLE,
@@ -20,7 +20,7 @@ import {
type CursorMessage, type CursorMessage,
type FormattedMessage, type FormattedMessage,
type MessageId, type MessageId,
} from "./cursor-protobuf-schema.js"; } from './cursor-protobuf-schema.js';
/** /**
* Encode a varint (variable-length integer) * Encode a varint (variable-length integer)
@@ -54,7 +54,7 @@ export function encodeField(
if (wireType === WIRE_TYPE.LEN) { if (wireType === WIRE_TYPE.LEN) {
const dataBytes = const dataBytes =
typeof value === "string" typeof value === 'string'
? new TextEncoder().encode(value) ? new TextEncoder().encode(value)
: value instanceof Uint8Array : value instanceof Uint8Array
? value ? value
@@ -85,10 +85,10 @@ function concatArrays(...arrays: Uint8Array[]): Uint8Array {
* Encode a tool result * Encode a tool result
*/ */
export function encodeToolResult(toolResult: CursorToolResult): Uint8Array { export function encodeToolResult(toolResult: CursorToolResult): Uint8Array {
const toolCallId = toolResult.tool_call_id || ""; const toolCallId = toolResult.tool_call_id || '';
const toolName = toolResult.name || ""; const toolName = toolResult.name || '';
const toolIndex = toolResult.index || 0; const toolIndex = toolResult.index || 0;
const rawArgs = toolResult.raw_args || "{}"; const rawArgs = toolResult.raw_args || '{}';
return concatArrays( return concatArrays(
encodeField(FIELD.TOOL_RESULT_CALL_ID, WIRE_TYPE.LEN, toolCallId), encodeField(FIELD.TOOL_RESULT_CALL_ID, WIRE_TYPE.LEN, toolCallId),
@@ -115,11 +115,7 @@ export function encodeMessage(
encodeField(FIELD.MSG_ID, WIRE_TYPE.LEN, messageId), encodeField(FIELD.MSG_ID, WIRE_TYPE.LEN, messageId),
...(toolResults.length > 0 ...(toolResults.length > 0
? toolResults.map((tr) => ? toolResults.map((tr) =>
encodeField( encodeField(FIELD.MSG_TOOL_RESULTS, WIRE_TYPE.LEN, encodeToolResult(tr))
FIELD.MSG_TOOL_RESULTS,
WIRE_TYPE.LEN,
encodeToolResult(tr)
)
) )
: []), : []),
encodeField(FIELD.MSG_IS_AGENTIC, WIRE_TYPE.VARINT, hasTools ? 1 : 0), encodeField(FIELD.MSG_IS_AGENTIC, WIRE_TYPE.VARINT, hasTools ? 1 : 0),
@@ -129,13 +125,7 @@ export function encodeMessage(
hasTools ? UNIFIED_MODE.AGENT : UNIFIED_MODE.CHAT hasTools ? UNIFIED_MODE.AGENT : UNIFIED_MODE.CHAT
), ),
...(isLast && hasTools ...(isLast && hasTools
? [ ? [encodeField(FIELD.MSG_SUPPORTED_TOOLS, WIRE_TYPE.LEN, encodeVarint(1))]
encodeField(
FIELD.MSG_SUPPORTED_TOOLS,
WIRE_TYPE.LEN,
encodeVarint(1)
),
]
: []) : [])
); );
} }
@@ -144,9 +134,7 @@ export function encodeMessage(
* Encode instruction text * Encode instruction text
*/ */
export function encodeInstruction(text: string): Uint8Array { export function encodeInstruction(text: string): Uint8Array {
return text return text ? encodeField(FIELD.INSTRUCTION_TEXT, WIRE_TYPE.LEN, text) : new Uint8Array(0);
? encodeField(FIELD.INSTRUCTION_TEXT, WIRE_TYPE.LEN, text)
: new Uint8Array(0);
} }
/** /**
@@ -169,7 +157,7 @@ export function encodeCursorSetting(): Uint8Array {
); );
return concatArrays( return concatArrays(
encodeField(FIELD.SETTING_PATH, WIRE_TYPE.LEN, "cursor\\aisettings"), encodeField(FIELD.SETTING_PATH, WIRE_TYPE.LEN, 'cursor\\aisettings'),
encodeField(FIELD.SETTING_UNKNOWN_3, WIRE_TYPE.LEN, new Uint8Array(0)), encodeField(FIELD.SETTING_UNKNOWN_3, WIRE_TYPE.LEN, new Uint8Array(0)),
encodeField(FIELD.SETTING_UNKNOWN_6, WIRE_TYPE.LEN, unknown6), encodeField(FIELD.SETTING_UNKNOWN_6, WIRE_TYPE.LEN, unknown6),
encodeField(FIELD.SETTING_UNKNOWN_8, WIRE_TYPE.VARINT, 1), encodeField(FIELD.SETTING_UNKNOWN_8, WIRE_TYPE.VARINT, 1),
@@ -182,10 +170,10 @@ export function encodeCursorSetting(): Uint8Array {
*/ */
export function encodeMetadata(): Uint8Array { export function encodeMetadata(): Uint8Array {
return concatArrays( return concatArrays(
encodeField(FIELD.META_PLATFORM, WIRE_TYPE.LEN, process.platform || "linux"), encodeField(FIELD.META_PLATFORM, WIRE_TYPE.LEN, process.platform || 'linux'),
encodeField(FIELD.META_ARCH, WIRE_TYPE.LEN, process.arch || "x64"), encodeField(FIELD.META_ARCH, WIRE_TYPE.LEN, process.arch || 'x64'),
encodeField(FIELD.META_VERSION, WIRE_TYPE.LEN, process.version || "v20.0.0"), encodeField(FIELD.META_VERSION, WIRE_TYPE.LEN, process.version || 'v20.0.0'),
encodeField(FIELD.META_CWD, WIRE_TYPE.LEN, process.cwd() || "/"), encodeField(FIELD.META_CWD, WIRE_TYPE.LEN, process.cwd() || '/'),
encodeField(FIELD.META_TIMESTAMP, WIRE_TYPE.LEN, new Date().toISOString()) encodeField(FIELD.META_TIMESTAMP, WIRE_TYPE.LEN, new Date().toISOString())
); );
} }
@@ -193,16 +181,10 @@ export function encodeMetadata(): Uint8Array {
/** /**
* Encode message ID * Encode message ID
*/ */
export function encodeMessageId( export function encodeMessageId(messageId: string, role: RoleType, summaryId?: string): Uint8Array {
messageId: string,
role: RoleType,
summaryId?: string
): Uint8Array {
return concatArrays( return concatArrays(
encodeField(FIELD.MSGID_ID, WIRE_TYPE.LEN, messageId), encodeField(FIELD.MSGID_ID, WIRE_TYPE.LEN, messageId),
...(summaryId ...(summaryId ? [encodeField(FIELD.MSGID_SUMMARY, WIRE_TYPE.LEN, summaryId)] : []),
? [encodeField(FIELD.MSGID_SUMMARY, WIRE_TYPE.LEN, summaryId)]
: []),
encodeField(FIELD.MSGID_ROLE, WIRE_TYPE.VARINT, role) encodeField(FIELD.MSGID_ROLE, WIRE_TYPE.VARINT, role)
); );
} }
@@ -211,37 +193,24 @@ export function encodeMessageId(
* Encode MCP tool * Encode MCP tool
*/ */
export function encodeMcpTool(tool: CursorTool): Uint8Array { export function encodeMcpTool(tool: CursorTool): Uint8Array {
const toolName = tool.function?.name || tool.name || ""; const toolName = tool.function?.name || tool.name || '';
const toolDesc = tool.function?.description || tool.description || ""; const toolDesc = tool.function?.description || tool.description || '';
const inputSchema = tool.function?.parameters || tool.input_schema || {}; const inputSchema = tool.function?.parameters || tool.input_schema || {};
return concatArrays( return concatArrays(
...(toolName ...(toolName ? [encodeField(FIELD.MCP_TOOL_NAME, WIRE_TYPE.LEN, toolName)] : []),
? [encodeField(FIELD.MCP_TOOL_NAME, WIRE_TYPE.LEN, toolName)] ...(toolDesc ? [encodeField(FIELD.MCP_TOOL_DESC, WIRE_TYPE.LEN, toolDesc)] : []),
: []),
...(toolDesc
? [encodeField(FIELD.MCP_TOOL_DESC, WIRE_TYPE.LEN, toolDesc)]
: []),
...(Object.keys(inputSchema).length > 0 ...(Object.keys(inputSchema).length > 0
? [ ? [encodeField(FIELD.MCP_TOOL_PARAMS, WIRE_TYPE.LEN, JSON.stringify(inputSchema))]
encodeField(
FIELD.MCP_TOOL_PARAMS,
WIRE_TYPE.LEN,
JSON.stringify(inputSchema)
),
]
: []), : []),
encodeField(FIELD.MCP_TOOL_SERVER, WIRE_TYPE.LEN, "custom") encodeField(FIELD.MCP_TOOL_SERVER, WIRE_TYPE.LEN, 'custom')
); );
} }
/** /**
* Wrap payload in ConnectRPC frame (5-byte header + payload) * Wrap payload in ConnectRPC frame (5-byte header + payload)
*/ */
export function wrapConnectRPCFrame( export function wrapConnectRPCFrame(payload: Uint8Array, compress = false): Uint8Array {
payload: Uint8Array,
compress = false
): Uint8Array {
let finalPayload = payload; let finalPayload = payload;
let flags: number = COMPRESS_FLAG.NONE; let flags: number = COMPRESS_FLAG.NONE;
src/cursor/cursor-protobuf-schema.ts
+1 -2
View File
@@ -143,8 +143,7 @@ export const FIELD = {
export type WireType = (typeof WIRE_TYPE)[keyof typeof WIRE_TYPE]; export type WireType = (typeof WIRE_TYPE)[keyof typeof WIRE_TYPE];
export type RoleType = (typeof ROLE)[keyof typeof ROLE]; export type RoleType = (typeof ROLE)[keyof typeof ROLE];
export type UnifiedModeType = (typeof UNIFIED_MODE)[keyof typeof UNIFIED_MODE]; export type UnifiedModeType = (typeof UNIFIED_MODE)[keyof typeof UNIFIED_MODE];
export type ThinkingLevelType = export type ThinkingLevelType = (typeof THINKING_LEVEL)[keyof typeof THINKING_LEVEL];
(typeof THINKING_LEVEL)[keyof typeof THINKING_LEVEL];
export type FieldNumber = (typeof FIELD)[keyof typeof FIELD]; export type FieldNumber = (typeof FIELD)[keyof typeof FIELD];
/** Cursor tool definition */ /** Cursor tool definition */
src/cursor/cursor-protobuf.ts
+15 -32
View File
@@ -3,7 +3,7 @@
* Exports encoder/decoder functions and builds complete requests * Exports encoder/decoder functions and builds complete requests
*/ */
import { randomUUID } from "crypto"; import { randomUUID } from 'crypto';
import { import {
ROLE, ROLE,
UNIFIED_MODE, UNIFIED_MODE,
@@ -14,7 +14,7 @@ import {
type FormattedMessage, type FormattedMessage,
type MessageId, type MessageId,
type ThinkingLevelType, type ThinkingLevelType,
} from "./cursor-protobuf-schema.js"; } from './cursor-protobuf-schema.js';
import { import {
encodeField, encodeField,
encodeVarint, encodeVarint,
@@ -26,15 +26,15 @@ import {
encodeMessageId, encodeMessageId,
encodeMcpTool, encodeMcpTool,
wrapConnectRPCFrame, wrapConnectRPCFrame,
} from "./cursor-protobuf-encoder.js"; } from './cursor-protobuf-encoder.js';
import { import {
decodeVarint, decodeVarint,
decodeField, decodeField,
decodeMessage, decodeMessage,
parseConnectRPCFrame, parseConnectRPCFrame,
extractTextFromResponse, extractTextFromResponse,
} from "./cursor-protobuf-decoder.js"; } from './cursor-protobuf-decoder.js';
import { WIRE_TYPE } from "./cursor-protobuf-schema.js"; import { WIRE_TYPE } from './cursor-protobuf-schema.js';
/** /**
* Build complete chat request protobuf * Build complete chat request protobuf
@@ -53,7 +53,7 @@ export function encodeRequest(
// Prepare messages // Prepare messages
for (let i = 0; i < messages.length; i++) { for (let i = 0; i < messages.length; i++) {
const msg = messages[i]; const msg = messages[i];
const role = msg.role === "user" ? ROLE.USER : ROLE.ASSISTANT; const role = msg.role === 'user' ? ROLE.USER : ROLE.ASSISTANT;
const msgId = randomUUID(); const msgId = randomUUID();
const isLast = i === messages.length - 1; const isLast = i === messages.length - 1;
@@ -71,38 +71,25 @@ export function encodeRequest(
// Map reasoning effort to thinking level // Map reasoning effort to thinking level
let thinkingLevel: ThinkingLevelType = THINKING_LEVEL.UNSPECIFIED; let thinkingLevel: ThinkingLevelType = THINKING_LEVEL.UNSPECIFIED;
if (reasoningEffort === "medium") thinkingLevel = THINKING_LEVEL.MEDIUM; if (reasoningEffort === 'medium') thinkingLevel = THINKING_LEVEL.MEDIUM;
else if (reasoningEffort === "high") thinkingLevel = THINKING_LEVEL.HIGH; else if (reasoningEffort === 'high') thinkingLevel = THINKING_LEVEL.HIGH;
// Build arrays for messages and tools // Build arrays for messages and tools
const messageFields = formattedMessages.map((fm) => const messageFields = formattedMessages.map((fm) =>
encodeField( encodeField(
FIELD.MESSAGES, FIELD.MESSAGES,
WIRE_TYPE.LEN, WIRE_TYPE.LEN,
encodeMessage( encodeMessage(fm.content, fm.role, fm.messageId, fm.isLast, fm.hasTools, fm.toolResults)
fm.content,
fm.role,
fm.messageId,
fm.isLast,
fm.hasTools,
fm.toolResults
)
) )
); );
const messageIdFields = messageIds.map((mid) => const messageIdFields = messageIds.map((mid) =>
encodeField( encodeField(FIELD.MESSAGE_IDS, WIRE_TYPE.LEN, encodeMessageId(mid.messageId, mid.role))
FIELD.MESSAGE_IDS,
WIRE_TYPE.LEN,
encodeMessageId(mid.messageId, mid.role)
)
); );
const toolFields = const toolFields =
tools?.length > 0 tools?.length > 0
? tools.map((tool) => ? tools.map((tool) => encodeField(FIELD.MCP_TOOLS, WIRE_TYPE.LEN, encodeMcpTool(tool)))
encodeField(FIELD.MCP_TOOLS, WIRE_TYPE.LEN, encodeMcpTool(tool))
)
: []; : [];
const supportedToolsField = isAgentic const supportedToolsField = isAgentic
@@ -113,10 +100,10 @@ export function encodeRequest(
const parts: Uint8Array[] = [ const parts: Uint8Array[] = [
...messageFields, ...messageFields,
encodeField(FIELD.UNKNOWN_2, WIRE_TYPE.VARINT, 1), encodeField(FIELD.UNKNOWN_2, WIRE_TYPE.VARINT, 1),
encodeField(FIELD.INSTRUCTION, WIRE_TYPE.LEN, encodeInstruction("")), encodeField(FIELD.INSTRUCTION, WIRE_TYPE.LEN, encodeInstruction('')),
encodeField(FIELD.UNKNOWN_4, WIRE_TYPE.VARINT, 1), encodeField(FIELD.UNKNOWN_4, WIRE_TYPE.VARINT, 1),
encodeField(FIELD.MODEL, WIRE_TYPE.LEN, encodeModel(modelName)), encodeField(FIELD.MODEL, WIRE_TYPE.LEN, encodeModel(modelName)),
encodeField(FIELD.WEB_TOOL, WIRE_TYPE.LEN, ""), encodeField(FIELD.WEB_TOOL, WIRE_TYPE.LEN, ''),
encodeField(FIELD.UNKNOWN_13, WIRE_TYPE.VARINT, 1), encodeField(FIELD.UNKNOWN_13, WIRE_TYPE.VARINT, 1),
encodeField(FIELD.CURSOR_SETTING, WIRE_TYPE.LEN, encodeCursorSetting()), encodeField(FIELD.CURSOR_SETTING, WIRE_TYPE.LEN, encodeCursorSetting()),
encodeField(FIELD.UNKNOWN_19, WIRE_TYPE.VARINT, 1), encodeField(FIELD.UNKNOWN_19, WIRE_TYPE.VARINT, 1),
@@ -133,16 +120,12 @@ export function encodeRequest(
WIRE_TYPE.VARINT, WIRE_TYPE.VARINT,
isAgentic ? UNIFIED_MODE.AGENT : UNIFIED_MODE.CHAT isAgentic ? UNIFIED_MODE.AGENT : UNIFIED_MODE.CHAT
), ),
encodeField(FIELD.UNKNOWN_47, WIRE_TYPE.LEN, ""), encodeField(FIELD.UNKNOWN_47, WIRE_TYPE.LEN, ''),
encodeField(FIELD.SHOULD_DISABLE_TOOLS, WIRE_TYPE.VARINT, isAgentic ? 0 : 1), encodeField(FIELD.SHOULD_DISABLE_TOOLS, WIRE_TYPE.VARINT, isAgentic ? 0 : 1),
encodeField(FIELD.THINKING_LEVEL, WIRE_TYPE.VARINT, thinkingLevel), encodeField(FIELD.THINKING_LEVEL, WIRE_TYPE.VARINT, thinkingLevel),
encodeField(FIELD.UNKNOWN_51, WIRE_TYPE.VARINT, 0), encodeField(FIELD.UNKNOWN_51, WIRE_TYPE.VARINT, 0),
encodeField(FIELD.UNKNOWN_53, WIRE_TYPE.VARINT, 1), encodeField(FIELD.UNKNOWN_53, WIRE_TYPE.VARINT, 1),
encodeField( encodeField(FIELD.UNIFIED_MODE_NAME, WIRE_TYPE.LEN, isAgentic ? 'Agent' : 'Ask'),
FIELD.UNIFIED_MODE_NAME,
WIRE_TYPE.LEN,
isAgentic ? "Agent" : "Ask"
),
]; ];
return concatArrays(...parts); return concatArrays(...parts);
src/cursor/cursor-translator.ts
+16 -20
View File
@@ -3,11 +3,7 @@
* Converts OpenAI messages to Cursor format * Converts OpenAI messages to Cursor format
*/ */
import type { import type { CursorMessage, CursorToolResult, CursorTool } from './cursor-protobuf-schema.js';
CursorMessage,
CursorToolResult,
CursorTool,
} from "./cursor-protobuf-schema.js";
/** OpenAI message format */ /** OpenAI message format */
interface OpenAIMessage { interface OpenAIMessage {
@@ -42,28 +38,28 @@ function convertMessages(messages: OpenAIMessage[]): CursorMessage[] {
for (let i = 0; i < messages.length; i++) { for (let i = 0; i < messages.length; i++) {
const msg = messages[i]; const msg = messages[i];
if (msg.role === "system") { if (msg.role === 'system') {
result.push({ result.push({
role: "user", role: 'user',
content: `[System Instructions]\n${msg.content}`, content: `[System Instructions]\n${msg.content}`,
}); });
continue; continue;
} }
if (msg.role === "tool") { if (msg.role === 'tool') {
let toolContent = ""; let toolContent = '';
if (typeof msg.content === "string") { if (typeof msg.content === 'string') {
toolContent = msg.content; toolContent = msg.content;
} else if (Array.isArray(msg.content)) { } else if (Array.isArray(msg.content)) {
for (const part of msg.content) { for (const part of msg.content) {
if (part.type === "text" && part.text) { if (part.type === 'text' && part.text) {
toolContent += part.text; toolContent += part.text;
} }
} }
} }
const toolName = msg.name || "tool"; const toolName = msg.name || 'tool';
const toolCallId = msg.tool_call_id || ""; const toolCallId = msg.tool_call_id || '';
// Accumulate tool result // Accumulate tool result
pendingToolResults.push({ pendingToolResults.push({
@@ -75,22 +71,22 @@ function convertMessages(messages: OpenAIMessage[]): CursorMessage[] {
continue; continue;
} }
if (msg.role === "user" || msg.role === "assistant") { if (msg.role === 'user' || msg.role === 'assistant') {
let content = ""; let content = '';
if (typeof msg.content === "string") { if (typeof msg.content === 'string') {
content = msg.content; content = msg.content;
} else if (Array.isArray(msg.content)) { } else if (Array.isArray(msg.content)) {
for (const part of msg.content) { for (const part of msg.content) {
if (part.type === "text" && part.text) { if (part.type === 'text' && part.text) {
content += part.text; content += part.text;
} }
} }
} }
// Keep tool_calls structure for assistant messages // Keep tool_calls structure for assistant messages
if (msg.role === "assistant" && msg.tool_calls && msg.tool_calls.length > 0) { if (msg.role === 'assistant' && msg.tool_calls && msg.tool_calls.length > 0) {
const assistantMsg: CursorMessage = { role: "assistant", content: "" }; const assistantMsg: CursorMessage = { role: 'assistant', content: '' };
if (content) { if (content) {
assistantMsg.content = content; assistantMsg.content = content;
} }
@@ -106,7 +102,7 @@ function convertMessages(messages: OpenAIMessage[]): CursorMessage[] {
} else if (content || pendingToolResults.length > 0) { } else if (content || pendingToolResults.length > 0) {
const msgObj: CursorMessage = { const msgObj: CursorMessage = {
role: msg.role, role: msg.role,
content: content || "", content: content || '',
}; };
// Attach pending tool results to this message // Attach pending tool results to this message
src/cursor/index.ts
+1 -6
View File
@@ -8,12 +8,7 @@
export * from './types'; export * from './types';
// Auth // Auth
export { export { autoDetectTokens, saveCredentials, loadCredentials, checkAuthStatus } from './cursor-auth';
autoDetectTokens,
saveCredentials,
loadCredentials,
checkAuthStatus,
} from './cursor-auth';
// Daemon // Daemon
export { isDaemonRunning, getDaemonStatus, startDaemon, stopDaemon } from './cursor-daemon'; export { isDaemonRunning, getDaemonStatus, startDaemon, stopDaemon } from './cursor-daemon';