Skip to content

Commit 419f769

Browse files
committed
Refactor security settings application in mcpproxy to only apply changes if explicitly set via command line flags. Update logging level for removed server menu items from Info to Debug in tray manager.
1 parent 7afd3c7 commit 419f769

File tree

2 files changed

+19
-7
lines changed

2 files changed

+19
-7
lines changed

cmd/mcpproxy/main.go

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -252,19 +252,31 @@ func runServer(cmd *cobra.Command, _ []string) error {
252252
cfg.ToolResponseLimit = cmdToolResponseLimit
253253
}
254254

255-
// Apply security settings from command line
256-
cfg.ReadOnlyMode = cmdReadOnlyMode
257-
cfg.DisableManagement = cmdDisableManagement
258-
cfg.AllowServerAdd = cmdAllowServerAdd
259-
cfg.AllowServerRemove = cmdAllowServerRemove
260-
cfg.EnablePrompts = cmdEnablePrompts
255+
// Apply security settings from command line ONLY if explicitly set
256+
if cmd.Flags().Changed("read-only") {
257+
cfg.ReadOnlyMode = cmdReadOnlyMode
258+
}
259+
if cmd.Flags().Changed("disable-management") {
260+
cfg.DisableManagement = cmdDisableManagement
261+
}
262+
if cmd.Flags().Changed("allow-server-add") {
263+
cfg.AllowServerAdd = cmdAllowServerAdd
264+
}
265+
if cmd.Flags().Changed("allow-server-remove") {
266+
cfg.AllowServerRemove = cmdAllowServerRemove
267+
}
268+
if cmd.Flags().Changed("enable-prompts") {
269+
cfg.EnablePrompts = cmdEnablePrompts
270+
}
261271

262272
logger.Info("Configuration loaded",
263273
zap.String("data_dir", cfg.DataDir),
264274
zap.Int("servers_count", len(cfg.Servers)),
265275
zap.Bool("tray_enabled", cfg.EnableTray),
266276
zap.Bool("read_only_mode", cfg.ReadOnlyMode),
267277
zap.Bool("disable_management", cfg.DisableManagement),
278+
zap.Bool("allow_server_add", cfg.AllowServerAdd),
279+
zap.Bool("allow_server_remove", cfg.AllowServerRemove),
268280
zap.Bool("enable_prompts", cfg.EnablePrompts))
269281

270282
// Create server

internal/tray/managers.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ func (m *MenuManager) UpdateUpstreamServersMenu(servers []map[string]interface{}
347347
// Hide servers that are no longer in the config
348348
for serverName, menuItem := range m.serverMenuItems {
349349
if _, exists := currentServerMap[serverName]; !exists {
350-
m.logger.Info("Hiding menu item for removed server", zap.String("server", serverName))
350+
m.logger.Debug("Hiding menu item for removed server", zap.String("server", serverName))
351351
menuItem.Hide()
352352
// Also hide its sub-menu items if they exist
353353
if actionItem, ok := m.serverActionItems[serverName]; ok {

0 commit comments

Comments
 (0)