[FIR] Add transformImports to FirFile

This commit is contained in:
Dmitriy Novozhilov
2020-05-19 11:26:15 +03:00
parent f60c8eac71
commit 0544369503
3 changed files with 9 additions and 2 deletions
@@ -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
}
@@ -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
@@ -391,7 +391,7 @@ object NodeConfigurator : AbstractFieldConfigurator<FirTreeBuilder>(FirTreeBuild
}
file.configure {
+fieldList(import)
+fieldList(import).withTransform()
+declarations.withTransform()
+stringField("name")
+field("packageFqName", fqNameType)