Support --mcp flag without value to use config default

- Implement custom mcpFlag type with IsBoolFlag() support
- Allow --mcp (uses config.yaml default) or --mcp=:8080 (custom port)
- Add MCPServerSet field to track if flag was explicitly set
- Update all MCP server checks to use MCPServerSet flag

Co-authored-by: keac <[email protected]>

Author: Copilot

common/config.go

@@ -47,6 +47,7 @@ type ENOptions struct {
 	OutPutType       string // 导出文件类型
 	IsApiMode        bool
 	MCPServer        string // MCP模式运行地址，如 :8080 或 http://localhost:8080
+	MCPServerSet     bool   // MCP flag是否被设置
 	IsPlugins        bool   // 是否作为后置插件查询
 	IsFast           bool // 是否快速查询
 	ENConfig         *ENConfig

common/flag.go

@@ -5,6 +5,38 @@ import (
 	"github.com/wgpsec/ENScan/common/gologger"
 )
 
+// mcpFlag is a custom flag type that supports optional values
+// When used as --mcp (without value), it will be set to "true"
+// When used as --mcp=:8080 or --mcp=http://..., it will be set to that value
+type mcpFlag struct {
+	value   string
+	changed bool
+}
+
+func (m *mcpFlag) String() string {
+	return m.value
+}
+
+func (m *mcpFlag) Set(s string) error {
+	// When IsBoolFlag returns true and flag is used without value,
+	// Go sets it to "true". We treat "true" as "use config default"
+	if s == "true" {
+		m.value = ""  // Empty means use config default
+	} else {
+		m.value = s
+	}
+	m.changed = true
+	return nil
+}
+
+func (m *mcpFlag) IsBoolFlag() bool {
+	return true
+}
+
+func (m *mcpFlag) Get() (string, bool) {
+	return m.value, m.changed
+}
+
 const banner = `
 
 ███████╗███╗   ██╗███████╗ ██████╗ █████╗ ███╗   ██╗
@@ -49,12 +81,19 @@ func Flag(Info *ENOptions) {
 	//其他设定
 	flag.BoolVar(&Info.IsGroup, "is-group", false, "查询关键词为集团")
 	flag.BoolVar(&Info.IsApiMode, "api", false, "API模式运行")
-	flag.StringVar(&Info.MCPServer, "mcp", "", "MCP模式运行，可指定监听地址，如 :8080 或 http://localhost:8080")
+	
+	// MCP flag with optional value support
+	mcpFlagVar := &mcpFlag{}
+	flag.Var(mcpFlagVar, "mcp", "MCP模式运行，可指定监听地址，如 :8080 或 http://localhost:8080，不指定则使用配置文件默认值")
+	
 	flag.BoolVar(&Info.ISKeyPid, "is-pid", false, "批量查询文件是否为公司PID")
 	flag.IntVar(&Info.DelayTime, "delay", 0, "每个请求延迟（S）-1为随机延迟1-5S")
 	flag.StringVar(&Info.Proxy, "proxy", "", "设置代理")
 	flag.IntVar(&Info.TimeOut, "timeout", 1, "每个请求默认1（分钟）超时")
 	flag.BoolVar(&Info.IsNoMerge, "no-merge", false, "开启后查询文件将单独导出")
 	flag.BoolVar(&Info.Version, "v", false, "版本信息")
 	flag.Parse()
+	
+	// Extract MCP flag value after parsing
+	Info.MCPServer, Info.MCPServerSet = mcpFlagVar.Get()
 }

common/parse.go

@@ -54,7 +54,7 @@ func (op *ENOptions) Parse() {
 		gologger.Fatal().Msgf("配置文件当前[V%.1f] 程序需要[V%.1f] 不匹配，请备份配置文件重新运行-v\n", conf.Version, cfgYV)
 	}
 
-	if op.KeyWord == "" && op.CompanyID == "" && op.InputFile == "" && !op.IsApiMode && op.MCPServer == "" {
+	if op.KeyWord == "" && op.CompanyID == "" && op.InputFile == "" && !op.IsApiMode && !op.MCPServerSet {
 		flag.PrintDefaults()
 		os.Exit(0)
 	}

runner/runner.go

@@ -116,7 +116,7 @@ func RunEnumeration(options *common.ENOptions) {
 
 	} else if options.IsApiMode {
 		api(options)
-	} else if options.MCPServer != "" {
+	} else if options.MCPServerSet {
 		McpServer(options)
 	} else {
 		esjob := NewENTaskQueue(1, options)
@@ -330,7 +330,7 @@ func (q *ESJob) StartENWorkers() {
 					gologger.Error().Msgf("保存缓存失败: %v", err)
 				}
 			case <-quitSig:
-				if !q.op.IsApiMode && q.op.MCPServer == "" {
+				if !q.op.IsApiMode && !q.op.MCPServerSet {
 					gologger.Error().Msgf("任务未完成退出，将自动保存过程文件！")
 					_ = q.OutFileByEnInfo("未完成任务结果")
 				}