i2_bot/lib/datacache/leagues.go

// i2_bot – Instinct PokememBro Bot
// Copyright (c) 2018 Vladimir "fat0troll" Hodakov

package datacache

import (
	"errors"
	"strconv"
	"strings"

	"github.com/fat0troll/i2_bot/lib/datamapping"
	"github.com/fat0troll/i2_bot/static"
	"gopkg.in/yaml.v2"
)

func (dc *DataCache) initLeagues() {
	c.Log.Info("Initializing Leagues storage...")
	dc.leagues = make(map[int]*datamapping.League)
}

func (dc *DataCache) loadLeagues() {
	c.Log.Info("Load current Leagues data to DataCache...")
	leagues := dc.getLeagues()

	for i := range leagues {
		dc.leagues[leagues[i].ID] = &leagues[i]
	}
	c.Log.Info("Loaded leagues in DataCache: " + strconv.Itoa(len(dc.leagues)))
}

func (dc *DataCache) getLeagues() []datamapping.League {
	leagues := []datamapping.League{}

	yamlFile, err := static.ReadFile("leagues.yml")
	if err != nil {
		c.Log.Error(err.Error())
		c.Log.Fatal("Can't read leagues data file")
	}

	err = yaml.Unmarshal(yamlFile, &leagues)
	if err != nil {
		c.Log.Error(err.Error())
		c.Log.Fatal("Can't parse leagues data file")
	}

	return leagues
}

// External functions

// GetLeagueByID returns league from datacache by ID
func (dc *DataCache) GetLeagueByID(leagueID int) (*datamapping.League, error) {
	if dc.leagues[leagueID] != nil {
		return dc.leagues[leagueID], nil
	}

	return nil, errors.New("There is no league with ID = " + strconv.Itoa(leagueID))
}

// GetLeagueByName returns league from datacache by name
func (dc *DataCache) GetLeagueByName(name string) (*datamapping.League, error) {
	for i := range dc.leagues {
		if strings.Contains(dc.leagues[i].Name, name) {
			return dc.leagues[i], nil
		}
	}

	return nil, errors.New("There is no league with name = " + name)
}

// GetLeagueByEnglishName returns league from datacache by english name
func (dc *DataCache) GetLeagueByEnglishName(name string) (*datamapping.League, error) {
	for i := range dc.leagues {
		if strings.Contains(dc.leagues[i].NameEnglish, name) {
			return dc.leagues[i], nil
		}
	}

	return nil, errors.New("There is no league with name = " + name)
}

// GetLeagueBySymbol returns league from datacache by emoji
func (dc *DataCache) GetLeagueBySymbol(symbol string) (*datamapping.League, error) {
	for i := range dc.leagues {
		if dc.leagues[i].Symbol == symbol {
			return dc.leagues[i], nil
		}
	}

	return nil, errors.New("There is no league with symbol = " + symbol)
}