probo/store/file/collection.go

package file

import (
	"encoding/json"
	"os"
	"path/filepath"

	"git.andreafazzi.eu/andrea/probo/models"
	"git.andreafazzi.eu/andrea/probo/store"
)

func NewCollectionFileStore() (*FileStore[*models.Collection, *store.Store[*models.Collection]], error) {
	return NewFileStore[*models.Collection](
		store.NewStore[*models.Collection](),
		filepath.Join(BaseDir, CollectionsDir),
		"collection",
		".json",
		func(s *store.Store[*models.Collection], filepath string, content []byte) (*models.Collection, error) {
			collection := new(models.Collection)
			err := json.Unmarshal(content, &collection)
			if err != nil {
				return nil, err
			}

			c, err := s.Create(collection)
			if err != nil {
				return nil, err
			}

			return c, nil
		},
		func(s *store.Store[*models.Collection], filePath string, collection *models.Collection) error {
			jsonData, err := json.Marshal(collection)
			if err != nil {
				return err
			}

			file, err := os.Create(filePath)
			if err != nil {
				return err
			}

			defer file.Close()

			_, err = file.Write(jsonData)
			if err != nil {
				return err
			}

			return nil
		},
	)
}