package bootstrap

import (
	"github.com/alist-org/alist/v3/internal/conf"
	"github.com/alist-org/alist/v3/internal/db"
	"github.com/alist-org/alist/v3/internal/fs"
	"github.com/alist-org/alist/v3/internal/offline_download/tool"
	"github.com/alist-org/alist/v3/pkg/tache"
)

func InitTaskManager() {
	fs.UploadTaskManager = tache.NewManager[*fs.UploadTask](tache.WithWorks(conf.Conf.Tasks.Upload.Workers), tache.WithMaxRetry(conf.Conf.Tasks.Upload.MaxRetry)) //upload will not support persist
	fs.CopyTaskManager = tache.NewManager[*fs.CopyTask](tache.WithWorks(conf.Conf.Tasks.Copy.Workers), tache.WithPersistFunction(db.GetTaskDataFunc("copy", conf.Conf.Tasks.Copy.TaskPersistant), db.UpdateTaskDataFunc("copy", conf.Conf.Tasks.Copy.TaskPersistant)), tache.WithMaxRetry(conf.Conf.Tasks.Copy.MaxRetry))
	tool.DownloadTaskManager = tache.NewManager[*tool.DownloadTask](tache.WithWorks(conf.Conf.Tasks.Download.Workers), tache.WithPersistFunction(db.GetTaskDataFunc("download", conf.Conf.Tasks.Download.TaskPersistant), db.UpdateTaskDataFunc("download", conf.Conf.Tasks.Download.TaskPersistant)), tache.WithMaxRetry(conf.Conf.Tasks.Download.MaxRetry))
	tool.TransferTaskManager = tache.NewManager[*tool.TransferTask](tache.WithWorks(conf.Conf.Tasks.Transfer.Workers), tache.WithPersistFunction(db.GetTaskDataFunc("transfer", conf.Conf.Tasks.Transfer.TaskPersistant), db.UpdateTaskDataFunc("transfer", conf.Conf.Tasks.Transfer.TaskPersistant)), tache.WithMaxRetry(conf.Conf.Tasks.Transfer.MaxRetry))
	if len(tool.TransferTaskManager.GetAll()) == 0 { //prevent offline downloaded files from being deleted
		CleanTempDir()
	}
}

// func InitTaskManager() {

// 	uploadTaskPersistPath := conf.Conf.Tasks.Upload.PersistPath
// 	copyTaskPersistPath := conf.Conf.Tasks.Copy.PersistPath
// 	downloadTaskPersistPath := conf.Conf.Tasks.Download.PersistPath
// 	transferTaskPersistPath := conf.Conf.Tasks.Transfer.PersistPath
// 	if !utils.Exists(uploadTaskPersistPath) {
// 		log.Infof("传输任务持久化文件")
// 		_, err := utils.CreateNestedFile(uploadTaskPersistPath)
// 		if err != nil {
// 			log.Fatalf("创建上传任务文件失败: %+v", err)
// 		}
// 	}

// 	if !utils.Exists(copyTaskPersistPath) {
// 		log.Infof("复制任务持久化文件")
// 		_, err := utils.CreateNestedFile(copyTaskPersistPath)
// 		if err != nil {
// 			log.Fatalf("创建复制任务文件失败: %+v", err)
// 		}

// 	}

// 	if !utils.Exists(downloadTaskPersistPath) {
// 		log.Infof("下载任务持久化文件")
// 		_, err := utils.CreateNestedFile(downloadTaskPersistPath)
// 		if err != nil {
// 			log.Fatalf("创建下载任务文件失败: %+v", err)
// 		}
// 	}

// 	if !utils.Exists(transferTaskPersistPath) {
// 		log.Infof("传输任务持久化文件")
// 		_, err := utils.CreateNestedFile(transferTaskPersistPath)
// 		if err != nil {
// 			log.Fatalf("创建传输任务文件失败: %+v", err)
// 		}
// 	}

// 	fs.UploadTaskManager = tache.NewManager[*fs.UploadTask](tache.WithWorks(conf.Conf.Tasks.Upload.Workers), tache.WithPersistPath(uploadTaskPersistPath), tache.WithMaxRetry(conf.Conf.Tasks.Upload.MaxRetry))
// 	fs.CopyTaskManager = tache.NewManager[*fs.CopyTask](tache.WithWorks(conf.Conf.Tasks.Copy.Workers), tache.WithPersistPath(copyTaskPersistPath), tache.WithMaxRetry(conf.Conf.Tasks.Copy.MaxRetry))
// 	tool.DownloadTaskManager = tache.NewManager[*tool.DownloadTask](tache.WithWorks(conf.Conf.Tasks.Download.Workers), tache.WithPersistPath(downloadTaskPersistPath), tache.WithMaxRetry(conf.Conf.Tasks.Download.MaxRetry))
// 	tool.TransferTaskManager = tache.NewManager[*tool.TransferTask](tache.WithWorks(conf.Conf.Tasks.Transfer.Workers), tache.WithPersistPath(transferTaskPersistPath), tache.WithMaxRetry(conf.Conf.Tasks.Transfer.MaxRetry))
// }