@@ -101,6 +101,7 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
101
101
.joinToString(" , " ) { it.name }
102
102
builder.addFunction(
103
103
FunSpec .builder(" getConcepts" )
104
+ .returns(List ::class .asClassName().parameterizedBy(IConcept ::class .asTypeName()))
104
105
.addModifiers(KModifier .OVERRIDE )
105
106
.addCode(language.getConcepts().map { it.conceptObjectType() }.toListLiteralCodeBlock())
106
107
.build()
@@ -139,7 +140,9 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
139
140
.build()
140
141
)
141
142
143
+ val enumType = ClassName (enum.language.name, enum.name)
142
144
val getLiteralFunBuilder = FunSpec .builder(" getLiteralByMemberId" )
145
+ .returns(enumType)
143
146
.addParameter(" uid" , String ::class )
144
147
val getLiteralCodeBuilder = CodeBlock .builder().beginControlFlow(" return when (uid) {" )
145
148
@@ -166,6 +169,7 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
166
169
val companion = TypeSpec .companionObjectBuilder()
167
170
.addFunction(
168
171
FunSpec .builder(" defaultValue" )
172
+ .returns(enumType)
169
173
.addCode(" return values()[%L]" , enum.defaultIndex)
170
174
.build())
171
175
.addFunction(
@@ -326,12 +330,12 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
326
330
)
327
331
}
328
332
333
+ val inputStepType = IMonoStep ::class .asTypeName()
334
+ .parameterizedBy(concept.nodeWrapperInterfaceType())
329
335
addFunction(
330
336
FunSpec .builder(feature.setterName())
331
- .receiver(
332
- IMonoStep ::class .asTypeName()
333
- .parameterizedBy(concept.nodeWrapperInterfaceType())
334
- )
337
+ .returns(inputStepType)
338
+ .receiver(inputStepType)
335
339
.addParameter(" value" , IMonoStep ::class .asTypeName().parameterizedBy(feature.asKotlinType()))
336
340
.addStatement(
337
341
" return %T.setProperty(this, %T.%N, value)" ,
@@ -417,12 +421,14 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
417
421
val instanceClassType = KClass ::class .asClassName().parameterizedBy(concept.nodeWrapperImplType())
418
422
addFunction(
419
423
FunSpec .builder(GeneratedConcept <* , * >::getInstanceClass.name)
424
+ .returns(KClass ::class .asClassName().parameterizedBy(concept.nodeWrapperImplType()))
420
425
.addModifiers(KModifier .OVERRIDE )
421
426
.addStatement(""" return %T::class""" , concept.nodeWrapperImplType())
422
427
.build()
423
428
)
424
429
addFunction(
425
430
FunSpec .builder(GeneratedConcept <* , * >::typed.name)
431
+ .returns(concept.conceptWrapperInterfaceType())
426
432
.addModifiers(KModifier .OVERRIDE )
427
433
.addStatement(""" return %T""" , concept.conceptWrapperInterfaceClass())
428
434
.build()
@@ -434,6 +440,7 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
434
440
)
435
441
addFunction(
436
442
FunSpec .builder(GeneratedConcept <* , * >::wrap.name)
443
+ .returns(concept.nodeWrapperImplType())
437
444
.addModifiers(KModifier .OVERRIDE )
438
445
.addParameter(" node" , INode ::class )
439
446
.addStatement(" return %T(node)" , concept.nodeWrapperImplType())
@@ -442,6 +449,7 @@ class MetaModelGenerator(val outputDir: Path, val nameConfig: NameConfig = NameC
442
449
concept.uid?.let { uid ->
443
450
addFunction(
444
451
FunSpec .builder(GeneratedConcept <* , * >::getUID.name)
452
+ .returns(String ::class )
445
453
.addModifiers(KModifier .OVERRIDE )
446
454
.addStatement(CodeBlock .of(" return %S" , uid).toString())
447
455
.build()
0 commit comments