Initial commit

2022-09-04 02:34:54 +03:00
commit d9de157ad8
1017 changed files with 407553 additions and 0 deletions
--- a/main.go
+++ b/main.go
@@ -0,0 +1,102 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/go-redis/redis/v8"
+	log "github.com/sirupsen/logrus"
+)
+
+// Производит некоторую инициализацию перед запуском main()
+func init() {
+	log.SetFormatter(&log.TextFormatter{
+		DisableQuote:           true,
+		DisableLevelTruncation: false,
+		DisableColors:          true,
+		FullTimestamp:          true,
+		TimestampFormat:        "2006-01-02 15:04:05",
+	})
+
+	readConfig()
+
+	// no panic, no trace
+	switch config.Loglevel {
+	case "fatal":
+		log.SetLevel(log.FatalLevel)
+	case "error":
+		log.SetLevel(log.ErrorLevel)
+	case "warn":
+		log.SetLevel(log.WarnLevel)
+	case "info":
+		log.SetLevel(log.InfoLevel)
+	case "debug":
+		log.SetLevel(log.DebugLevel)
+	default:
+		log.SetLevel(log.InfoLevel)
+	}
+
+	// Проинициализируем мапку MODE-ами пользователей произвольным значением и удалим его
+	var usersModes = map[string]map[string]map[string]bool{
+		"user": {
+			"mode": {
+				"v": true,
+			},
+		},
+	}
+
+	userMode.Set("channel", usersModes)
+	userMode.Delete("channel")
+}
+
+// Собственно, какбэ "точка входа" - основная процедура в нашем боте
+func main() {
+	// Main context
+	var ctx = context.Background()
+
+	// Откроем лог и скормим его логгеру
+	if config.Log != "" {
+		logfile, err := os.OpenFile(config.Log, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
+
+		if err != nil {
+			log.Fatalf("Unable to open log file %s: %s", config.Log, err)
+		}
+
+		log.SetOutput(logfile)
+	}
+
+	// Иницализируем redis-клиента
+	redisClient = redis.NewClient(&redis.Options{
+		Addr: fmt.Sprintf("%s:%d", config.Redis.Server, config.Redis.Port),
+	})
+
+	log.Debugf("Lazy connect() to redis at %s:%d", config.Redis.Server, config.Redis.Port)
+	subscriber = redisClient.Subscribe(ctx, config.Redis.MyChannel)
+	redisMsgChan := subscriber.Channel()
+
+	go ircClientRun()
+	go ircSend()
+	go ircSendUnrestricted()
+
+	// Самое время поставить траппер сигналов
+	signal.Notify(sigChan,
+		syscall.SIGINT,
+		syscall.SIGTERM,
+		syscall.SIGQUIT)
+
+	go sigHandler()
+
+	// Обработчик событий от редиски
+	for msg := range redisMsgChan {
+		if shutdown {
+			continue
+		}
+
+		redisMsgParser(msg.Payload)
+	}
+}
+
+/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */