package task

import (
	distsvc "gitlink.org.cn/cloudream/common/pkg/distlock/service"
	"gitlink.org.cn/cloudream/common/pkg/task"
	"gitlink.org.cn/cloudream/common/utils/ipfs"
	coorcli "gitlink.org.cn/cloudream/rabbitmq/client/coordinator"
)

type TaskContext struct {
	IPFS        *ipfs.IPFS
	DistLock    *distsvc.Service
	Coordinator *coorcli.Client
}

// 需要在Task结束后主动调用，completing函数将在Manager加锁期间被调用，
// 因此适合进行执行结果的设置
type CompleteFn = task.CompleteFn

type Manager = task.Manager[TaskContext]

type TaskBody = task.TaskBody[TaskContext]

type Task = task.Task[TaskContext]

type CompleteOption = task.CompleteOption

func NewManager(ipfs *ipfs.IPFS, distlock *distsvc.Service, coorCli *coorcli.Client) Manager {
	return task.NewManager(TaskContext{
		IPFS:        ipfs,
		DistLock:    distlock,
		Coordinator: coorCli,
	})
}