Добавить комментарий перед функцией, используя TypeScript Compiler API

Question

У меня есть файл TypeScript, который я хочу перенести в JavaScript.В рамках этого перевода я хочу добавить комментарий перед каждой функцией, и я надеялся сделать это с помощью API-интерфейса компилятора TypeScript.

Я пробовал два разных подхода.Одним из них было захватить SourceFile и изменить его statements, например, так:

const program = ts.createProgram([args.input], {});
const srcFile = find(program.getSourceFiles(), (sourceFile) => !sourceFile.isDeclarationFile);
srcFile.statements = ts.createNodeArray(srcFile.statements.map((statement) => {
    if (!ts.isFunctionDeclaration(statement)) {
        return statement;
    }
    return ts.addSyntheticLeadingComment(
        statement,
        ts.SyntaxKind.MultiLineCommentTrivia,
        "My long desired comment",
        true,
    );
}));

, что дает мне следующую ошибку:

TypeError: Cannot read property 'emitNode' of undefined
at getOrCreateEmitNode (/Users/.../node_modules/typescript/lib/typescript.js:52792:19)
at getOrCreateEmitNode (/Users/.../node_modules/typescript/lib/typescript.js:52801:17)
at setSyntheticLeadingComments (/Users/.../node_modules/typescript/lib/typescript.js:52918:9)
at Object.addSyntheticLeadingComment (/Users/.../node_modules/typescript/lib/typescript.js:52923:16)
at /Users/.../dist/index.js:26:15
at Array.map (<anonymous>)
at Object.<anonymous> (/Users/.../dist/index.js:21:60)
at Module._compile (internal/modules/cjs/loader.js:654:30)
at Object.Module._extensions..js (internal/modules/cjs/loader.js:665:10)
at Module.load (internal/modules/cjs/loader.js:566:32)

Я попытался напечатать statement прямо перед ts.addSyntheticLeadingComment и statement - это FunctionDeclaration, как и ожидалось, хотя и пропущено поле emitNode, которое, как я ожидаю, будет создано функцией getOrCreateEmitNode.

Второй подход, который я попробовал, похож, но сталкивается с той же проблемой;вместо перезаписи оригинала srcFile.statement я работаю с принтером следующим образом:

const printer = ts.createPrinter(undefined, {
    substituteNode: (hint, node) => {
        if (ts.isFunctionDeclaration(node)) {
            return ts.addSyntheticLeadingComment(
                node,
                ts.SyntaxKind.MultiLineCommentTrivia,
                "My long desired comment",
                true,
           );
        }
    },
});

console.log(printer.printFile(srcFile));

, что выдает ту же ошибку, что и предыдущий код.

Файл TypeScriptпопытаться изменить это очень просто:

function myFunc(a: number, b: number): number {
    return a + b;
}

Я был бы очень признателен за помощь в этом.

С уважением, Раду

Answer 1

Подставлять узел не обязательно.Помните, что комментарии не являются частью AST, поэтому не добавляйте их в массив операторов вместо объявления существующей функции.Вместо этого просто вызовите addSyntheticLeadingComment на узле без использования возвращаемого значения.

Например, следующий код работает нормально:

import * as ts from "typescript";

const file = ts.createSourceFile("test.ts", `function myFunc(a: number, b: number): number {
    return a + b;
}`, ts.ScriptTarget.Latest, true);
const functionDec = file.statements.find(ts.isFunctionDeclaration)!;

ts.addSyntheticLeadingComment(functionDec, ts.SyntaxKind.MultiLineCommentTrivia,
    "My long desired comment", true);

const printer = ts.createPrinter({ removeComments: false });
console.log(printer.printFile(file));

Выходы:

/*My long desired comment*/
function myFunc(a: number, b: number): number {
    return a + b;
}