feat(database): add DbContext and DbContextProvider classes #80

This commit adds the `DbContext` and `DbContextProvider` classes to the `GenSqlScriptBySelection.kt` file in the `exts/database/src/main/kotlin/cc/unitmesh/database/actions/` directory.

The `DbContext` class is a data class that holds information about the database version, schema name, and table names.

The `DbContextProvider` class is responsible for retrieving the columns of specified tables. It has a `getTableColumns` function that takes a list of table names and returns a list of column names from those tables.

Author: phodal

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

@@ -3,7 +3,9 @@ package cc.unitmesh.database.actions
 import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
 import com.intellij.database.model.DasColumn
+import com.intellij.database.model.DasTable
 import com.intellij.database.model.ObjectKind
+import com.intellij.database.model.RawDataSource
 import com.intellij.database.psi.DbElement
 import com.intellij.database.psi.DbPsiFacade
 import com.intellij.database.util.DasUtil
@@ -39,12 +41,42 @@ class GenSqlScriptBySelection : AbstractChatIntention() {
             tables.filter { table -> table.kind == ObjectKind.TABLE && table.dasParent?.name == schemaName }
         }.toList()
 
+        val tableColumns = DbContextProvider(dasTables).getTableColumns(dasTables.map { it.name })
+
         val prompt = """
             |Database: $databaseVersion
             |Requirement: $selectedText
             |Tables: ${dasTables.joinToString { it.name }}
+            |Columns: ${tableColumns.joinToString { it }}
         """.trimMargin()
 
         println(prompt)
     }
 }
+
+data class DbContext(
+    val databaseVersion: String,
+    val schemaName: String,
+    val tableNames: List<String>,
+) {
+}
+
+data class DbContextProvider(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.flatMap { tableName ->
+            if (tables.contains(tableName.name)) {
+                DasUtil.getColumns(tableName).map {
+                    "${it.name}: ${it.dasType.toDataType()}"
+                }
+            } else {
+                emptyList()
+            }
+        }
+    }
+}