diff options
Diffstat (limited to '')
-rw-r--r-- | main.go | 30 | ||||
-rw-r--r-- | template.go | 54 |
2 files changed, 69 insertions, 15 deletions
@@ -53,8 +53,9 @@ func (c club) engarde() (string, error) { {[classe club] [nom "%s"] [modifie vrai] [date_oed "332"] [cle %d]}`, c.name, c.id), nil } -func prepareParticipants(participants []participant) { - clubs := map[string]club{} +func prepareParticipants(participants []participant) []club { + clubMap := map[string]club{} + var clubs []club var clubId, participantId uint for i := range participants { @@ -68,27 +69,30 @@ func prepareParticipants(participants []participant) { pClub = strings.Split(pClub, ", ")[0] } - if c, ok := clubs[pClub]; ok { + if c, ok := clubMap[pClub]; ok { p.club = c } else { clubId++ c = club{pClub, clubId} p.club = c - clubs[p.Club] = c + clubMap[p.Club] = c + clubs = append(clubs, c) } } + + return clubs } -func parseOphardtInput(fileName string) ([]participant, error) { +func parseOphardtInput(fileName string) ([]participant, []club, error) { f, err := os.Open(fileName) if err != nil { - return nil, fmt.Errorf("opening input file '%s': %w", fileName, err) + return nil, nil, fmt.Errorf("opening input file '%s': %w", fileName, err) } encReader, err := encodedReader(f) if err != nil { - return nil, fmt.Errorf("cannot determine encoding of file '%s': %w", fileName, err) + return nil, nil, fmt.Errorf("cannot determine encoding of file '%s': %w", fileName, err) } csvReader := csv.NewReader(encReader) @@ -96,17 +100,17 @@ func parseOphardtInput(fileName string) ([]participant, error) { dec, err := csvutil.NewDecoder(csvReader) if err != nil { - return nil, fmt.Errorf("reading from file '%s': %w", fileName, err) + 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, fmt.Errorf("decoding file '%s': %w", fileName, err) + return nil, nil, fmt.Errorf("decoding file '%s': %w", fileName, err) } - prepareParticipants(participants) - return participants, nil + clubs := prepareParticipants(participants) + return participants, clubs, nil } func usage() string { @@ -155,12 +159,12 @@ func run() error { return err } - _, err = parseOphardtInput(cfg.inputFile) + participants, clubs, err := parseOphardtInput(cfg.inputFile) if err != nil { return err } - return write(cfg) + return write(cfg, participants, clubs) } func main() { diff --git a/template.go b/template.go index ce1127a..1e43f08 100644 --- a/template.go +++ b/template.go @@ -10,11 +10,12 @@ import ( ) const verbatimPath = "templates/verbatim" +const participantsName = "tireur.txt" +const clubsName = "club.txt" //go:embed templates/verbatim var verbatim embed.FS -//goland:noinspection GoUnhandledErrorResult func writeVerbatimFile(outputDir string, entry fs.DirEntry) error { inName := path.Join(verbatimPath, entry.Name()) @@ -53,7 +54,7 @@ func writeVerbatim(outputDir string) error { return nil } -func write(config EngardeConfig) error { +func write(config EngardeConfig, participants []participant, clubs []club) error { if err := os.MkdirAll(config.outputDir, 0755); err != nil { return fmt.Errorf("creating output directory '%s': %w", config.outputDir, err) } @@ -62,5 +63,54 @@ func write(config EngardeConfig) error { return fmt.Errorf("copying default files: %w", err) } + if err := writeParticipants(config.outputDir, participants); err != nil { + return fmt.Errorf("writing participant data: %w", err) + } + + if err := writeClubs(config.outputDir, clubs); err != nil { + return fmt.Errorf("writing club data: %w", err) + } + + return nil +} + +func writeClubs(outputDir string, clubs []club) error { + entries := make([]engarde, len(clubs)) + for i := range clubs { + entries[i] = clubs[i] + } + + return writeEngarde(outputDir, clubsName, entries) +} + +func writeParticipants(outputDir string, participants []participant) error { + entries := make([]engarde, len(participants)) + for i := range participants { + entries[i] = participants[i] + } + + return writeEngarde(outputDir, participantsName, entries) +} + +func writeEngarde(outputDir, fileName string, entries []engarde) error { + outName := path.Join(outputDir, fileName) + output, err := os.Create(outName) + if err != nil { + return fmt.Errorf("creating '%s': %w", outName, err) + } + defer output.Close() + + encodedOutput := encodedWriter(output) + + for _, entry := range entries { + str, err := entry.engarde() + if err != nil { + return err + } + if _, err = encodedOutput.Write([]byte(str)); err != nil { + return fmt.Errorf("writing to '%s': %w", outName, err) + } + } + return nil } |