feat(javascript): add mostPopularFrameworks method to JsDependenciesSnapshot #81

This commit adds a new method called `mostPopularFrameworks` to the `JsDependenciesSnapshot` class in the JavaScript codebase. This method filters the `packages` map to include only the most popular packages and excludes any packages starting with "@type". It then maps each entry to a string representation of the package and its version, if available. The resulting list of dependencies is returned. This method is used in the `getMostPopularPackagesContext` function in the `JavaScriptContextProvider` class to retrieve the most popular packages for the chat context.

Author: phodal

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/JsDependenciesSnapshot.kt

@@ -1,12 +1,13 @@
 package cc.unitmesh.ide.javascript
 
-import cc.unitmesh.devti.provider.context.ChatCreationContext
+import cc.unitmesh.ide.javascript.provider.MOST_POPULAR_PACKAGES
 import com.intellij.javascript.nodejs.PackageJsonData
 import com.intellij.javascript.nodejs.packageJson.PackageJsonFileManager
 import com.intellij.lang.javascript.buildTools.npm.PackageJsonUtil
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.guessProjectDir
 import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.psi.PsiFile
 
 /**
  * Represents a snapshot of JavaScript dependencies in a Kotlin language project.
@@ -21,19 +22,29 @@ import com.intellij.openapi.vfs.VirtualFile
  */
 class JsDependenciesSnapshot(
     val packageJsonFiles: Set<VirtualFile>,
-    val resolvedPackageJson: Boolean,
-    val tsConfigs: Set<VirtualFile>,
+    private val resolvedPackageJson: Boolean,
+    private val tsConfigs: Set<VirtualFile>,
     val packages: Map<String, PackageJsonData.PackageJsonDependencyEntry>
 ) {
+    fun mostPopularFrameworks(): List<String> {
+        val dependencies = this.packages
+            .asSequence()
+            .filter { entry -> MOST_POPULAR_PACKAGES.contains(entry.key) && !entry.key.startsWith("@type") }
+            .map { entry ->
+                val dependency = entry.key
+                val version = entry.value.parseVersion()
+                if (version != null) "$dependency: $version" else dependency
+            }
+            .toList()
+        return dependencies
+    }
+
     companion object {
-        fun create(
-            project: Project,
-            creationContext: ChatCreationContext?
-        ): JsDependenciesSnapshot {
+        fun create(project: Project, psiFile: PsiFile?): JsDependenciesSnapshot {
             var packageJsonFiles = emptySet<VirtualFile>()
             var resolvedPackageJson = false
 
-            val virtualFile = creationContext?.sourceFile?.virtualFile
+            val virtualFile = psiFile?.virtualFile
             if (virtualFile != null) {
                 val packageJson = PackageJsonUtil.findUpPackageJson(virtualFile)
                 if (packageJson != null) {

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/flow/AutoPage.kt

@@ -17,6 +17,10 @@ import cc.unitmesh.ide.javascript.flow.model.DsComponent
 interface AutoPage : TaskFlow<String> {
     var userTask: String
 
+    fun getFrameworks(): List<String> {
+        return listOf("React", "Vue", "Angular")
+    }
+
     /**
      * Retrieves all routes in the project, including the routes in the submodules.
      *

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/flow/ReactAutoPage.kt

@@ -77,7 +77,6 @@ class ReactAutoPage(
         }
     }
 
-
     override fun getPages(): List<DsComponent> = pages
 
     override fun getComponents(): List<DsComponent> = components

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/JavaScriptBuildSystemProvider.kt

@@ -12,7 +12,7 @@ import com.intellij.openapi.project.guessProjectDir
 
 class JavaScriptBuildSystemProvider : BuildSystemProvider() {
     override fun collect(project: Project): DockerfileContext? {
-        val snapshot = JsDependenciesSnapshot.create(project, null)
+        val snapshot = JsDependenciesSnapshot.create(project, null?.sourceFile)
         if (snapshot.packageJsonFiles.isEmpty()) {
             return null
         }

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/JavaScriptContextProvider.kt

@@ -22,15 +22,16 @@ class JavaScriptContextProvider : ChatContextProvider {
 
     override suspend fun collect(project: Project, creationContext: ChatCreationContext): List<ChatContextItem> {
         val results = mutableListOf<ChatContextItem>()
-        val snapshot = JsDependenciesSnapshot.create(project, creationContext)
-
+        val snapshot = JsDependenciesSnapshot.create(project, creationContext?.sourceFile)
         val typeScriptLanguageContext = getTypeScriptLanguageContext(snapshot)
+        val mostPopularPackagesContext = getMostPopularPackagesContext(snapshot)
+
+        val techStack = prepareStack(snapshot)
+
         if (typeScriptLanguageContext != null) results.add(typeScriptLanguageContext)
 
-        val mostPopularPackagesContext = getMostPopularPackagesContext(snapshot)
         if (mostPopularPackagesContext != null) results.add(mostPopularPackagesContext)
 
-        val techStack = prepareStack(snapshot)
         log.info("Tech stack: $techStack")
         if (techStack.coreFrameworks().isNotEmpty()) {
             val element = ChatContextItem(
@@ -70,15 +71,7 @@ class JavaScriptContextProvider : ChatContextProvider {
      *         or null if no popular packages are found
      */
     private fun getMostPopularPackagesContext(snapshot: JsDependenciesSnapshot): ChatContextItem? {
-        val dependencies = snapshot.packages
-            .asSequence()
-            .filter { entry -> MOST_POPULAR_PACKAGES.contains(entry.key) && !entry.key.startsWith("@type") }
-            .map { entry ->
-                val dependency = entry.key
-                val version = entry.value.parseVersion()
-                if (version != null) "$dependency: $version" else dependency
-            }
-            .toList()
+        val dependencies = snapshot.mostPopularFrameworks()
 
         if (dependencies.isEmpty()) return null
 
@@ -133,3 +126,4 @@ class JavaScriptContextProvider : ChatContextProvider {
     }
 }
 
+