monocart-coverage-reports/lib/istanbul/istanbul.js

A code coverage tool to generate native V8 reports or Istanbul reports.

const fs = require('fs');
const path = require('path');
const istanbulReports = require('istanbul-reports');
const istanbulLibCoverage = require('istanbul-lib-coverage');
const istanbulLibReport = require('istanbul-lib-report');
const EC = require('eight-colors');

const Util = require('../utils/util.js');
const IstanbulSummary = require('./istanbul-summary.js');

const { initIstanbulSourcePath } = require('../utils/source-path.js');
const { getUntestedList } = require('../converter/untested.js');

const findHtmlPath = (outputDir) => {
    const defaultHtml = path.resolve(outputDir, 'index.html');
    if (fs.existsSync(defaultHtml)) {
        return defaultHtml;
    }

    const htmlList = [];
    Util.forEachFile(outputDir, ['.html'], (name, dir) => {
        const htmlPath = path.resolve(dir, name);
        if (name === 'index.html') {
            htmlList.unshift(htmlPath);
            return 'break';
        }
        htmlList.push(htmlPath);
    });

    if (htmlList.length) {
        return htmlList[0];
    }

    return defaultHtml;

};

const saveIstanbulReports = (coverageData, fileSources, options) => {

    coverageData = initIstanbulSourcePath(coverageData, fileSources, options);

    const coverageMap = istanbulLibCoverage.createCoverageMap(coverageData);

    const defaultWatermarks = {
        statements: [50, 80],
        branches: [50, 80],
        functions: [50, 80],
        lines: [50, 80]
    };
    const watermarks = Util.resolveWatermarks(defaultWatermarks, options.watermarks);

    // https://github.com/istanbuljs/istanbuljs/tree/master/packages/istanbul-lib-report

    // defaultSummarizer and sourceFinder can be passed from options
    const contextOptions = {
        coverageMap,
        // The summarizer to default to (may be overridden by some reports)
        // values can be nested/flat/pkg. Defaults to 'pkg'
        defaultSummarizer: options.defaultSummarizer || 'nested',

        dir: options.outputDir,
        watermarks,
        sourceFinder: (filePath) => {

            // console.log(`find file source: ${filePath}`);

            if (fileSources) {
                const source = fileSources[filePath];
                if (source) {
                    return source;
                }
            }

            if (typeof options.sourceFinder === 'function') {
                const source = options.sourceFinder(filePath);
                if (source) {
                    return source;
                }
            }

            if (fs.existsSync(filePath)) {
                return fs.readFileSync(filePath, 'utf8');
            }

            // console.log('Not found source file:', filePath);

            return `Not found source file: ${filePath}`;
        }
    };

    // create a context for report generation
    const context = istanbulLibReport.createContext(contextOptions);

    const istanbulGroup = options.reportGroup.get('istanbul');
    // already has
    if (istanbulGroup) {
        for (const [reportName, reportOptions] of istanbulGroup) {
            const t1 = Date.now();
            const report = istanbulReports.create(reportName, reportOptions);
            report.execute(context);
            Util.logTime(`${EC.magenta('├')} [generate] saved report: ${reportName}`, t1);
        }
    }

    // add watermarks and color
    const coverageReport = new IstanbulSummary();
    coverageReport.execute(context);

    const reportPath = Util.resolveReportPath(options, () => {
        // find first index.html path for preview
        const htmlPath = findHtmlPath(options.outputDir);
        return Util.relativePath(htmlPath);
    });

    const coverageResults = {
        type: 'istanbul',
        reportPath,
        version: Util.version,
        name: options.name,
        watermarks: contextOptions.watermarks,
        // summary and files
        ... coverageReport.getReport()
    };

    return coverageResults;
};

const addUntestedFiles = async (istanbulData, options) => {
    const testedMap = new Map();
    Object.keys(istanbulData).forEach((filePath) => {
        const sourcePath = Util.relativePath(filePath);
        testedMap.set(sourcePath, true);
    });

    // untested files
    const untestedList = await getUntestedList(testedMap, options, 'istanbul');
    if (!untestedList) {
        return;
    }

    // console.log('istanbul untestedList', untestedList);
    untestedList.forEach((item) => {
        if (!istanbulData[item.path]) {
            istanbulData[item.path] = istanbulLibCoverage.createFileCoverage(item);
        }
    });

};

const mergeIstanbulDataList = (dataList, options) => {
    const istanbulCoverageList = dataList.map((it) => it.data);
    const coverageMap = istanbulLibCoverage.createCoverageMap();
    istanbulCoverageList.forEach((coverage) => {
        coverageMap.merge(coverage);
    });
    const istanbulData = coverageMap.toJSON();
    return istanbulData;
};

const mergeIstanbulCoverage = async (dataList, options) => {
    const istanbulData = mergeIstanbulDataList(dataList, options);
    await addUntestedFiles(istanbulData, options);
    return istanbulData;
};

module.exports = {
    saveIstanbulReports,
    mergeIstanbulDataList,
    mergeIstanbulCoverage
};