@angular-experts/hawkeye/index.mjs

CLI to launch Hawkeye, a bundle analyzer / visualizer

#!/usr/bin/env node
import httpServer from 'http-server';
import {program} from 'commander';
import path from 'path';
import {readFileSync, writeFileSync} from 'fs';
import inquirer from 'inquirer';
import chalk from 'chalk';
import { fileURLToPath } from 'url';
import { dirname, join } from 'path';
import open from 'open';

import packageJson from './package.json' with {type: "json"};

program
    .name('hawkeye-cli')
    .description('CLI to launch the Hawkeye app')
    .version(packageJson.version)
    .argument('<string>', 'path to the stats.json file')
    .action((pathToStatsJsonFile) => {
        try {
            const statsJson = readFileSync(pathToStatsJsonFile, 'utf-8');
            const __filename = fileURLToPath(import.meta.url);
            const __dirname = dirname(__filename);

            const filePath = join(__dirname, 'client', 'assets', 'stats.json');
            writeFileSync(filePath, statsJson);

            const server = httpServer.createServer({
                root: join(__dirname, 'client'),
                cache: -1
            });

            const PORT = 1420; // Replace with your desired port
            server.listen(PORT, () => {
                logHawkeyeMessage(`Server is running at http://localhost:${PORT}?launchMode=cli`, 'info')
            });

            open(`http://localhost:${PORT}/?launchMode=cli`);

        } catch (error) {
            logHawkeyeMessage(error, 'error');
        }
    });

program.command('init').action(async () => {
    try {
        const PACKAGE_JSON_PATH = path.join(process.cwd(), 'package.json');
        const PACKAGE_JSON = JSON.parse(readFileSync(PACKAGE_JSON_PATH, 'utf-8'));
        let multiProjectWorkspace = false;

        const {nxWorkspace} = await inquirer.prompt([
            {
                type: 'confirm',
                name: 'nxWorkspace',
                message: 'Is this an Nx workspace?'
            }
        ]);

        if(!nxWorkspace) {
            const multiProjectWorkspaceAnswer = await inquirer.prompt([
                {
                    type: 'confirm',
                    name: 'multiProjectWorkspace',
                    message: 'Do you have a multi-project Angular CLI workspace?'
                }
            ]);
            multiProjectWorkspace = multiProjectWorkspaceAnswer.multiProjectWorkspace;
        }

        const {projectName} = await inquirer.prompt([
            {
                type: 'input',
                name: 'projectName',
                message: 'What is the name of your project?'
            }
        ]);

        if(nxWorkspace){
            const ANALYSE_SCRIPT_NX_WORKSPACE = `nx build ${projectName} --stats-json --named-chunks && npx -y @angular-experts/hawkeye dist/apps/${projectName}/stats.json`;
            PACKAGE_JSON.scripts = {
                ...PACKAGE_JSON.scripts,
                [`analyze:${projectName}`]: ANALYSE_SCRIPT_NX_WORKSPACE
            };
        } else if (multiProjectWorkspace) {
            const ANALYSE_SCRIPT_MULTI_PROJECT_WORKSPACE = `ng build  ${projectName} --stats-json --named-chunks && npx -y @angular-experts/hawkeye dist/${projectName}/stats.json`;
            PACKAGE_JSON.scripts = {
                ...PACKAGE_JSON.scripts,
                [`analyze:${projectName}`]: ANALYSE_SCRIPT_MULTI_PROJECT_WORKSPACE
            };
        } else if(!multiProjectWorkspace) {
            const ANALYSE_SCRIPT_SINGLE_PROJECT_WORKSPACE = `ng build --stats-json --named-chunks && npx -yü @angular-experts/hawkeye dist/${projectName}/stats.json`;
            PACKAGE_JSON.scripts = {...PACKAGE_JSON.scripts, analyze: ANALYSE_SCRIPT_SINGLE_PROJECT_WORKSPACE};
        }

        writeFileSync(PACKAGE_JSON_PATH, JSON.stringify(PACKAGE_JSON, null, 2));

        const generatedAnalyzeScriptCommand = nxWorkspace || multiProjectWorkspace ? `analyze:${projectName}` : `analyze`;

        logHawkeyeMessage(`Analyze script successfully added to your package.json`, 'success');
        logHawkeyeMessage(`Go ahead and npm run ${generatedAnalyzeScriptCommand}`, 'info');
    } catch (error) {
        if(error.code === 'EISDIR'){
            logHawkeyeMessage('You specified a path to a directory but Hawkeye expects a path to a stats.json file', 'error');
        }

        logHawkeyeMessage(error, 'error');
    }
});
program.parse();

function logHawkeyeMessage(message, level) {
    switch(level) {
        case 'info':
            console.log(chalk.blue(`𓅃: ${message}`));
            break;
        case 'success':
            console.log(chalk.green(`𓅃: ${message}`));
            break;
        case 'warn':
            console.warn(chalk.yellow(`𓅃️: ${message}`));
            break;
        case 'error':
            console.error(chalk.red(`𓅃: ${message}`));
            break;
        default:
            console.log(`𓅃 ${message}`);
    }
}