upstream-mattermost/webapp/scripts/utils.mjs

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

import fs from 'node:fs';
import path from 'node:path';

import chalk from 'chalk';

function getWorkspaces() {
    const packageFile = fs.readFileSync('package.json');
    const packageJson = JSON.parse(packageFile, 'utf8');

    return packageJson.workspaces;
}

function getPlatformPackagesContainingCommand(scriptName) {
    return getWorkspaces().filter((workspace) => {
        if (!workspace.startsWith('platform/')) {
            return false;
        }

        const workspacePackageFile = fs.readFileSync(path.join(workspace, 'package.json'));
        const workspacePackageJson = JSON.parse(workspacePackageFile, 'utf8');

        return workspacePackageJson?.scripts?.[scriptName];
    });
}

/**
 * Returns an array of concurrently commands to run a given script on every platform workspace that contains it.
 */
export function getPlatformCommands(scriptName) {
    return getPlatformPackagesContainingCommand(scriptName).map((workspace) => ({
        command: `npm:${scriptName} --workspace=${workspace}`,
        name: workspace.substring(workspace.lastIndexOf('/') + 1),
        prefixColor: getColorForWorkspace(workspace),
    }));
}

const workspaceColors = ['green', 'magenta', 'yellow', 'red', 'blue'];
function getColorForWorkspace(workspace) {
    const index = getWorkspaces().indexOf(workspace);

    return index === -1 ? chalk.white : workspaceColors[index % workspaceColors.length];
}

/**
 * @param {import("concurrently").CloseEvent[]} closeEvents - An array of CloseEvents thrown by concurrently when waiting on a result
 * @param {number} codeOnSignal - Which error code to return when the process is interrupted
 */
export function getExitCode(closeEvents, codeOnSignal = 1) {
    const exitCode = closeEvents.find((event) => !event.killed && event.exitCode > 0)?.exitCode;

    if (typeof exitCode === 'string') {
        return codeOnSignal
    } else {
        return exitCode;
    }
}