aleesa-phrases-go/aleesa-phrases.go

package main

import (
	"context"
	"crypto/sha256"
	"encoding/json"
	"errors"
	"fmt"
	"io/fs"
	"math"
	"os"
	"os/signal"
	"strconv"
	"strings"
	"syscall"
	"time"

	"github.com/cockroachdb/errors/oserror"
	"github.com/cockroachdb/pebble"
	"github.com/go-redis/redis/v8"
	log "github.com/sirupsen/logrus"
)

// Входящее сообщение из pubsub-канала redis-ки
type rMsg struct {
	From    string `json:"from,omitempty"`
	ChatID  string `json:"chatid,omitempty"`
	UserID  string `json:"userid,omitempty"`
	Message string `json:"message,omitempty"`
	Plugin  string `json:"plugin,omitempty"`
	Mode    string `json:"mode,omitempty"`
	Misc    struct {
		Answer      int64  `json:"answer,omitempty"`
		BotNick     string `json:"bot_nick,omitempty"`
		CSign       string `json:"csign,omitempty"`
		FwdCnt      int64  `json:"fwd_cnt,omitempty"`
		GoodMorning int64  `json:"good_morning,omitempty"`
		MsgFormat   int64  `json:"msg_format,omitempty"`
		Username    string `json:"username,omitempty"`
	} `json:"Misc"`
}

// Исходящее сообщение в pubsub-канал redis-ки
type sMsg struct {
	From    string `json:"from"`
	ChatID  string `json:"chatid"`
	Userid  string `json:"userid"`
	Message string `json:"message"`
	Plugin  string `json:"plugin"`
	Mode    string `json:"mode"`
	Misc    struct {
		Answer      int64  `json:"answer"`
		BotNick     string `json:"bot_nick"`
		CSign       string `json:"csign"`
		FwdCnt      int64  `json:"fwd_cnt"`
		GoodMorning int64  `json:"good_morning"`
		MsgFormat   int64  `json:"msg_format"`
		Username    string `json:"username"`
	} `json:"misc"`
}

// Дефолтный конфиг клиента-редиски
var redisClient *redis.Client
var subscriber *redis.PubSub

// Мапка с открытыми дескрипторами баз с кармой
var karmaDB = make(map[string]*pebble.DB)

// Канал, в который приходят уведомления для хэндлера сигналов от траппера сигналов
var sigChan = make(chan os.Signal, 1)

// Main context
var ctx = context.Background()

// Ставится в true, если мы получили сигнал на выключение
var shutdown = false

// Изменим карму фразе
func changeKarma(chatID string, phrase string, action string) string {
	var chatHash = sha256.Sum256([]byte(chatID))
	var database = fmt.Sprintf("karma_db/%x", chatHash)
	var karmaInt int
	var answer string
	var err error

	if phrase = strings.TrimSpace(phrase); phrase == "" {
		phrase = " "
	}

	if _, ok := karmaDB[database]; !ok {
		var options pebble.Options
		// По дефолту ограничение ставится на мегабайты данных, а не на количество файлов, поэтому с дефолтными
		// настройками порождается огромное количество файлов. Умолчальное ограничение на количество файлов - 500 штук,
		// что нас не устраивает, поэтому немного снизим эту цифру до более приемлемых значений
		options.L0CompactionFileThreshold = 8

		karmaDB[database], err = pebble.Open(config.DataDir+"/"+database, &options)

		if err != nil {
			log.Errorf("Unable to open karma db, %s: %s\n", database, err)

			if phrase == " " {
				return "Карма пустоты равна 0"
			} else {
				return fmt.Sprintf("Карма %s равна 0", phrase)
			}
		}
	}

	karmaString, err := FetchV(karmaDB[database], phrase)

	if err != nil {
		// Такого ключа в базе нет
		if errors.Is(err, pebble.ErrNotFound) {
			karmaInt = 0
			// Такой бд нет
		} else if errors.Is(err, fs.ErrNotExist) {
			karmaInt = 0
		} else if errors.Is(err, oserror.ErrNotExist) {
			karmaInt = 0
			// Какая-то ещё ошибка
		} else {
			log.Errorf("Unable to Get karma of %s in database %s: %s", phrase, database, err)

			if phrase == "" {
				return fmt.Sprint("Карма пустоты равна 0")
			} else {
				return fmt.Sprintf("Карма %s равна 0", phrase)
			}
		}
	} else {
		if karmaInt, err = strconv.Atoi(karmaString); err != nil {
			karmaInt = 0
			log.Errorf("Got garbage instead of karma for %s in database %s: %s", phrase, database, err)
		}
	}

	switch action {
	case "++":
		karmaInt += 1
	case "--":
		karmaInt -= 1
	default:
		log.Errorf("Function changeKarma got strange action neither ++ nor --: %s", action)

		if phrase == "" {
			return fmt.Sprint("Карма пустоты равна 0")
		} else {
			return fmt.Sprintf("Карма %s равна 0", phrase)
		}
	}

	// Поменяли карму, теперь пора превратить значение int в string :)
	karmaString = fmt.Sprintf("%d", karmaInt)

	if err := StoreKV(karmaDB[database], "", phrase, karmaString, false); err != nil {
		log.Errorf("Unable to Set karma of %s in database %s: %s", phrase, database, err)
	}

	// Каждые -5 пунктов пробивают дно :)
	if karmaInt < -5 && math.Mod(float64(karmaInt), 5) == -1 {
		answer = "Зарегистрировано пробитие дна: к"
	} else {
		answer = "K"
	}

	if phrase == " " {
		return fmt.Sprintf("%sарма пустоты равна %s", answer, karmaString)
	} else {
		return fmt.Sprintf("%sарма %s равна %s", answer, phrase, karmaString)
	}
}

// Достанем карму фразы
func getKarma(chatID string, phrase string) string {
	var err error

	chatHash := sha256.Sum256([]byte(chatID))
	database := fmt.Sprintf("karma_db/%x", chatHash)

	if phrase = strings.Trim(phrase, "\n\r\t "); phrase == "" {
		phrase = " "
	}

	if _, ok := karmaDB[database]; !ok {
		var options pebble.Options
		// По дефолту ограничение ставится на мегабайты данных, а не на количество файлов, поэтому с дефолтными
		// настройками порождается огромное количество файлов. Умолчальное ограничение на количество файлов - 500 штук,
		// что нас не устраивает, поэтому немного снизим эту цифру до более приемлемых значений
		options.L0CompactionFileThreshold = 8
		karmaDB[database], err = pebble.Open(config.DataDir+"/"+database, &options)

		if err != nil {
			log.Errorf("Unable to open karma db, %s: %s\n", database, err)

			if phrase == " " {
				return "Карма пустоты равна 0"
			} else {
				return fmt.Sprintf("Карма %s равна 0", phrase)
			}
		}
	}

	answer, err := FetchV(karmaDB[database], phrase)

	// Если из базы ничего не вынулось, по каким-то причинам, то просто соврём, что карма равна нулю
	if err != nil {
		if errors.Is(err, pebble.ErrNotFound) {
			log.Debugf("Unable to get karma for %s: no record found in db %s", phrase, database)
		} else if errors.Is(err, fs.ErrNotExist) {
			log.Debugf("Unable to get karma for %s: db dir %s does not exist", phrase, database)
		} else if errors.Is(err, oserror.ErrNotExist) {
			log.Debugf("Unable to get karma for %s: db dir %s does not exist", phrase, database)
		} else {
			log.Errorf("Unable to get karma for %s in db dir %s: %s", phrase, database, err)
		}

		if phrase == " " {
			return "Карма пустоты равна 0"
		} else {
			return fmt.Sprintf("Карма %s равна 0", phrase)
		}
	}

	if phrase == " " {
		return fmt.Sprintf("Карма пустоты равна %s", answer)
	} else {
		return fmt.Sprintf("Карма %s равна %s", phrase, answer)
	}
}

// А не пятница ли сейчас?
func friday() string {
	var answer = fmt.Sprintf("Today is %s", time.Now().Weekday().String())
	var phrase string
	var err error

	// Но нам-то интересно какое-то более доругое высказывание относительно текущего дня недели...
	dayNum := int(time.Now().Weekday())

	switch dayNum {
	case 1:
		phrase, err = FetchRandomV(pebbleDB.monday.db, pebbleDB.monday.name)
	case 2:
		phrase, err = FetchRandomV(pebbleDB.tuesday.db, pebbleDB.tuesday.name)
	case 3:
		phrase, err = FetchRandomV(pebbleDB.wednesday.db, pebbleDB.wednesday.name)
	case 4:
		phrase, err = FetchRandomV(pebbleDB.thursday.db, pebbleDB.thursday.name)
	case 5:
		phrase, err = FetchRandomV(pebbleDB.friday.db, pebbleDB.friday.name)
	case 6:
		phrase, err = FetchRandomV(pebbleDB.saturday.db, pebbleDB.saturday.name)
	case 7:
		phrase, err = FetchRandomV(pebbleDB.sunday.db, pebbleDB.sunday.name)
	}

	if err != nil {
		log.Errorln(err)
		return answer
	}

	return phrase
}

// Пословица
func proverb() string {
	var answer = "There is no knowledge that is not power."
	phrase, err := FetchRandomV(pebbleDB.proverb.db, pebbleDB.proverb.name)

	if err != nil {
		log.Errorln(err)
		return answer
	}

	return phrase
}

// Фортунка
func fortune() string {
	var answer = "There is no knowledge that is not power."
	phrase, err := FetchRandomV(pebbleDB.fortune.db, pebbleDB.fortune.name)

	if err != nil {
		log.Errorln(err)
		return answer
	}

	return phrase
}

// Горутинка, которая парсит json-чики прилетевшие из REDIS-ки
func msgParser(ctx context.Context, msg string) {
	var sendTo string
	var j rMsg
	var message sMsg

	log.Debugf("Incomming raw json: %s", msg)

	if err := json.Unmarshal([]byte(msg), &j); err != nil {
		log.Warnf("Unable to to parse message from redis channel: %s", err)
		return
	}

	// Validate our j
	if exist := j.From; exist == "" {
		log.Warnf("Incorrect msg from redis, no from field: %s", msg)
		return
	}

	if exist := j.ChatID; exist == "" {
		log.Warnf("Incorrect msg from redis, no chatid field: %s", msg)
		return
	}

	if exist := j.UserID; exist == "" {
		log.Warnf("Incorrect msg from redis, no userid field: %s", msg)
		return
	}

	if exist := j.Message; exist == "" {
		log.Warnf("Incorrect msg from redis, no message field: %s", msg)
		return
	}

	if exist := j.Plugin; exist == "" {
		log.Warnf("Incorrect msg from redis, no plugin field: %s", msg)
		return
	} else {
		sendTo = j.Plugin
	}

	if exist := j.Mode; exist == "" {
		log.Warnf("Incorrect msg from redis, no mode field: %s", msg)
		return
	}

	// j.Misc.Answer может и не быть, тогда ответа на такое сообщение не будет
	if j.Misc.Answer == 0 {
		log.Debug("Field Misc->Answer = 0, skipping message")
		return
	}

	// j.Misc.BotNick тоже можно не передавать, тогда будет записана пустая строка
	// j.Misc.CSign если нам его не передали, возьмём значение из конфига
	if exist := j.Misc.CSign; exist == "" {
		j.Misc.CSign = config.CSign
	}

	// j.Misc.FwdCnt если нам его не передали, то будет 0
	if exist := j.Misc.FwdCnt; exist == 0 {
		j.Misc.FwdCnt = 1
	}

	// j.Misc.GoodMorning может быть быть 1 или 0, по-умолчанию 0
	// j.Misc.MsgFormat может быть быть 1 или 0, по-умолчанию 0
	// j.Misc.Username можно не передавать, тогда будет пустая строка

	// Отвалидировались, теперь вернёмся к нашим баранам.

	// Если у нас циклическая пересылка сообщения, попробуем её тут разорвать, отбросив сообщение
	if j.Misc.FwdCnt > config.ForwardsMax {
		log.Warnf("Discarding msg with fwd_cnt exceeding max value: %s", msg)
		return
	} else {
		j.Misc.FwdCnt++
	}

	// Классифицирем входящие сообщения. Первым делом, попробуем определить команды
	msgLen := len(j.Message)

	if j.Message[0:len(j.Misc.CSign)] == j.Misc.CSign { //nolint:gocritic
		var cmd = j.Message[len(j.Misc.CSign):]

		switch cmd {
		case "friday":
			j.Message = friday()
		case "пятница":
			j.Message = friday()
		case "proverb":
			j.Message = proverb()
		case "пословица":
			j.Message = proverb()
		case "f":
			j.Message = fortune()
		case "ф":
			j.Message = fortune()
		case "fortune":
			j.Message = fortune()
		case "фортунка":
			j.Message = fortune()
		case "karma":
			j.Message = getKarma(j.ChatID, "")
		case "карма":
			j.Message = getKarma(j.ChatID, "")
		default:
			cmdLen := len(cmd)
			cmds := []string{"karma ", "карма "}

			for _, command := range cmds {
				if cmdLen > len(command) && cmd[0:len(command)] == command {
					j.Message = getKarma(j.ChatID, cmd[len(command):])
					break
				}
			}
		}

		// Это последняя проверка, поэтому во всех else мы исходящее сообщение можем смело ставить в пустую строку
	} else if msgLen > len("++") {
		if j.Message[msgLen-len("--"):msgLen] == "--" || j.Message[msgLen-len("++"):msgLen] == "++" {

			// Предполагается, что менять карму мы будем для одной фразы, то есть для 1 строки
			if len(strings.Split(j.Message, "\n")) == 1 {
				if j.Message[msgLen-len("--"):msgLen] == "--" {
					j.Message = changeKarma(j.ChatID, j.Message[0:msgLen-len("--")], "--")
				} else if j.Message[msgLen-len("++"):msgLen] == "++" {
					j.Message = changeKarma(j.ChatID, j.Message[0:msgLen-len("++")], "++")
				} else {
					j.Message = ""
				}
			} else {
				j.Message = ""
			}
		} else {
			j.Message = ""
		}
	} else {
		j.Message = ""
	}

	if j.Message != "" {
		// Настало время формировать json и засылать его в дальше
		message.From = j.From
		message.Userid = j.UserID
		message.ChatID = j.ChatID
		message.Message = j.Message
		message.Plugin = j.Plugin
		message.Mode = j.Mode
		message.Misc.FwdCnt = j.Misc.FwdCnt
		message.Misc.CSign = j.Misc.CSign
		message.Misc.Username = j.Misc.Username
		message.Misc.Answer = j.Misc.Answer
		message.Misc.BotNick = j.Misc.BotNick
		message.Misc.MsgFormat = j.Misc.MsgFormat

		data, err := json.Marshal(message)

		if err != nil {
			log.Warnf("Unable to to serialize message for redis: %s", err)
			return
		}

		// Заталкиваем наш json в редиску
		if err := redisClient.Publish(ctx, sendTo, data).Err(); err != nil {
			log.Warnf("Unable to send data to redis channel %s: %s", sendTo, err)
		} else {
			log.Debugf("Send msg to redis channel %s: %s", sendTo, string(data))
		}
	}
}

// Хэндлер сигналов закрывает все бд и сваливает из приложения
func sigHandler() {
	var err error

	for {
		var s = <-sigChan
		switch s {
		case syscall.SIGINT:
			log.Infoln("Got SIGINT, quitting")
		case syscall.SIGTERM:
			log.Infoln("Got SIGTERM, quitting")
		case syscall.SIGQUIT:
			log.Infoln("Got SIGQUIT, quitting")

		// Заходим на новую итерацию, если у нас "неинтересный" сигнал
		default:
			continue
		}

		// Чтобы не срать в логи ошибками от редиски, проставим shutdown state приложения в true
		shutdown = true

		// Отпишемся от всех каналов и закроем коннект к редиске
		if err = subscriber.Unsubscribe(ctx); err != nil {
			log.Errorf("Unable to unsubscribe from redis channels cleanly: %s", err)
		}

		if err = subscriber.Close(); err != nil {
			log.Errorf("Unable to close redis connection cleanly: %s", err)
		}

		// Закроем статичные базы с фразами, тут их многовато :(
		if err = pebbleDB.fortune.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.fortune.name, err)
		}

		if err = pebbleDB.proverb.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.proverb.name, err)
		}

		if err = pebbleDB.count.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.count.name, err)
		}

		if err = pebbleDB.monday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.monday.name, err)
		}

		if err = pebbleDB.tuesday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.tuesday.name, err)
		}

		if err = pebbleDB.wednesday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.wednesday.name, err)
		}

		if err = pebbleDB.thursday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.thursday.name, err)
		}

		if err = pebbleDB.friday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.friday.name, err)
		}

		if err = pebbleDB.saturday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.saturday.name, err)
		}

		if err = pebbleDB.sunday.db.Close(); err != nil {
			log.Errorf("Unable to close %s db: %s", pebbleDB.sunday.name, err)
		}

		// Закроем базы с кармой
		for dbName, karma := range karmaDB {
			if err := karma.Close(); err != nil {
				log.Errorf("Unable to close %s karma db: %s", dbName, err)
			}
		}

		os.Exit(0)
	}
}

// Производит некоторую инициализацию перед запуском main()
func init() {
	log.SetFormatter(&log.TextFormatter{
		DisableQuote:           true,
		DisableLevelTruncation: false,
		DisableColors:          true,
		FullTimestamp:          true,
		TimestampFormat:        "2006-01-02 15:04:05",
	})

	config = 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)
	}

	var err error
	// Самое гнусное - открыть базки со статичными фразами, конкретно тут - в RO-режиме
	var options pebble.Options
	options.ReadOnly = true

	pebbleDB.count.name = "count_db"
	pebbleDB.count.db, err = pebble.Open(config.DataDir+"/"+pebbleDB.count.name, &options)
	if err != nil {
		log.Errorf("Unable to open count_db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.fortune.name = "fortune_db"
	pebbleDB.fortune.db, err = pebble.Open(config.DataDir+"/"+pebbleDB.fortune.name, &options)
	if err != nil {
		log.Errorf("Unable to open fortune db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.proverb.name = "proverb_db"
	pebbleDB.proverb.db, err = pebble.Open(config.DataDir+"/"+pebbleDB.proverb.name, &options)
	if err != nil {
		log.Errorf("Unable to open proverb db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.monday.name = "monday"
	pebbleDB.monday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.monday.name, &options)
	if err != nil {
		log.Errorf("Unable to open monday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.tuesday.name = "tuesday"
	pebbleDB.tuesday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.tuesday.name, &options)
	if err != nil {
		log.Errorf("Unable to open tuesday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.wednesday.name = "wednesday"
	pebbleDB.wednesday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.wednesday.name, &options)
	if err != nil {
		log.Errorf("Unable to open wednesday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.thursday.name = "thursday"
	pebbleDB.thursday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.thursday.name, &options)
	if err != nil {
		log.Errorf("Unable to open thursday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.friday.name = "friday"
	pebbleDB.friday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.friday.name, &options)
	if err != nil {
		log.Errorf("Unable to open friday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.saturday.name = "saturday"
	pebbleDB.saturday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.saturday.name, &options)
	if err != nil {
		log.Errorf("Unable to open saturday db: %s\n", err)
		os.Exit(1)
	}

	pebbleDB.sunday.name = "sunday"
	pebbleDB.sunday.db, err = pebble.Open(config.DataDir+"/friday_db/"+pebbleDB.sunday.name, &options)
	if err != nil {
		log.Errorf("Unable to open sunday db: %s\n", err)
		os.Exit(1)
	}
}

// Основная функция программы, не добавить и не убавить
func main() {

	// Откроем лог и скормим его логгеру
	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)
	}

	// Иницализируем клиента
	redisClient = redis.NewClient(&redis.Options{
		Addr: fmt.Sprintf("%s:%d", config.Server, config.Port),
	})

	log.Debugf("Lazy connect() to redis at %s:%d", config.Server, config.Port)
	subscriber = redisClient.Subscribe(ctx, config.Channel)

	// Самое время поставить траппер сигналов
	signal.Notify(sigChan,
		syscall.SIGINT,
		syscall.SIGTERM,
		syscall.SIGQUIT)

	go sigHandler()

	// Обработчик событий от редиски
	for {
		if shutdown {
			time.Sleep(1 * time.Second)
			continue
		}

		msg, err := subscriber.ReceiveMessage(ctx)

		if err != nil {
			if !shutdown {
				log.Warnf("Unable to connect to redis at %s:%d: %s", config.Server, config.Port, err)
			}

			time.Sleep(1 * time.Second)
			continue
		}

		go msgParser(ctx, msg.Payload)
	}
}