Merge branch 'dev' into feature/20260602_connection_driver_i18n

This commit is contained in:
tianqijiuyun-latiao
2026-06-23 23:34:16 +08:00
39 changed files with 3045 additions and 1840 deletions

View File

@@ -8,6 +8,15 @@ import (
"GoNavi-Wails/internal/connection"
)
type connectionProtectionKey string
const (
connectionProtectionDataEdit connectionProtectionKey = "restrictDataEdit"
connectionProtectionStructureEdit connectionProtectionKey = "restrictStructureEdit"
connectionProtectionScriptExecution connectionProtectionKey = "restrictScriptExecution"
connectionProtectionDataImport connectionProtectionKey = "restrictDataImport"
)
var connectionReadOnlySupportedTypes = map[string]struct{}{
"clickhouse": {},
"dameng": {},
@@ -88,40 +97,41 @@ var mongoMetaCommandKeys = map[string]struct{}{
}
var readOnlyConnectionActionTextKeys = map[string]string{
"创建数据库": "connection.backend.action.create_database",
"connection.backend.action.create_database": "connection.backend.action.create_database",
"创建模式": "connection.backend.action.create_schema",
"connection.backend.action.create_schema": "connection.backend.action.create_schema",
"重命名模式": "connection.backend.action.rename_schema",
"connection.backend.action.rename_schema": "connection.backend.action.rename_schema",
"删除模式": "connection.backend.action.drop_schema",
"connection.backend.action.drop_schema": "connection.backend.action.drop_schema",
"重命名数据库": "connection.backend.action.rename_database",
"connection.backend.action.rename_database": "connection.backend.action.rename_database",
"删除数据库": "connection.backend.action.drop_database",
"connection.backend.action.drop_database": "connection.backend.action.drop_database",
"重命名表": "connection.backend.action.rename_table",
"connection.backend.action.rename_table": "connection.backend.action.rename_table",
"删除表": "connection.backend.action.drop_table",
"connection.backend.action.drop_table": "connection.backend.action.drop_table",
"删除视图": "connection.backend.action.drop_view",
"connection.backend.action.drop_view": "connection.backend.action.drop_view",
"删除函数或存储过程": "connection.backend.action.drop_function_or_procedure",
"创建数据库": "connection.backend.action.create_database",
"connection.backend.action.create_database": "connection.backend.action.create_database",
"创建模式": "connection.backend.action.create_schema",
"connection.backend.action.create_schema": "connection.backend.action.create_schema",
"重命名模式": "connection.backend.action.rename_schema",
"connection.backend.action.rename_schema": "connection.backend.action.rename_schema",
"删除模式": "connection.backend.action.drop_schema",
"connection.backend.action.drop_schema": "connection.backend.action.drop_schema",
"重命名数据库": "connection.backend.action.rename_database",
"connection.backend.action.rename_database": "connection.backend.action.rename_database",
"删除数据库": "connection.backend.action.drop_database",
"connection.backend.action.drop_database": "connection.backend.action.drop_database",
"重命名表": "connection.backend.action.rename_table",
"connection.backend.action.rename_table": "connection.backend.action.rename_table",
"删除表": "connection.backend.action.drop_table",
"connection.backend.action.drop_table": "connection.backend.action.drop_table",
"删除视图": "connection.backend.action.drop_view",
"connection.backend.action.drop_view": "connection.backend.action.drop_view",
"删除函数或存储过程": "connection.backend.action.drop_function_or_procedure",
"connection.backend.action.drop_function_or_procedure": "connection.backend.action.drop_function_or_procedure",
"重命名视图": "connection.backend.action.rename_view",
"connection.backend.action.rename_view": "connection.backend.action.rename_view",
"导入数据": "connection.backend.action.import_data",
"connection.backend.action.import_data": "connection.backend.action.import_data",
"提交结果修改": "connection.backend.action.apply_result_changes",
"connection.backend.action.apply_result_changes": "connection.backend.action.apply_result_changes",
"预览结果修改": "connection.backend.action.preview_result_changes",
"connection.backend.action.preview_result_changes": "connection.backend.action.preview_result_changes",
"clear_table": "connection.backend.action.clear_table",
"connection.backend.action.clear_table": "connection.backend.action.clear_table",
"truncate_table": "connection.backend.action.truncate_table",
"connection.backend.action.truncate_table": "connection.backend.action.truncate_table",
"数据同步写入": "connection.backend.action.data_sync_write",
"connection.backend.action.data_sync_write": "connection.backend.action.data_sync_write",
"重命名视图": "connection.backend.action.rename_view",
"connection.backend.action.rename_view": "connection.backend.action.rename_view",
"导入数据": "connection.backend.action.import_data",
"connection.backend.action.import_data": "connection.backend.action.import_data",
"提交结果修改": "connection.backend.action.apply_result_changes",
"connection.backend.action.apply_result_changes": "connection.backend.action.apply_result_changes",
"预览结果修改": "connection.backend.action.preview_result_changes",
"connection.backend.action.preview_result_changes": "connection.backend.action.preview_result_changes",
"clear_table": "connection.backend.action.clear_table",
"connection.backend.action.clear_table": "connection.backend.action.clear_table",
"truncate_table": "connection.backend.action.truncate_table",
"connection.backend.action.truncate_table": "connection.backend.action.truncate_table",
"connection.backend.action.data_sync_structure": "connection.backend.action.data_sync_structure",
"数据同步写入": "connection.backend.action.data_sync_write",
"connection.backend.action.data_sync_write": "connection.backend.action.data_sync_write",
}
func supportsConnectionReadOnlyMode(config connection.ConnectionConfig) bool {
@@ -129,8 +139,57 @@ func supportsConnectionReadOnlyMode(config connection.ConnectionConfig) bool {
return ok
}
func hasAnyConnectionProtection(config connection.ConnectionProtectionConfig) bool {
return config.RestrictDataEdit ||
config.RestrictStructureEdit ||
config.RestrictScriptExecution ||
config.RestrictDataImport
}
func resolveConnectionProtectionConfig(config connection.ConnectionConfig) connection.ConnectionProtectionConfig {
if !supportsConnectionReadOnlyMode(config) {
return connection.ConnectionProtectionConfig{}
}
if hasAnyConnectionProtection(config.Protection) {
return config.Protection
}
if config.ReadOnly {
return connection.ConnectionProtectionConfig{
RestrictDataEdit: true,
RestrictStructureEdit: true,
RestrictScriptExecution: true,
RestrictDataImport: true,
}
}
return connection.ConnectionProtectionConfig{}
}
func isConnectionProtectionEnabled(config connection.ConnectionConfig, key connectionProtectionKey) bool {
protection := resolveConnectionProtectionConfig(config)
switch key {
case connectionProtectionDataEdit:
return protection.RestrictDataEdit
case connectionProtectionStructureEdit:
return protection.RestrictStructureEdit
case connectionProtectionScriptExecution:
return protection.RestrictScriptExecution
case connectionProtectionDataImport:
return protection.RestrictDataImport
default:
return false
}
}
func isConnectionForcedReadOnly(config connection.ConnectionConfig) bool {
return config.ReadOnly && supportsConnectionReadOnlyMode(config)
protection := resolveConnectionProtectionConfig(config)
return protection.RestrictDataEdit &&
protection.RestrictStructureEdit &&
protection.RestrictScriptExecution &&
protection.RestrictDataImport
}
func isConnectionScriptExecutionRestricted(config connection.ConnectionConfig) bool {
return isConnectionProtectionEnabled(config, connectionProtectionScriptExecution)
}
func normalizeReadOnlyConnectionText(text func(string, map[string]any) string) func(string, map[string]any) string {
@@ -174,9 +233,9 @@ func readOnlyConnectionActionBlockedMessage(action string) string {
return readOnlyConnectionActionBlockedMessageWithText(action, defaultAppText)
}
func ensureReadOnlyConnectionAllowsQueryWithText(config connection.ConnectionConfig, query string, text func(string, map[string]any) string) error {
func ensureConnectionAllowsQueryWithText(config connection.ConnectionConfig, query string, text func(string, map[string]any) string) error {
text = normalizeReadOnlyConnectionText(text)
if !isConnectionForcedReadOnly(config) {
if !isConnectionScriptExecutionRestricted(config) {
return nil
}
for _, statement := range splitSQLStatements(query) {
@@ -187,20 +246,32 @@ func ensureReadOnlyConnectionAllowsQueryWithText(config connection.ConnectionCon
return nil
}
func ensureReadOnlyConnectionAllowsQuery(config connection.ConnectionConfig, query string) error {
return ensureReadOnlyConnectionAllowsQueryWithText(config, query, defaultAppText)
func ensureConnectionAllowsQuery(config connection.ConnectionConfig, query string) error {
return ensureConnectionAllowsQueryWithText(config, query, defaultAppText)
}
func ensureReadOnlyConnectionAllowsActionWithText(config connection.ConnectionConfig, action string, text func(string, map[string]any) string) error {
func ensureConnectionAllowsActionWithText(config connection.ConnectionConfig, key connectionProtectionKey, action string, text func(string, map[string]any) string) error {
text = normalizeReadOnlyConnectionText(text)
if !isConnectionForcedReadOnly(config) {
if !isConnectionProtectionEnabled(config, key) {
return nil
}
return errors.New(readOnlyConnectionActionBlockedMessageWithText(action, text))
}
func ensureReadOnlyConnectionAllowsAction(config connection.ConnectionConfig, action string) error {
return ensureReadOnlyConnectionAllowsActionWithText(config, action, defaultAppText)
func ensureConnectionAllowsAction(config connection.ConnectionConfig, key connectionProtectionKey, action string) error {
return ensureConnectionAllowsActionWithText(config, key, action, defaultAppText)
}
func ensureConnectionAllowsDataEdit(config connection.ConnectionConfig, action string) error {
return ensureConnectionAllowsAction(config, connectionProtectionDataEdit, action)
}
func ensureConnectionAllowsStructureEdit(config connection.ConnectionConfig, action string) error {
return ensureConnectionAllowsAction(config, connectionProtectionStructureEdit, action)
}
func ensureConnectionAllowsDataImport(config connection.ConnectionConfig, action string) error {
return ensureConnectionAllowsAction(config, connectionProtectionDataImport, action)
}
func isReadOnlyMongoCommand(query string) bool {

View File

@@ -32,6 +32,7 @@ func TestConnectionReadOnlyMessagesUseLocalizedText(t *testing.T) {
"connection.backend.error.readonly_action_blocked",
"connection.backend.action.create_database",
"connection.backend.action.import_data",
"connection.backend.action.data_sync_structure",
"connection.backend.action.data_sync_write",
"connection.backend.action.clear_table",
"connection.backend.action.truncate_table",
@@ -67,6 +68,7 @@ func TestConnectionReadOnlyCatalogKeysExist(t *testing.T) {
"connection.backend.action.preview_result_changes",
"connection.backend.action.clear_table",
"connection.backend.action.truncate_table",
"connection.backend.action.data_sync_structure",
"connection.backend.action.data_sync_write",
}

View File

@@ -22,18 +22,18 @@ func TestSupportsConnectionReadOnlyMode(t *testing.T) {
func TestEnsureReadOnlyConnectionAllowsQuery(t *testing.T) {
sqlConfig := connection.ConnectionConfig{Type: "postgres", ReadOnly: true}
if err := ensureReadOnlyConnectionAllowsQuery(sqlConfig, "SELECT * FROM users"); err != nil {
if err := ensureConnectionAllowsQuery(sqlConfig, "SELECT * FROM users"); err != nil {
t.Fatalf("read-only postgres connection should allow select: %v", err)
}
if err := ensureReadOnlyConnectionAllowsQuery(sqlConfig, "UPDATE users SET name = 'next'"); err == nil {
if err := ensureConnectionAllowsQuery(sqlConfig, "UPDATE users SET name = 'next'"); err == nil {
t.Fatal("read-only postgres connection should block update")
}
mongoConfig := connection.ConnectionConfig{Type: "mongodb", ReadOnly: true}
if err := ensureReadOnlyConnectionAllowsQuery(mongoConfig, `{"find":"users","filter":{"active":true}}`); err != nil {
if err := ensureConnectionAllowsQuery(mongoConfig, `{"find":"users","filter":{"active":true}}`); err != nil {
t.Fatalf("read-only mongodb connection should allow find: %v", err)
}
if err := ensureReadOnlyConnectionAllowsQuery(mongoConfig, `{"delete":"users","deletes":[{"q":{"active":false},"limit":0}]}`); err == nil {
if err := ensureConnectionAllowsQuery(mongoConfig, `{"delete":"users","deletes":[{"q":{"active":false},"limit":0}]}`); err == nil {
t.Fatal("read-only mongodb connection should block delete")
}
}
@@ -45,7 +45,7 @@ func TestEnsureReadOnlyConnectionAllowsAction(t *testing.T) {
})
config := connection.ConnectionConfig{Type: "postgres", ReadOnly: true}
err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_database")
err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_database")
if err == nil {
t.Fatal("read-only connection should block mutating actions")
}
@@ -53,3 +53,27 @@ func TestEnsureReadOnlyConnectionAllowsAction(t *testing.T) {
t.Fatalf("blocked action message should include action label, got %q", err.Error())
}
}
func TestEnsureConnectionProtectionSeparatesActionCategories(t *testing.T) {
config := connection.ConnectionConfig{
Type: "postgres",
Protection: connection.ConnectionProtectionConfig{
RestrictDataEdit: true,
RestrictDataImport: true,
RestrictStructureEdit: false,
},
}
if err := ensureConnectionAllowsQuery(config, "UPDATE users SET name = 'next'"); err != nil {
t.Fatalf("script execution should remain allowed when only data-edit/import restrictions are enabled: %v", err)
}
if err := ensureConnectionAllowsDataEdit(config, "connection.backend.action.apply_result_changes"); err == nil {
t.Fatal("data edit restriction should block result changes")
}
if err := ensureConnectionAllowsDataImport(config, "connection.backend.action.import_data"); err == nil {
t.Fatal("data import restriction should block imports")
}
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_database"); err != nil {
t.Fatalf("structure edits should remain allowed when structure restriction is disabled: %v", err)
}
}

View File

@@ -195,7 +195,7 @@ func (a *App) CreateDatabase(config connection.ConnectionConfig, dbName string)
if dbName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.database_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.create_database"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.create_database"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
@@ -368,7 +368,7 @@ func resolveSchemaDDLTargetDatabaseWithText(config connection.ConnectionConfig,
}
func (a *App) CreateSchema(config connection.ConnectionConfig, dbName string, schemaName string) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.create_schema"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.create_schema"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
dbType := resolveDDLDBType(config)
@@ -396,7 +396,7 @@ func (a *App) CreateSchema(config connection.ConnectionConfig, dbName string, sc
}
func (a *App) RenameSchema(config connection.ConnectionConfig, dbName string, oldSchemaName string, newSchemaName string) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.rename_schema"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.rename_schema"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
dbType := resolveDDLDBType(config)
@@ -422,7 +422,7 @@ func (a *App) RenameSchema(config connection.ConnectionConfig, dbName string, ol
}
func (a *App) DropSchema(config connection.ConnectionConfig, dbName string, schemaName string) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_schema"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_schema"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
dbType := resolveDDLDBType(config)
@@ -680,7 +680,7 @@ func (a *App) RenameDatabase(config connection.ConnectionConfig, oldName string,
if oldName == "" || newName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.database_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.rename_database"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.rename_database"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
if strings.EqualFold(oldName, newName) {
@@ -726,7 +726,7 @@ func (a *App) DropDatabase(config connection.ConnectionConfig, dbName string) co
if dbName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.database_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_database"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_database"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
@@ -763,7 +763,7 @@ func (a *App) RenameTable(config connection.ConnectionConfig, dbName string, old
if oldTableName == "" || newTableName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.table_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.rename_table"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.rename_table"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
if strings.EqualFold(oldTableName, newTableName) {
@@ -818,7 +818,7 @@ func (a *App) DropTable(config connection.ConnectionConfig, dbName string, table
if tableName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.table_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_table"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_table"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
@@ -885,7 +885,7 @@ func (a *App) DBQueryWithCancel(config connection.ConnectionConfig, dbName strin
}
query = sanitizeSQLForPgLike(resolveDDLDBType(config), query)
if err := ensureReadOnlyConnectionAllowsQuery(config, query); err != nil {
if err := ensureConnectionAllowsQuery(config, query); err != nil {
return connection.QueryResult{Success: false, Message: err.Error(), QueryID: queryID}
}
@@ -1021,7 +1021,7 @@ func (a *App) DBQueryMulti(config connection.ConnectionConfig, dbName string, qu
}
query = sanitizeSQLForPgLike(resolveDDLDBType(config), query)
if err := ensureReadOnlyConnectionAllowsQuery(config, query); err != nil {
if err := ensureConnectionAllowsQuery(config, query); err != nil {
return connection.QueryResult{Success: false, Message: err.Error(), QueryID: queryID}
}
@@ -1429,7 +1429,7 @@ func (a *App) DBQueryIsolated(config connection.ConnectionConfig, dbName string,
runConfig := normalizeRunConfig(config, dbName)
query = sanitizeSQLForPgLike(resolveDDLDBType(config), query)
if err := ensureReadOnlyConnectionAllowsQuery(config, query); err != nil {
if err := ensureConnectionAllowsQuery(config, query); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
@@ -2318,7 +2318,7 @@ func (a *App) DropView(config connection.ConnectionConfig, dbName string, viewNa
if viewName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.view_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_view"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_view"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
@@ -2353,7 +2353,7 @@ func (a *App) DropFunction(config connection.ConnectionConfig, dbName string, ro
if routineName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.routine_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.drop_function_or_procedure"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.drop_function_or_procedure"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
if routineType != "FUNCTION" && routineType != "PROCEDURE" {
@@ -2398,7 +2398,7 @@ func (a *App) RenameView(config connection.ConnectionConfig, dbName string, oldN
if oldName == "" || newName == "" {
return connection.QueryResult{Success: false, Message: a.appText("db.backend.error.view_name_required", nil)}
}
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.rename_view"); err != nil {
if err := ensureConnectionAllowsStructureEdit(config, "connection.backend.action.rename_view"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
if strings.EqualFold(oldName, newName) {

View File

@@ -45,7 +45,7 @@ func (a *App) DBQueryMultiTransactional(config connection.ConnectionConfig, dbNa
}
query = sanitizeSQLForPgLike(transactionDBType, query)
if err := ensureReadOnlyConnectionAllowsQuery(config, query); err != nil {
if err := ensureConnectionAllowsQuery(config, query); err != nil {
return connection.QueryResult{Success: false, Message: err.Error(), QueryID: queryID}
}
if !shouldUseManagedSQLTransaction(transactionDBType, query) {

View File

@@ -1938,7 +1938,7 @@ func (a *App) PreviewImportFile(filePath string) connection.QueryResult {
}
func (a *App) ImportData(config connection.ConnectionConfig, dbName, tableName string) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.import_data"); err != nil {
if err := ensureConnectionAllowsDataImport(config, "connection.backend.action.import_data"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
selection, err := runtime.OpenFileDialog(a.ctx, runtime.OpenDialogOptions{
@@ -2284,7 +2284,7 @@ func formatImportSQLValue(dbType, columnType string, value interface{}) string {
// ImportDataWithProgress 执行导入并发送进度事件
func (a *App) ImportDataWithProgress(config connection.ConnectionConfig, dbName, tableName, filePath string) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.import_data"); err != nil {
if err := ensureConnectionAllowsDataImport(config, "connection.backend.action.import_data"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
runConfig := normalizeRunConfig(config, dbName)
@@ -2338,7 +2338,7 @@ func (a *App) ImportDataWithProgress(config connection.ConnectionConfig, dbName,
}
func (a *App) ApplyChanges(config connection.ConnectionConfig, dbName, tableName string, changes connection.ChangeSet) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.apply_result_changes"); err != nil {
if err := ensureConnectionAllowsDataEdit(config, "connection.backend.action.apply_result_changes"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
runConfig := normalizeRunConfig(config, dbName)
@@ -2367,7 +2367,7 @@ type ChangePreview struct {
}
func (a *App) PreviewChanges(config connection.ConnectionConfig, dbName, tableName string, changes connection.ChangeSet) connection.QueryResult {
if err := ensureReadOnlyConnectionAllowsAction(config, "connection.backend.action.preview_result_changes"); err != nil {
if err := ensureConnectionAllowsDataEdit(config, "connection.backend.action.preview_result_changes"); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
runConfig := normalizeRunConfig(config, dbName)
@@ -2903,7 +2903,7 @@ func tableDataClearMessageKeys(mode tableDataClearMode, partial bool) (failureKe
func (a *App) runTableDataClear(config connection.ConnectionConfig, dbName string, tableNames []string, mode tableDataClearMode) connection.QueryResult {
actionLabel, progressLabel := tableDataClearActionLabels(mode)
if err := ensureReadOnlyConnectionAllowsAction(config, actionLabel); err != nil {
if err := ensureConnectionAllowsDataEdit(config, actionLabel); err != nil {
return connection.QueryResult{Success: false, Message: err.Error()}
}
runConfig := normalizeRunConfig(config, dbName)

View File

@@ -11,6 +11,29 @@ import (
"github.com/wailsapp/wails/v2/pkg/runtime"
)
func ensureDataSyncTargetProtection(config sync.SyncConfig) error {
content := strings.ToLower(strings.TrimSpace(config.Content))
strategy := strings.ToLower(strings.TrimSpace(config.TargetTableStrategy))
touchesStructure := content == "schema" ||
content == "both" ||
config.AutoAddColumns ||
config.CreateIndexes ||
(strategy != "" && strategy != "existing_only")
touchesData := content == "" || content == "data" || content == "both"
if touchesStructure {
if err := ensureConnectionAllowsStructureEdit(config.TargetConfig, "connection.backend.action.data_sync_structure"); err != nil {
return err
}
}
if touchesData {
if err := ensureConnectionAllowsDataImport(config.TargetConfig, "connection.backend.action.data_sync_write"); err != nil {
return err
}
}
return nil
}
func (a *App) resolveDataSyncConfigSecrets(config sync.SyncConfig) (sync.SyncConfig, error) {
resolved := config
sourceConfig, sourceDatabase, err := a.resolveDataSyncEndpointConfig(config.SourceConfig, config.SourceDatabase)
@@ -60,7 +83,7 @@ func (a *App) resolveDataSyncEndpointConfig(raw connection.ConnectionConfig, sel
// DataSync executes a data synchronization task
func (a *App) DataSync(config sync.SyncConfig) sync.SyncResult {
if err := ensureReadOnlyConnectionAllowsAction(config.TargetConfig, "connection.backend.action.data_sync_write"); err != nil {
if err := ensureDataSyncTargetProtection(config); err != nil {
return sync.SyncResult{
Success: false,
Message: err.Error(),