openclaw/Xiaji-go
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

fix(cf): dnsproxy name support and dnsadd on/off

Browse Source
This commit is contained in:
openclaw
2026-03-19 19:56:27 +08:00
parent 36f11fa846
commit 73a829a4e9
9 changed files with 636 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
internal/core/ops/service.go Normal file
View File

@@ -0,0 +1,100 @@
package ops
import (
"fmt"
"strings"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"ops-assistant/internal/core/command"
coremodule "ops-assistant/internal/core/module"
"ops-assistant/internal/core/registry"
)
type Service struct {
dbPath string
baseDir string
registry *registry.Registry
db *gorm.DB
}
func NewService(dbPath, baseDir string, reg *registry.Registry) *Service {
db, _ := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
return &Service{dbPath: dbPath, baseDir: baseDir, registry: reg, db: db}
}
func (s *Service) Handle(userID int64, text string) (bool, string) {
if !strings.HasPrefix(strings.TrimSpace(text), "/") {
return false, ""
}
cmd, _, err := command.ParseWithInputs(text)
if err != nil {
return false, ""
}
// 通用帮助
if cmd.Module == "help" || cmd.Name == "/help" || cmd.Name == "/start" {
return true, s.helpText()
}
if cmd.Module == "ops" && (len(cmd.Args) == 0 || cmd.Args[0] == "help") {
return true, s.helpText()
}
if cmd.Module == "ops" && len(cmd.Args) > 0 && cmd.Args[0] == "modules" {
return true, s.modulesStatusText()
}
if cmd.Module != "" && cmd.Module != "ops" && s.db != nil {
if !coremodule.IsEnabled(s.db, cmd.Module) {
return true, fmt.Sprintf("[ERR_FEATURE_DISABLED] 模块未启用: %s（开关: enable_module_%s）", cmd.Module, cmd.Module)
}
}
out, handled, err := s.registry.Handle(userID, cmd)
if !handled {
return false, ""
}
if err != nil {
return true, "❌ OPS 执行失败: " + err.Error()
}
return true, out
}
func (s *Service) helpText() string {
lines := []string{
"🛠️ OPS 交互命令：",
"- /ops modules （查看模块启用状态）",
"- /cpa help",
"- /cpa status",
"- /cpa usage backup",
"- /cpa usage restore <backup_id> [--confirm YES_RESTORE] [--dry-run]",
"- /cf status （需要 enable_module_cf）",
"- /cf zones （需要 enable_module_cf）",
"- /cf dns list <zone_id> （需要 enable_module_cf）",
"- /cf dns update <zone_id> <record_id> <type> <name> <content> [ttl] [proxied:true|false] （需要 enable_module_cf）",
"- /cf dnsadd <name> <content> [on|off] [type] （需要 enable_module_cf）",
"- /cf dnsset <record_id> <content> [true] （需要 enable_module_cf）",
"- /cf dnsdel <record_id> YES （需要 enable_module_cf）",
"- /cf dnsproxy <record_id|name> on|off （需要 enable_module_cf）",
"- /cf workers list （需要 enable_module_cf）",
"- /mail status （需要 enable_module_mail）",
}
return strings.Join(lines, "\n")
}
func (s *Service) modulesStatusText() string {
mods := s.registry.ListModules()
if len(mods) == 0 {
return "暂无已注册模块"
}
lines := []string{"🧩 模块状态："}
for _, m := range mods {
enabled := false
if s.db != nil {
enabled = coremodule.IsEnabled(s.db, m)
}
state := "disabled"
if enabled {
state = "enabled"
}
lines = append(lines, fmt.Sprintf("- %s: %s", m, state))
}
lines = append(lines, "\n可用命令：/ops modules")
return strings.Join(lines, "\n")
}

254
internal/module/cf/commands.go Normal file
View File

@@ -0,0 +1,254 @@
package cf
import (
"fmt"
"strings"
"ops-assistant/internal/core/ecode"
coremodule "ops-assistant/internal/core/module"
)
func commandSpecs() []coremodule.CommandSpec {
return []coremodule.CommandSpec{
{
Prefixes: []string{"/cf status"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_status",
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf status（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf status 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf status 执行失败: ",
},
{
Prefixes: []string{"/cf zones"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_zones",
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf zones（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf zones 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf zones 执行失败: ",
},
{
Prefixes: []string{"/cf dns list"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_list",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dns list <zone_id>"))
}
return map[string]string{"zone_id": parts[3]}, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf dns list <zone_id>（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dns list 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dns list 执行失败: ",
ErrHint: "/cf dns list <zone_id>",
},
{
Prefixes: []string{"/cf dns update"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_update",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 8 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dns update <zone_id> <record_id> <type> <name> <content> [ttl] [proxied]"))
}
inputs := map[string]string{
"zone_id": parts[3],
"record_id": parts[4],
"type": parts[5],
"name": parts[6],
"content": parts[7],
}
if len(parts) >= 9 {
inputs["ttl"] = parts[8]
}
if len(parts) >= 10 {
inputs["proxied"] = parts[9]
}
return inputs, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf dns update（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dns update 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dns update 执行失败: ",
ErrHint: "/cf dns update <zone_id> <record_id> <type> <name> <content> [ttl] [proxied:true|false]",
},
{
Prefixes: []string{"/cf dnsadd"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_add",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dnsadd <name> <content> [on|off] [type]"))
}
name := parts[2]
content := parts[3]
proxied := "false"
recType := "A"
if len(parts) >= 5 {
switch strings.ToLower(parts[4]) {
case "on":
proxied = "true"
if len(parts) >= 6 {
recType = parts[5]
}
case "off":
proxied = "false"
if len(parts) >= 6 {
recType = parts[5]
}
case "true":
proxied = "true"
if len(parts) >= 6 {
recType = parts[5]
}
case "false":
proxied = "false"
if len(parts) >= 6 {
recType = parts[5]
}
default:
// treat as type when no on/off provided
recType = parts[4]
}
}
inputs := map[string]string{
"name": name,
"content": content,
"type": strings.ToUpper(recType),
"proxied": strings.ToLower(proxied),
}
return inputs, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf dnsadd（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dnsadd 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dnsadd 执行失败: ",
ErrHint: "/cf dnsadd <name> <content> [on|off] [type]",
},
{
Prefixes: []string{"/cf dnsset"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_set",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dnsset <record_id> <content> [true]"))
}
proxied := "false"
if len(parts) >= 5 && strings.EqualFold(parts[4], "true") {
proxied = "true"
}
return map[string]string{
"record_id": parts[2],
"content": parts[3],
"proxied": strings.ToLower(proxied),
}, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf dnsset（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dnsset 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dnsset 执行失败: ",
ErrHint: "/cf dnsset <record_id> <content> [true]",
},
{
Prefixes: []string{"/cf dnsdel"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_del",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dnsdel <record_id> YES"))
}
if len(parts) < 4 || !strings.EqualFold(parts[3], "YES") {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "缺少确认词 YES，示例：/cf dnsdel <record_id> YES"))
}
return map[string]string{
"record_id": parts[2],
}, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: false,
},
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dnsdel 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dnsdel 执行失败: ",
ErrHint: "/cf dnsdel <record_id> YES",
},
{
Prefixes: []string{"/cf dnsproxy"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_dns_proxy",
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数不足，示例：/cf dnsproxy <record_id|name> on|off"))
}
mode := strings.ToLower(parts[3])
if mode != "on" && mode != "off" {
return nil, fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "参数无效，示例：/cf dnsproxy <record_id|name> on|off"))
}
proxied := "false"
if mode == "on" {
proxied = "true"
}
inputs := map[string]string{
"proxied": proxied,
"record_id": "__empty__",
"name": "__empty__",
}
target := parts[2]
if strings.Contains(target, ".") {
inputs["name"] = target
} else {
inputs["record_id"] = target
}
return inputs, nil
},
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf dnsproxy（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf dnsproxy 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf dnsproxy 执行失败: ",
ErrHint: "/cf dnsproxy <record_id|name> on|off",
},
{
Prefixes: []string{"/cf workers list"},
Template: coremodule.CommandTemplate{
RunbookName: "cf_workers_list",
Gate: coremodule.Gate{
NeedFlag: "enable_module_cf",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /cf workers list（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cf workers list 已执行，job=%d", jobID) },
},
ErrPrefix: "/cf workers list 执行失败: ",
},
}
}

40
internal/module/cf/module.go Normal file
View File

@@ -0,0 +1,40 @@
package cf
import (
"strings"
"ops-assistant/internal/core/command"
"ops-assistant/internal/core/ecode"
coremodule "ops-assistant/internal/core/module"
"ops-assistant/internal/core/runbook"
"gorm.io/gorm"
)
type Module struct {
db *gorm.DB
exec *runbook.Executor
runner *coremodule.Runner
}
func New(db *gorm.DB, exec *runbook.Executor) *Module {
return &Module{db: db, exec: exec, runner: coremodule.NewRunner(db, exec)}
}
func (m *Module) Handle(userID int64, cmd *command.ParsedCommand) (string, error) {
text := strings.TrimSpace(cmd.Raw)
if text == "/cf" || strings.HasPrefix(text, "/cf help") {
return "CF 模块\n- /cf status [--dry-run]\n- /cf zones\n- /cf dns list <zone_id>\n- /cf dns update <zone_id> <record_id> <type> <name> <content> [ttl] [proxied:true|false]\n- /cf dnsadd <name> <content> [on|off] [type]\n- /cf dnsset <record_id> <content> [true]\n- /cf dnsdel <record_id> YES\n- /cf dnsproxy <record_id|name> on|off\n- /cf workers list", nil
}
specs := commandSpecs()
if sp, ok := coremodule.MatchCommand(text, specs); ok {
jobID, out, err := coremodule.ExecTemplate(m.runner, userID, cmd.Raw, sp.Template)
if err != nil {
return ecode.Tag(ecode.ErrStepFailed, coremodule.FormatExecError(sp, err)), nil
}
if out == "dry-run" {
return ecode.Tag("OK", coremodule.FormatDryRunMessage(sp.Template)), nil
}
return ecode.Tag("OK", coremodule.FormatSuccessMessage(sp.Template, jobID)), nil
}
return ecode.Tag(ecode.ErrStepFailed, "CF 模块已接入，当前支持：/cf status, /cf help"), nil
}