retain-first-comment.md

Retain comment on first line

Problem

When removing or replacing the first statement in a file, it is possible for leading comments at the top of the file to be removed.

Solution

To retain the leading comments during a transformation, the comments array on the statement's node must be copied to the next statement's node that will be at the top of the file.

Examples

Bad

Transform

export default function transformer(file, api) {
  const j = api.jscodeshift;

  return j(file.source)
    .find(j.VariableDeclaration)
    .replaceWith(
        j.expressionStatement(j.callExpression(
                j.identifier('foo'), []
            )
        )
    )
    .toSource();
};

In

// Comment on first line
const firstStatement = require('some-module');

Out

foo();

Good

Transform

export default function transformer(file, api) {
  const j = api.jscodeshift;
  const root = j(file.source);
  
  const { comments } = root.find(j.Program).get('body', 0).node;
  root.find(j.VariableDeclaration).replaceWith(
    j.expressionStatement(j.callExpression(
        j.identifier('foo'),
        []
    ))
  );
  root.get().node.comments = comments;
  
  return root.toSource();
};

In

// Comment on first line
const firstStatement = require('some-module');

Out

// Comment on first line
foo();