javascript - 如何使用jscodeshift在文件开头插入一行

Question

https://astexplorer.net/#/gist/ad90272020dd0bfb15619d93cca81b66/28d3cf7178271f4f99b10bc9352daa873c2f2b20

// file
var a = "a" // what if this is import statement?

// jscodeshift
export default (file, api) => {
  const j = api.jscodeshift;

  const root = j(file.source);

  root.find(j.VariableDeclaration)
    .insertBefore("use strict");
  return root.toSource();
}

如果文件的第一行代码不同，insertBefore 如何工作。例如（变量​​声明、导入语句）

Answer 1

看起来您必须"cast"将节点连接到jscodeshift.

一个解决方案是：

export default (file, api) => {
  const j = api.jscodeshift

  const root = j(file.source)

  j(root.find(j.VariableDeclaration).at(0).get())
    .insertBefore(
      '"use strict";'
    )
  return root.toSource()
}

编辑

为了您的澄清。

如果你想use strict在文件的开头插入无论如何：

export default (file, api) => {
    const j = api.jscodeshift
    const s = '"use strict";';
    const root = j(file.source)

    root.get().node.program.body.unshift(s);  

    return root.toSource()
}

如果要use strict在import声明后添加，如果有：

export default (file, api) => {
    const j = api.jscodeshift
    const s = '"use strict";';
    const root = j(file.source);
    const imports = root.find(j.ImportDeclaration);
    const n = imports.length;

    if(n){
        //j(imports.at(0).get()).insertBefore(s); // before the imports
       j(imports.at(n-1).get()).insertAfter(s); // after the imports
    }else{
       root.get().node.program.body.unshift(s); // begining of file
    }         

    return root.toSource();
}

Answer 2

到目前为止我最好的解决方案是在第一个之前插入j.Declaration，

j(root.find(j.Declaration).at(0).get())
 .insertBefore('"use strict";')