diff options
Diffstat (limited to 'packages/cli/src/config')
| -rw-r--r-- | packages/cli/src/config/args.ts | 34 | ||||
| -rw-r--r-- | packages/cli/src/config/env.ts | 46 |
2 files changed, 80 insertions, 0 deletions
diff --git a/packages/cli/src/config/args.ts b/packages/cli/src/config/args.ts new file mode 100644 index 00000000..45f654db --- /dev/null +++ b/packages/cli/src/config/args.ts @@ -0,0 +1,34 @@ +import yargs from 'yargs/yargs'; +import { hideBin } from 'yargs/helpers'; + +export interface CliArgs { + target_dir: string | undefined; + _: (string | number)[]; // Captures positional arguments + // Add other expected args here if needed + // e.g., verbose?: boolean; +} + +export async function parseArguments(): Promise<CliArgs> { + const argv = await yargs(hideBin(process.argv)) + .option('target_dir', { + alias: 'd', + type: 'string', + description: + 'The target directory for Gemini operations. Defaults to the current working directory.', + }) + .help() + .alias('h', 'help') + .strict() // Keep strict mode to error on unknown options + .parseAsync(); + + // Handle warnings for extra arguments here + if (argv._ && argv._.length > 0) { + console.warn( + `Warning: Additional arguments provided (${argv._.join(', ')}), but will be ignored.` + ); + } + + // Cast to the interface to ensure the structure aligns with expectations + // Use `unknown` first for safer casting if types might not perfectly match + return argv as unknown as CliArgs; +}
\ No newline at end of file diff --git a/packages/cli/src/config/env.ts b/packages/cli/src/config/env.ts new file mode 100644 index 00000000..8a15fd6e --- /dev/null +++ b/packages/cli/src/config/env.ts @@ -0,0 +1,46 @@ +import * as dotenv from 'dotenv'; +import * as fs from 'node:fs'; +import * as path from 'node:path'; +import process from 'node:process'; + +function findEnvFile(startDir: string): string | null { + // Start search from the provided directory (e.g., current working directory) + let currentDir = path.resolve(startDir); // Ensure absolute path + 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 { + // Start searching from the current working directory by default + const envFilePath = findEnvFile(process.cwd()); + + if (!envFilePath) { + return; + } + + dotenv.config({ path: envFilePath }); + + if (!process.env.GEMINI_API_KEY) { + console.error('Error: GEMINI_API_KEY environment variable is not set in the loaded .env file.'); + process.exit(1); + } +} + +export function getApiKey(): string { + loadEnvironment(); + const apiKey = process.env.GEMINI_API_KEY; + if (!apiKey) { + throw new Error('GEMINI_API_KEY is missing. Ensure loadEnvironment() was called successfully.'); + } + return apiKey; +}
\ No newline at end of file |
