pgn.go

package pgn

type Game struct {
	tags   map[string]string
	moves  map[int]*Move
	result string
}

func New(pgn string) (*Game, error) {
	l := newLexer(pgn)
	p := newParser(l)
	game, err := p.ParsePGN()

	if err != nil {
		return nil, err
	}

	return game, nil
}

func (g *Game) GetTag(name string) string {
	return g.tags[name]
}

func (g *Game) SetTag(tag, value string) {
	g.tags[tag] = value
}

func (g *Game) TagPairs() map[string]string {
	return g.tags
}

func (g *Game) Event() string {
	return g.tags["Event"]
}

func (g *Game) Site() string {
	return g.tags["Site"]
}

func (g *Game) Round() string {
	return g.tags["Round"]
}

func (g *Game) Date() string {
	return g.tags["Date"]
}

func (g *Game) White() string {
	return g.tags["White"]
}

func (g *Game) Black() string {
	return g.tags["Black"]
}

func (g *Game) Result() string {
	return g.result
}

func (g *Game) SetResult(result string) {
	g.result = result
}

func (g *Game) GetMove(number int) *Move {
	return g.moves[number]
}

func (g *Game) SetMove(number int, move *Move) {
	g.moves[number] = move
}

func (g *Game) Moves() map[int]*Move {
	return g.moves
}

func (g *Game) IsDraw() bool {
	if g.result == "1/2-1/2" {
		return true
	}

	return false
}

func (g *Game) Winner() string {
	if g.result == "1-0" {
		return "White"
	}

	if g.result == "0-1" {
		return "Black"
	}

	if g.result == "1/2-1/2" {
		return "Draw"
	}

	return "Unknown"
}