mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-06-20 21:43:56 +08:00
3da3a3fb13
- 扩展数据库名字段识别，兼容 SCHEMA、database_name 等返回列名 - 按驱动返回列顺序兜底提取单列结果，避免非标准列名导致误判为空 - 补充 MyCAT 风格回归测试，覆盖 SHOW DATABASES 与当前库回退逻辑 Close #552
116 lines
4.4 KiB
Go
116 lines
4.4 KiB
Go
package mcpserver
|
|
|
|
import (
|
|
"context"
|
|
|
|
"GoNavi-Wails/internal/ai"
|
|
aiservice "GoNavi-Wails/internal/ai/service"
|
|
appcore "GoNavi-Wails/internal/app"
|
|
"GoNavi-Wails/internal/appdata"
|
|
"GoNavi-Wails/internal/connection"
|
|
"GoNavi-Wails/internal/logger"
|
|
)
|
|
|
|
// Backend 抽象 GoNavi 后端能力，便于复用真实 App 和单元测试替身。
|
|
type Backend interface {
|
|
Close(context.Context) error
|
|
GetSavedConnections() ([]connection.SavedConnectionView, error)
|
|
GetEditableSavedConnection(id string) (connection.SavedConnectionView, error)
|
|
DBGetDatabases(config connection.ConnectionConfig) connection.QueryResult
|
|
DBGetTables(config connection.ConnectionConfig, dbName string) connection.QueryResult
|
|
DBGetAllColumns(config connection.ConnectionConfig, dbName string) connection.QueryResult
|
|
DBGetColumns(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult
|
|
DBGetIndexes(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult
|
|
DBGetForeignKeys(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult
|
|
DBGetTriggers(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult
|
|
DBShowCreateTable(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult
|
|
DBQueryMulti(config connection.ConnectionConfig, dbName string, query string, queryID string) connection.QueryResult
|
|
InspectSQL(dbType string, sql string) appcore.SQLInspection
|
|
GetSQLSafetyLevel() ai.SQLPermissionLevel
|
|
}
|
|
|
|
// AppBackend 基于现有 internal/app.App 暴露 MCP 所需数据库能力。
|
|
type AppBackend struct {
|
|
app *appcore.App
|
|
}
|
|
|
|
func NewAppBackend(ctx context.Context) *AppBackend {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
a := appcore.NewApp()
|
|
appcore.InitializeLifecycle(a, ctx)
|
|
return &AppBackend{app: a}
|
|
}
|
|
|
|
func (b *AppBackend) Close(ctx context.Context) error {
|
|
if b == nil || b.app == nil {
|
|
return nil
|
|
}
|
|
b.app.Shutdown()
|
|
return nil
|
|
}
|
|
|
|
func (b *AppBackend) GetSavedConnections() ([]connection.SavedConnectionView, error) {
|
|
return b.app.GetSavedConnections()
|
|
}
|
|
|
|
func (b *AppBackend) GetEditableSavedConnection(id string) (connection.SavedConnectionView, error) {
|
|
return b.app.GetEditableSavedConnection(id)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetDatabases(config connection.ConnectionConfig) connection.QueryResult {
|
|
return b.app.DBGetDatabases(config)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetTables(config connection.ConnectionConfig, dbName string) connection.QueryResult {
|
|
return b.app.DBGetTables(config, dbName)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetAllColumns(config connection.ConnectionConfig, dbName string) connection.QueryResult {
|
|
return b.app.DBGetAllColumns(config, dbName)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetColumns(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBGetColumns(config, dbName, tableName)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetIndexes(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBGetIndexes(config, dbName, tableName)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetForeignKeys(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBGetForeignKeys(config, dbName, tableName)
|
|
}
|
|
|
|
func (b *AppBackend) DBGetTriggers(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBGetTriggers(config, dbName, tableName)
|
|
}
|
|
|
|
func (b *AppBackend) DBShowCreateTable(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBShowCreateTable(config, dbName, tableName)
|
|
}
|
|
|
|
func (b *AppBackend) DBQueryMulti(config connection.ConnectionConfig, dbName string, query string, queryID string) connection.QueryResult {
|
|
return b.app.DBQueryMulti(config, dbName, query, queryID)
|
|
}
|
|
|
|
func (b *AppBackend) InspectSQL(dbType string, sql string) appcore.SQLInspection {
|
|
return appcore.InspectSQL(dbType, sql)
|
|
}
|
|
|
|
func (b *AppBackend) GetSQLSafetyLevel() ai.SQLPermissionLevel {
|
|
inspection, err := aiservice.NewProviderConfigStore(appdata.MustResolveActiveRoot(), nil).Inspect()
|
|
if err != nil {
|
|
logger.Error(err, "加载 MCP SQL 安全控制失败，按只读模式回退")
|
|
return ai.PermissionReadOnly
|
|
}
|
|
|
|
switch inspection.Snapshot.SafetyLevel {
|
|
case ai.PermissionReadOnly, ai.PermissionReadWrite, ai.PermissionFull:
|
|
return inspection.Snapshot.SafetyLevel
|
|
default:
|
|
return ai.PermissionReadOnly
|
|
}
|
|
}
|