package cron_runner

import (
	"sync"

	"gitee.com/carlmax_my/console-core-go/pkg/cache"
	"gitee.com/carlmax_my/console-core-go/pkg/errors"

	"github.com/jakecoffman/cron"
	"go.uber.org/zap"
)

var _ Server = (*server)(nil)

type taskCount struct {
	wg   sync.WaitGroup
	exit chan struct{}
}

// func (tc *taskCount) i() {}

func (tc *taskCount) Add() {
	tc.wg.Add(1)
}

func (tc *taskCount) Done() {
	tc.wg.Done()
}

func (tc *taskCount) Exit() {
	tc.wg.Done()
	<-tc.exit
}

func (tc *taskCount) Wait() {
	tc.Add()
	tc.wg.Wait()
	close(tc.exit)
}

type server struct {
	logger    *zap.Logger
	cron      *cron.Cron
	taskCount *taskCount
	cache     cache.Cache
}

// singleton
type Server interface {
	i()

	Start()

	TaskCountWait()

	// Stop 停止 cron 服务
	Stop()

	// AddTask 增加定时任务
	AddTask(task ITaskModel)

	// RemoveTask 删除定时任务
	RemoveTask(taskId string)

	// AddJob 增加定时任务执行的工作内容
	AddJob(task ITaskModel) cron.FuncJob
}

func New(logger *zap.Logger, cache cache.Cache) (Server, error) {
	if logger == nil {
		return nil, errors.New("logger required")
	}

	return &server{
		logger: logger,
		cron:   cron.New(),
		taskCount: &taskCount{
			wg:   sync.WaitGroup{},
			exit: make(chan struct{}),
		},
		cache: cache,
	}, nil
}

func (s *server) i() {}