azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[FIR] setup origin for generated enum functions

Browse Source
This commit is contained in:
Anna Kozlova
2023-04-27 09:38:12 +02:00
committed by teamcity
parent 68b94b07b8
commit 3f6432cdc9
1 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/tree/src/org/jetbrains/kotlin/fir/EnumClassUtils.kt
+20 -8
View File
@@ -34,12 +34,16 @@ import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.name.StandardClassIds import org.jetbrains.kotlin.name.StandardClassIds
fun FirRegularClassBuilder.generateValuesFunction( fun FirRegularClassBuilder.generateValuesFunction(
moduleData: FirModuleData, packageFqName: FqName, classFqName: FqName, makeExpect: Boolean = false moduleData: FirModuleData,
packageFqName: FqName,
classFqName: FqName,
makeExpect: Boolean = false,
origin: FirDeclarationOrigin = FirDeclarationOrigin.Source
) { ) {
val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration) val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration)
declarations += buildSimpleFunction { declarations += buildSimpleFunction {
source = sourceElement source = sourceElement
origin = FirDeclarationOrigin.Source this.origin = origin
this.moduleData = moduleData this.moduleData = moduleData
returnTypeRef = buildResolvedTypeRef { returnTypeRef = buildResolvedTypeRef {
source = sourceElement source = sourceElement
@@ -67,12 +71,16 @@ fun FirRegularClassBuilder.generateValuesFunction(
} }
fun FirRegularClassBuilder.generateValueOfFunction( fun FirRegularClassBuilder.generateValueOfFunction(
moduleData: FirModuleData, packageFqName: FqName, classFqName: FqName, makeExpect: Boolean = false moduleData: FirModuleData,
packageFqName: FqName,
classFqName: FqName,
makeExpect: Boolean = false,
origin: FirDeclarationOrigin = FirDeclarationOrigin.Source
) { ) {
val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration) val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration)
declarations += buildSimpleFunction { declarations += buildSimpleFunction {
source = sourceElement source = sourceElement
origin = FirDeclarationOrigin.Source this.origin = origin
this.moduleData = moduleData this.moduleData = moduleData
returnTypeRef = buildResolvedTypeRef { returnTypeRef = buildResolvedTypeRef {
source = sourceElement source = sourceElement
@@ -92,7 +100,7 @@ fun FirRegularClassBuilder.generateValueOfFunction(
valueParameters += buildValueParameter vp@{ valueParameters += buildValueParameter vp@{
source = sourceElement source = sourceElement
containingFunctionSymbol = this@buildSimpleFunction.symbol containingFunctionSymbol = this@buildSimpleFunction.symbol
origin = FirDeclarationOrigin.Source this.origin = origin
this.moduleData = moduleData this.moduleData = moduleData
returnTypeRef = buildResolvedTypeRef { returnTypeRef = buildResolvedTypeRef {
source = sourceElement source = sourceElement
@@ -119,14 +127,18 @@ fun FirRegularClassBuilder.generateValueOfFunction(
} }
fun FirRegularClassBuilder.generateEntriesGetter( fun FirRegularClassBuilder.generateEntriesGetter(
moduleData: FirModuleData, packageFqName: FqName, classFqName: FqName, makeExpect: Boolean = false moduleData: FirModuleData,
packageFqName: FqName,
classFqName: FqName,
makeExpect: Boolean = false,
origin: FirDeclarationOrigin = FirDeclarationOrigin.Source
) { ) {
val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration) val sourceElement = source?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration)
declarations += buildProperty { declarations += buildProperty {
source = sourceElement source = sourceElement
isVar = false isVar = false
isLocal = false isLocal = false
origin = FirDeclarationOrigin.Source this.origin = origin
this.moduleData = moduleData this.moduleData = moduleData
returnTypeRef = buildResolvedTypeRef { returnTypeRef = buildResolvedTypeRef {
source = sourceElement source = sourceElement
@@ -147,7 +159,7 @@ fun FirRegularClassBuilder.generateEntriesGetter(
resolvePhase = this@generateEntriesGetter.resolvePhase resolvePhase = this@generateEntriesGetter.resolvePhase
getter = FirDefaultPropertyGetter( getter = FirDefaultPropertyGetter(
sourceElement?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration), sourceElement?.fakeElement(KtFakeSourceElementKind.EnumGeneratedDeclaration),
moduleData, FirDeclarationOrigin.Source, returnTypeRef.copyWithNewSourceKind(KtFakeSourceElementKind.EnumGeneratedDeclaration), moduleData, origin, returnTypeRef.copyWithNewSourceKind(KtFakeSourceElementKind.EnumGeneratedDeclaration),
Visibilities.Public, symbol Visibilities.Public, symbol
).apply { ).apply {
this.status = createStatus(this@generateEntriesGetter.status).apply { this.status = createStatus(this@generateEntriesGetter.status).apply {