package main import ( "flag" "fmt" "net/http" _ "net/http/pprof" "os" "os/signal" "dev.narayana.im/narayana/telegabber/config" "dev.narayana.im/narayana/telegabber/xmpp" log "github.com/sirupsen/logrus" "github.com/zelenin/go-tdlib/client" goxmpp "gosrc.io/xmpp" ) var version string = "1.9.0" var commit string var sm *goxmpp.StreamManager var component *goxmpp.Component var err error var cleanupDone chan struct{} var sigintChannel chan os.Signal func main() { if commit != "" { version = fmt.Sprintf("%v-%v", version, commit) } var profilingPort = flag.Int("profiling-port", 0, "The port for pprof server") // YAML config, compatible with the format of Zhabogram 2.0.0 var configPath = flag.String("config", "config.yml", "Config file path") // JSON schema (not for editing by a user) var schemaPath = flag.String("schema", "./config_schema.json", "Schema file path") // Folder for Badger DB of message ids var idsPath = flag.String("ids", "ids", "Ids folder path") var versionFlag = flag.Bool("version", false, "Print the version and exit") flag.Parse() if *versionFlag { fmt.Printf("%v\n", version) os.Exit(0) } if *profilingPort > 0 { go func() { log.Println(http.ListenAndServe(fmt.Sprintf("localhost:%v", *profilingPort), nil)) }() } cleanupDone = make(chan struct{}) sigintChannel = make(chan os.Signal, 1) signal.Notify(sigintChannel, os.Interrupt) config, err := config.ReadConfig(*configPath, *schemaPath) if err != nil { log.Fatal(err) } client.SetLogVerbosityLevel(&client.SetLogVerbosityLevelRequest{ NewVerbosityLevel: stringToTdlibLogConstant(config.Telegram.Loglevel), }) SetLogrusLevel(config.XMPP.Loglevel) log.Infof("Starting telegabber version %v", version) sm, component, err = xmpp.NewComponent(config.XMPP, config.Telegram, *idsPath) if err != nil { log.Fatal(err) } go func() { <-sigintChannel log.Error("Interrupting...") exit() os.Exit(0) }() // reconnect automatically err = sm.Run() exit() if err != nil { <-cleanupDone log.Fatal(err) // bye os.Exit(-1) } } var tdlibLogConstants = map[string]int32{ ":fatal": 0, ":error": 1, ":warn": 2, ":info": 3, ":debug": 4, ":verbose": 5, ":all": 1023, } func stringToTdlibLogConstant(c string) int32 { level, ok := tdlibLogConstants[c] if !ok { level = 0 } return level } func exit() { xmpp.Close(component) close(cleanupDone) }