probo/pkg/store/participant.go

package store

import (
	"math/rand"
	"os"
	"strconv"

	"git.andreafazzi.eu/andrea/probo/pkg/models"
	"github.com/gocarina/gocsv"
)

type ParticipantStore struct {
	*FilterStore[*models.Participant]
}

func NewParticipantStore() *ParticipantStore {
	store := new(ParticipantStore)
	store.FilterStore = NewFilterStore[*models.Participant]()

	return store
}

func (s *ParticipantStore) ImportCSV(path string) ([]*models.Participant, error) {
	file, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, os.ModePerm)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	participants := make([]*models.Participant, 0)

	if err := gocsv.UnmarshalFile(file, &participants); err != nil {
		return nil, err
	}

	memParticipants := make([]*models.Participant, 0)

	for _, p := range participants {
		if p.Token == "" {
			p.Token = generateToken()
		}
		memParticipant, err := s.Create(p)
		if err != nil {
			return nil, err
		}
		memParticipants = append(memParticipants, memParticipant)
	}

	return memParticipants, nil

}

func generateToken() string {
	// Generate six random numbers from 1 to 9
	var token string
	for i := 0; i < 6; i++ {
		randomNumber := rand.Intn(9) + 1
		token += strconv.Itoa(randomNumber)
	}

	return token
}