telegabber/persistence/sessions.go

142 lines
2.9 KiB
Go
Raw Normal View History

package persistence
import (
"github.com/pkg/errors"
"io/ioutil"
2019-12-11 22:48:35 +00:00
"time"
"dev.narayana.im/narayana/telegabber/yamldb"
2019-11-19 20:25:14 +00:00
log "github.com/sirupsen/logrus"
"gopkg.in/yaml.v2"
)
2019-12-11 22:48:35 +00:00
var zeroLocation *time.Location
func init() {
var err error
zeroLocation, err = time.LoadLocation("")
if err != nil {
log.Fatal("Wrong hardcoded timezone")
}
}
// SessionsYamlDB wraps YamlDB with Session
type SessionsYamlDB struct {
yamldb.YamlDB
Data *SessionsMap
}
// SessionsMap is for :sessions: subtree
type SessionsMap struct {
Sessions map[string]Session `yaml:":sessions"`
}
// Session is a key-values subtree
type Session struct {
2019-12-05 19:56:12 +00:00
Login string `yaml:":login"`
Timezone string `yaml:":timezone"`
}
2019-12-04 15:55:15 +00:00
var sessionDB *SessionsYamlDB
// SessionMarshaller implementation for YamlDB
func SessionMarshaller() ([]byte, error) {
2019-12-18 21:00:23 +00:00
cleanedMap := SessionsMap{}
emptySessionsMap(&cleanedMap)
for jid, session := range sessionDB.Data.Sessions {
if session.Login != "" {
cleanedMap.Sessions[jid] = session
}
}
return yaml.Marshal(&cleanedMap)
}
// LoadSessions restores TDlib sessions from the previous run
2019-12-04 15:55:15 +00:00
func LoadSessions(path string) (*SessionsYamlDB, error) {
var sessionData SessionsMap
2019-12-04 15:55:15 +00:00
var err error
2019-12-04 15:55:15 +00:00
sessionDB, err = initYamlDB(path, &sessionData)
if err != nil {
return sessionDB, errors.Wrap(err, "Sessions restore error")
}
return sessionDB, nil
}
2019-11-19 20:25:14 +00:00
func emptySessionsMap(dataPtr *SessionsMap) {
dataPtr.Sessions = make(map[string]Session)
}
2019-12-04 15:55:15 +00:00
func initYamlDB(path string, dataPtr *SessionsMap) (*SessionsYamlDB, error) {
file, err := ioutil.ReadFile(path)
if err == nil {
err = yaml.Unmarshal(file, dataPtr)
if err != nil {
2019-12-04 15:55:15 +00:00
return nil, errors.Wrap(err, "YamlDB is corrupted")
}
2019-11-19 20:25:14 +00:00
if dataPtr.Sessions == nil {
emptySessionsMap(dataPtr)
}
log.Debugf("Unmarshalled YAML: %#v", *dataPtr)
} else {
// DB file does not exist, create an empty DB
2019-11-19 20:25:14 +00:00
emptySessionsMap(dataPtr)
}
2019-12-04 15:55:15 +00:00
return &SessionsYamlDB{
YamlDB: yamldb.YamlDB{
Path: path,
PathNew: path + ".new",
},
Data: dataPtr,
}, nil
}
2019-12-05 19:56:12 +00:00
// Get retrieves a session value
func (s *Session) Get(key string) (string, error) {
switch key {
case "timezone":
return s.Timezone, nil
}
return "", errors.New("Unknown session property")
}
// ToMap converts the session to a map
func (s *Session) ToMap() map[string]string {
m := make(map[string]string)
for _, configKey := range []string{"timezone"} {
value, _ := s.Get(configKey)
m[configKey] = value
}
return m
}
// Set sets a session value
func (s *Session) Set(key string, value string) (string, error) {
switch key {
case "timezone":
s.Timezone = value
return value, nil
}
return "", errors.New("Unknown session property")
}
2019-12-11 22:48:35 +00:00
2019-12-12 00:24:35 +00:00
// TimezoneToLocation tries to convert config timezone to location
2019-12-11 22:48:35 +00:00
func (s *Session) TimezoneToLocation() *time.Location {
time, err := time.Parse("-07:00", s.Timezone)
if err == nil {
return time.Location()
}
// default
return zeroLocation
}