aeb4d29c3b
There is nothing to do really if any of those fail. The program will terminate anyway as the defer happens in the main() functions.
110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/42wim/matterbridge/bridge/config"
|
|
"github.com/42wim/matterbridge/gateway"
|
|
"github.com/42wim/matterbridge/gateway/bridgemap"
|
|
"github.com/google/gops/agent"
|
|
prefixed "github.com/matterbridge/logrus-prefixed-formatter"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
version = "1.17.6-dev"
|
|
githash string
|
|
|
|
flagConfig = flag.String("conf", "matterbridge.toml", "config file")
|
|
flagDebug = flag.Bool("debug", false, "enable debug")
|
|
flagVersion = flag.Bool("version", false, "show version")
|
|
flagGops = flag.Bool("gops", false, "enable gops agent")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *flagVersion {
|
|
fmt.Printf("version: %s %s\n", version, githash)
|
|
return
|
|
}
|
|
|
|
rootLogger := setupLogger()
|
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "main"})
|
|
|
|
if *flagGops {
|
|
if err := agent.Listen(agent.Options{}); err != nil {
|
|
logger.Errorf("Failed to start gops agent: %#v", err)
|
|
} else {
|
|
defer agent.Close()
|
|
}
|
|
}
|
|
|
|
logger.Printf("Running version %s %s", version, githash)
|
|
if strings.Contains(version, "-dev") {
|
|
logger.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.")
|
|
}
|
|
|
|
cfg := config.NewConfig(rootLogger, *flagConfig)
|
|
cfg.BridgeValues().General.Debug = *flagDebug
|
|
|
|
// if logging to a file, ensure it is closed when the program terminates
|
|
// nolint:errcheck
|
|
defer func() {
|
|
if f, ok := rootLogger.Out.(*os.File); ok {
|
|
f.Sync()
|
|
f.Close()
|
|
}
|
|
}()
|
|
|
|
r, err := gateway.NewRouter(rootLogger, cfg, bridgemap.FullMap)
|
|
if err != nil {
|
|
logger.Fatalf("Starting gateway failed: %s", err)
|
|
}
|
|
if err = r.Start(); err != nil {
|
|
logger.Fatalf("Starting gateway failed: %s", err)
|
|
}
|
|
logger.Printf("Gateway(s) started succesfully. Now relaying messages")
|
|
select {}
|
|
}
|
|
|
|
func setupLogger() *logrus.Logger {
|
|
logger := &logrus.Logger{
|
|
Out: os.Stdout,
|
|
Formatter: &prefixed.TextFormatter{
|
|
PrefixPadding: 13,
|
|
DisableColors: true,
|
|
},
|
|
Level: logrus.InfoLevel,
|
|
}
|
|
if *flagDebug || os.Getenv("DEBUG") == "1" {
|
|
logger.SetReportCaller(true)
|
|
logger.Formatter = &prefixed.TextFormatter{
|
|
PrefixPadding: 13,
|
|
DisableColors: true,
|
|
FullTimestamp: false,
|
|
|
|
CallerFormatter: func(function, file string) string {
|
|
return fmt.Sprintf(" [%s:%s]", function, file)
|
|
},
|
|
CallerPrettyfier: func(f *runtime.Frame) (string, string) {
|
|
sp := strings.SplitAfter(f.File, "/matterbridge/")
|
|
filename := f.File
|
|
if len(sp) > 1 {
|
|
filename = sp[1]
|
|
}
|
|
s := strings.Split(f.Function, ".")
|
|
funcName := s[len(s)-1]
|
|
return funcName, fmt.Sprintf("%s:%d", filename, f.Line)
|
|
},
|
|
}
|
|
|
|
logger.Level = logrus.DebugLevel
|
|
logger.WithFields(logrus.Fields{"prefix": "main"}).Info("Enabling debug logging.")
|
|
}
|
|
return logger
|
|
}
|