telegabber/persistence/sessions.go

package persistence

import (
	"github.com/pkg/errors"
	"io/ioutil"

	"dev.narayana.im/narayana/telegabber/yamldb"

	"gopkg.in/yaml.v2"
)

// 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 {
	Login string `yaml:":login"`
}

var sessionDB SessionsYamlDB

// SessionMarshaller implementation for YamlDB
func SessionMarshaller() ([]byte, error) {
	return yaml.Marshal(sessionDB.Data)
}

// LoadSessions restores TDlib sessions from the previous run
func LoadSessions(path string) (SessionsYamlDB, error) {
	var sessionData SessionsMap

	sessionDB, err := initYamlDB(path, &sessionData)
	if err != nil {
		return sessionDB, errors.Wrap(err, "Sessions restore error")
	}

	return sessionDB, nil
}

func initYamlDB(path string, dataPtr *SessionsMap) (SessionsYamlDB, error) {
	file, err := ioutil.ReadFile(path)
	if err == nil {
		err = yaml.Unmarshal(file, dataPtr)
		if err != nil {
			return SessionsYamlDB{}, errors.Wrap(err, "YamlDB is corrupted")
		}
	} else {
		// DB file does not exist, create an empty DB
		dataPtr.Sessions = make(map[string]Session)
	}

	return SessionsYamlDB{
		YamlDB: yamldb.YamlDB{
			Path:    path,
			PathNew: path + ".new",
		},
		Data: dataPtr,
	}, nil
}