summaryrefslogtreecommitdiff
path: root/packages/cli/src/config
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/config')
-rw-r--r--packages/cli/src/config/args.ts34
-rw-r--r--packages/cli/src/config/env.ts46
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