Przeglądaj źródła

fix: insertcode support

陈帅 6 lat temu
rodzic
commit
68d9765dab
1 zmienionych plików z 2 dodań i 2 usunięć
  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.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 => {
           if (node.type === 'ReturnStatement') {
             const index = node.argument.children.findIndex(item => {