unified-push-xmpp/gateway/main.go

package main

import (
	"flag"

	"dev.sum7.eu/genofire/golang-lib/file"
	"dev.sum7.eu/genofire/golang-lib/web"
	apiStatus "dev.sum7.eu/genofire/golang-lib/web/api/status"
	webM "dev.sum7.eu/genofire/golang-lib/web/metrics"
	"github.com/bdlm/log"
	"github.com/bdlm/std/logger"
)

var VERSION = "development"

type configData struct {
	JWTSecret   JWTSecret   `toml"jwt_secret"`
	EndpointURL string      `toml:"endpoint_url"`
	XMPP        XMPPService `toml:"xmpp"`
	Webserver   web.Service `toml:"webserver"`
	Log         struct {
		Level     uint32 `toml:"level"`
		Timestamp bool   `toml:"timestamp"`
	} `toml:"log"`
}

func main() {
	webM.VERSION = VERSION
	apiStatus.VERSION = VERSION
	webM.NAMESPACE = "unifiedpush"

	configPath := "config.toml"
	showVersion := false

	flag.StringVar(&configPath, "c", configPath, "path to configuration file")
	flag.BoolVar(&showVersion, "version", showVersion, "show current version")

	flag.Parse()

	if showVersion {
		log.WithField("version", VERSION).Info("Version")
		return
	}

	config := &configData{}
	if err := file.ReadTOML(configPath, config); err != nil {
		log.Panicf("open config file: %s", err)
	}

	log.SetLevel(logger.Level(config.Log.Level))
	log.SetFormatter(&log.TextFormatter{
		DisableTimestamp: !config.Log.Timestamp,
	})

	go func() {
		if err := config.XMPP.Run(config.JWTSecret, config.EndpointURL); err != nil {
			log.Panicf("startup xmpp: %v", err)
		}
	}()

	config.Webserver.ModuleRegister(Bind(&config.XMPP, config.JWTSecret))

	log.Info("startup")
	if err := config.Webserver.Run(); err != nil {
		log.Fatal(err)
	}
}