feat(database): add DbContextActionProvider class #80

This commit adds the `DbContextActionProvider` class, which retrieves the columns of specified tables. It includes a `getTableColumns` function that takes a list of table names and returns a list of column names from those tables. The implementation uses the `DasUtil.getColumns` method to retrieve the columns and formats them as "TableName: ColumnName: DataType". This class is added to the `database` package in both the `222` and `233` directories. Additionally, the `GenSqlScriptBySelection.kt` file is modified to import and use the `DbContextActionProvider` class.

Author: phodal

exts/database/src/222/main/kotlin/cc/unitmesh/database/DbContextActionProvider.kt

@@ -0,0 +1,26 @@
+package cc.unitmesh.database
+
+import com.intellij.database.model.DasTable
+import com.intellij.database.util.DasUtil
+
+data class DbContextActionProvider(val dasTables: List<DasTable>) {
+    /**
+     * Retrieves the columns of the specified tables.
+     *
+     * @param tables A list of table names to retrieve the columns from.
+     * @return A list of column names from the specified tables.
+     */
+    fun getTableColumns(tables: List<String>): List<String> {
+        return dasTables.mapNotNull { tableName ->
+            if (tables.contains(tableName.name)) {
+                val columns = DasUtil.getColumns(tableName).map {
+                    "${it.name}: ${it.dataType}"
+                }.joinToString(", ")
+
+                "TableName: ${tableName.name}, Columns: $columns"
+            } else {
+                null
+            }
+        }
+    }
+}

exts/database/src/233/main/kotlin/cc/unitmesh/database/DbContextActionProvider.kt

@@ -0,0 +1,26 @@
+package cc.unitmesh.database
+
+import com.intellij.database.model.DasTable
+import com.intellij.database.util.DasUtil
+
+data class DbContextActionProvider(val dasTables: List<DasTable>) {
+    /**
+     * Retrieves the columns of the specified tables.
+     *
+     * @param tables A list of table names to retrieve the columns from.
+     * @return A list of column names from the specified tables.
+     */
+    fun getTableColumns(tables: List<String>): List<String> {
+        return dasTables.mapNotNull { tableName ->
+            if (tables.contains(tableName.name)) {
+                val columns = DasUtil.getColumns(tableName).map {
+                    "${it.name}: ${it.dasType.toDataType()}"
+                }.joinToString(", ")
+
+                "TableName: ${tableName.name}, Columns: $columns"
+            } else {
+                null
+            }
+        }
+    }
+}

exts/database/src/main/kotlin/cc/unitmesh/database/actions/GenSqlScriptBySelection.kt

@@ -1,5 +1,6 @@
 package cc.unitmesh.database.actions
 
+import cc.unitmesh.database.DbContextActionProvider
 import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.gui.sendToChatPanel
@@ -9,7 +10,6 @@ import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.template.TemplateRender
 import cc.unitmesh.devti.util.LLMCoroutineScope
 import cc.unitmesh.devti.util.parser.parseCodeFromString
-import com.intellij.database.model.DasTable
 import com.intellij.database.model.ObjectKind
 import com.intellij.database.psi.DbPsiFacade
 import com.intellij.database.util.DasUtil
@@ -22,9 +22,6 @@ import com.intellij.openapi.progress.Task
 import com.intellij.openapi.progress.impl.BackgroundableProcessIndicator
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiFile
-import com.intellij.util.awaitCancellationAndInvoke
-import kotlinx.coroutines.async
-import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
 
 
@@ -209,24 +206,3 @@ data class DbContext(
     var tableInfos: List<String> = emptyList(),
 )
 
-data class DbContextActionProvider(val dasTables: List<DasTable>) {
-    /**
-     * Retrieves the columns of the specified tables.
-     *
-     * @param tables A list of table names to retrieve the columns from.
-     * @return A list of column names from the specified tables.
-     */
-    fun getTableColumns(tables: List<String>): List<String> {
-        return dasTables.mapNotNull { tableName ->
-            if (tables.contains(tableName.name)) {
-                val columns = DasUtil.getColumns(tableName).map {
-                    "${it.name}: ${it.dasType.toDataType()}"
-                }.joinToString(", ")
-
-                "TableName: ${tableName.name}, Columns: $columns"
-            } else {
-                null
-            }
-        }
-    }
-}

scala/src/test/kotlin/DbContextActionProvider.kt

@@ -0,0 +1,24 @@
+import com.intellij.database.model.DasTable
+import com.intellij.database.util.DasUtil
+
+data class DbContextActionProvider(val dasTables: List<DasTable>) {
+    /**
+     * Retrieves the columns of the specified tables.
+     *
+     * @param tables A list of table names to retrieve the columns from.
+     * @return A list of column names from the specified tables.
+     */
+    fun getTableColumns(tables: List<String>): List<String> {
+        return dasTables.mapNotNull { tableName ->
+            if (tables.contains(tableName.name)) {
+                val columns = DasUtil.getColumns(tableName).map {
+                    "${it.name}: ${it.dataType.typeName}"
+                }.joinToString(", ")
+
+                "TableName: ${tableName.name}, Columns: $columns"
+            } else {
+                null
+            }
+        }
+    }
+}