mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-06-16 11:39:38 +08:00
5b843ee25b
- 新增 GoNavi MCP stdio server 与 Claude/Codex 用户级安装入口 - 增加安装状态检测、刷新复制能力和浏览器联调 mock - 外部 execute_sql 对齐 GoNavi safetyLevel 并补充前端/后端验证
99 lines
3.3 KiB
Go
99 lines
3.3 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
|
|
DBGetColumns(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
|
|
}
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
b.app.Shutdown(ctx)
|
|
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) DBGetColumns(config connection.ConnectionConfig, dbName string, tableName string) connection.QueryResult {
|
|
return b.app.DBGetColumns(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
|
|
}
|
|
}
|