typescript-transform-paths/lib/index.js

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.normalizePath = void 0;
var path_1 = require("path");
var typescript_1 = __importDefault(require("typescript"));
var url_1 = require("url");
var fs_1 = require("fs");
/* ****************************************************************************************************************** *
 * Helpers
 * ****************************************************************************************************************** */
exports.normalizePath = function (p) {
    // Is extended length or has non-ascii chars (respectively)
    return /^\\\\\?\\/.test(p) || /[^\u0000-\u0080]+/.test(p)
        ? p
        : // Normalize to forward slash and remove repeating slashes
            p.replace(/[\\\/]+/g, "/");
};
/* ****************************************************************************************************************** *
 * Transformer
 * ****************************************************************************************************************** */
var transformer = function (_) { return function (context) { return function (sourceFile) {
    var _a, _b;
    var resolver = typeof context.getEmitResolver === "function"
        ? context.getEmitResolver()
        : undefined;
    var compilerOptions = context.getCompilerOptions();
    var sourceDir = path_1.dirname(sourceFile.fileName);
    var implicitExtensions = [".ts", ".d.ts"];
    var allowJs = compilerOptions.allowJs === true;
    var allowJsx = compilerOptions.jsx !== undefined &&
        compilerOptions.jsx !== typescript_1.default.JsxEmit.None;
    var allowJson = compilerOptions.resolveJsonModule === true;
    allowJs && implicitExtensions.push(".js");
    allowJsx && implicitExtensions.push(".tsx");
    allowJs && allowJsx && implicitExtensions.push(".jsx");
    allowJson && implicitExtensions.push(".json");
    var isDeclarationFile = sourceFile.isDeclarationFile;
    var _c = compilerOptions.baseUrl, baseUrl = _c === void 0 ? "" : _c, _d = compilerOptions.paths, paths = _d === void 0 ? {} : _d;
    paths["*"] = (_b = (_a = paths["*"]) === null || _a === void 0 ? void 0 : _a.concat("*")) !== null && _b !== void 0 ? _b : ["*"];
    var binds = Object.keys(paths)
        .filter(function (key) { return paths[key].length; })
        .map(function (key) { return ({
        regexp: new RegExp("^" + key.replace("*", "(.*)") + "$"),
        paths: paths[key],
    }); });
    if (!baseUrl || binds.length === 0) {
        // There is nothing we can do without baseUrl and paths specified.
        return sourceFile;
    }
    function isRelative(s) {
        return s[0] === ".";
    }
    function isUrl(s) {
        return url_1.parse(s).protocol !== null;
    }
    function fileExists(s) {
        // check for implicit extensions .ts, .dts, etc...
        for (var _i = 0, implicitExtensions_1 = implicitExtensions; _i < implicitExtensions_1.length; _i++) {
            var ext = implicitExtensions_1[_i];
            if (fs_1.existsSync(s + ext))
                return true;
        }
        // else if has extensions, file must exist
        if (path_1.extname(s) !== "")
            return fs_1.existsSync(s);
        return false;
    }
    function bindModuleToFile(moduleName) {
        if (isRelative(moduleName)) {
            // if it's relative path do not transform
            return moduleName;
        }
        for (var _i = 0, binds_1 = binds; _i < binds_1.length; _i++) {
            var _a = binds_1[_i], regexp = _a.regexp, paths_2 = _a.paths;
            var match = regexp.exec(moduleName);
            if (match) {
                for (var _b = 0, paths_1 = paths_2; _b < paths_1.length; _b++) {
                    var p = paths_1[_b];
                    var out = p.replace(/\*/g, match[1]);
                    if (isUrl(out))
                        return out;
                    var filepath = path_1.resolve(baseUrl, out);
                    if (!fileExists(filepath + "/index") && !fileExists(filepath))
                        continue;
                    var resolved = fixupImportPath(path_1.relative(sourceDir, filepath));
                    return isRelative(resolved) ? resolved : "./" + resolved;
                }
            }
        }
        return undefined;
    }
    var isRequire = function (node) {
        return typescript_1.default.isCallExpression(node) &&
            typescript_1.default.isIdentifier(node.expression) &&
            node.expression.text === "require" &&
            typescript_1.default.isStringLiteral(node.arguments[0]) &&
            node.arguments.length === 1;
    };
    var isAsyncImport = function (node) {
        return typescript_1.default.isCallExpression(node) &&
            node.expression.kind === typescript_1.default.SyntaxKind.ImportKeyword &&
            typescript_1.default.isStringLiteral(node.arguments[0]) &&
            node.arguments.length === 1;
    };
    function visit(node) {
        if (isRequire(node) || isAsyncImport(node)) {
            return unpathRequireAndAsyncImport(node);
        }
        if (typescript_1.default.isExternalModuleReference(node)) {
            return unpathImportEqualsDeclaration(node);
        }
        if (typescript_1.default.isImportDeclaration(node)) {
            return unpathImportDeclaration(node);
        }
        if (typescript_1.default.isExportDeclaration(node)) {
            return unpathExportDeclaration(node);
        }
        if (typescript_1.default.isImportTypeNode(node)) {
            return unpathImportTypeNode(node);
        }
        return typescript_1.default.visitEachChild(node, visit, context);
    }
    function unpathRequireAndAsyncImport(node) {
        var firstArg = node.arguments[0];
        var file = bindModuleToFile(firstArg.text);
        if (!file) {
            return node;
        }
        var fileLiteral = typescript_1.default.createLiteral(file);
        return typescript_1.default.updateCall(node, node.expression, node.typeArguments, [
            fileLiteral,
        ]);
    }
    function unpathImportTypeNode(node) {
        var argument = node.argument;
        var literal = argument.literal;
        if (!typescript_1.default.isStringLiteral(literal)) {
            return node;
        }
        var file = bindModuleToFile(literal.text);
        if (!file) {
            return node;
        }
        var fileLiteral = typescript_1.default.createLiteral(file);
        var fileArgument = typescript_1.default.updateLiteralTypeNode(argument, fileLiteral);
        return typescript_1.default.updateImportTypeNode(node, fileArgument, node.qualifier, node.typeArguments, node.isTypeOf);
    }
    function unpathImportEqualsDeclaration(node) {
        if (!typescript_1.default.isStringLiteral(node.expression)) {
            return node;
        }
        var file = bindModuleToFile(node.expression.text);
        if (!file) {
            return node;
        }
        var fileLiteral = typescript_1.default.createLiteral(file);
        return typescript_1.default.updateExternalModuleReference(node, fileLiteral);
    }
    function unpathImportDeclaration(node) {
        if (!typescript_1.default.isStringLiteral(node.moduleSpecifier)) {
            return node;
        }
        var file = bindModuleToFile(node.moduleSpecifier.text);
        if (!file) {
            return node;
        }
        var fileLiteral = typescript_1.default.createLiteral(file);
        var importClause = typescript_1.default.visitNode(node.importClause, visitImportClause, typescript_1.default.isImportClause);
        return node.importClause === importClause ||
            importClause ||
            isDeclarationFile
            ? typescript_1.default.updateImportDeclaration(node, node.decorators, node.modifiers, node.importClause, fileLiteral)
            : undefined;
    }
    function visitImportClause(node) {
        var name = resolver.isReferencedAliasDeclaration(node)
            ? node.name
            : undefined;
        var namedBindings = typescript_1.default.visitNode(node.namedBindings, visitNamedImportBindings, typescript_1.default.isNamedImports);
        return name || namedBindings
            ? typescript_1.default.updateImportClause(node, name, namedBindings, node.isTypeOnly)
            : undefined;
    }
    function visitNamedImportBindings(node) {
        if (node.kind === typescript_1.default.SyntaxKind.NamespaceImport) {
            return resolver.isReferencedAliasDeclaration(node) ? node : undefined;
        }
        else {
            var elements = typescript_1.default.visitNodes(node.elements, visitImportSpecifier, typescript_1.default.isImportSpecifier);
            return elements.some(function (e) { return e; })
                ? typescript_1.default.updateNamedImports(node, elements)
                : undefined;
        }
    }
    function visitImportSpecifier(node) {
        return resolver.isReferencedAliasDeclaration(node) ? node : undefined;
    }
    function unpathExportDeclaration(node) {
        if (!node.moduleSpecifier || !typescript_1.default.isStringLiteral(node.moduleSpecifier)) {
            return node;
        }
        var file = bindModuleToFile(node.moduleSpecifier.text);
        if (!file) {
            return node;
        }
        var fileLiteral = typescript_1.default.createLiteral(file);
        if ((!node.exportClause &&
            !compilerOptions.isolatedModules &&
            !resolver.moduleExportsSomeValue(node.moduleSpecifier)) ||
            (node.exportClause && resolver.isValueAliasDeclaration(node))) {
            return typescript_1.default.updateExportDeclaration(node, node.decorators, node.modifiers, node.exportClause, fileLiteral, node.isTypeOnly);
        }
        var exportClause = typescript_1.default.visitNode(node.exportClause, visitNamedExports, typescript_1.default.isNamedExports);
        return node.exportClause === exportClause ||
            exportClause ||
            isDeclarationFile
            ? typescript_1.default.updateExportDeclaration(node, node.decorators, node.modifiers, node.exportClause, fileLiteral, node.isTypeOnly)
            : undefined;
    }
    function visitNamedExports(node) {
        var elements = typescript_1.default.visitNodes(node.elements, visitExportSpecifier, typescript_1.default.isExportSpecifier);
        return elements.some(function (e) { return e; })
            ? typescript_1.default.updateNamedExports(node, elements)
            : undefined;
    }
    function visitExportSpecifier(node) {
        return resolver.isValueAliasDeclaration(node) ? node : undefined;
    }
    function fixupImportPath(p) {
        var res = exports.normalizePath(p);
        /* Remove implicit extension */
        var ext = path_1.extname(res);
        if (ext && implicitExtensions.includes(ext.replace(/^\./, "")))
            res = res.slice(0, -ext.length);
        return res;
    }
    return typescript_1.default.visitNode(sourceFile, visit);
}; }; };
exports.default = transformer;