-
Notifications
You must be signed in to change notification settings - Fork 3
/
main.go
73 lines (60 loc) · 1.75 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package onlineconfbot
import (
"context"
"flag"
stdlog "log"
"os"
"os/signal"
"path/filepath"
"syscall"
"github.com/onlineconf/onlineconf-go"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
var commandName = filepath.Base(flag.CommandLine.Name())
var configDir = flag.String("config-dir", "", "onlineconf configuration files directory")
var configModule = flag.String("config-module", commandName, "onlineconf module name")
var logLevel = flag.String("log-level", "debug", "log level")
var config *onlineconf.Module
var db *database
func BotMain[botType Bot](newBot func(*onlineconf.Module, SubscriptionStorage) (botType, error)) {
flag.Parse()
stdlog.SetFlags(0)
onlineconf.SetOutput(log.Logger)
level, err := zerolog.ParseLevel(*logLevel)
if err != nil {
log.Fatal().Err(err).Msg("unknown log level specified")
}
zerolog.SetGlobalLevel(level)
if *configDir != "" {
onlineconf.Initialize(*configDir)
}
config = onlineconf.GetModule(*configModule)
db, err = openDatabase()
if err != nil {
log.Fatal().Err(err).Msg("failed to open database")
}
err = db.Ping()
if err != nil {
log.Fatal().Err(err).Msg("failed to ping database")
}
bot, err := newBot(config, db)
if err != nil {
log.Fatal().Err(err).Msg("failed to initialize bot")
}
ctx, cancel := context.WithCancel(log.Logger.WithContext(context.Background()))
defer cancel()
log.Info().Msg("onlineconf-bot started")
sigC := make(chan os.Signal, 1)
signal.Notify(sigC, syscall.SIGINT, syscall.SIGTERM)
defer signal.Stop(sigC)
go func() {
sig := <-sigC
log.Info().Str("signal", sig.String()).Msg("signal received, terminating")
signal.Stop(sigC)
cancel()
}()
go bot.UpdatesProcessor(ctx)
notificationsReceiver(ctx, bot)
log.Info().Msg("onlineconf-bot stopped")
}