model.go

package main

import (
	"fmt"
	"encoding/json"
	"bufio"
	"os"
	"strconv"
	"bytes"
	"io/ioutil"
	"strings"
)

type RoundType string

const (
	RoundTypeFixed        RoundType = "Fixed"
	RoundTypeBestTogether RoundType = "BestTogether"
)

type Basic struct {
	FileName string
	LongName string
	AmountPerRound float64
	Currency string
	Rounds []RoundType
}

type Player struct {
	Name string
	Club string	 `json:",omitempty`
}

type Players map[int] Player

type Tournament struct {
	Basic Basic
	Players Players
}

func load_tournament(filename string) (*Tournament,error) {
	if strings.Contains(filename, not_allowed_in_name()) {
		return nil, fmt.Errorf(gettext("Nedovolené znaky v názve turnaja"))
	}
	f, err := os.Open("data/" + filename + ".json")
	if err != nil {
		return nil, err
	}
	defer f.Close()
	reader := bufio.NewReader(f)
	decoder := json.NewDecoder(reader)
	var t *Tournament = NewTournament(filename)
	if err := decoder.Decode(t); err != nil {
		return nil, err
	}
	return  t, nil
}

func NewTournament(filename string) *Tournament {
	var t Tournament = Tournament{
		Basic: Basic{
			FileName: filename,
			Rounds: []RoundType {RoundTypeFixed, RoundTypeFixed, RoundTypeFixed, RoundTypeBestTogether, RoundTypeBestTogether},

		},
		Players: make(Players),
	}
	return &t
}

func (players *Players)UnmarshalJSON(b []byte) error {
	var tmp map[string]Player
	if err := json.Unmarshal(b, &tmp); err!=nil {
		return err
	}
	for str,player := range(tmp) {
		n, err := strconv.Atoi(str)
		if err != nil {
			return err
		}
		if val,ok :=(*players)[n]; ok {
			return fmt.Errorf(gettext(`Kľúč "%s" bol použitý viac krát`), val)
		}
		(*players)[n]=player
	}
	return nil
}

func (self Player)empty() bool {
	return self.Name=="" && self.Club==""
}

func (self Players)MarshalJSON() ([]byte, error) {
	if len(self) == 0 {
		return []byte("{}"), nil
	}
	var start rune = '{'
	var b bytes.Buffer
	for i:=0; i<len(self); i++ {
		txt,err := json.Marshal((self)[i])
		if err!=nil {
			return nil,err
		}
		b.WriteString(fmt.Sprintf(`%c"%d":%s`, start, i, txt))
		start=','
	}
	b.WriteRune('}')
	return b.Bytes(), nil
}

func getTournaments() ([]*Tournament, error) {
	result := make([]*Tournament, 0)
	fileinfos, err := ioutil.ReadDir("data")
	if err!=nil {
		return nil, err
	}
	for _,f := range fileinfos {
		filename:=f.Name()
		if (f.Mode()&os.ModeType==0) && strings.HasSuffix(filename, ".json") {
			t,err := load_tournament(filename[:len(filename)-len(".json")])
			if err==nil {
				result=append(result, t)
			}
		}
	}
	return result, nil
}

func not_allowed_in_name() string {
	return `."/\:'`;
}