Refactor config file handling in InitCommand

Author: naheel0

git-ai/src/commands/InitCommand.ts

@@ -12,7 +12,6 @@ import { logger } from '../utils/logger.js';
 async function readSecretInput(rl: readline.Interface, prompt: string): Promise<string> {
   const rlAny = rl as any;
   const originalWrite = rlAny._writeToOutput;
-  // Suppress character echoing: only allow the initial prompt to be written
   let promptWritten = false;
   rlAny._writeToOutput = function _writeToOutput(str: string) {
     if (!promptWritten) {
@@ -38,6 +37,7 @@ export async function initCommand() {
   console.log('🚀 Welcome to AI-Git-Terminal Setup\n');
 
   try {
+    // --- Step 1: Read API Key ---
     let apiKey = '';
     while (!apiKey) {
       const apiKeyInput = await readSecretInput(rl, '🔑 Enter your Gemini API Key: ');
@@ -47,56 +47,63 @@ export async function initCommand() {
       }
     }
 
+    // --- Step 2: Read model name ---
     const modelInput = await rl.question('🤖 Enter model name (default: gemini-1.5-flash): ');
     const model = modelInput.trim() || 'gemini-1.5-flash';
 
+    // --- Step 3: Build config object ---
     const newConfig: Config = {
-      ai: {
-        provider: 'gemini',
-        apiKey,
-        model: model,
-      },
-      git: {
-        autoStage: false,
-      },
-      ui: {
-        theme: 'dark',
-        showIcons: true,
-      },
+      ai: { provider: 'gemini', apiKey, model },
+      git: { autoStage: false },
+      ui: { theme: 'dark', showIcons: true },
     };
 
-    // Validate with Zod and persist with restricted permissions (mode 0o600)
+    // Validate with Zod
     ConfigSchema.parse(newConfig);
 
     const configPath = path.join(os.homedir(), '.aigitrc');
 
-    if (fs.existsSync(configPath)) {
-      const overwriteChoice = (await rl.question(
-        '⚠️ Existing config found. Choose [o]verwrite, [b]ackup then replace, or [c]ancel: '
-      )).trim().toLowerCase();
+    // --- Step 4: Attempt atomic creation ---
+    try {
+      const fd = fs.openSync(configPath, fs.O_CREAT | fs.O_EXCL | fs.O_RDWR, 0o600);
+      fs.writeFileSync(fd, JSON.stringify(newConfig, null, 2));
+      fs.closeSync(fd);
+      console.log(`\n✅ Configuration saved to ${configPath}`);
+      console.log('Try running: ai-git commit');
+      return;
+    } catch (err: any) {
+      if (err.code !== 'EEXIST') throw err;
+      // File already exists, proceed to backup/overwrite prompt
+    }
 
-      if (overwriteChoice === 'b' || overwriteChoice === 'backup') {
-        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
-        const backupPath = `${configPath}.bak-${timestamp}`;
-        fs.renameSync(configPath, backupPath);
-        console.log(`📦 Existing config backed up to ${backupPath}`);
-      } else if (overwriteChoice === 'o' || overwriteChoice === 'overwrite') {
-        console.log('📝 Overwriting existing config file.');
-      } else {
-        console.log('🚫 Initialization canceled. Existing config left unchanged.');
-        return;
-      }
+    // --- Step 5: Handle existing file ---
+    const overwriteChoice = (await rl.question(
+      '⚠️ Existing config found. Choose [o]verwrite, [b]ackup then replace, or [c]ancel: '
+    )).trim().toLowerCase();
+
+    if (overwriteChoice === 'b' || overwriteChoice === 'backup') {
+      const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+      const backupPath = `${configPath}.bak-${timestamp}`;
+      fs.renameSync(configPath, backupPath);
+      console.log(`📦 Existing config backed up to ${backupPath}`);
+      // Now write new config
+      fs.writeFileSync(configPath, JSON.stringify(newConfig, null, 2), { mode: 0o600 });
+    } else if (overwriteChoice === 'o' || overwriteChoice === 'overwrite') {
+      fs.writeFileSync(configPath, JSON.stringify(newConfig, null, 2), { mode: 0o600 });
+      console.log('📝 Overwriting existing config file.');
+    } else {
+      console.log('🚫 Initialization canceled. Existing config left unchanged.');
+      return;
     }
 
-    fs.writeFileSync(configPath, JSON.stringify(newConfig, null, 2), { mode: 0o600 });
     fs.chmodSync(configPath, 0o600);
-
     console.log(`\n✅ Configuration saved to ${configPath}`);
     console.log('Try running: ai-git commit');
+
   } catch (error) {
     logger.error('Failed to save configuration: ' + (error instanceof Error ? error.message : String(error)));
     console.error('\n❌ Invalid input or failed to write config file.');
   } finally {
     rl.close();
   }
-}
+}