MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-03-10 22:21:22 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix(ide): Update IDE extension to write auth token in env var (#14999)

Browse Source
This commit is contained in:
Shreya Keshive
2025-12-12 11:42:49 -05:00
committed by GitHub
parent 86134e9970
commit bb33e281c0
2 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
packages/vscode-ide-companion/src/ide-server.test.ts
View File

@@ -108,7 +108,7 @@ describe('IDEServer', () => {
await ideServer.start(mockContext);
const replaceMock = mockContext.environmentVariableCollection.replace;
expect(replaceMock).toHaveBeenCalledTimes(2);
expect(replaceMock).toHaveBeenCalledTimes(3);
expect(replaceMock).toHaveBeenNthCalledWith(
1,
@@ -127,6 +127,12 @@ describe('IDEServer', () => {
expectedWorkspacePaths,
);
expect(replaceMock).toHaveBeenNthCalledWith(
3,
'GEMINI_CLI_IDE_AUTH_TOKEN',
'test-auth-token',
);
const port = getPortFromMock(replaceMock);
const expectedPortFile = path.join(
'/tmp',
@@ -254,6 +260,10 @@ describe('IDEServer', () => {
'GEMINI_CLI_IDE_WORKSPACE_PATH',
expectedWorkspacePaths,
);
expect(replaceMock).toHaveBeenCalledWith(
'GEMINI_CLI_IDE_AUTH_TOKEN',
'test-auth-token',
);
const port = getPortFromMock(replaceMock);
const expectedPortFile = path.join(

5
packages/vscode-ide-companion/src/ide-server.ts
View File

@@ -38,6 +38,7 @@ class CORSError extends Error {
const MCP_SESSION_ID_HEADER = 'mcp-session-id';
const IDE_SERVER_PORT_ENV_VAR = 'GEMINI_CLI_IDE_SERVER_PORT';
const IDE_WORKSPACE_PATH_ENV_VAR = 'GEMINI_CLI_IDE_WORKSPACE_PATH';
const IDE_AUTH_TOKEN_ENV_VAR = 'GEMINI_CLI_IDE_AUTH_TOKEN';
interface WritePortAndWorkspaceArgs {
context: vscode.ExtensionContext;
@@ -70,6 +71,10 @@ async function writePortAndWorkspace({
IDE_WORKSPACE_PATH_ENV_VAR,
workspacePath,
);
context.environmentVariableCollection.replace(
IDE_AUTH_TOKEN_ENV_VAR,
authToken,
);
const content = JSON.stringify({
port,