package cron

import (
	"fmt"
	"log"
	"sync"

	"github.com/robfig/cron/v3"
)

var (
	cronManager *CronManager
	once        sync.Once
)

// CronManager 定时任务管理器
type CronManager struct {
	cron      *cron.Cron
	tasks     map[string]cron.EntryID
	mu        sync.RWMutex
	isRunning bool
}

// GetCronManager 获取定时任务管理器单例
func GetCronManager() *CronManager {
	once.Do(func() {
		cronManager = &CronManager{
			cron:  cron.New(cron.WithSeconds()), // 支持秒级定时
			tasks: make(map[string]cron.EntryID),
		}
	})
	return cronManager
}

// AddTask 添加定时任务
// name: 任务名称，用于标识任务
// spec: cron表达式，例如：
func (cm *CronManager) AddTask(name string, spec string, job func()) error {
	cm.mu.Lock()
	defer cm.mu.Unlock()
	if _, exists := cm.tasks[name]; exists {
		return fmt.Errorf("任务 %s 已存在", name)
	}
	// 添加任务
	entryID, err := cm.cron.AddFunc(spec, func() {
		defer func() {
			if r := recover(); r != nil {
				log.Printf("定时任务 [%s] 执行panic: %v", name, r)
			}
		}()
		log.Printf("定时任务 [%s] 开始执行", name)
		job()
		log.Printf("定时任务 [%s] 执行完成", name)
	})

	if err != nil {
		return fmt.Errorf("添加任务失败: %v", err)
	}
	cm.tasks[name] = entryID
	log.Printf("定时任务 [%s] 添加成功, cron表达式: %s", name, spec)
	return nil
}

// RemoveTask 移除定时任务
func (cm *CronManager) RemoveTask(name string) error {
	cm.mu.Lock()
	defer cm.mu.Unlock()

	entryID, exists := cm.tasks[name]
	if !exists {
		return fmt.Errorf("任务 %s 不存在", name)
	}

	cm.cron.Remove(entryID)
	delete(cm.tasks, name)
	log.Printf("定时任务 [%s] 已移除", name)
	return nil
}

// Start 启动定时任务管理器
func (cm *CronManager) Start() {
	cm.mu.Lock()
	defer cm.mu.Unlock()

	if cm.isRunning {
		log.Println("定时任务管理器已经在运行中")
		return
	}

	cm.cron.Start()
	cm.isRunning = true
	log.Printf("定时任务管理器已启动，当前任务数: %d", len(cm.tasks))
}

// Stop 停止定时任务管理器
func (cm *CronManager) Stop() {
	cm.mu.Lock()
	defer cm.mu.Unlock()

	if !cm.isRunning {
		return
	}

	ctx := cm.cron.Stop()
	<-ctx.Done()
	cm.isRunning = false
	log.Println("定时任务管理器已停止")
}

// IsRunning 检查是否在运行
func (cm *CronManager) IsRunning() bool {
	cm.mu.RLock()
	defer cm.mu.RUnlock()
	return cm.isRunning
}

// GetTaskCount 获取任务数量
func (cm *CronManager) GetTaskCount() int {
	cm.mu.RLock()
	defer cm.mu.RUnlock()
	return len(cm.tasks)
}

// ListTasks 列出所有任务名称
func (cm *CronManager) ListTasks() []string {
	cm.mu.RLock()
	defer cm.mu.RUnlock()

	taskNames := make([]string, 0, len(cm.tasks))
	for name := range cm.tasks {
		taskNames = append(taskNames, name)
	}
	return taskNames
}