92 lines
2.3 KiB
Go
92 lines
2.3 KiB
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)
|
||
}
|