add support for nile auth env and nile local connect

Author: sriramsub

src/commands/auth.ts

@@ -41,9 +41,66 @@ export function createAuthCommand(getOptions: GetOptions): Command {
     .addHelpText('after', `
 Examples:
   ${formatCommand('nile auth quickstart --nextjs')}     Create a Next.js app with Nile auth
+  ${formatCommand('nile auth env')}                     Generate Nile environment variables
   
 ${getGlobalOptionsHelp()}`);
 
+  auth
+    .command('env')
+    .description('Generate Nile environment variables')
+    .option('--output <file>', 'Output to a file (e.g. .env.local)')
+    .action(async (cmdOptions) => {
+      try {
+        const step = ora({
+          text: 'Fetching credentials from Nile...',
+          color: 'yellow'
+        }).start();
+
+        try {
+          const options = getOptions();
+          const configManager = new ConfigManager(options);
+          const api = new NileAPI({
+            token: configManager.getToken(),
+            dbHost: configManager.getDbHost(),
+            controlPlaneUrl: configManager.getGlobalHost(),
+          });
+
+          const { workspaceSlug, databaseName } = await getWorkspaceAndDatabase(options);
+          const credentials = await api.createDatabaseCredentials(workspaceSlug, databaseName);
+          const database = await api.getDatabase(workspaceSlug, databaseName);
+          
+          step.succeed('Received credentials from Nile database');
+          await sleep(300);
+
+          const envContent = `NILEDB_USER=${credentials.id}
+NILEDB_PASSWORD=${credentials.password}
+NILEDB_API_URL=https://us-west-2.api.thenile.dev/v2/databases/${database.id}
+NILEDB_POSTGRES_URL=postgres://us-west-2.db.thenile.dev:5432/${databaseName}`;
+
+          if (cmdOptions.output) {
+            await fs.writeFile(cmdOptions.output, envContent);
+            console.log(styles.success(`\n✓ Environment variables written to ${cmdOptions.output}`));
+          } else {
+            console.log(styles.info('\nNile Environment Variables:'));
+            console.log(styles.info('```env'));
+            console.log(envContent);
+            console.log(styles.info('```'));
+            console.log(styles.info('\nTo save these variables to a file, use --output flag:'));
+            console.log(styles.command('nile auth env --output .env.local'));
+          }
+        } catch (error: any) {
+          step.fail('Failed to generate environment variables');
+          if (getOptions().debug) {
+            console.error(theme.error('Error details:'), error.message);
+          }
+          process.exit(1);
+        }
+      } catch (error: any) {
+        console.error(theme.error('\nFailed to generate environment variables:'), error.message || 'Unknown error');
+        process.exit(1);
+      }
+    });
+
   auth
     .command('quickstart')
     .description('Set up a new application with Nile authentication')

src/commands/local.ts

@@ -43,7 +43,8 @@ Examples:
   ${formatCommand('nile local start', '--no-prompt')} Start without prompting for psql connection
   ${formatCommand('nile local stop')}                 Stop local development environment
   ${formatCommand('nile local info')}                 Show connection information
-
+  ${formatCommand('nile local connect --psql')}       Connect to local database using psql
+  
 ${getGlobalOptionsHelp()}`);
 
   local
@@ -314,5 +315,86 @@ ${getGlobalOptionsHelp()}`);
       }
     });
 
+  local
+    .command('connect')
+    .description('Connect to local database')
+    .requiredOption('--psql', 'Connect using PostgreSQL CLI')
+    .action(async (cmdOptions) => {
+      try {
+        // Check if container is running
+        const { stdout } = await execAsync('docker ps --filter name=nile-local --format {{.Names}}');
+        if (!stdout.includes('nile-local')) {
+          console.error(theme.error('\nNo Nile local environment is currently running.'));
+          console.log(theme.dim('Start it with: nile local start'));
+          process.exit(1);
+        }
+
+        const connectSpinner = ora({
+          text: 'Connecting to local database...',
+          color: 'cyan'
+        }).start();
+
+        try {
+          // Add a delay to ensure the database is ready for connections
+          await new Promise(resolve => setTimeout(resolve, 1000));
+          
+          // Connect using psql with individual parameters
+          const psql = spawn('psql', [
+            '-h', 'localhost',
+            '-p', '5432',
+            '-U', '00000000-0000-0000-0000-000000000000',
+            '-d', 'test',
+            '-w'  // Never prompt for password
+          ], {
+            stdio: 'inherit',
+            env: {
+              ...process.env,
+              PGPASSWORD: 'password'  // Set password via environment variable
+            }
+          });
+
+          // Stop the spinner immediately as psql will take over the terminal
+          connectSpinner.stop();
+
+          // Handle psql exit
+          psql.on('exit', (code) => {
+            if (code !== 0) {
+              console.error(theme.error('\nFailed to connect using psql. Please check if psql is installed.'));
+              if (getOptions().debug) {
+                console.error(theme.dim('Try connecting directly with:'));
+                console.error(theme.dim('PGPASSWORD=password psql -h localhost -p 5432 -U 00000000-0000-0000-0000-000000000000 -d test'));
+              }
+              process.exit(1);
+            }
+          });
+
+          // Handle psql error
+          psql.on('error', (error) => {
+            connectSpinner.fail('Failed to launch psql');
+            console.error(theme.error('\nError launching psql:'), error.message);
+            if (error.message.includes('ENOENT')) {
+              console.error(theme.error('Please make sure psql is installed and available in your PATH'));
+            }
+            process.exit(1);
+          });
+
+        } catch (error: any) {
+          connectSpinner.fail('Failed to connect to database');
+          if (getOptions().debug) {
+            console.error(theme.error('Error details:'), error.message);
+          }
+          process.exit(1);
+        }
+      } catch (error: any) {
+        const options = getOptions();
+        if (options.debug) {
+          console.error(theme.error('\nFailed to connect to database:'), error);
+        } else {
+          console.error(theme.error('\nFailed to connect to database:'), error.message || 'Unknown error');
+        }
+        process.exit(1);
+      }
+    });
+
   return local;
 }