package runner

import (
	"fmt"

	"github.com/ssdomei232/goodBaby/model"
)

// RuleExecutor 规则执行器接口
type RuleExecutor interface {
	// Execute 执行规则
	Execute(rule *model.Rule) error
	// GetType 获取执行器支持的规则类型
	GetType() string
}

// ExecutorRegistry 执行器注册表
type ExecutorRegistry struct {
	executors map[string]RuleExecutor
}

// NewExecutorRegistry 创建新的执行器注册表
func NewExecutorRegistry() *ExecutorRegistry {
	return &ExecutorRegistry{
		executors: make(map[string]RuleExecutor),
	}
}

// Register 注册规则执行器
func (er *ExecutorRegistry) Register(executor RuleExecutor) {
	er.executors[executor.GetType()] = executor
}

// Execute 根据规则类型执行规则
func (er *ExecutorRegistry) Execute(rule *model.Rule) error {
	executor, exists := er.executors[rule.Type]
	if !exists {
		return fmt.Errorf("不支持的规则类型: %s", rule.Type)
	}

	return executor.Execute(rule)
}

// GetSupportedTypes 获取所有支持的规则类型
func (er *ExecutorRegistry) GetSupportedTypes() []string {
	types := make([]string, 0, len(er.executors))
	for t := range er.executors {
		types = append(types, t)
	}
	return types
}

var globalExecutorRegistry *ExecutorRegistry

// InitExecutorRegistry 初始化执行器注册表并注册所有执行器
func InitExecutorRegistry() *ExecutorRegistry {
	registry := NewExecutorRegistry()

	// 注册所有规则执行器
	registry.Register(&BilibiliDynamicExecutor{})
	registry.Register(&EmailExecutor{})
	registry.Register(&GithubMakeRepoPublicExecutor{})
	registry.Register(&OneBotExecutor{})
	// 未来添加新规则类型时，在这里注册即可

	globalExecutorRegistry = registry
	return registry
}

// GetGlobalExecutorRegistry 获取全局执行器注册表
func GetGlobalExecutorRegistry() *ExecutorRegistry {
	if globalExecutorRegistry == nil {
		panic("执行器注册表未初始化，请先调用 InitExecutorRegistry()")
	}
	return globalExecutorRegistry
}