#!/usr/bin/env node

import { input } from '@inquirer/prompts';
import select from '@inquirer/select';
import chalk from 'chalk';
import childProcess from 'child_process';
import { Command } from 'commander';
import { execa, execaCommand } from 'execa';
import fs from 'fs';
import open from 'open';
import ora from 'ora';
import path from 'path';

const program = new Command();
const log = console.log;
const repoUrl = 'https://github.com/fal-ai/fal-nextjs-template.git';
const green = chalk.green;
const purple = chalk.hex('#6e40c9');

async function main() {
  const spinner = ora({
    text: 'Creating codebase',
  });
  try {
    const kebabRegez = /^([a-z]+)(-[a-z0-9]+)*$/;

    program
      .name('The fal.ai App Generator')
      .description('Generate full stack AI apps integrated with fal.ai');

    program.parse(process.argv);

    const args = program.args;
    let appName = args[0];

    if (!appName || !kebabRegez.test(args[0])) {
      appName = await input({
        message: 'Enter your app name',
        default: 'model-playground',
        validate: (d) => {
          if (!kebabRegez.test(d)) {
            return 'please enter your app name in the format of my-app-name';
          }
          return true;
        },
      });
    }

    const hasFalEnv = await select({
      message: 'Do you have a fal.ai API key?',
      choices: [
        {
          name: 'Yes',
          value: true,
        },
        {
          name: 'No',
          value: false,
        },
      ],
    });

    if (!hasFalEnv) {
      await open('https://www.fal.ai/dashboard');
    }

    const fal_api_key = await input({ message: 'Fal AI API Key' });

    const envs = `
    # environment, either PRODUCTION or DEVELOPMENT
    ENVIRONMENT="PRODUCTION"

    # FAL AI API Key
    FAL_KEY="${fal_api_key}"
    `;

    log(`\nInitializing project. \n`);

    spinner.start();
    await execa('git', ['clone', repoUrl, appName]);

    let packageJson = fs.readFileSync(`${appName}/package.json`, 'utf8');
    const packageObj = JSON.parse(packageJson);
    packageObj.name = appName;
    packageJson = JSON.stringify(packageObj, null, 2);
    fs.writeFileSync(`${appName}/package.json`, packageJson);
    fs.writeFileSync(`${appName}/.env.local`, envs);

    process.chdir(path.join(process.cwd(), appName));
    await execa('rm', ['-rf', '.git']);
    await execa('git', ['init']);

    spinner.text = '';
    let startCommand = '';

    if (isBunInstalled()) {
      spinner.text = 'Installing dependencies';
      await execaCommand('bun install').pipeStdout(process.stdout);
      spinner.text = '';
      startCommand = 'bun dev';
      console.log('\n');
    } else if (isYarnInstalled()) {
      await execaCommand('yarn').pipeStdout(process.stdout);
      startCommand = 'yarn dev';
    } else {
      spinner.text = 'Installing dependencies';
      await execa('npm', ['install', '--verbose']).pipeStdout(process.stdout);
      spinner.text = '';
      startCommand = 'npm run dev';
    }

    spinner.stop();
    await execa('git', ['add', '.']);
    await execa('git', ['commit', '-m', 'Initial commit']);

    process.chdir('../');
    log(
      `${green.bold('Success!')} Created ${purple.bold(
        appName
      )} at ${process.cwd()} \n`
    );
    log(
      `To get started, change into the new directory and run ${chalk.cyan(
        startCommand
      )}\n`
    );
  } catch (err) {
    log('\n');
    if (err.exitCode == 128) {
      log('Error: directory already exists.');
    }
    spinner.stop();
  }
}

main();

function isYarnInstalled() {
  try {
    childProcess.execSync('yarn --version');
    return true;
  } catch {
    return false;
  }
}

function isBunInstalled() {
  try {
    childProcess.execSync('bun --version');
    return true;
  } catch (err) {
    return false;
  }
}