Fix so that we can process cases 'export default class Class {...}' and 'export class Class {...}' correctly

Zvonimir Sabljic · Zvonimir Sabljic · commit 8d409588dc6d · 2023-06-21T12:25:33.000+02:00
diff --git a/src/utils/code.js b/src/utils/code.js
@@ -260,6 +260,8 @@ function processAst(ast, cb) {
                             return cb(prop.key.name, null, 'exportObj');
                         }
                     });
+                } else if (declaration.type === 'ClassDeclaration') {
+                    return cb(declaration.id ? declaration.id.name : declaration.name, null, 'exportFnDef');
                 }
             } else if (path.isExportNamedDeclaration()) {
                 if (path.node.declaration) {
@@ -269,6 +271,8 @@ function processAst(ast, cb) {
                         path.node.declaration.declarations.forEach(declaration => {
                             return cb(declaration.id.name, null, 'exportFn');
                         });
+                    } else if (path.node.declaration.type === 'ClassDeclaration') {
+                        return cb(path.node.declaration.id.name, null, 'exportFnDef');
                     }
                 } else if (path.node.specifiers.length > 0) {
                     path.node.specifiers.forEach(spec => {

Original file line number	Diff line number	Diff line change
`@@ -260,6 +260,8 @@ function processAst(ast, cb) {`
`260`	`260`	`return cb(prop.key.name, null, 'exportObj');`
`261`	`261`	`}`
`262`	`262`	`});`
	`263`	`+ } else if (declaration.type === 'ClassDeclaration') {`
	`264`	`+ return cb(declaration.id ? declaration.id.name : declaration.name, null, 'exportFnDef');`
`263`	`265`	`}`
`264`	`266`	`} else if (path.isExportNamedDeclaration()) {`
`265`	`267`	`if (path.node.declaration) {`
`@@ -269,6 +271,8 @@ function processAst(ast, cb) {`
`269`	`271`	`path.node.declaration.declarations.forEach(declaration => {`
`270`	`272`	`return cb(declaration.id.name, null, 'exportFn');`
`271`	`273`	`});`
	`274`	`+ } else if (path.node.declaration.type === 'ClassDeclaration') {`
	`275`	`+ return cb(path.node.declaration.id.name, null, 'exportFnDef');`
`272`	`276`	`}`
`273`	`277`	`} else if (path.node.specifiers.length > 0) {`
`274`	`278`	`path.node.specifiers.forEach(spec => {`