From 05443695030954d36996edd7df28c2e991377c60 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Tue, 19 May 2020 11:26:15 +0300
Subject: [PATCH] [FIR] Add `transformImports` to FirFile

---
 .../gen/org/jetbrains/kotlin/fir/declarations/FirFile.kt   | 2 ++
 .../jetbrains/kotlin/fir/declarations/impl/FirFileImpl.kt  | 7 ++++++-
 .../kotlin/fir/tree/generator/NodeConfigurator.kt          | 2 +-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/FirFile.kt b/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/FirFile.kt
index c136461e1bb..c4c46efcaf1 100644
--- a/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/FirFile.kt
+++ b/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/FirFile.kt
@@ -36,5 +36,7 @@ abstract class FirFile : FirPureAbstractElement(), FirAnnotationContainer, FirDe
 
     abstract override fun <D> transformAnnotations(transformer: FirTransformer<D>, data: D): FirFile
 
+    abstract fun <D> transformImports(transformer: FirTransformer<D>, data: D): FirFile
+
     abstract fun <D> transformDeclarations(transformer: FirTransformer<D>, data: D): FirFile
 }
diff --git a/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/impl/FirFileImpl.kt b/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/impl/FirFileImpl.kt
index 105e63216f2..a6cd3fbd72c 100644
--- a/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/impl/FirFileImpl.kt
+++ b/compiler/fir/tree/gen/org/jetbrains/kotlin/fir/declarations/impl/FirFileImpl.kt
@@ -43,7 +43,7 @@ internal class FirFileImpl(
 
     override fun <D> transformChildren(transformer: FirTransformer<D>, data: D): FirFileImpl {
         transformAnnotations(transformer, data)
-        imports.transformInplace(transformer, data)
+        transformImports(transformer, data)
         transformDeclarations(transformer, data)
         return this
     }
@@ -53,6 +53,11 @@ internal class FirFileImpl(
         return this
     }
 
+    override fun <D> transformImports(transformer: FirTransformer<D>, data: D): FirFileImpl {
+        imports.transformInplace(transformer, data)
+        return this
+    }
+
     override fun <D> transformDeclarations(transformer: FirTransformer<D>, data: D): FirFileImpl {
         declarations.transformInplace(transformer, data)
         return this
diff --git a/compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/NodeConfigurator.kt b/compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/NodeConfigurator.kt
index 7f3674dc652..72db8b6e9ab 100644
--- a/compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/NodeConfigurator.kt
+++ b/compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/NodeConfigurator.kt
@@ -391,7 +391,7 @@ object NodeConfigurator : AbstractFieldConfigurator<FirTreeBuilder>(FirTreeBuild
         }
 
         file.configure {
-            +fieldList(import)
+            +fieldList(import).withTransform()
             +declarations.withTransform()
             +stringField("name")
             +field("packageFqName", fqNameType)