46 lines
1.5 KiB
Go
46 lines
1.5 KiB
Go
// Package config 解析 aio-mcp 的启动配置（命令行参数）。
|
||
package config
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"io"
|
||
)
|
||
|
||
// 支持的传输方式。
|
||
const (
|
||
TransportStdio = "stdio"
|
||
TransportHTTP = "http"
|
||
)
|
||
|
||
// Config 是 aio-mcp 的运行时配置。
|
||
type Config struct {
|
||
Transport string // MCP 传输方式：stdio 或 http
|
||
Addr string // http 模式下的监听地址
|
||
LogLevel string // 日志级别：debug / info / warn / error
|
||
DataDir string // 数据目录：SQLite 库文件与工具产物的存放位置
|
||
}
|
||
|
||
// Parse 解析命令行参数。args 不含程序名（通常传 os.Args[1:]），
|
||
// 用法与错误信息写到 out。
|
||
func Parse(args []string, out io.Writer) (*Config, error) {
|
||
cfg := &Config{}
|
||
|
||
fs := flag.NewFlagSet("aio-mcp", flag.ContinueOnError)
|
||
fs.SetOutput(out)
|
||
fs.StringVar(&cfg.Transport, "transport", TransportHTTP, "传输方式：http（默认，streamable HTTP）或 stdio（本地 Agent 拉起）")
|
||
fs.StringVar(&cfg.Addr, "addr", "localhost:8000", "http 模式下的监听地址")
|
||
fs.StringVar(&cfg.LogLevel, "log-level", "info", "日志级别：debug / info / warn / error")
|
||
fs.StringVar(&cfg.DataDir, "data-dir", "data", "数据目录：SQLite 库文件与工具产物的存放位置")
|
||
|
||
if err := fs.Parse(args); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
if cfg.Transport != TransportStdio && cfg.Transport != TransportHTTP {
|
||
return nil, fmt.Errorf("未知 transport: %q（支持: %s, %s）", cfg.Transport, TransportStdio, TransportHTTP)
|
||
}
|
||
|
||
return cfg, nil
|
||
}
|