gte/internal/process/fetch.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
}