ai-vault 3.3.0 → 3.5.0

package/dist/commands/setup.js

@@ -1,33 +1,39 @@
 /**
  * Setup Command - Interactive provider configuration
  */
-import * as clack from '@clack/prompts';
 import chalk from 'chalk';
 import { readFileSync } from 'fs';
 import { resolve } from 'path';
 import { saveProviderConfig, loadConfig, isProviderConfigured } from '../utils/config.js';
 import { getProvider } from '../providers/index.js';
-const VALID_PROVIDERS = ['grok-web', 'grok-x', 'chatgpt', 'claude'];
+import { createCliUI } from '../utils/cli-ui.js';
+import * as clack from '@clack/prompts';
+const VALID_PROVIDERS = ['grok-web', 'grok-x', 'chatgpt', 'claude', 'gemini'];
 export async function setupCommand(options = {}) {
-    clack.intro(chalk.bold.blue('AI Vault Setup'));
+    const ui = createCliUI();
+    ui.intro(chalk.bold.blue('AI Vault Setup'));
     // Check existing configuration
     const config = await loadConfig();
     const configuredProviders = Object.keys(config.providers);
     if (configuredProviders.length > 0) {
-        clack.log.info(`Currently configured: ${configuredProviders.join(', ')}`);
+        ui.log.info(`Currently configured: ${configuredProviders.join(', ')}`);
     }
     let providerName;
     // Use provider from argument if provided
     if (options.provider) {
         if (!VALID_PROVIDERS.includes(options.provider)) {
-            clack.log.error(`Invalid provider: ${options.provider}`);
-            clack.log.info(`Valid providers: ${VALID_PROVIDERS.join(', ')}`);
+            ui.log.error(`Invalid provider: ${options.provider}`);
+            ui.log.info(`Valid providers: ${VALID_PROVIDERS.join(', ')}`);
             process.exit(1);
         }
         providerName = options.provider;
-        clack.log.info(`Configuring provider: ${providerName}`);
+        ui.log.info(`Configuring provider: ${providerName}`);
     }
     else {
+        if (!ui.isInteractive) {
+            ui.log.error('Provider selection requires a TTY. Pass a provider name explicitly.');
+            process.exit(1);
+        }
         // Select provider interactively
         const provider = await clack.select({
             message: 'Which provider do you want to configure?',
@@ -36,22 +42,32 @@ export async function setupCommand(options = {}) {
                 { value: 'grok-x', label: 'Grok on X (x.com)', hint: 'x.com/grok' },
                 { value: 'chatgpt', label: 'ChatGPT (OpenAI)', hint: 'chatgpt.com' },
                 { value: 'claude', label: 'Claude (Anthropic)', hint: 'claude.ai' },
+                { value: 'gemini', label: 'Gemini (Google)', hint: 'gemini.google.com' },
             ],
         });
         if (clack.isCancel(provider)) {
-            clack.cancel('Setup cancelled');
+            ui.cancel('Setup cancelled');
             process.exit(0);
         }
         providerName = provider;
     }
     // Check if already configured
     if (await isProviderConfigured(providerName)) {
-        const overwrite = await clack.confirm({
-            message: `${providerName} is already configured. Overwrite?`,
-        });
-        if (clack.isCancel(overwrite) || !overwrite) {
-            clack.cancel('Setup cancelled');
-            process.exit(0);
+        if (!ui.isInteractive) {
+            if (!options.cookiesFile) {
+                ui.log.error(`${providerName} is already configured. Run setup in a TTY to overwrite or pass --cookies-file.`);
+                process.exit(1);
+            }
+            ui.log.warn(`${providerName} is already configured. Overwriting with provided cookies file.`);
+        }
+        else {
+            const overwrite = await clack.confirm({
+                message: `${providerName} is already configured. Overwrite?`,
+            });
+            if (clack.isCancel(overwrite) || !overwrite) {
+                ui.cancel('Setup cancelled');
+                process.exit(0);
+            }
         }
     }
     // Configure based on provider
@@ -66,22 +82,30 @@ export async function setupCommand(options = {}) {
         case 'claude':
             await setupClaude(options);
             break;
+        case 'gemini':
+            await setupGemini(options);
+            break;
         default:
-            clack.log.error(`${providerName} provider is not yet implemented`);
+            ui.log.error(`${providerName} provider is not yet implemented`);
             process.exit(1);
     }
-    clack.outro(chalk.green('✓ Setup complete! Run `ai-vault archive` to start archiving.'));
+    ui.outro(chalk.green('✓ Setup complete! Run `ai-vault archive` to start archiving.'));
 }
 async function setupGrok(providerName, options) {
+    const ui = createCliUI();
     const displayName = providerName === 'grok-web' ? 'Grok (grok.com)' : 'Grok on X (x.com/grok)';
-    clack.log.step(`Configuring ${displayName}`);
+    ui.log.step(`Configuring ${displayName}`);
     // If cookies file provided, skip to cookie auth
     let authMethod;
     if (options.cookiesFile) {
         authMethod = 'cookies';
-        clack.log.info(`Using cookies from file: ${options.cookiesFile}`);
+        ui.log.info(`Using cookies from file: ${options.cookiesFile}`);
     }
     else {
+        if (!ui.isInteractive) {
+            ui.log.error('Interactive auth selection requires a TTY. Pass --cookies-file.');
+            process.exit(1);
+        }
         // Choose auth method
         const selected = await clack.select({
             message: 'Authentication method:',
@@ -91,7 +115,7 @@ async function setupGrok(providerName, options) {
             ],
         });
         if (clack.isCancel(selected)) {
-            clack.cancel('Setup cancelled');
+            ui.cancel('Setup cancelled');
             process.exit(0);
         }
         authMethod = selected;
@@ -143,21 +167,25 @@ async function setupGrok(providerName, options) {
                     // Already in object format
                     cookies = parsed;
                 }
-                clack.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
+                ui.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
             }
             catch (error) {
-                clack.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
+                ui.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
                 process.exit(1);
             }
         }
         else {
+            if (!ui.isInteractive) {
+                ui.log.error('Interactive cookie entry requires a TTY. Pass --cookies-file.');
+                process.exit(1);
+            }
             // Interactive prompt for each cookie
             const websiteUrl = providerName === 'grok-web' ? 'grok.com' : 'x.com/grok';
-            clack.log.info(`To get cookies from ${displayName}:`);
-            clack.log.info(`1. Open ${websiteUrl} in your browser and log in`);
-            clack.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
-            clack.log.info(`3. Go to Application → Cookies → https://${websiteUrl.split('/')[0]}`);
-            clack.log.info('4. Find each cookie below and copy its VALUE');
+            ui.log.info(`To get cookies from ${displayName}:`);
+            ui.log.info(`1. Open ${websiteUrl} in your browser and log in`);
+            ui.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
+            ui.log.info(`3. Go to Application → Cookies → https://${websiteUrl.split('/')[0]}`);
+            ui.log.info('4. Find each cookie below and copy its VALUE');
             console.log();
             cookies = {};
             // Prompt for sso cookie first
@@ -175,7 +203,7 @@ async function setupGrok(providerName, options) {
                 },
             });
             if (clack.isCancel(ssoValue)) {
-                clack.cancel('Setup cancelled');
+                ui.cancel('Setup cancelled');
                 process.exit(0);
             }
             cookies['sso'] = ssoValue.trim();
@@ -196,14 +224,14 @@ async function setupGrok(providerName, options) {
                 },
             });
             if (clack.isCancel(ssoRwValue)) {
-                clack.cancel('Setup cancelled');
+                ui.cancel('Setup cancelled');
                 process.exit(0);
             }
             // Use sso value if sso-rw is empty
             const ssoRwTrimmed = ssoRwValue ? ssoRwValue.trim() : '';
             cookies['sso-rw'] = ssoRwTrimmed.length > 0 ? ssoRwTrimmed : cookies['sso'];
             if (ssoRwTrimmed.length === 0) {
-                clack.log.info('Using same value as "sso" for "sso-rw"');
+                ui.log.info('Using same value as "sso" for "sso-rw"');
             }
             // Prompt for stblid cookie
             const stblidValue = await clack.text({
@@ -221,11 +249,11 @@ async function setupGrok(providerName, options) {
                 },
             });
             if (clack.isCancel(stblidValue)) {
-                clack.cancel('Setup cancelled');
+                ui.cancel('Setup cancelled');
                 process.exit(0);
             }
             cookies['stblid'] = stblidValue.trim();
-            clack.log.success('All cookies collected successfully!');
+            ui.log.success('All cookies collected successfully!');
         }
         providerConfig = {
             providerName: providerName,
@@ -234,7 +262,7 @@ async function setupGrok(providerName, options) {
         };
     }
     // Test authentication
-    const spinner = clack.spinner();
+    const spinner = ui.spinner();
     spinner.start('Testing authentication...');
     try {
         const provider = getProvider(providerConfig.providerName);
@@ -243,23 +271,27 @@ async function setupGrok(providerName, options) {
         await provider.cleanup?.();
         if (!isAuth) {
             spinner.stop('Authentication failed');
-            clack.log.error('Could not authenticate with provided credentials');
+            ui.log.error('Could not authenticate with provided credentials');
             process.exit(1);
         }
         spinner.stop('✓ Authentication successful!');
     }
     catch (error) {
         spinner.stop('Authentication failed');
-        clack.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
+        ui.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
         process.exit(1);
     }
     // Save configuration
     await saveProviderConfig(providerConfig);
-    clack.log.success('Configuration saved to ~/.ai-vault/config.json');
+    ui.log.success('Configuration saved to ~/.ai-vault/config.json');
     // Configure archive directory
     const config = await loadConfig();
     if (!config.settings?.archiveDir) {
         console.log();
+        if (!ui.isInteractive) {
+            ui.log.info('Using default archive directory (~/ai-vault-data)');
+            return;
+        }
         const customDir = await clack.text({
             message: 'Archive directory (press Enter for default):',
             placeholder: '~/ai-vault-data',
@@ -269,12 +301,13 @@ async function setupGrok(providerName, options) {
             config.settings.archiveDir = customDir;
             const { saveConfig } = await import('../utils/config.js');
             await saveConfig(config);
-            clack.log.info(`Archive directory set to: ${customDir}`);
+            ui.log.info(`Archive directory set to: ${customDir}`);
         }
     }
 }
 async function setupChatGPT(options) {
-    clack.log.step('Configuring ChatGPT (OpenAI)');
+    const ui = createCliUI();
+    ui.log.step('Configuring ChatGPT (OpenAI)');
     let cookies;
     // Read from file if provided
     if (options.cookiesFile) {
@@ -290,33 +323,28 @@ async function setupChatGPT(options) {
                 // Already in object format
                 cookies = parsed;
             }
-            clack.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
+            ui.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
         }
         catch (error) {
-            clack.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            ui.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
             process.exit(1);
         }
     }
     else {
+        if (!ui.isInteractive) {
+            ui.log.error('Interactive cookie entry requires a TTY. Pass --cookies-file.');
+            process.exit(1);
+        }
         // Interactive prompt for each cookie
-        clack.log.info('To get cookies from ChatGPT:');
-        clack.log.info('1. Open chatgpt.com in your browser and log in');
-        clack.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
-        clack.log.info('3. Go to Application → Cookies → https://chatgpt.com');
-        clack.log.info('4. Find the cookies below and copy their VALUES');
-        clack.log.info('');
-        clack.log.warn('⚠️  Important: Some cookies may not be present for all users.');
-        clack.log.warn('   Press Enter to skip optional cookies.');
+        ui.log.info('To get cookies from ChatGPT:');
+        ui.log.info('1. Open chatgpt.com in your browser and log in');
+        ui.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
+        ui.log.info('3. Go to Application → Cookies → https://chatgpt.com');
+        ui.log.info('4. Find the cookies below and copy their VALUES');
+        ui.log.info('');
+        ui.log.warn('⚠️  Important: Some cookies may not be present for all users.');
+        ui.log.warn('   Press Enter to skip optional cookies.');
         console.log();
-        // Define required cookies for ChatGPT
-        const requiredCookies = [
-            {
-                name: '__Secure-next-auth.session-token',
-                description: 'Session token (REQUIRED)',
-                hint: 'Long JWT-like token',
-                required: true,
-            },
-        ];
         const optionalCookies = [
             {
                 name: '__Host-next-auth.csrf-token',
@@ -332,23 +360,48 @@ async function setupChatGPT(options) {
             },
         ];
         cookies = {};
-        // Prompt for required cookies
-        for (const cookieInfo of requiredCookies) {
-            const value = await clack.text({
-                message: `Enter value for cookie "${cookieInfo.name}" (${cookieInfo.description}):`,
-                placeholder: cookieInfo.hint,
-                validate: (val) => {
-                    if (!val || val.trim().length === 0) {
-                        return `${cookieInfo.name} is required`;
-                    }
-                    return undefined;
-                },
-            });
-            if (clack.isCancel(value)) {
-                clack.cancel('Setup cancelled');
-                process.exit(0);
-            }
-            cookies[cookieInfo.name] = value.trim();
+        ui.log.warn('   Some accounts now split the session token into .0 and .1 cookies.');
+        ui.log.warn('   Provide either the single session token or both .0 and .1 parts.');
+        console.log();
+        const baseTokenValue = await clack.text({
+            message: 'Enter value for cookie "__Secure-next-auth.session-token" (optional if using .0/.1):',
+            placeholder: 'Long JWT-like token - press Enter to use .0/.1 cookies',
+        });
+        if (clack.isCancel(baseTokenValue)) {
+            ui.cancel('Setup cancelled');
+            process.exit(0);
+        }
+        const tokenPart0Value = await clack.text({
+            message: 'Enter value for cookie "__Secure-next-auth.session-token.0" (optional):',
+            placeholder: 'First chunk - press Enter to skip if not found',
+        });
+        if (clack.isCancel(tokenPart0Value)) {
+            ui.cancel('Setup cancelled');
+            process.exit(0);
+        }
+        const tokenPart1Value = await clack.text({
+            message: 'Enter value for cookie "__Secure-next-auth.session-token.1" (optional):',
+            placeholder: 'Second chunk - press Enter to skip if not found',
+        });
+        if (clack.isCancel(tokenPart1Value)) {
+            ui.cancel('Setup cancelled');
+            process.exit(0);
+        }
+        const baseToken = baseTokenValue?.trim();
+        const tokenPart0 = tokenPart0Value?.trim();
+        const tokenPart1 = tokenPart1Value?.trim();
+        if (!baseToken && (!tokenPart0 || !tokenPart1)) {
+            ui.log.error('Missing required session token. Provide "__Secure-next-auth.session-token" or both ".0" and ".1" parts.');
+            process.exit(1);
+        }
+        if (baseToken) {
+            cookies['__Secure-next-auth.session-token'] = baseToken;
+        }
+        if (tokenPart0) {
+            cookies['__Secure-next-auth.session-token.0'] = tokenPart0;
+        }
+        if (tokenPart1) {
+            cookies['__Secure-next-auth.session-token.1'] = tokenPart1;
         }
         // Prompt for optional cookies
         for (const cookieInfo of optionalCookies) {
@@ -357,7 +410,7 @@ async function setupChatGPT(options) {
                 placeholder: `${cookieInfo.hint} (press Enter to skip)`,
             });
             if (clack.isCancel(value)) {
-                clack.cancel('Setup cancelled');
+                ui.cancel('Setup cancelled');
                 process.exit(0);
             }
             const trimmedValue = value?.trim();
@@ -365,7 +418,7 @@ async function setupChatGPT(options) {
                 cookies[cookieInfo.name] = trimmedValue;
             }
         }
-        clack.log.success(`Collected ${Object.keys(cookies).length} cookie${Object.keys(cookies).length > 1 ? 's' : ''} successfully!`);
+        ui.log.success(`Collected ${Object.keys(cookies).length} cookie${Object.keys(cookies).length > 1 ? 's' : ''} successfully!`);
     }
     const providerConfig = {
         providerName: 'chatgpt',
@@ -373,7 +426,7 @@ async function setupChatGPT(options) {
         cookies,
     };
     // Test authentication
-    const spinner = clack.spinner();
+    const spinner = ui.spinner();
     spinner.start('Testing authentication...');
     try {
         const provider = getProvider(providerConfig.providerName);
@@ -382,23 +435,27 @@ async function setupChatGPT(options) {
         await provider.cleanup?.();
         if (!isAuth) {
             spinner.stop('Authentication failed');
-            clack.log.error('Could not authenticate with provided credentials');
+            ui.log.error('Could not authenticate with provided credentials');
             process.exit(1);
         }
         spinner.stop('✓ Authentication successful!');
     }
     catch (error) {
         spinner.stop('Authentication failed');
-        clack.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
+        ui.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
         process.exit(1);
     }
     // Save configuration
     await saveProviderConfig(providerConfig);
-    clack.log.success('Configuration saved to ~/.ai-vault/config.json');
+    ui.log.success('Configuration saved to ~/.ai-vault/config.json');
     // Configure archive directory
     const config = await loadConfig();
     if (!config.settings?.archiveDir) {
         console.log();
+        if (!ui.isInteractive) {
+            ui.log.info('Using default archive directory (~/ai-vault-data)');
+            return;
+        }
         const customDir = await clack.text({
             message: 'Archive directory (press Enter for default):',
             placeholder: '~/ai-vault-data',
@@ -408,12 +465,147 @@ async function setupChatGPT(options) {
             config.settings.archiveDir = customDir;
             const { saveConfig } = await import('../utils/config.js');
             await saveConfig(config);
-            clack.log.info(`Archive directory set to: ${customDir}`);
+            ui.log.info(`Archive directory set to: ${customDir}`);
+        }
+    }
+}
+async function setupGemini(options) {
+    const ui = createCliUI();
+    ui.log.step('Configuring Gemini (Google)');
+    let cookies;
+    // Read from file if provided
+    if (options.cookiesFile) {
+        try {
+            const filePath = resolve(options.cookiesFile);
+            const fileContent = readFileSync(filePath, 'utf-8');
+            const parsed = JSON.parse(fileContent);
+            if (Array.isArray(parsed)) {
+                cookies = Object.fromEntries(parsed.map((cookie) => [cookie.name, cookie.value]));
+            }
+            else {
+                cookies = parsed;
+            }
+            ui.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
+        }
+        catch (error) {
+            ui.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            process.exit(1);
+        }
+    }
+    else {
+        if (!ui.isInteractive) {
+            ui.log.error('Interactive cookie entry requires a TTY. Pass --cookies-file.');
+            process.exit(1);
+        }
+        ui.log.info('Gemini requires the following cookies from gemini.google.com:');
+        console.log();
+        ui.log.info('  Required:');
+        ui.log.info('    SID, HSID, SSID, APISID, SAPISID, SIDCC, NID');
+        ui.log.info('    __Secure-1PSID, __Secure-1PSIDTS, __Secure-1PSIDCC, __Secure-1PAPISID');
+        ui.log.info('    __Secure-3PSID, __Secure-3PSIDTS, __Secure-3PSIDCC, __Secure-3PAPISID');
+        ui.log.info('  Optional (but recommended):');
+        ui.log.info('    SOCS (required in EU), AEC, S, NID, __Secure-ENID, __Secure-STRP,');
+        ui.log.info('    SEARCH_SAMESITE, __Secure-BUCKET, _ga, _gcl_au');
+        console.log();
+        ui.log.info('How to export all cookies at once:');
+        ui.log.info('  1. Open gemini.google.com in your browser and log in');
+        ui.log.info('  2. Install "Cookie-Editor" or "EditThisCookie" browser extension');
+        ui.log.info('  3. Click the extension icon → Export as JSON');
+        ui.log.info('  4. Paste the JSON below, OR save to a file and use:');
+        ui.log.info('     ai-vault setup gemini --cookies-file <path>');
+        console.log();
+        const jsonInput = await clack.text({
+            message: 'Paste cookies JSON (array or {"name":"value"} object):',
+            placeholder: '[{"name":"SID","value":"..."},...]  or  {"SID":"...","HSID":"..."}',
+            validate: (val) => {
+                if (!val || val.trim().length === 0)
+                    return 'Cookies JSON is required';
+                try {
+                    const parsed = JSON.parse(val.trim());
+                    if (typeof parsed !== 'object' || parsed === null)
+                        return 'Must be a JSON object or array';
+                    const keys = Array.isArray(parsed)
+                        ? parsed.map((c) => c?.name).filter(Boolean)
+                        : Object.keys(parsed);
+                    if (keys.length === 0)
+                        return 'No cookies found in JSON';
+                    const required = ['SID', '__Secure-1PSID', 'SAPISID'];
+                    const missing = required.filter((k) => !keys.includes(k));
+                    if (missing.length === required.length) {
+                        return `Missing key cookies: ${required.join(', ')}`;
+                    }
+                }
+                catch {
+                    return 'Invalid JSON — copy the full export from your browser extension';
+                }
+            },
+        });
+        if (clack.isCancel(jsonInput)) {
+            ui.cancel('Setup cancelled');
+            process.exit(0);
+        }
+        // JSON validity already confirmed by the validator above
+        const parsed = JSON.parse(jsonInput.trim());
+        if (Array.isArray(parsed)) {
+            cookies = Object.fromEntries(parsed.map((c) => [c.name, c.value]));
+        }
+        else {
+            cookies = parsed;
+        }
+        ui.log.success(`Loaded ${Object.keys(cookies).length} cookies`);
+    }
+    const providerConfig = {
+        providerName: 'gemini',
+        authMethod: 'cookies',
+        cookies,
+    };
+    // Test authentication
+    const spinner = ui.spinner();
+    spinner.start('Testing authentication...');
+    try {
+        const provider = getProvider(providerConfig.providerName);
+        await provider.authenticate(providerConfig);
+        const isAuth = await provider.isAuthenticated();
+        await provider.cleanup?.();
+        if (!isAuth) {
+            spinner.stop('Authentication failed');
+            ui.log.error('Could not authenticate with provided credentials');
+            process.exit(1);
+        }
+        spinner.stop('✓ Authentication successful!');
+    }
+    catch (error) {
+        spinner.stop('Authentication failed');
+        ui.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
+        process.exit(1);
+    }
+    // Save configuration
+    await saveProviderConfig(providerConfig);
+    ui.log.success('Configuration saved to ~/.ai-vault/config.json');
+    // Configure archive directory
+    const config = await loadConfig();
+    if (!config.settings?.archiveDir) {
+        console.log();
+        if (!ui.isInteractive) {
+            ui.log.info('Using default archive directory (~/ai-vault-data)');
+            return;
+        }
+        const customDir = await clack.text({
+            message: 'Archive directory (press Enter for default):',
+            placeholder: '~/ai-vault-data',
+        });
+        if (!clack.isCancel(customDir) && customDir) {
+            config.settings = config.settings || {};
+            config.settings.archiveDir = customDir;
+            const { saveConfig } = await import('../utils/config.js');
+            await saveConfig(config);
+            ui.log.info(`Archive directory set to: ${customDir}`);
         }
     }
 }
 async function setupClaude(options) {
-    clack.log.step('Configuring Claude (Anthropic)');
+    const ui = createCliUI();
+    ui.log.step('Configuring Claude (Anthropic)');
     let cookies;
     // Read from file if provided
     if (options.cookiesFile) {
@@ -429,20 +621,24 @@ async function setupClaude(options) {
                 // Already in object format
                 cookies = parsed;
             }
-            clack.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
+            ui.log.success(`Loaded ${Object.keys(cookies).length} cookies from file`);
         }
         catch (error) {
-            clack.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            ui.log.error(`Failed to read cookies file: ${error instanceof Error ? error.message : 'Unknown error'}`);
             process.exit(1);
         }
     }
     else {
+        if (!ui.isInteractive) {
+            ui.log.error('Interactive cookie entry requires a TTY. Pass --cookies-file.');
+            process.exit(1);
+        }
         // Interactive prompt for sessionKey cookie
-        clack.log.info('To get your sessionKey cookie from Claude:');
-        clack.log.info('1. Open claude.ai in your browser and log in');
-        clack.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
-        clack.log.info('3. Go to Application → Cookies → https://claude.ai');
-        clack.log.info('4. Find the "sessionKey" cookie and copy its VALUE');
+        ui.log.info('To get your sessionKey cookie from Claude:');
+        ui.log.info('1. Open claude.ai in your browser and log in');
+        ui.log.info('2. Open Developer Tools (F12 or Cmd+Option+I)');
+        ui.log.info('3. Go to Application → Cookies → https://claude.ai');
+        ui.log.info('4. Find the "sessionKey" cookie and copy its VALUE');
         console.log();
         const sessionKey = await clack.text({
             message: 'Enter value for cookie "sessionKey":',
@@ -462,13 +658,13 @@ async function setupClaude(options) {
             },
         });
         if (clack.isCancel(sessionKey)) {
-            clack.cancel('Setup cancelled');
+            ui.cancel('Setup cancelled');
             process.exit(0);
         }
         cookies = {
             sessionKey: sessionKey.trim(),
         };
-        clack.log.success('sessionKey collected successfully!');
+        ui.log.success('sessionKey collected successfully!');
     }
     const providerConfig = {
         providerName: 'claude',
@@ -476,7 +672,7 @@ async function setupClaude(options) {
         cookies,
     };
     // Test authentication
-    const spinner = clack.spinner();
+    const spinner = ui.spinner();
     spinner.start('Testing authentication...');
     try {
         const provider = getProvider(providerConfig.providerName);
@@ -485,23 +681,27 @@ async function setupClaude(options) {
         await provider.cleanup?.();
         if (!isAuth) {
             spinner.stop('Authentication failed');
-            clack.log.error('Could not authenticate with provided credentials');
+            ui.log.error('Could not authenticate with provided credentials');
             process.exit(1);
         }
         spinner.stop('✓ Authentication successful!');
     }
     catch (error) {
         spinner.stop('Authentication failed');
-        clack.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
+        ui.log.error('Error: ' + (error instanceof Error ? error.message : 'Unknown error'));
         process.exit(1);
     }
     // Save configuration
     await saveProviderConfig(providerConfig);
-    clack.log.success('Configuration saved to ~/.ai-vault/config.json');
+    ui.log.success('Configuration saved to ~/.ai-vault/config.json');
     // Configure archive directory
     const config = await loadConfig();
     if (!config.settings?.archiveDir) {
         console.log();
+        if (!ui.isInteractive) {
+            ui.log.info('Using default archive directory (~/ai-vault-data)');
+            return;
+        }
         const customDir = await clack.text({
             message: 'Archive directory (press Enter for default):',
             placeholder: '~/ai-vault-data',
@@ -511,7 +711,7 @@ async function setupClaude(options) {
             config.settings.archiveDir = customDir;
             const { saveConfig } = await import('../utils/config.js');
             await saveConfig(config);
-            clack.log.info(`Archive directory set to: ${customDir}`);
+            ui.log.info(`Archive directory set to: ${customDir}`);
         }
     }
 }