Описание

Привет всем,

Я пытаюсь создать что-то, что добавляло бы комментарии поверх функции. К сожалению, похоже, что ts.setSyntheticLeadingComments не позволяет мне заменять существующие комментарии.

Я пытался:

  • ts.setSyntheticLeadingComments(node, [])
  • ts.setSyntheticLeadingComments(node, undefined)
  • node = ts.setSyntheticLeadingComments(node, [])

Но ничего из этого не работает. В конечном итоге моей целью было заменить существующие комментарии, которые я мог бы создать, новыми.

Любая идея ? Спасибо 🙏

Размножение

const transformFactory = (context: ts.TransformationContext) => (
  rootNode: ts.SourceFile
): ts.SourceFile => {
  const visit = (node: ts.Node) => {
    node = ts.visitEachChild(node, visit, context);

    ts.setSyntheticLeadingComments(node, []);

    return node;
  };

  return ts.visitNode(rootNode, visit);
};

const sourceFile = ts.createSourceFile(
  path,
  source,
  ts.ScriptTarget.ESNext,
  true,
  ts.ScriptKind.TS
);
const result = ts.transform(sourceFile, [transformFactory]);
const resultPrinter = ts.createPrinter({ removeComments: false });

console.log(resultPrinter.printFile(result.transformed[0]));

Попробуйте следующий преобразователь и посмотрите, как никакие комментарии не удаляются вообще

Использование ts.createPrinter(..., { substituteNode(hint, node) { ... } }) тоже не помогает

Примечание

Похоже, что ts.getSyntheticLeadingComments() тоже не работает так, как я ожидал. Он всегда возвращает undefined, что побудило меня использовать следующие утилиты, хотя я не уверен, что полностью понимаю их назначение (заимствовано из https://github.com/angular/tsickle/blob/6f5835a644fcd3d3c8c8c8c08c8c8c08c8c08c8c08c08c08c08c08c8c08c08c08c08c08c08c08c08c08c06d6 )

/**
 * A replacement for ts.getLeadingCommentRanges that returns the union of synthetic and
 * non-synthetic comments on the given node, with their text included. The returned comments must
 * not be mutated, as their content might or might not be reflected back into the AST.
 */
export function getAllLeadingComments(node: ts.Node):
    ReadonlyArray<Readonly<ts.CommentRange&{text: string}>> {
  const allRanges: Array<Readonly<ts.CommentRange&{text: string}>> = [];
  const nodeText = node.getFullText();
  const cr = ts.getLeadingCommentRanges(nodeText, 0);
  if (cr) allRanges.push(...cr.map(c => ({...c, text: nodeText.substring(c.pos, c.end)})));
  const synthetic = ts.getSyntheticLeadingComments(node);
  if (synthetic) allRanges.push(...synthetic);
  return allRanges;
}
2
Flavian Desverne 15 Мар 2019 в 01:15

1 ответ

Лучший ответ

Проблема в том, что вы ожидаете, что функции *SyntheticLeadingComments повлияют на комментарии источника. Они не будут. Они будут влиять только на комментарии, которые были ранее синтезированы (то есть добавлены вами в код).

Фактические комментарии не хранятся в виде узлов в AST. Вы можете получить комментарии к фактическим источникам, используя getLeadingCommentRanges и getTrailingCommentRanges.

Узел имеет позиции start и end, которые не содержат никаких комментариев. Также существует fullStart для узла, который представляет собой позицию, включающую любые ведущие комментарии. Когда выводится узел, машинописный текст знает, как копировать комментарии к выводу.

Если мы используем setTextRange для установки диапазона узла, чтобы исключить эти существующие комментарии, в результате мы эффективно удаляем их из вывода и можем добавлять новые комментарии, используя setSyntheticLeadingComments:

import * as ts from 'typescript'

const transformFactory = (context: ts.TransformationContext) => (
    rootNode: ts.SourceFile
): ts.SourceFile => {
    const visit = (node: ts.Node) => {
        node = ts.visitEachChild(node, visit, context);
            if(ts.isFunctionDeclaration(node)) {
            let sourceFileText = node.getSourceFile().text;
            const existingComments = ts.getLeadingCommentRanges(sourceFileText, node.pos);
            if (existingComments) {
                // Log existing comments just for fun 
                for (const comment of existingComments) {
                    console.log(sourceFileText.substring(comment.pos, comment.end))
                }
                // Comment also attaches to the first child, we must remove it recursively.
                let removeComments =  (c: ts.Node) => {
                    if (c.getFullStart() === node.getFullStart()) {
                        ts.setTextRange(c, { pos: c.getStart(), end: c.getEnd() });
                    }
                    c = ts.visitEachChild(c, removeComments, context);
                    return c;
                }
                ts.visitEachChild(node, removeComments, context);
                ts.setTextRange(node, { pos: node.getStart(), end: node.getEnd() })
                ts.setSyntheticLeadingComments(node, [{
                    pos: -1,
                    end: -1,
                    hasTrailingNewLine: false,
                    text: "Improved comment",
                    kind: ts.SyntaxKind.SingleLineCommentTrivia
                }]);

            }
        }
        return node;
    };

    return ts.visitNode(rootNode, visit);
};

const sourceFile = ts.createSourceFile(
    "path.ts",
    `
// Original comment
function test () {

}
`,
    ts.ScriptTarget.ESNext,
    true,
    ts.ScriptKind.TS
);
const result = ts.transform(sourceFile, [transformFactory]);
const resultPrinter = ts.createPrinter({ removeComments: false });

console.log("!");
console.log(resultPrinter.printFile(result.transformed[0]));
3
Titian Cernicova-Dragomir 15 Мар 2019 в 13:12