浏览代码

fix: insertcode support

陈帅 6 年之前
父节点
当前提交
68d9765dab
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      scripts/insertCode.js

+ 2 - 2
scripts/insertCode.js

@@ -126,8 +126,8 @@ const insertRightContent = configPath => {
     body.splice(index, 0, parseCode(`import NoticeIconView from './NoticeIconView';`));
     body.splice(index, 0, parseCode(`import NoticeIconView from './NoticeIconView';`));
 
 
     body.forEach(item => {
     body.forEach(item => {
-      if (item.type === 'ClassDeclaration') {
-        const classBody = item.body.body[0].body;
+      if (item.type === 'VariableDeclaration') {
+        const classBody = item.declarations[0].init.body;
         classBody.body.forEach(node => {
         classBody.body.forEach(node => {
           if (node.type === 'ReturnStatement') {
           if (node.type === 'ReturnStatement') {
             const index = node.argument.children.findIndex(item => {
             const index = node.argument.children.findIndex(item => {