SaveAny-Bot/core/tasks/batchimport/taskinfo.go

package batchimport

type TaskElementInfo interface {
	FileName() string
	FileSize() int64
	GetSourcePath() string
	SourceStorageName() string
}

func (e *TaskElement) FileName() string {
	return e.FileInfo.Name
}

func (e *TaskElement) FileSize() int64 {
	return e.FileInfo.Size
}

func (e *TaskElement) GetSourcePath() string {
	return e.SourcePath
}

func (e *TaskElement) SourceStorageName() string {
	return e.SourceStorage.Name()
}

type TaskInfo interface {
	TaskID() string
	TotalSize() int64
	Uploaded() int64
	Count() int
	Processing() []TaskElementInfo
	FailedFiles() []string
}

func (t *Task) TotalSize() int64 {
	return t.totalSize
}

func (t *Task) Uploaded() int64 {
	return t.uploaded.Load()
}

func (t *Task) Count() int {
	return len(t.elems)
}

func (t *Task) Processing() []TaskElementInfo {
	t.processingMu.RLock()
	defer t.processingMu.RUnlock()

	result := make([]TaskElementInfo, 0, len(t.processing))
	for _, elem := range t.processing {
		result = append(result, elem)
	}
	return result
}

func (t *Task) FailedFiles() []string {
	t.processingMu.RLock()
	defer t.processingMu.RUnlock()

	result := make([]string, 0, len(t.failed))
	for id := range t.failed {
		// Find the element by ID
		for _, elem := range t.elems {
			if elem.ID == id {
				result = append(result, elem.FileInfo.Name)
				break
			}
		}
	}
	return result
}