nmdc-telegramfrontend/NTFConfig.go

61 lines
1.4 KiB
Go

package main
import (
"encoding/json"
"io/ioutil"
)
// NTFConfig keeps track of setup properties as well as runtime state.
// FIXME separate compile-time and run-time properties into separate files
type NTFConfig struct {
HubAddr string
HubDescription string
HubIgnoreNicks []string // Nicknames of Hub-Security/bots to exclude (e.g. "PtokaX").
HubNickMinChars int
BotAPIKey string
GroupChatID int64
// Map of telegram user IDs to NMDC nicks
KnownUsers map[int64]string
// Map of telegram users known to be in the group chat ==> telegram displayname
GroupChatMembers map[int64]string
// Map of telegram users to their direct conversation ID with the bot
DirectMessageChats map[int64]int64
}
func LoadConfig(configFile string) (NTFConfig, error) {
b, err := ioutil.ReadFile(configFile)
if err != nil {
return NTFConfig{}, err
}
ret := NTFConfig{}
err = json.Unmarshal(b, &ret)
if err != nil {
return NTFConfig{}, err
}
if ret.KnownUsers == nil {
ret.KnownUsers = make(map[int64]string)
}
if ret.GroupChatMembers == nil {
ret.GroupChatMembers = make(map[int64]string)
}
if ret.DirectMessageChats == nil {
ret.DirectMessageChats = make(map[int64]int64)
}
return ret, nil
}
func (this *NTFConfig) Save(configFile string) error {
b, err := json.MarshalIndent(this, "", " ")
if err != nil {
return err
}
return ioutil.WriteFile(configFile, b, 0644)
}