openclaw/ops-assistant
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init: ops-assistant codebase

Browse Source
This commit is contained in:
OpenClaw Agent
2026-03-19 21:23:28 +08:00
commit 81deba4766
94 changed files with 10767 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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
}

62
internal/module/cpa/commands.go Normal file
View File

@@ -0,0 +1,62 @@
package cpa
import (
"fmt"
"strings"
coremodule "ops-assistant/internal/core/module"
"ops-assistant/internal/core/runbook"
)
func commandSpecs() []coremodule.CommandSpec {
return []coremodule.CommandSpec{
{
Prefixes: []string{"/cpa status"},
Template: coremodule.CommandTemplate{
RunbookName: "cpa_status",
Gate: coremodule.Gate{AllowDryRun: true},
DryRunMsg: "🧪 dry-run: 将执行 /cpa status（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cpa status 已执行，job=%d", jobID) },
},
ErrPrefix: "/cpa status 执行失败: ",
},
{
Prefixes: []string{"/cpa usage backup"},
Template: coremodule.CommandTemplate{
RunbookName: "cpa_usage_backup",
Gate: coremodule.Gate{AllowDryRun: true},
DryRunMsg: "🧪 dry-run: 将执行 /cpa usage backup（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cpa usage backup 已执行，job=%d", jobID) },
},
ErrPrefix: "/cpa usage backup 执行失败: ",
},
{
Prefixes: []string{"/cpa usage restore "},
ErrHint: "--confirm YES_RESTORE",
Template: coremodule.CommandTemplate{
RunbookName: "cpa_usage_restore",
Gate: coremodule.Gate{NeedFlag: "allow_ops_restore", RequireConfirm: true, ExpectedToken: "YES_RESTORE", AllowDryRun: true},
InputsFn: func(_ string, parts []string) (map[string]string, error) {
if len(parts) < 4 {
return nil, fmt.Errorf("❌ 用法：/cpa usage restore <backup_id>")
}
backupID := strings.TrimSpace(parts[3])
if backupID == "" {
return nil, fmt.Errorf("❌ backup_id 不能为空")
}
return map[string]string{"backup_id": backupID}, nil
},
MetaFn: func(userID int64, confirmToken string, inputs map[string]string) runbook.RunMeta {
meta := runbook.NewMeta()
meta.Target = "hwsg"
meta.RiskLevel = "high"
meta.ConfirmHash = hashConfirmToken(confirmToken)
return meta
},
DryRunMsg: "🧪 dry-run: 将执行 restore（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /cpa usage restore 已执行，job=%d", jobID) },
},
ErrPrefix: "/cpa usage restore 执行失败: ",
},
}
}

16
internal/module/cpa/crypto.go Normal file
View File

@@ -0,0 +1,16 @@
package cpa
import (
"crypto/sha256"
"encoding/hex"
"strings"
)
func hashConfirmToken(token string) string {
t := strings.TrimSpace(token)
if t == "" {
return ""
}
sum := sha256.Sum256([]byte(t))
return hex.EncodeToString(sum[:])
}

13
internal/module/cpa/guards.go Normal file
View File

@@ -0,0 +1,13 @@
package cpa
import (
"ops-assistant/internal/core/ecode"
)
func formatErr(code, msg string) string {
return ecode.Tag(code, msg)
}
func formatOK(msg string) string {
return ecode.Tag("OK", msg)
}

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

@@ -0,0 +1,40 @@
package cpa
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 == "/cpa" || strings.HasPrefix(text, "/cpa help") {
return "CPA 模块\n- /cpa status\n- /cpa usage backup\n- /cpa usage restore <backup_id> [--confirm YES_RESTORE] [--dry-run]", 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 formatErr(ecode.ErrStepFailed, coremodule.FormatExecError(sp, err)), nil
}
if out == "dry-run" {
return formatOK(coremodule.FormatDryRunMessage(sp.Template)), nil
}
return formatOK(coremodule.FormatSuccessMessage(sp.Template, jobID)), nil
}
return "❓ 暂不支持该 CPA 命令。当前支持：/cpa status, /cpa usage backup, /cpa usage restore <backup_id>", nil
}

25
internal/module/mail/commands.go Normal file
View File

@@ -0,0 +1,25 @@
package mail
import (
"fmt"
coremodule "ops-assistant/internal/core/module"
)
func commandSpecs() []coremodule.CommandSpec {
return []coremodule.CommandSpec{
{
Prefixes: []string{"/mail status"},
Template: coremodule.CommandTemplate{
RunbookName: "mail_status",
Gate: coremodule.Gate{
NeedFlag: "enable_module_mail",
AllowDryRun: true,
},
DryRunMsg: "🧪 dry-run: 将执行 /mail status（未实际执行）",
SuccessMsg: func(jobID uint) string { return fmt.Sprintf("✅ /mail status 已执行，job=%d", jobID) },
},
ErrPrefix: "/mail status 执行失败: ",
},
}
}

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

@@ -0,0 +1,40 @@
package mail
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 == "/mail" || strings.HasPrefix(text, "/mail help") {
return "Mail 模块\n- /mail status [--dry-run]", 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, "Mail 模块已接入，当前支持：/mail status, /mail help"), nil
}