gottdad/config.go

125 lines
2.5 KiB
Go
Raw Normal View History

2021-11-06 16:33:16 +01:00
package main
import (
_ "embed"
"encoding/json"
"io/ioutil"
"time"
"github.com/tidwall/pretty"
)
//go:embed config.sample.json
var cfgSample []byte
type ServerConfig struct {
Addr string `json:"addr"`
Passwd string `json:"passwd"`
}
type TelegramConfig struct {
AdminID int64 `json:"admin_id"`
ChatID int64 `json:"chat_id"`
URL string `json:"url"`
Token string `json:"token"`
}
type GameConfig struct {
TimeZone string `json:"timezone"`
DailyAllotment time.Duration `json:"daily_allotment"`
StartingAllotment time.Duration `json:"starting_allotment"`
Threshold time.Duration `json:"threshold"`
StartDate time.Time `json:"start_date"`
2021-12-04 05:49:20 +01:00
Started bool `json:"started"`
2021-11-06 16:33:16 +01:00
}
type ClientConfig struct {
UserID int `json:"user_id"`
Username string `json:"username"`
2021-11-10 05:56:53 +01:00
Passwd string `json:"passwd"`
2021-11-06 16:33:16 +01:00
Online bool `json:"online"`
TimeLeft time.Duration `json:"time_left"`
CompanyID uint8 `json:"company_id`
2021-12-04 05:49:20 +01:00
Ready bool `json:"ready"`
2021-11-06 16:33:16 +01:00
}
type Config struct {
Server *ServerConfig `json:"server"`
Telegram *TelegramConfig `json:"telegram"`
Game *GameConfig `json:"game"`
Clients map[int]*ClientConfig `json:"clients"`
}
// Init values for a config based on package defaults
func (c *Config) Init() error {
err := json.Unmarshal(cfgSample, &c)
c.Clients = make(map[int]*ClientConfig)
if err != nil {
return err
}
return nil
}
// Load config values from a given file
func (c *Config) Load(path string) error {
err := c.Init()
if err != nil {
return err
}
b, err := ioutil.ReadFile(path)
if err != nil {
2021-11-06 17:13:45 +01:00
return c.Save(path)
2021-11-06 16:33:16 +01:00
}
err = json.Unmarshal(b, &c)
if err != nil {
return err
}
2021-11-06 17:13:45 +01:00
return c.Save(path)
2021-11-06 16:33:16 +01:00
}
// Save config values to a given file
func (c *Config) Save(path string) error {
b, err := c.Export()
if err != nil {
return nil
}
err = ioutil.WriteFile(path, b, 0644)
return err
}
// Export config values as a bytestream
func (c *Config) Export() ([]byte, error) {
b, err := json.Marshal(c)
if err != nil {
return b, err
}
return pretty.Pretty(b), nil
}
func (c *Config) CompanyIsRegistered(id uint8) bool {
2021-11-28 14:41:00 +01:00
if id == 255 {
return false
}
2021-11-06 16:33:16 +01:00
for _, cc := range c.Clients {
if cc.CompanyID == id {
return true
}
}
return false
}
func (c *Config) GetCompanyClient(id uint8) *ClientConfig {
for _, cc := range c.Clients {
if cc.CompanyID == id {
return cc
}
}
return nil
}