package main
import (
"encoding/csv"
"errors"
"fmt"
"log"
"os"
"strings"
"time"
"github.com/jszwec/csvutil"
)
type Participant struct {
LastName string `csv:"lastname"`
FirstName string `csv:"firstname"`
DateOfBirth string `csv:"dateofbirth"`
Gender Gender `csv:"gender"`
Nation string `csv:"nation"`
Region string `csv:"region"`
ClubStr string `csv:"club"`
Id uint `csv:"-"`
ClubId uint `csv:"-"`
}
type Club struct {
Name string
Id uint
}
func prepareParticipants(participants []Participant) []Club {
clubMap := map[string]uint{}
var clubs []Club
var clubId, participantId uint
for i := range participants {
p := &participants[i]
participantId++
p.Id = participantId
pClub := p.ClubStr
if strings.Contains(pClub, ", ") {
pClub = strings.Split(pClub, ", ")[0]
}
if cId, ok := clubMap[pClub]; ok {
p.ClubId = cId
} else {
clubId++
p.ClubId = clubId
clubMap[pClub] = clubId
clubs = append(clubs, Club{pClub, clubId})
}
}
return clubs
}
func parseOphardtInput(fileName string) ([]Participant, []Club, error) {
f, err := os.Open(fileName)
if err != nil {
return nil, nil, fmt.Errorf("opening input file '%s': %w", fileName, err)
}
encReader, err := encodedReader(f)
if err != nil {
return nil, nil, fmt.Errorf("cannot determine encoding of file '%s': %w", fileName, err)
}
csvReader := csv.NewReader(encReader)
csvReader.Comma = ';'
dec, err := csvutil.NewDecoder(csvReader)
if err != nil {
return nil, nil, fmt.Errorf("reading from file '%s': %w", fileName, err)
}
dec.DisallowMissingColumns = true
var participants []Participant
if err = dec.Decode(&participants); err != nil {
return nil, nil, fmt.Errorf("decoding file '%s': %w", fileName, err)
}
clubs := prepareParticipants(participants)
return participants, clubs, nil
}
func usage() string {
return fmt.Sprintf("Usage: %s