nobalmako 1.0.0

package/src/bin/nobalmako.ts

@@ -0,0 +1,341 @@
+#!/usr/bin/env node
+import { Command } from 'commander';
+import fs from 'fs';
+import path from 'path';
+import dotenv from 'dotenv';
+import os from 'os';
+import { spawn } from 'child_process';
+// @ts-ignore
+import { prompt } from 'enquirer';
+
+const program = new Command();
+const CONFIG_FILE = path.join(os.homedir(), '.nobalmako.json');
+const PROJECT_CONFIG = path.join(process.cwd(), 'nobalmako.json');
+
+function saveConfig(config: any) {
+  const current = loadConfig();
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify({ ...current, ...config }, null, 2));
+}
+
+function loadConfig() {
+  if (fs.existsSync(CONFIG_FILE)) {
+    return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf-8'));
+  }
+  return {};
+}
+
+function loadProjectConfig() {
+  if (fs.existsSync(PROJECT_CONFIG)) {
+    return JSON.parse(fs.readFileSync(PROJECT_CONFIG, 'utf-8'));
+  }
+  return {};
+}
+
+function getToken(optionsToken?: string) {
+  const config = loadConfig();
+  return optionsToken || process.env.NOBALMAKO_TOKEN || config.token;
+}
+
+program
+  .name('nobalmako')
+  .description('Securing your environment variables')
+  .version('1.0.0');
+
+program
+  .command('login')
+  .description('Login to Nobalmako')
+  .option('--email <email>', 'User email')
+  .option('--password <password>', 'User password')
+  .option('--api-url <url>', 'Base API URL', 'http://localhost:3000/api')
+  .action(async (options) => {
+    let { email, password } = options;
+    
+    if (!email) {
+      const response = await prompt({
+        type: 'input',
+        name: 'email',
+        message: 'Enter your email:'
+      }) as any;
+      email = response.email;
+    }
+
+    if (!password) {
+      const response = await prompt({
+        type: 'password',
+        name: 'password',
+        message: 'Enter your password:'
+      }) as any;
+      password = response.password;
+    }
+
+    try {
+      const response = await fetch(`${options.apiUrl}/auth/login`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ email, password })
+      });
+
+      if (!response.ok) {
+        throw new Error('Login failed. Please check your credentials.');
+      }
+
+      const data = await response.json();
+      saveConfig({ token: data.token, email: data.user.email });
+
+      console.log(`\x1b[32mSuccessfully logged in as ${data.user.email}!\x1b[0m`);
+    } catch (error: any) {
+      console.error(`\x1b[31mError:\x1b[0m ${error.message}`);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('logout')
+  .description('Clear local credentials')
+  .action(() => {
+    if (fs.existsSync(CONFIG_FILE)) {
+      fs.unlinkSync(CONFIG_FILE);
+      console.log('\x1b[32mSuccessfully logged out.\x1b[0m');
+    } else {
+      console.log('You are not logged in.');
+    }
+  });
+
+program
+  .command('init')
+  .description('Initialize a local project configuration')
+  .option('-p, --project <project>', 'Project name')
+  .option('-e, --env <environment>', 'Default environment name')
+  .action(async (options) => {
+    let { project, env } = options;
+
+    if (!project) {
+      const response = await prompt({
+        type: 'input',
+        name: 'project',
+        message: 'Project name (slug):'
+      }) as any;
+      project = response.project;
+    }
+
+    if (!env) {
+      const response = await prompt({
+        type: 'input',
+        name: 'env',
+        message: 'Default environment (e.g. production):'
+      }) as any;
+      env = response.env;
+    }
+
+    fs.writeFileSync(PROJECT_CONFIG, JSON.stringify({ project, environment: env }, null, 2));
+    console.log(`\x1b[32mCreated nobalmako.json with defaults.\x1b[0m`);
+  });
+
+program
+  .command('pull')
+  .description('Pull environment variables from Nobalmako and save to .env')
+  .option('-p, --project <project>', 'Project name')
+  .option('-e, --env <environment>', 'Environment name (e.g. production, staging)')
+  .option('-f, --file <filename>', 'Output filename')
+  .option('--api-url <url>', 'Base API URL', 'http://localhost:3000/api')
+  .option('-t, --token <token>', 'API Token (overrides login)')
+  .action(async (options) => {
+    const token = getToken(options.token);
+    const pConfig = loadProjectConfig();
+    
+    const project = options.project || pConfig.project;
+    const env = options.env || pConfig.environment;
+    const file = options.file || pConfig.file || '.env';
+
+    if (!token) {
+      console.error('\x1b[31mError: Not logged in.\x1b[0m Run \x1b[33mnobalmako login\x1b[0m or set \x1b[33mNOBALMAKO_TOKEN\x1b[0m.');
+      process.exit(1);
+    }
+
+    if (!project || !env) {
+      console.error('\x1b[31mError: Project and Environment are required.\x1b[0m Use \x1b[33mnobalmako init\x1b[0m or provide flags.');
+      process.exit(1);
+    }
+
+    console.log(`\x1b[34m[Nobalmako]\x1b[0m Fetching secrets for project \x1b[36m${project}\x1b[0m (\x1b[35m${env}\x1b[0m)...`);
+
+    try {
+      const response = await fetch(`${options.apiUrl}/variables?team=${encodeURIComponent(project)}&environment=${encodeURIComponent(env)}`, {
+        headers: {
+          'Authorization': `Bearer ${token}`
+        }
+      });
+
+      if (!response.ok) {
+        const errorData = await response.json();
+        throw new Error(errorData.error || `HTTP error! status: ${response.status}`);
+      }
+
+      const data = await response.json();
+      const variables = data.variables || [];
+
+      if (variables.length === 0) {
+        console.log('\x1b[33mNo variables found for this project and environment.\x1b[0m');
+        return;
+      }
+
+      let envContent = `# Generated by Nobalmako on ${new Date().toISOString()}\n`;
+      variables.forEach((v: any) => {
+        if (v.description) {
+          envContent += `# ${v.description}\n`;
+        }
+        envContent += `${v.key}=${v.value}\n`;
+      });
+
+      const outputPath = path.resolve(process.cwd(), file);
+      fs.writeFileSync(outputPath, envContent);
+
+      console.log(`\x1b[32mSuccess!\x1b[0m Pulled ${variables.length} variables into \x1b[33m${file}\x1b[0m`);
+    } catch (error: any) {
+      console.error(`\x1b[31mPull failed:\x1b[0m ${error.message}`);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('push')
+  .description('Push environment variables from a .env file to Nobalmako')
+  .option('-p, --project <project>', 'Project name')
+  .option('-e, --env <environment>', 'Environment name (e.g. production, staging)')
+  .option('-f, --file <filename>', 'Input filename')
+  .option('-s, --secret', 'Mark all variables as secret', false)
+  .option('--api-url <url>', 'Base API URL', 'http://localhost:3000/api')
+  .option('-t, --token <token>', 'API Token (overrides login)')
+  .action(async (options) => {
+    const token = getToken(options.token);
+    const pConfig = loadProjectConfig();
+    
+    const project = options.project || pConfig.project;
+    const env = options.env || pConfig.environment;
+    const file = options.file || pConfig.file || '.env';
+
+    if (!token) {
+      console.error('\x1b[31mError: Not logged in.\x1b[0m Run \x1b[33mnobalmako login\x1b[0m or set \x1b[33mNOBALMAKO_TOKEN\x1b[0m.');
+      process.exit(1);
+    }
+
+    if (!project || !env) {
+      console.error('\x1b[31mError: Project and Environment are required.\x1b[0m Use \x1b[33mnobalmako init\x1b[0m or provide flags.');
+      process.exit(1);
+    }
+
+    const inputPath = path.resolve(process.cwd(), file);
+    if (!fs.existsSync(inputPath)) {
+      console.error(`\x1b[31mError: File not found: ${file}\x1b[0m`);
+      process.exit(1);
+    }
+
+    const envContent = fs.readFileSync(inputPath, 'utf-8');
+    const envVars = dotenv.parse(envContent);
+    const keys = Object.keys(envVars);
+
+    console.log(`\x1b[34m[Nobalmako]\x1b[0m Pushing ${keys.length} variables to \x1b[36m${project}\x1b[0m (\x1b[35m${env}\x1b[0m)...`);
+
+    try {
+      // 1. Get Project ID and Environment ID
+      const envsResponse = await fetch(`${options.apiUrl}/environments`, {
+        headers: { 'Authorization': `Bearer ${token}` }
+      });
+      if (!envsResponse.ok) throw new Error('Failed to fetch environments');
+      const envsData = await envsResponse.json();
+      const targetEnv = envsData.environments.find((e: any) => 
+        e.teamName === project && e.name === env
+      );
+
+      if (!targetEnv) {
+        throw new Error(`Could not find environment "${env}" for project "${project}".`);
+      }
+
+      // 2. Push each variable
+      let successCount = 0;
+      for (const [key, value] of Object.entries(envVars)) {
+        const response = await fetch(`${options.apiUrl}/variables`, {
+          method: 'POST',
+          headers: {
+            'Authorization': `Bearer ${token}`,
+            'Content-Type': 'application/json'
+          },
+          body: JSON.stringify({
+            key,
+            value,
+            teamId: targetEnv.teamId,
+            environmentId: targetEnv.id,
+            isSecret: options.secret
+          })
+        });
+
+        if (response.ok) {
+          successCount++;
+          process.stdout.write(`\rProgress: ${successCount}/${keys.length}`);
+        }
+      }
+
+      console.log(`\n\x1b[32mSuccess!\x1b[0m Pushed ${successCount} variables to \x1b[33m${project}\x1b[0m`);
+    } catch (error: any) {
+      console.error(`\x1b[31mPush failed:\x1b[0m ${error.message}`);
+      process.exit(1);
+    }
+  });
+
+program
+  .command('run')
+  .description('Run a command with environment variables injected')
+  .argument('<command...>', 'The command to execute')
+  .option('-p, --project <project>', 'Project name')
+  .option('-e, --env <environment>', 'Environment name')
+  .option('--api-url <url>', 'Base API URL', 'http://localhost:3000/api')
+  .option('-t, --token <token>', 'API Token')
+  .action(async (commandArgs, options) => {
+    const token = getToken(options.token);
+    const pConfig = loadProjectConfig();
+    
+    const project = options.project || pConfig.project;
+    const env = options.env || pConfig.environment;
+
+    if (!token) {
+      console.error('\x1b[31mError: Not logged in.\x1b[0m');
+      process.exit(1);
+    }
+
+    if (!project || !env) {
+      console.error('\x1b[31mError: Project and Environment are required.\x1b[0m');
+      process.exit(1);
+    }
+
+    try {
+      const response = await fetch(`${options.apiUrl}/variables?team=${encodeURIComponent(project)}&environment=${encodeURIComponent(env)}`, {
+        headers: { 'Authorization': `Bearer ${token}` }
+      });
+
+      if (!response.ok) throw new Error('Failed to fetch variables');
+      const data = await response.json();
+      const variables = data.variables || [];
+
+      // Create new env object
+      const newEnv = { ...process.env };
+      variables.forEach((v: any) => {
+        newEnv[v.key] = v.value;
+      });
+
+      const [cmd, ...args] = commandArgs;
+      const child = spawn(cmd, args, {
+        stdio: 'inherit',
+        env: newEnv,
+        shell: true
+      });
+
+      child.on('exit', (code) => {
+        process.exit(code || 0);
+      });
+    } catch (error: any) {
+      console.error(`\x1b[31mRun failed:\x1b[0m ${error.message}`);
+      process.exit(1);
+    }
+  });
+
+program.parse(process.argv);