@terascope/scripts/dist/src/helpers/misc.js

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.writePkgHeader = exports.writeHeader = exports.formatList = exports.writeIfChanged = exports.listMdFiles = exports.getName = exports.getDevDockerImage = exports.getServicesForSuite = exports.getAvailableTestSuites = exports.getRootInfo = exports.getRootDir = void 0;
const path_1 = __importDefault(require("path"));
const pkg_up_1 = __importDefault(require("pkg-up"));
const fs_extra_1 = __importDefault(require("fs-extra"));
const lodash_defaultsdeep_1 = __importDefault(require("lodash.defaultsdeep"));
const utils_1 = require("@terascope/utils");
const sort_package_json_1 = __importDefault(require("sort-package-json"));
const interfaces_1 = require("./interfaces");
const config_1 = require("./config");
const signale_1 = __importDefault(require("./signale"));
let rootDir;
function getRootDir(cwd = process.cwd()) {
    if (rootDir)
        return rootDir;
    const rootPkgJSON = pkg_up_1.default.sync({ cwd });
    if (!rootPkgJSON) {
        throw new Error('Unable to find root directory, run in the root of the repo');
    }
    if (_getRootInfo(rootPkgJSON) != null) {
        rootDir = path_1.default.dirname(rootPkgJSON);
        return rootDir;
    }
    const upOne = path_1.default.join(path_1.default.dirname(rootPkgJSON), '..');
    if (!fs_extra_1.default.existsSync(upOne) || !fs_extra_1.default.statSync(upOne).isDirectory()) {
        throw new Error('Unable to find root directory');
    }
    return getRootDir(upOne);
}
exports.getRootDir = getRootDir;
function _getRootInfo(pkgJSONPath) {
    const pkg = fs_extra_1.default.readJSONSync(pkgJSONPath);
    const isRoot = utils_1.get(pkg, 'terascope.root', false);
    if (!isRoot)
        return undefined;
    const dir = path_1.default.dirname(pkgJSONPath);
    const folderName = path_1.default.basename(dir);
    return sort_package_json_1.default(lodash_defaultsdeep_1.default(pkg, {
        dir,
        relativeDir: '.',
        folderName,
        displayName: getName(pkg.name),
        documentation: '',
        homepage: '',
        bugs: {
            url: '',
        },
        terascope: {
            root: isRoot,
            type: 'monorepo',
            target: 'es2018',
            tests: {
                suites: {}
            },
            docker: {
                registries: [`terascope/${folderName}`],
            },
            npm: {
                registry: config_1.NPM_DEFAULT_REGISTRY
            }
        },
    }));
}
let _rootInfo;
function getRootInfo() {
    if (_rootInfo)
        return _rootInfo;
    _rootInfo = _getRootInfo(path_1.default.join(getRootDir(), 'package.json'));
    return _rootInfo;
}
exports.getRootInfo = getRootInfo;
function getAvailableTestSuites() {
    return Object.keys(getRootInfo().terascope.tests.suites);
}
exports.getAvailableTestSuites = getAvailableTestSuites;
function getServicesForSuite(suite) {
    const services = getRootInfo().terascope.tests.suites[suite] || [];
    const invalidServices = services.filter((name) => !Object.values(interfaces_1.Service).includes(name));
    if (invalidServices.length) {
        const actual = invalidServices.join(', ');
        const expected = Object.values(interfaces_1.Service).join(', ');
        throw new Error(`Unsupported service(s) ${actual}, expected ${expected}`);
    }
    return services;
}
exports.getServicesForSuite = getServicesForSuite;
function getDevDockerImage() {
    const rootInfo = getRootInfo();
    const [registry] = rootInfo.terascope.docker.registries;
    return `${registry}:dev-${config_1.DEV_TAG}`;
}
exports.getDevDockerImage = getDevDockerImage;
function getName(input) {
    return utils_1.toTitleCase(input);
}
exports.getName = getName;
function listMdFiles(dir, levels = 10) {
    if (levels < 1)
        return [];
    if (!dir || !fs_extra_1.default.statSync(dir).isDirectory()) {
        throw new Error(`Invalid directory "${dir}" given to listMdFiles`);
    }
    const files = [];
    for (const fileName of fs_extra_1.default.readdirSync(dir)) {
        const filePath = path_1.default.join(dir, fileName);
        if (fs_extra_1.default.statSync(filePath).isDirectory()) {
            files.push(...listMdFiles(filePath, levels - 1));
        }
        else if (path_1.default.extname(fileName) === '.md') {
            files.push(filePath);
        }
    }
    return files;
}
exports.listMdFiles = listMdFiles;
async function writeIfChanged(filePath, contents, options = {}) {
    if (options.mkdir) {
        await fs_extra_1.default.ensureDir(path_1.default.dirname(filePath));
    }
    const exists = fs_extra_1.default.existsSync(filePath);
    if (exists && !fs_extra_1.default.statSync(filePath).isFile()) {
        throw new Error('Unable to write to non-file type');
    }
    if (!contents) {
        if (exists) {
            await fs_extra_1.default.unlink(filePath);
        }
        return true;
    }
    if (typeof contents === 'string') {
        const _contents = `${contents.trim()}\n`;
        if (exists) {
            const existing = await fs_extra_1.default.readFile(filePath, 'utf8');
            if (existing === _contents) {
                return false;
            }
        }
        if (options.log !== false) {
            signale_1.default.debug(`wrote ${path_1.default.relative(getRootDir(), filePath)} file`);
        }
        await fs_extra_1.default.writeFile(filePath, _contents, 'utf8');
        return true;
    }
    if (utils_1.isPlainObject(contents) || Array.isArray(contents)) {
        if (exists) {
            const existing = await fs_extra_1.default.readJSON(filePath);
            if (JSON.stringify(existing) === JSON.stringify(contents)) {
                return false;
            }
        }
        if (options.log !== false) {
            signale_1.default.debug(`wrote ${path_1.default.relative(getRootDir(), filePath)} JSON file`);
        }
        await fs_extra_1.default.writeJSON(filePath, contents, {
            spaces: 4,
        });
        return true;
    }
    throw new Error('Invalid contents given to writeIfChanged');
}
exports.writeIfChanged = writeIfChanged;
function formatList(list) {
    return `\n - ${list.join('\n - ')}`;
}
exports.formatList = formatList;
function writeHeader(msg, prefixNewline) {
    if (prefixNewline)
        process.stderr.write('\n');
    signale_1.default.star(`${msg}`);
}
exports.writeHeader = writeHeader;
function writePkgHeader(prefix, pkgInfos, prefixNewline) {
    const names = pkgInfos.map(({ name }) => name).join(', ');
    writeHeader(`${prefix} for ${names}`, prefixNewline);
}
exports.writePkgHeader = writePkgHeader;
//# sourceMappingURL=misc.js.map