49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package module
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"ops-assistant/internal/core/ecode"
|
|
"ops-assistant/internal/core/policy"
|
|
"ops-assistant/internal/core/runbook"
|
|
)
|
|
|
|
type Runner struct {
|
|
db *gorm.DB
|
|
exec *runbook.Executor
|
|
}
|
|
|
|
func NewRunner(db *gorm.DB, exec *runbook.Executor) *Runner {
|
|
return &Runner{db: db, exec: exec}
|
|
}
|
|
|
|
func (r *Runner) Run(commandText string, operator int64, req Request) (uint, string, error) {
|
|
if strings.TrimSpace(req.RunbookName) == "" {
|
|
return 0, "", fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "runbook 不能为空"))
|
|
}
|
|
if req.DryRun {
|
|
if !req.Gate.AllowDryRun {
|
|
return 0, "", fmt.Errorf(ecode.Tag(ecode.ErrStepFailed, "当前命令不允许 dry-run"))
|
|
}
|
|
return 0, "dry-run", nil
|
|
}
|
|
if err := policy.CheckGate(r.db, policy.GateRequest{
|
|
NeedFlag: req.Gate.NeedFlag,
|
|
RequireConfirm: req.Gate.RequireConfirm,
|
|
ConfirmToken: req.ConfirmToken,
|
|
ExpectedToken: req.Gate.ExpectedToken,
|
|
AllowDryRun: req.Gate.AllowDryRun,
|
|
DryRun: req.DryRun,
|
|
}); err != nil {
|
|
code := ecode.ErrFeatureDisabled
|
|
if strings.Contains(err.Error(), "confirm") || strings.Contains(err.Error(), "确认") {
|
|
code = ecode.ErrConfirmRequired
|
|
}
|
|
return 0, "", fmt.Errorf(ecode.Tag(code, err.Error()))
|
|
}
|
|
return r.exec.RunWithInputsAndMeta(commandText, req.RunbookName, operator, req.Inputs, req.Meta)
|
|
}
|