Working on supporting BaseNode in compiler plugin

Author: ftomassetti

ast/src/commonMain/kotlin/com/strumenta/kolasu/model/BaseNode.kt

@@ -33,7 +33,10 @@ abstract class BaseNode : NodeLike {
         TODO("Not yet implemented")
     }
 
-    override fun <T : NodeLike> addToContainment(containmentName: String, child: T) {
+    override fun <T : NodeLike> addToContainment(
+        containmentName: String,
+        child: T,
+    ) {
         TODO("Not yet implemented")
     }
 
@@ -65,19 +68,28 @@ abstract class BaseNode : NodeLike {
         TODO("Not yet implemented")
     }
 
-    override fun <T : NodeLike> removeFromContainment(containmentName: String, child: T) {
+    override fun <T : NodeLike> removeFromContainment(
+        containmentName: String,
+        child: T,
+    ) {
         TODO("Not yet implemented")
     }
 
-    override fun <T> setAttribute(attributeName: String, value: T) {
+    override fun <T> setAttribute(
+        attributeName: String,
+        value: T,
+    ) {
         TODO("Not yet implemented")
     }
 
-    override fun <T : PossiblyNamed> setReferenceReferred(referenceName: String, referred: T) {
+    override fun <T : PossiblyNamed> setReferenceReferred(
+        referenceName: String,
+        referred: T,
+    ) {
         TODO("Not yet implemented")
     }
 
     override fun subscribe(observer: ObservableObserver<NodeNotification<in NodeLike>>) {
         TODO("Not yet implemented")
     }
-}
+}

ast/src/commonMain/kotlin/com/strumenta/kolasu/model/FeatureDescription.kt

@@ -4,18 +4,18 @@ data class FeatureDescription(
     val name: String,
     val provideNodes: Boolean,
     val multiplicity: Multiplicity,
-    val valueProvider: ()->Any?,
+    val valueProvider: () -> Any?,
     val featureType: FeatureType,
-    val derived: Boolean = false
+    val derived: Boolean = false,
 ) {
     constructor(
         name: String,
         provideNodes: Boolean,
         multiplicity: Multiplicity,
         value: Any?,
         featureType: FeatureType,
-        derived: Boolean = false
-    ) : this(name, provideNodes, multiplicity, {value}, featureType, derived)
+        derived: Boolean = false,
+    ) : this(name, provideNodes, multiplicity, { value }, featureType, derived)
 
     val isMultiple: Boolean
         get() = multiplicity == Multiplicity.MANY
@@ -25,5 +25,4 @@ data class FeatureDescription(
 
     val value: Any?
         get() = valueProvider.invoke()
-
 }

gradle.properties

@@ -1,5 +1,5 @@
 version=1.6.0-RC1-SNAPSHOT
-kotlinVersion=2.0.0-Beta2
+kotlinVersion=2.0.0-Beta3
 dokkaVersion=1.9.10
 antlr_version=4.9.3
 clikt_version=3.5.2

kotlin-ir-plugin/src/main/kotlin/com/strumenta/kolasu/kcp/StarLasuComponentRegistrar.kt

@@ -15,7 +15,7 @@ import org.jetbrains.kotlin.fir.extensions.FirExtensionRegistrarAdapter
 @AutoService(CompilerPluginRegistrar::class)
 class StarLasuComponentRegistrar : CompilerPluginRegistrar() {
     override val supportsK2: Boolean
-        get() = false
+        get() = true
 
     override fun ExtensionStorage.registerExtensions(configuration: CompilerConfiguration) {
         val messageCollector = configuration.get(CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY, MessageCollector.NONE)

kotlin-ir-plugin/src/main/kotlin/com/strumenta/kolasu/kcp/StarLasuIrGenerationExtension.kt

@@ -2,6 +2,7 @@
 
 package com.strumenta.kolasu.kcp
 
+import com.strumenta.kolasu.model.BaseNode
 import com.strumenta.kolasu.model.Node
 import org.jetbrains.kotlin.backend.common.extensions.FirIncompatiblePluginAPI
 import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
@@ -21,6 +22,7 @@ class StarLasuIrGenerationExtension(
     private fun checkASTNode(
         irClass: IrClass,
         pluginContext: IrPluginContext,
+        isBaseNode: Boolean,
     ) {
         irClass.primaryConstructor?.valueParameters?.forEach { param ->
             if (param.isVal() && (param.isSingleContainment() || param.isSingleAttribute())) {
@@ -33,25 +35,45 @@ class StarLasuIrGenerationExtension(
         }
         irClass.accept(FieldObservableExtension(pluginContext), null)
         irClass.accept(SettingParentExtension(pluginContext, messageCollector), null)
+        if (isBaseNode) {
+//            override val properties: List<FeatureDescription>
+//            get() = TODO("Not yet implemented")
+        }
     }
 
     override fun generate(
         moduleFragment: IrModuleFragment,
         pluginContext: IrPluginContext,
     ) {
+        messageCollector.report(
+            CompilerMessageSeverity.WARNING,
+            "COMPILER PLUGIN IS GENERATING",
+        )
         moduleFragment.files.forEach { irFile ->
-            irFile.declarations.filterIsInstance(IrClass::class.java).forEach { irClass ->
+            irFile.declarations.filterIsInstance<IrClass>().forEach { irClass ->
                 val isASTNode =
                     irClass.getAllSuperclasses().any {
                         it.kotlinFqName.toString() == Node::class.qualifiedName
                     }
-                if (isASTNode) {
-                    messageCollector.report(
-                        CompilerMessageSeverity.INFO,
-                        "AST class ${irClass.kotlinFqName} identified",
-                        irClass.compilerSourceLocation,
-                    )
-                    checkASTNode(irClass, pluginContext)
+                val isBaseNode =
+                    irClass.getAllSuperclasses().any {
+                        it.kotlinFqName.toString() == BaseNode::class.qualifiedName
+                    }
+                if (isASTNode || isBaseNode) {
+                    if (isASTNode) {
+                        messageCollector.report(
+                            CompilerMessageSeverity.INFO,
+                            "AST class ${irClass.kotlinFqName} identified",
+                            irClass.compilerSourceLocation,
+                        )
+                    } else {
+                        messageCollector.report(
+                            CompilerMessageSeverity.INFO,
+                            "BaseNode subclass ${irClass.kotlinFqName} identified",
+                            irClass.compilerSourceLocation,
+                        )
+                    }
+                    checkASTNode(irClass, pluginContext, isBaseNode)
                 }
             }
         }