mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-18 10:01:29 -07:00
* Wire through passthrough commands * Add default passthrough commands * Clean up config passing to useGeminiStream
95 lines
1.9 KiB
TypeScript
95 lines
1.9 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import * as dotenv from 'dotenv';
|
|
import * as fs from 'node:fs';
|
|
import * as path from 'node:path';
|
|
import process from 'node:process';
|
|
|
|
const DEFAULT_PASSTHROUGH_COMMANDS = ['ls', 'git', 'npm'];
|
|
|
|
export class Config {
|
|
private apiKey: string;
|
|
private model: string;
|
|
private targetDir: string;
|
|
private debugMode: boolean;
|
|
private passthroughCommands: string[];
|
|
|
|
constructor(
|
|
apiKey: string,
|
|
model: string,
|
|
targetDir: string,
|
|
debugMode: boolean,
|
|
passthroughCommands?: string[],
|
|
) {
|
|
this.apiKey = apiKey;
|
|
this.model = model;
|
|
this.targetDir = targetDir;
|
|
this.debugMode = debugMode;
|
|
this.passthroughCommands =
|
|
passthroughCommands || DEFAULT_PASSTHROUGH_COMMANDS;
|
|
}
|
|
|
|
getApiKey(): string {
|
|
return this.apiKey;
|
|
}
|
|
|
|
getModel(): string {
|
|
return this.model;
|
|
}
|
|
|
|
getTargetDir(): string {
|
|
return this.targetDir;
|
|
}
|
|
|
|
getDebugMode(): boolean {
|
|
return this.debugMode;
|
|
}
|
|
|
|
getPassthroughCommands(): string[] {
|
|
return this.passthroughCommands;
|
|
}
|
|
}
|
|
|
|
function findEnvFile(startDir: string): string | null {
|
|
let currentDir = path.resolve(startDir);
|
|
while (true) {
|
|
const envPath = path.join(currentDir, '.env');
|
|
if (fs.existsSync(envPath)) {
|
|
return envPath;
|
|
}
|
|
const parentDir = path.dirname(currentDir);
|
|
if (parentDir === currentDir || !parentDir) {
|
|
return null;
|
|
}
|
|
currentDir = parentDir;
|
|
}
|
|
}
|
|
|
|
export function loadEnvironment(): void {
|
|
const envFilePath = findEnvFile(process.cwd());
|
|
if (!envFilePath) {
|
|
return;
|
|
}
|
|
dotenv.config({ path: envFilePath });
|
|
}
|
|
|
|
export function createServerConfig(
|
|
apiKey: string,
|
|
model: string,
|
|
targetDir: string,
|
|
debugMode: boolean,
|
|
passthroughCommands?: string[],
|
|
): Config {
|
|
return new Config(
|
|
apiKey,
|
|
model,
|
|
path.resolve(targetDir),
|
|
debugMode,
|
|
passthroughCommands,
|
|
);
|
|
}
|