2015-10-23 22:34:37 +02:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2015-12-18 20:54:28 +01:00
|
|
|
"flag"
|
2016-06-23 20:31:12 +02:00
|
|
|
"fmt"
|
2018-06-08 22:30:35 +02:00
|
|
|
"os"
|
2020-05-24 13:58:15 +02:00
|
|
|
"runtime"
|
2018-06-08 22:30:35 +02:00
|
|
|
"strings"
|
|
|
|
|
2016-08-14 21:48:51 +02:00
|
|
|
"github.com/42wim/matterbridge/bridge/config"
|
2016-09-18 19:21:15 +02:00
|
|
|
"github.com/42wim/matterbridge/gateway"
|
2018-11-30 23:53:00 +01:00
|
|
|
"github.com/42wim/matterbridge/gateway/bridgemap"
|
2017-03-23 23:28:55 +01:00
|
|
|
"github.com/google/gops/agent"
|
2018-08-06 21:10:54 +02:00
|
|
|
prefixed "github.com/matterbridge/logrus-prefixed-formatter"
|
2018-12-26 15:16:09 +01:00
|
|
|
"github.com/sirupsen/logrus"
|
2015-10-23 22:34:37 +02:00
|
|
|
)
|
|
|
|
|
2017-02-17 22:32:42 +01:00
|
|
|
var (
|
2021-06-01 23:36:53 +02:00
|
|
|
version = "1.22.2"
|
2017-02-17 22:32:42 +01:00
|
|
|
githash string
|
2019-02-23 22:51:27 +01:00
|
|
|
|
|
|
|
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")
|
2017-02-17 22:32:42 +01:00
|
|
|
)
|
2016-06-23 20:31:12 +02:00
|
|
|
|
2015-10-23 22:34:37 +02:00
|
|
|
func main() {
|
2016-06-23 20:31:12 +02:00
|
|
|
flag.Parse()
|
2019-02-23 22:51:27 +01:00
|
|
|
if *flagVersion {
|
|
|
|
fmt.Printf("version: %s %s\n", version, githash)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
rootLogger := setupLogger()
|
|
|
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "main"})
|
|
|
|
|
2017-03-23 23:28:55 +01:00
|
|
|
if *flagGops {
|
2018-12-08 17:04:10 +01:00
|
|
|
if err := agent.Listen(agent.Options{}); err != nil {
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Errorf("Failed to start gops agent: %#v", err)
|
2018-12-08 17:04:10 +01:00
|
|
|
} else {
|
|
|
|
defer agent.Close()
|
|
|
|
}
|
2017-03-23 23:28:55 +01:00
|
|
|
}
|
2019-02-23 22:51:27 +01:00
|
|
|
|
|
|
|
logger.Printf("Running version %s %s", version, githash)
|
2017-04-01 17:24:19 +02:00
|
|
|
if strings.Contains(version, "-dev") {
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.")
|
2016-09-30 23:19:47 +02:00
|
|
|
}
|
2019-02-23 22:51:27 +01:00
|
|
|
|
|
|
|
cfg := config.NewConfig(rootLogger, *flagConfig)
|
2018-11-15 20:43:43 +01:00
|
|
|
cfg.BridgeValues().General.Debug = *flagDebug
|
2019-02-23 22:51:27 +01:00
|
|
|
|
2020-07-18 15:46:19 +02:00
|
|
|
// 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()
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2019-02-23 22:51:27 +01:00
|
|
|
r, err := gateway.NewRouter(rootLogger, cfg, bridgemap.FullMap)
|
2017-07-25 20:11:52 +02:00
|
|
|
if err != nil {
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Fatalf("Starting gateway failed: %s", err)
|
2017-04-01 17:24:19 +02:00
|
|
|
}
|
2019-02-23 22:51:27 +01:00
|
|
|
if err = r.Start(); err != nil {
|
|
|
|
logger.Fatalf("Starting gateway failed: %s", err)
|
2016-07-11 21:23:33 +02:00
|
|
|
}
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Printf("Gateway(s) started succesfully. Now relaying messages")
|
2016-09-18 19:21:15 +02:00
|
|
|
select {}
|
2015-10-23 22:34:37 +02:00
|
|
|
}
|
2019-02-23 22:51:27 +01:00
|
|
|
|
|
|
|
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" {
|
2020-05-24 13:58:15 +02:00
|
|
|
logger.SetReportCaller(true)
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Formatter = &prefixed.TextFormatter{
|
2020-05-24 13:58:15 +02:00
|
|
|
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)
|
|
|
|
},
|
2019-02-23 22:51:27 +01:00
|
|
|
}
|
2020-05-24 13:58:15 +02:00
|
|
|
|
2019-02-23 22:51:27 +01:00
|
|
|
logger.Level = logrus.DebugLevel
|
|
|
|
logger.WithFields(logrus.Fields{"prefix": "main"}).Info("Enabling debug logging.")
|
|
|
|
}
|
|
|
|
return logger
|
|
|
|
}
|