2018-01-29 23:50:25 +04:00
|
|
|
|
// i2_bot – Instinct PokememBro Bot
|
|
|
|
|
// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
|
|
|
|
|
|
|
|
|
|
package datacache
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"errors"
|
2018-04-30 19:46:28 +04:00
|
|
|
|
"gopkg.in/yaml.v2"
|
2018-03-31 08:34:27 +04:00
|
|
|
|
"source.wtfteam.pro/i2_bot/i2_bot/lib/datamapping"
|
2018-04-30 19:46:28 +04:00
|
|
|
|
"source.wtfteam.pro/i2_bot/i2_bot/static"
|
2018-01-29 23:50:25 +04:00
|
|
|
|
"strconv"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func (dc *DataCache) initElements() {
|
|
|
|
|
c.Log.Info("Initializing Elements storage...")
|
2018-03-31 08:34:27 +04:00
|
|
|
|
dc.elements = make(map[int]*datamapping.Element)
|
2018-01-29 23:50:25 +04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (dc *DataCache) loadElements() {
|
2018-03-31 08:34:27 +04:00
|
|
|
|
c.Log.Info("Load current Elements data to DataCache...")
|
|
|
|
|
elements := dc.getElements()
|
2018-01-29 23:50:25 +04:00
|
|
|
|
|
|
|
|
|
for i := range elements {
|
|
|
|
|
dc.elements[elements[i].ID] = &elements[i]
|
|
|
|
|
}
|
|
|
|
|
c.Log.Info("Loaded elements in DataCache: " + strconv.Itoa(len(dc.elements)))
|
2018-03-31 08:34:27 +04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (dc *DataCache) getElements() []datamapping.Element {
|
|
|
|
|
elements := []datamapping.Element{}
|
|
|
|
|
|
2018-04-30 19:46:28 +04:00
|
|
|
|
yamlFile, err := static.ReadFile("elements.yml")
|
|
|
|
|
if err != nil {
|
|
|
|
|
c.Log.Error(err.Error())
|
|
|
|
|
c.Log.Fatal("Can't read elements data file")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
err = yaml.Unmarshal(yamlFile, &elements)
|
|
|
|
|
if err != nil {
|
|
|
|
|
c.Log.Error(err.Error())
|
|
|
|
|
c.Log.Fatal("Can't parse elements data file")
|
|
|
|
|
}
|
2018-03-31 08:34:27 +04:00
|
|
|
|
|
|
|
|
|
return elements
|
2018-01-29 23:50:25 +04:00
|
|
|
|
}
|
|
|
|
|
|
2018-03-31 08:34:27 +04:00
|
|
|
|
// External functions
|
|
|
|
|
|
|
|
|
|
// GetElementByID returns element with given ID
|
|
|
|
|
func (dc *DataCache) GetElementByID(elementID int) (*datamapping.Element, error) {
|
|
|
|
|
if dc.elements[elementID] != nil {
|
|
|
|
|
return dc.elements[elementID], nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil, errors.New("There is no element with ID = " + strconv.Itoa(elementID))
|
|
|
|
|
}
|
2018-05-02 00:47:55 +04:00
|
|
|
|
|
|
|
|
|
// FindElementIDBySymbol returns element ID for given symbol
|
|
|
|
|
func (dc *DataCache) FindElementIDBySymbol(symbol string) (int, error) {
|
|
|
|
|
for i := range dc.elements {
|
|
|
|
|
if dc.elements[i].Symbol == symbol {
|
|
|
|
|
return i, nil
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0, errors.New("There is no element with symbol = " + symbol)
|
|
|
|
|
}
|