unified-push-xmpp/distributor/main.go

package main

import (
	"flag"
	"os"
	"path/filepath"

	"dev.sum7.eu/genofire/golang-lib/file"
	"dev.sum7.eu/genofire/unified-push-xmpp/distributor/storage"
	"github.com/bdlm/log"
	"github.com/bdlm/std/logger"
	"unifiedpush.org/go/np2p_dbus/distributor"
)

var dbus *distributor.DBus

type configData struct {
	StoragePath string      `toml:"storage_path"`
	XMPP        XMPPService `toml:"xmpp"`
}

func defaultPath(given, filename string) string {
	if given != "" {
		return given
	}
	basedir := os.Getenv("XDG_CONFIG_HOME")
	if len(basedir) == 0 {
		basedir = os.Getenv("HOME")
		if len(basedir) == 0 {
			basedir = "./" // FIXME: set to cwd if dunno wth is going on
		}
		basedir = filepath.Join(basedir, ".config")
	}
	basedir = filepath.Join(basedir, "unifiedpushxmpp")
	os.MkdirAll(basedir, 0o700)
	return filepath.Join(basedir, filename)
}

func main() {
	configPath := ""
	loglevel := 40
	flag.StringVar(&configPath, "c", configPath, "path to configuration file")
	flag.IntVar(&loglevel, "l", loglevel, "loglevel")
	flag.Parse()

	log.SetLevel(logger.Level(loglevel))

	config := &configData{}
	configPath = defaultPath(configPath, "config.toml")
	if err := file.ReadTOML(configPath, config); err != nil {
		if os.IsNotExist(err) {
			if err := file.SaveTOML(configPath, config); err != nil {
				log.Panicf("save example config file: %s", err)
			}
			log.Panicf("open config file, we generate one here %s", configPath)
		}
		log.Panicf("open config file: %s", err)
	}

	store, err := storage.InitStorage(defaultPath(config.StoragePath, "database.db"))
	if err != nil {
		log.Panicf("open storage: %s", err)
	}

	dbus = distributor.NewDBus("org.unifiedpush.Distributor.xmpp")
	dbus.StartHandling(&config.XMPP)

	log.Debug("startup ...")
	if err := config.XMPP.Run(dbus, store); err != nil {
		log.Errorf("startup xmpp: %v", err)
	}
}