mlh-tsd/dist/lib/parser.js

Check TypeScript type definitions

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.parseErrorAssertionToLocation = exports.extractAssertions = void 0;
const typescript_1 = require("../../libraries/typescript");
const assertions_1 = require("./assertions");
// TODO: Use Object.values() when targetting Node.js >= 8
const assertionTypes = new Set(Object.keys(assertions_1.Assertion).map(key => assertions_1.Assertion[key]));
/**
 * Extract all assertions.
 *
 * @param program - TypeScript program.
 */
exports.extractAssertions = (program) => {
    const assertions = new Map();
    /**
     * Recursively loop over all the nodes and extract all the assertions out of the source files.
     */
    function walkNodes(node) {
        if (typescript_1.isCallExpression(node)) {
            const text = node.expression.getText();
            // Check if the call type is a valid assertion
            if (assertionTypes.has(text)) {
                const assertion = text;
                const nodes = assertions.get(assertion) || new Set();
                nodes.add(node);
                assertions.set(assertion, nodes);
            }
        }
        typescript_1.forEachChild(node, walkNodes);
    }
    for (const sourceFile of program.getSourceFiles()) {
        walkNodes(sourceFile);
    }
    return assertions;
};
/**
 * Loop over all the error assertion nodes and convert them to a location map.
 *
 * @param assertions - Assertion map.
 */
exports.parseErrorAssertionToLocation = (assertions) => {
    const nodes = assertions.get(assertions_1.Assertion.EXPECT_ERROR);
    const expectedErrors = new Map();
    if (!nodes) {
        // Bail out if we don't have any error nodes
        return expectedErrors;
    }
    // Iterate over the nodes and add the node range to the map
    for (const node of nodes) {
        const location = {
            fileName: node.getSourceFile().fileName,
            start: node.getStart(),
            end: node.getEnd()
        };
        const pos = node
            .getSourceFile()
            .getLineAndCharacterOfPosition(node.getStart());
        expectedErrors.set(location, {
            fileName: location.fileName,
            line: pos.line + 1,
            column: pos.character
        });
    }
    return expectedErrors;
};