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
}