@moonwall/cli/dist/types/src/internal/deriveTestIds.js

Testing framework for the Moon family of projects

import chalk from "chalk";
import fs from "node:fs";
import { confirm } from "@inquirer/prompts";
import path from "node:path";
export async function deriveTestIds(params) {
    const usedPrefixes = new Set();
    const { rootDir, singlePrefix } = params;
    try {
        await fs.promises.access(rootDir, fs.constants.R_OK);
    }
    catch (error) {
        console.error(`🔴 Error accessing directory ${chalk.bold(`/${rootDir}`)}, please sure this exists`);
        process.exitCode = 1;
        return;
    }
    console.log(`🟢 Processing ${rootDir} ...`);
    const topLevelDirs = getTopLevelDirs(rootDir);
    const foldersToRename = [];
    if (singlePrefix) {
        const prefix = generatePrefix(rootDir, usedPrefixes, params.prefixPhrase);
        foldersToRename.push({ prefix, dir: "." });
    }
    else {
        for (const dir of topLevelDirs) {
            const prefix = generatePrefix(dir, usedPrefixes, params.prefixPhrase);
            foldersToRename.push({ prefix, dir });
        }
    }
    const result = await confirm({
        message: `This will rename ${foldersToRename.length} suites IDs in ${rootDir}, continue?`,
    });
    if (!result) {
        console.log("🔴 Aborted");
        return;
    }
    for (const folder of foldersToRename) {
        const { prefix, dir } = folder;
        process.stdout.write(`🟢 Changing suite ${dir} to use prefix ${chalk.bold(`(${prefix})`)} ....`);
        generateId(path.join(rootDir, dir), rootDir, prefix);
        process.stdout.write(" Done ✅\n");
    }
    console.log(`🏁 Finished renaming rootdir ${chalk.bold(`/${rootDir}`)}`);
}
function getTopLevelDirs(rootDir) {
    return fs
        .readdirSync(rootDir)
        .filter((dir) => fs.statSync(path.join(rootDir, dir)).isDirectory());
}
function generatePrefix(directory, usedPrefixes, rootPrefix) {
    const sanitizedDir = directory.replace(/[-_ ]/g, "").toUpperCase();
    let prefix = rootPrefix ?? sanitizedDir[0];
    let additionalIndex = 1;
    while (usedPrefixes.has(prefix) && additionalIndex < sanitizedDir.length) {
        prefix += rootPrefix?.[additionalIndex] ?? sanitizedDir[additionalIndex];
        additionalIndex++;
    }
    let numericSuffix = 0;
    while (usedPrefixes.has(prefix)) {
        if (numericSuffix < 10) {
            numericSuffix++;
            prefix = sanitizedDir[0] + numericSuffix.toString();
        }
        else {
            let lastChar = prefix.slice(-1).charCodeAt(0);
            if (lastChar >= 90) {
                // 'Z'
                lastChar = 65; // 'A'
            }
            else {
                lastChar++;
            }
            prefix = sanitizedDir[0] + String.fromCharCode(lastChar);
        }
    }
    usedPrefixes.add(prefix);
    return prefix;
}
function generateId(directory, rootDir, prefix) {
    const contents = fs.readdirSync(directory);
    contents.sort((a, b) => {
        const aIsDir = fs.statSync(path.join(directory, a)).isDirectory();
        const bIsDir = fs.statSync(path.join(directory, b)).isDirectory();
        if (aIsDir && !bIsDir)
            return -1;
        if (!aIsDir && bIsDir)
            return 1;
        return customFileSort(a, b);
    });
    let fileCount = 1;
    let subDirCount = 1;
    for (const item of contents) {
        const fullPath = path.join(directory, item);
        if (fs.statSync(fullPath).isDirectory()) {
            const subDirPrefix = `0${subDirCount}`.slice(-2);
            generateId(fullPath, rootDir, prefix + subDirPrefix);
            subDirCount++;
        }
        else {
            const fileContent = fs.readFileSync(fullPath, "utf-8");
            if (fileContent.includes("describeSuite")) {
                const newId = prefix + `0${fileCount}`.slice(-2);
                const updatedContent = fileContent.replace(/(describeSuite\s*?\(\s*?\{\s*?id\s*?:\s*?['"])[^'"]+(['"])/, `$1${newId}$2`);
                fs.writeFileSync(fullPath, updatedContent);
            }
            fileCount++;
        }
    }
}
function hasSpecialCharacters(filename) {
    return /[ \t!@#$%^&*()_+\-=[\]{};':"\\|,.<>/?]+/.test(filename);
}
function customFileSort(a, b) {
    const aHasSpecialChars = hasSpecialCharacters(a);
    const bHasSpecialChars = hasSpecialCharacters(b);
    if (aHasSpecialChars && !bHasSpecialChars)
        return -1;
    if (!aHasSpecialChars && bHasSpecialChars)
        return 1;
    return a.localeCompare(b, undefined, { sensitivity: "accent" });
}
//# sourceMappingURL=deriveTestIds.js.map