66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package process
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"ewintr.nl/gte/internal/storage"
|
|
"ewintr.nl/gte/internal/task"
|
|
)
|
|
|
|
var (
|
|
ErrFetchProcess = errors.New("could not fetch tasks")
|
|
)
|
|
|
|
// Fetch fetches all tasks in regular folders from the remote repository and overwrites what is stored locally
|
|
type Fetch struct {
|
|
remote *storage.RemoteRepository
|
|
local storage.LocalRepository
|
|
folders []string
|
|
}
|
|
|
|
type FetchResult struct {
|
|
Duration string `json:"duration"`
|
|
Count int `json:"count"`
|
|
}
|
|
|
|
func NewFetch(remote *storage.RemoteRepository, local storage.LocalRepository, folders ...string) *Fetch {
|
|
if len(folders) == 0 {
|
|
folders = task.KnownFolders
|
|
}
|
|
|
|
return &Fetch{
|
|
remote: remote,
|
|
local: local,
|
|
folders: folders,
|
|
}
|
|
}
|
|
|
|
func (s *Fetch) Process() (*FetchResult, error) {
|
|
start := time.Now()
|
|
tasks := []*task.Task{}
|
|
for _, folder := range s.folders {
|
|
if folder == task.FOLDER_INBOX {
|
|
continue
|
|
}
|
|
folderTasks, err := s.remote.FindAll(folder)
|
|
if err != nil {
|
|
return &FetchResult{}, fmt.Errorf("%w: %v", ErrFetchProcess, err)
|
|
}
|
|
|
|
for _, t := range folderTasks {
|
|
tasks = append(tasks, t)
|
|
}
|
|
}
|
|
|
|
if err := s.local.SetTasks(tasks); err != nil {
|
|
return &FetchResult{}, fmt.Errorf("%w: %v", ErrFetchProcess, err)
|
|
}
|
|
|
|
return &FetchResult{
|
|
Duration: time.Since(start).String(),
|
|
Count: len(tasks),
|
|
}, nil
|
|
}
|