From 0428916d3c499b67584a2e958a2970c325aac193 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Thu, 31 Oct 2019 13:52:41 +0300
Subject: [PATCH] [FIR] Add error nodes to default transformer and visitors

---
 .../fir/visitors/FirDefaultTransformer.kt     | 20 ++++++++++++++++++-
 .../kotlin/fir/visitors/FirDefaultVisitor.kt  | 18 +++++++++++++++++
 .../fir/visitors/FirDefaultVisitorVoid.kt     | 18 +++++++++++++++++
 3 files changed, 55 insertions(+), 1 deletion(-)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultTransformer.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultTransformer.kt
index 240999d73e4..1873761ae5d 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultTransformer.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultTransformer.kt
@@ -5,10 +5,12 @@
 
 package org.jetbrains.kotlin.fir.visitors
 
-import org.jetbrains.kotlin.fir.declarations.FirDeclaration
 import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.declarations.FirErrorFunction
 import org.jetbrains.kotlin.fir.declarations.FirSealedClass
 import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.references.FirErrorNamedReference
+import org.jetbrains.kotlin.fir.references.FirReference
 import org.jetbrains.kotlin.fir.types.*
 
 abstract class FirDefaultTransformer<D> : FirTransformer<D>() {
@@ -90,5 +92,21 @@ abstract class FirDefaultTransformer<D> : FirTransformer<D>() {
     override fun transformSealedClass(sealedClass: FirSealedClass, data: D): CompositeTransformResult<FirStatement> {
         return transformRegularClass(sealedClass, data)
     }
+
+    override fun transformErrorExpression(errorExpression: FirErrorExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformExpression(errorExpression, data)
+    }
+
+    override fun transformErrorLoop(errorLoop: FirErrorLoop, data: D): CompositeTransformResult<FirStatement> {
+        return transformLoop(errorLoop, data)
+    }
+
+    override fun transformErrorNamedReference(errorNamedReference: FirErrorNamedReference, data: D): CompositeTransformResult<FirReference> {
+        return transformNamedReference(errorNamedReference, data)
+    }
+
+    override fun transformErrorFunction(errorFunction: FirErrorFunction, data: D): CompositeTransformResult<FirStatement> {
+        return transformFunction(errorFunction, data)
+    }
 }
 
diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitor.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitor.kt
index d06e23d4885..27ad53b61cf 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitor.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitor.kt
@@ -6,8 +6,10 @@
 package org.jetbrains.kotlin.fir.visitors
 
 import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.declarations.FirErrorFunction
 import org.jetbrains.kotlin.fir.declarations.FirSealedClass
 import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.references.FirErrorNamedReference
 import org.jetbrains.kotlin.fir.types.*
 
 abstract class FirDefaultVisitor<R, D> : FirVisitor<R, D>() {
@@ -86,5 +88,21 @@ abstract class FirDefaultVisitor<R, D> : FirVisitor<R, D>() {
     override fun visitSealedClass(sealedClass: FirSealedClass, data: D): R {
         return visitRegularClass(sealedClass, data)
     }
+
+    override fun visitErrorExpression(errorExpression: FirErrorExpression, data: D): R {
+        return visitExpression(errorExpression, data)
+    }
+
+    override fun visitErrorLoop(errorLoop: FirErrorLoop, data: D): R {
+        return visitLoop(errorLoop, data)
+    }
+
+    override fun visitErrorNamedReference(errorNamedReference: FirErrorNamedReference, data: D): R {
+        return visitNamedReference(errorNamedReference, data)
+    }
+
+    override fun visitErrorFunction(errorFunction: FirErrorFunction, data: D): R {
+        return visitFunction(errorFunction, data)
+    }
 }
 
diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitorVoid.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitorVoid.kt
index d1cf7e39635..d1194462382 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitorVoid.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/visitors/FirDefaultVisitorVoid.kt
@@ -6,8 +6,10 @@
 package org.jetbrains.kotlin.fir.visitors
 
 import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.declarations.FirErrorFunction
 import org.jetbrains.kotlin.fir.declarations.FirSealedClass
 import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.references.FirErrorNamedReference
 import org.jetbrains.kotlin.fir.types.*
 
 abstract class FirDefaultVisitorVoid : FirVisitorVoid() {
@@ -86,4 +88,20 @@ abstract class FirDefaultVisitorVoid : FirVisitorVoid() {
     override fun visitSealedClass(sealedClass: FirSealedClass) {
         return visitRegularClass(sealedClass)
     }
+
+    override fun visitErrorExpression(errorExpression: FirErrorExpression) {
+        return visitExpression(errorExpression)
+    }
+
+    override fun visitErrorLoop(errorLoop: FirErrorLoop) {
+        return visitLoop(errorLoop)
+    }
+
+    override fun visitErrorNamedReference(errorNamedReference: FirErrorNamedReference) {
+        return visitNamedReference(errorNamedReference)
+    }
+
+    override fun visitErrorFunction(errorFunction: FirErrorFunction) {
+        return visitFunction(errorFunction)
+    }
 }
\ No newline at end of file