nitro-codegen/lib/index.js

The code-generator for react-native-nitro-modules.

#!/usr/bin/env node
import yargs from 'yargs/yargs';
import { hideBin } from 'yargs/helpers';
import chalk from 'chalk';
import { prettifyDirectory } from './prettifyDirectory.js';
import { getFiles } from './getFiles.js';
import { runNitrogen } from './nitrogen.js';
import { promises as fs } from 'fs';
import { isValidLogLevel, setLogLevel } from './Logger.js';
import { initNewNitroModule } from './init.js';
const commandName = 'nitro-codegen';
// Maximum of 100 col width
const cliWidth = Math.min(process.stdout.columns * 0.9, 100);
// Set up yargs CLI app
await yargs(hideBin(process.argv))
    .option('log-level', {
    type: 'string',
    description: 'Configures the log-level of nitrogen.',
    default: 'info',
    choices: ['debug', 'info', 'warning', 'error'],
})
    .middleware((args) => {
    if (!isValidLogLevel(args.logLevel)) {
        throw new Error(`Invalid log-level ("${args.logLevel}")!`);
    }
    setLogLevel(args.logLevel);
})
    // 🔥 nitrogen [path]
    .command('$0 [basePath]', `Usage: ${chalk.bold(`${commandName} <basePath> [options]`)}\n` +
    `Run the nitro code-generator on all ${chalk.underline('**/*.nitro.ts')} files found ` +
    `in the current directory and generate C++, Swift or Kotlin outputs in ${chalk.underline('./nitrogen/generated')}.`, (y) => y
    .positional('basePath', {
    type: 'string',
    description: `The base path of where Nitrogen will start looking for ${chalk.underline('**/*.nitro.ts')} specs.`,
    default: process.cwd(),
})
    .option('out', {
    type: 'string',
    description: 'Configures the output path of the generated C++, Swift or Kotlin files.',
    default: './nitrogen/generated',
})
    .option('config', {
    type: 'string',
    description: `A custom path to a ${chalk.underline('nitro.json')} config file.`,
    default: './nitro.json',
}), async (argv) => {
    const basePath = argv.basePath;
    const outputDirectory = argv.out;
    await runNitrogenCommand(basePath, outputDirectory);
})
    // 🔥 nitrogen init <moduleName>
    .command('init <moduleName>', `Usage: ${chalk.bold(`${commandName} init <moduleName> [options]`)}\n` +
    `Create a new Nitro Module.`, (y) => y
    .positional('moduleName', {
    type: 'string',
    description: 'The name of the Nitro Module that will be created.',
    demandOption: true,
})
    .option('path', {
    type: 'string',
    description: `A custom path to create the new Nitro Module in - instead of the current working directory.`,
    default: process.cwd(),
}), async (argv) => {
    await initNewNitroModule(argv.path, argv.moduleName);
})
    .usage(`Usage: ${chalk.bold('$0 [options]')}\n` +
    `$0 is a code-generater for Nitro Modules (${chalk.underline('https://github.com/mrousavy/nitro')})\n` +
    `It converts all TypeScript specs found in ${chalk.underline('**/*.nitro.ts')} to C++, Swift or Kotlin specs.\n` +
    `Each library/module must have a ${chalk.underline('nitro.json')} configuration file in it's root directory.`)
    .help()
    .strict()
    .wrap(cliWidth).argv;
async function runNitrogenCommand(baseDirectory, outputDirectory) {
    // 1. Prepare output folders
    const filesBefore = await getFiles(outputDirectory);
    const start = performance.now();
    // 2. Run Nitrogen
    const { generatedFiles, generatedSpecsCount, targetSpecsCount } = await runNitrogen({
        baseDirectory: baseDirectory,
        outputDirectory: outputDirectory,
    });
    const end = performance.now();
    const timeS = ((end - start) / 1000).toFixed(1);
    console.log(`🎉  Generated ${generatedSpecsCount}/${targetSpecsCount} HybridObject${generatedSpecsCount === 1 ? '' : 's'} in ${timeS}s!`);
    console.log(`💡  Your code is in ${chalk.underline(prettifyDirectory(outputDirectory))}`);
    // 3. Delete all old dangling files
    const addedFiles = generatedFiles.filter((f) => !filesBefore.includes(f));
    const removedFiles = filesBefore.filter((f) => !generatedFiles.includes(f));
    if (addedFiles.length > 0 || removedFiles.length > 0) {
        let text = '';
        const as = addedFiles.length > 1 ? 's' : '';
        const rs = removedFiles.length > 1 ? 's' : '';
        if (addedFiles.length > 0 && removedFiles.length === 0) {
            text = `Added ${addedFiles.length} file${as}`;
        }
        else if (addedFiles.length === 0 && removedFiles.length > 0) {
            text = `Removed ${removedFiles.length} file${rs}`;
        }
        else {
            text = `Added ${addedFiles.length} file${as} and removed ${removedFiles.length} file${rs}`;
        }
        console.log(`‼️   ${text} - ${chalk.bold('you need to run `pod install`/sync gradle to update files!')}`);
    }
    const promises = removedFiles.map(async (file) => {
        const stat = await fs.stat(file);
        if (stat.isFile()) {
            await fs.rm(file);
        }
    });
    await Promise.all(promises);
}