feat(scala): add ScalaClassContextBuilder and test case

- Added `ScalaClassContextBuilder` class to handle building class context for Scala classes.
- Added a test case `ScalaClassContextBuilderTest` to test the functionality of `ScalaClassContextBuilder`.
- The test case verifies that the `getClassContext` method correctly retrieves the class name from a Scala file.
- The `ScalaClassContextBuilder` class extracts information such as functions, fields, superclass, and annotations from the Scala class.
- The class context is used for further processing and analysis in the plugin.

Author: phodal

build.gradle.kts

@@ -72,7 +72,7 @@ val cppPlugins = listOf(
     "com.intellij.cidr.lang",
     "com.intellij.clion",
     "com.intellij.cidr.base",
-    "com.intellij.nativeDebug",
+//    "com.intellij.nativeDebug",
     "org.jetbrains.plugins.clion.test.google",
     "org.jetbrains.plugins.clion.test.catch"
 )
@@ -83,6 +83,7 @@ val rustPlugins = listOf(
 
 val riderVersion = prop("riderVersion")
 val riderPlugins: List<String> = listOf()
+val scalaPlugin = prop("scalaPlugin")
 
 val pluginProjects: List<Project> get() = rootProject.allprojects.toList()
 val ideaPlugins =
@@ -224,6 +225,9 @@ project(":plugin") {
             "idea" -> {
                 pluginList += javaPlugins
             }
+            "scala" -> {
+                pluginList += javaPlugins + scalaPlugin
+            }
 
             "python" -> {
                 pluginList += pycharmPlugins
@@ -424,6 +428,7 @@ project(":") {
         kover(project(":kotlin"))
         kover(project(":pycharm"))
         kover(project(":rust"))
+        kover(project(":scala"))
     }
 
     task("resolveDependencies") {
@@ -478,6 +483,17 @@ project(":kotlin") {
     }
 }
 
+project(":scala") {
+    intellij {
+        version.set(ideaVersion)
+        plugins.set(ideaPlugins + scalaPlugin)
+    }
+    dependencies {
+        implementation(project(":"))
+        implementation(project(":java"))
+    }
+}
+
 project(":rust") {
     intellij {
         version.set(ideaVersion)

gradle-222.properties

@@ -9,6 +9,9 @@ pluginUntilBuild = 232.*
 
 clionVersion=CL-2022.2.4
 
+# check latest available version here https://plugins.jetbrains.com/plugin/1347-scala/versions
+scalaPlugin=org.intellij.scala:2022.3.2
+
 # https://plugins.jetbrains.com/plugin/9568-go/versions
 goPlugin=org.jetbrains.plugins.go:222.4459.24
 

gradle-233.properties

@@ -10,7 +10,7 @@ pluginUntilBuild=241.*
 clionVersion=CL-2023.3
 
 # check latest available version here https://plugins.jetbrains.com/plugin/1347-scala/versions
-#scalaPlugin=org.intellij.scala:2023.3.19
+scalaPlugin=org.intellij.scala:2023.3.19
 
 # check latest available version here https://plugins.jetbrains.com/plugin/22407-rust/versions
 rustPlugin=com.jetbrains.rust:233.21799.284

plugin/src/main/resources/META-INF/plugin.xml

@@ -28,5 +28,6 @@
         <module name="cc.unitmesh.go"/>
         <module name="cc.unitmesh.rust"/>
         <module name="cc.unitmesh.cpp"/>
+        <module name="cc.unitmesh.scala"/>
     </content>
 </idea-plugin>

scala/src/main/kotlin/cc/unitmesh/scala/context/ScalaClassContextBuilder.kt

@@ -0,0 +1,58 @@
+package cc.unitmesh.scala.context
+
+import cc.unitmesh.devti.context.ClassContext
+import cc.unitmesh.devti.context.builder.ClassContextBuilder
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiMethod
+import com.intellij.psi.PsiNameIdentifierOwner
+import com.intellij.psi.PsiReference
+import com.intellij.psi.search.GlobalSearchScope
+import com.intellij.psi.search.SearchScope
+import com.intellij.psi.search.searches.MethodReferencesSearch
+import com.intellij.psi.search.searches.ReferencesSearch
+import org.jetbrains.plugins.scala.lang.psi.api.toplevel.typedef.ScClass
+
+class ScalaClassContextBuilder : ClassContextBuilder {
+    override fun getClassContext(psiElement: PsiElement, gatherUsages: Boolean): ClassContext? {
+        if (psiElement !is ScClass) return null
+
+        val name = psiElement.name ?: return null
+
+        val functions = psiElement.methods.toList()
+        val allFields = psiElement.fields.toList()
+        val superClass = psiElement.supers.mapNotNull { it.qualifiedName }.toList()
+        val usages = findUsages(psiElement as PsiNameIdentifierOwner)
+
+        val annotations = psiElement.annotations.mapNotNull {
+            it.text
+        }
+
+        val displayName = psiElement.qualifiedName ?: psiElement.name ?: ""
+        return ClassContext(
+            psiElement,
+            psiElement.text,
+            name,
+            functions,
+            allFields,
+            superClass,
+            usages,
+            displayName = displayName,
+            annotations
+        )
+    }
+}
+
+fun findUsages(nameIdentifierOwner: PsiNameIdentifierOwner): List<PsiReference> {
+    val project = nameIdentifierOwner.project
+    val searchScope = GlobalSearchScope.allScope(project) as SearchScope
+
+    return when (nameIdentifierOwner) {
+        is PsiMethod -> {
+            MethodReferencesSearch.search(nameIdentifierOwner, searchScope, true)
+        }
+
+        else -> {
+            ReferencesSearch.search((nameIdentifierOwner as PsiElement), searchScope, true)
+        }
+    }.findAll().map { it as PsiReference }
+}

scala/src/main/resources/cc.unitmesh.scala.xml

@@ -0,0 +1,16 @@
+<idea-plugin package="cc.unitmesh.scala">
+    <!--suppress PluginXmlValidity -->
+    <dependencies>
+        <!--suppress PluginXmlValidity -->
+        <module name="jvm-core"/>
+        <plugin id="org.intellij.scala"/>
+<!--        <plugin id="org.jetbrains.plugins.gradle"/>-->
+    </dependencies>
+
+    <extensions defaultExtensionNs="cc.unitmesh">
+        <classContextBuilder
+                language="Scala"
+                implementationClass="cc.unitmesh.scala.context.ScalaClassContextBuilder"
+                />
+    </extensions>
+</idea-plugin>

scala/src/test/kotlin/cc/unitmesh/scala/context/ScalaClassContextBuilderTest.kt

@@ -0,0 +1,29 @@
+package cc.unitmesh.scala.context
+
+import com.intellij.psi.PsiFileFactory
+import com.intellij.testFramework.LightPlatformTestCase
+
+class ScalaClassContextBuilderTest: LightPlatformTestCase() {
+    fun testShouldGetClassName() {
+        val fileFactory: PsiFileFactory = PsiFileFactory.getInstance(project)
+
+        val classCode = """
+            class Point(var x: Int, var y: Int):
+              def move(dx: Int, dy: Int): Unit =
+                x = x + dx
+                y = y + dy
+            
+              override def toString: String =
+                s"(${'$'}x, ${'$'}y)"
+            end Point
+            """.trimIndent()
+
+        val psiFile = fileFactory.createFileFromText("Point.scala", classCode)
+        val psiElement = psiFile.children[0]
+        val classContext = ScalaClassContextBuilder().getClassContext(psiElement, false)!!
+
+        assertEquals(classContext.name, "Point")
+//        TestCase.assertEquals(classContext.methods.size, 2)
+//        TestCase.assertEquals(classContext.format(), """""")
+    }
+}

settings.gradle.kts

@@ -14,4 +14,5 @@ include(
     "rust",
     "csharp",
     "cpp",
+    "scala",
 )