fix: node.parent undefined error

searchfe · May 22, 2019 · fd47c1f · fd47c1f
1 parent 3d20979
commit fd47c1f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/emitter.ts b/src/emitter.ts
@@ -800,7 +800,7 @@ export function emitFile(
         // emitDecorators(node, node.decorators);
         emitModifiers(node, node.modifiers);
         // emit(node.asteriskToken);
-        if (ts.isClassDeclaration(node.parent)) {
+        if (ts.isClassDeclaration(node.parent || node.original.parent)) {
             writeKeyword("function");
             writeSpace();
             emit(node.name);
@@ -1718,7 +1718,7 @@ export function emitFile(
         if (
             node.kind === ts.SyntaxKind.ArrowFunction
             || node.kind === ts.SyntaxKind.FunctionExpression
-            || (node.kind === ts.SyntaxKind.MethodDeclaration && !ts.isClassDeclaration(node.parent))
+            || (node.kind === ts.SyntaxKind.MethodDeclaration && !ts.isClassDeclaration(node.parent || node.original.parent))
         ) {
             const wrappedNode = createWrappedNode(node, {typeChecker});
             const body = wrappedNode.getBody();