hdkv
/
i2_bot
Archived
1
Fork 0
This repository has been archived on 2022-11-04. You can view files and clone it, but cannot push or open issues/pull-requests.
i2_bot/lib/datacache/leagues.go

81 lines
2.0 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

// i2_bot Instinct PokememBro Bot
// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
package datacache
import (
"errors"
"strconv"
"strings"
"gopkg.in/yaml.v2"
"github.com/fat0troll/i2_bot/lib/datamapping"
"github.com/fat0troll/i2_bot/static"
)
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)
}
// 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)
}