Compare commits
3 Commits
b8ffe727ee
...
fcd8ec8b5b
| Author | SHA1 | Date | |
|---|---|---|---|
|
fcd8ec8b5b | ||
|
|
1606e89a4a | ||
|
|
9943dbad17 |
111
README
111
README
@@ -1,111 +0,0 @@
|
|||||||
aleesa-misc-go
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
Основной роутер команд для aleesa-bot. Все протокольные плагины шлют команды сюда, а отсюда команды разлетаются в
|
|
||||||
модули-исполнители и ответ от исполнителей идёт обратно заказчику.
|
|
||||||
Работает по протоколу Redis pubsub.
|
|
||||||
|
|
||||||
|
|
||||||
Механика работы
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
Роутер подписан на канал misc (тот, который указан в поле channel конфига).
|
|
||||||
|
|
||||||
Модули, реализующие "клиентские протоколы" - telegram, irc, jabber и проч, подписаны на канал своего протокола.
|
|
||||||
Входящие сообщения они получают только через этот канал.
|
|
||||||
|
|
||||||
Модули, реализующие иную функциональность также подписаны каждый на свой канал. Кроме того, для реализации
|
|
||||||
дополнительных синхронных возможностей, они могут слушать например http запросы на отдельных - ip:портах. Но это уже
|
|
||||||
модуль-специфичная настройка.
|
|
||||||
|
|
||||||
Формат протокола - это json с полями:
|
|
||||||
from - модуль (по сути, redis-канал), в который надо ответить
|
|
||||||
chatid - чат, из которого прилетело сообщение
|
|
||||||
userid - пользователь, на чью фразу мы отвечаем, или пользователь, для которого предназначается ответ
|
|
||||||
message - текст сообщения
|
|
||||||
plugin - первоначальный источник (по сути, redis-канал) сообщения.
|
|
||||||
mode - public или private
|
|
||||||
|
|
||||||
misc - объект с дополнительными полями:
|
|
||||||
|
|
||||||
misc->answer - 0 если ответ не требуется и 1 если ответ требуется (использован в модуле craniac), значение
|
|
||||||
по-умолчанию 1, если поля нет.
|
|
||||||
|
|
||||||
misc->bot_nick - ник бота. Типа как, для разных чятов он может быть разный. (Использовано в модуле craniac.) Если
|
|
||||||
поле отсутствует, проверка на обращение к боту по нику не происходит.
|
|
||||||
|
|
||||||
misc->csign - command sign - значок, которым предваряется команда в чятике. Исторически сложилось так, что для
|
|
||||||
разных ботов этот символ был разный, например, eggdrop использовал точку, а sulci - восклицательный
|
|
||||||
знак. По-умолчанию, aleesa-bot вездепроставляет восклицательный знак, если во входящем сообщении
|
|
||||||
такой настройки нету. Изначально настройка задаётся в конфиге чат-модуля (aleesa-telegram,
|
|
||||||
например).
|
|
||||||
|
|
||||||
misc->fwd_cnt - количество пересылок сообщения. Значение по-умолчанию 1. Предполагается, что при пересылке
|
|
||||||
сообщения, модуль будет инкрементировать это значение и заодно проверять, что оно не превышает
|
|
||||||
настройку forwards_max в конфиге или 5, если в конфиге такой настройки нету.
|
|
||||||
|
|
||||||
misc->good_morning - флажок признака сообщения-фразы дня, которая по идее осылается в ~8 часов по мск. Используется в
|
|
||||||
модуле phrases.
|
|
||||||
|
|
||||||
misc->msg_format - Использовать форматирование в сообщении. На текущий момент этот флажок выставляется в модуле
|
|
||||||
telegram и благополучно проходит через модуль misc, далее попадает в модуль phrases, и, наконец,
|
|
||||||
возвращается в модуль telegram, чтобы выходное сообщение было отослано как отформатированное.
|
|
||||||
|
|
||||||
misc->username - имя пользователя, по которому к нему надо обратиться в ответе (используется в модуле games). Для
|
|
||||||
telegram это фактически ссылка на пользователя, как правило с его firstname и lastname (или
|
|
||||||
@username, если firstname и lastname у него нету), чтобы его можно было захайлатить.
|
|
||||||
|
|
||||||
|
|
||||||
Как это собрать?
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
Понадобится golang версии 1.19 или более новая.
|
|
||||||
|
|
||||||
Сам процесс сборки простой, достаточно запустить:
|
|
||||||
make
|
|
||||||
|
|
||||||
и через некоторое время мы получаем бинарник aleesa-misc-go. Это и есть наше приложение.
|
|
||||||
|
|
||||||
|
|
||||||
Как это запустить?
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
Приложение тестировалось на alpine linux x86_64 3.16, но должно работать на любом (более свежем) linux-е.
|
|
||||||
|
|
||||||
После создания data/config.json из data/config_sample.json, можно запускать приложение.
|
|
||||||
|
|
||||||
Предполагается, что конфиг написан в формате human json (https://hjson.github.io), который допускает коментарии и
|
|
||||||
некоторые другие вольности в форматировании json-а. Конфиг ищется последовательно по следующим путям по принципу "кто
|
|
||||||
первый того и тапки":
|
|
||||||
~/.aleesa-misc-go.json
|
|
||||||
~/aleesa-misc-go.json
|
|
||||||
/etc/aleesa-misc-go.json
|
|
||||||
data/config.json
|
|
||||||
|
|
||||||
В каталоге contrib находится скрипт для alpine linux, системы инициализации openrc. Этот скрипт достаточно положить в
|
|
||||||
/etc/init.d/aleesa-misc-go. После чего скопировать бинарник в каталог /var/lib/aleesa-misc-go, положить конфиг в одну
|
|
||||||
из дефолтных локаций (тестировалась /var/lib/aleesa-misc-go/data/config.json) и после этого сервис запускается
|
|
||||||
командой:
|
|
||||||
service aleesa-misc-go start
|
|
||||||
|
|
||||||
В автозагрузку его можно прописать через команду:
|
|
||||||
rc-update add aleesa-misc-go default
|
|
||||||
|
|
||||||
|
|
||||||
Nota Bene
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
По механике, в процессе работы мы форкаем горутинки на каждое входящее сообщение, особо не контроллируя этот процесс.
|
|
||||||
Предполагается, что на этот роутер не будет большой нагрузки, даже если будет несколько ботов на входе - ограничения
|
|
||||||
телеграма и других мессенджеров не должны дать возможности развить достаточно большой message rate, чтобы об это
|
|
||||||
споткнуться.
|
|
||||||
|
|
||||||
Формально, более удачным было бы иметь автоматический пул воркеров-горутинок и раскидывать им через channel-ы
|
|
||||||
сообщения. Внутри каждой горутинки можно было бы повиcнуть в бесконечном цикле на канале входящих сообщений и так и
|
|
||||||
работать. Но на практике оно того не стОит.
|
|
||||||
|
|
||||||
|
|
||||||
Nota Bene #2
|
|
||||||
-----------------------------------------------------------------------------------------------------------------------
|
|
||||||
Go не поддерживает системный вызов fork() из-за чего демонизация программ на гошке средствами самой гошки - это в
|
|
||||||
некотором роде проблема. Решается она с помощью утилиты daemonize (https://github.com/bmc/daemonize).
|
|
||||||
|
|
||||||
Соответствено, если для запуска этой программы предполагается использовать возможности демонизации, то можно
|
|
||||||
воспользоваться этой программой.
|
|
||||||
|
|
||||||
В alpine linux есть средства openrc для супервайзинга foreground-сервисов.
|
|
||||||
73
README.md
Normal file
73
README.md
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
# aleesa-misc-go
|
||||||
|
|
||||||
|
Основной роутер команд для aleesa-bot. Все протокольные плагины шлют команды сюда, а отсюда команды разлетаются в
|
||||||
|
модули-исполнители и ответ от исполнителей идёт обратно заказчику.
|
||||||
|
|
||||||
|
Работает по протоколу Redis pubsub.
|
||||||
|
|
||||||
|
## Механика работы
|
||||||
|
|
||||||
|
Роутер подписан на канал misc (тот, который указан в поле channel конфига).
|
||||||
|
|
||||||
|
Модули, реализующие "клиентские протоколы" - telegram, irc, jabber и проч, подписаны на канал своего протокола.
|
||||||
|
Входящие сообщения они получают только через этот канал.
|
||||||
|
|
||||||
|
Модули, реализующие иную функциональность также подписаны каждый на свой канал. Роутер распознаёт команду и направляет
|
||||||
|
её в канал соответствующего бэкэнд-сервиса, который уже дальше работает в соответствии со своей бизнес-логикой.
|
||||||
|
|
||||||
|
## Как это собрать?
|
||||||
|
|
||||||
|
Понадобится компилятор go версии 1.22 или более новый.
|
||||||
|
|
||||||
|
Сам процесс сборки простой, достаточно запустить:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
make
|
||||||
|
```
|
||||||
|
|
||||||
|
и через некоторое время мы получаем бинарник **aleesa-misc-go**. Это и есть наше приложение.
|
||||||
|
|
||||||
|
## Как это запустить?
|
||||||
|
|
||||||
|
Приложение тестировалось на Alpine linux x86_64 3.16, но должно работать на любом (более свежем) linux-е.
|
||||||
|
|
||||||
|
После создания **data/config.json** из **data/config_sample.json**, можно запускать приложение.
|
||||||
|
|
||||||
|
Предполагается, что конфиг написан в формате [human json](https://hjson.github.io), который допускает коментарии и
|
||||||
|
некоторые другие вольности в форматировании json-а. Конфиг ищется последовательно по следующим путям по принципу "кто
|
||||||
|
первый того и тапки":
|
||||||
|
|
||||||
|
```
|
||||||
|
~/.aleesa-misc-go.json
|
||||||
|
~/aleesa-misc-go.json
|
||||||
|
/etc/aleesa-misc-go.json
|
||||||
|
data/config.json
|
||||||
|
```
|
||||||
|
|
||||||
|
В каталоге contrib находится скрипт для alpine linux, системы инициализации openrc. Этот скрипт достаточно положить в
|
||||||
|
**/etc/init.d/aleesa-misc-go**. После чего скопировать бинарник в каталог **/var/lib/aleesa-misc-go**, положить конфиг в
|
||||||
|
одну из дефолтных локаций (тестировалась **/var/lib/aleesa-misc-go/data/config.json**) и после этого сервис запускается
|
||||||
|
командой:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
service aleesa-misc-go start
|
||||||
|
```
|
||||||
|
|
||||||
|
В автозагрузку его можно прописать через команду:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
rc-update add aleesa-misc-go default
|
||||||
|
```
|
||||||
|
|
||||||
|
## Nota Bene #1
|
||||||
|
|
||||||
|
Go не поддерживает системный вызов fork() из-за чего демонизация программ на гошке средствами самой гошки - это в
|
||||||
|
некотором роде проблема. Решается она с помощью утилиты [daemonize](https://github.com/bmc/daemonize).
|
||||||
|
|
||||||
|
Соответствено, если для запуска этой программы предполагается использовать возможности демонизации, то можно
|
||||||
|
воспользоваться этой программой.
|
||||||
|
|
||||||
|
В alpine linux есть средства openrc для супервайзинга foreground-сервисов.
|
||||||
|
|
||||||
|
Технически, можно было бы решить эту проблему fork()-а с помощью "github.com/sevlyar/go-daemon", но на текущий момент
|
||||||
|
она не стоит остро, поэтому решения не требуется.
|
||||||
@@ -1,259 +1,16 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
|
||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
"github.com/hjson/hjson-go"
|
"github.com/hjson/hjson-go"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
// msgParser горутинка, которая парсит json-чики прилетевшие из REDIS-ки.
|
|
||||||
func msgParser(ctx context.Context, msg string) {
|
|
||||||
var (
|
|
||||||
sendTo = "craniac"
|
|
||||||
j rMsg
|
|
||||||
)
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
|
|
||||||
// j.Threadid может быть пустым, значит либо нам его не дали, либо дали пустым. Это нормально.
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
|
|
||||||
if exist := j.Mode; exist == "" {
|
|
||||||
log.Warnf("Incorrect msg from redis, no mode field: %s", msg)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// j.Misc.Answer может и не быть, тогда ответа на такое сообщение не будет.
|
|
||||||
// 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
|
|
||||||
}
|
|
||||||
|
|
||||||
j.Misc.Fwdcnt++
|
|
||||||
|
|
||||||
// Классифицирем входящие сообщения. Первым делом, попробуем определить команды.
|
|
||||||
if j.Message[0:len(j.Misc.Csign)] == j.Misc.Csign {
|
|
||||||
// Может быть, это команда модуля phrases?
|
|
||||||
var (
|
|
||||||
done = false
|
|
||||||
cmd = j.Message[len(j.Misc.Csign):]
|
|
||||||
)
|
|
||||||
|
|
||||||
cmds := []string{"friday", "пятница", "proverb", "пословица", "пословиться", "fortune", "фортунка", "f", "ф",
|
|
||||||
"karma", "карма", "rum", "ром", "vodka", "водка", "beer", "пиво", "tequila", "текила", "whisky", "виски",
|
|
||||||
"absinthe", "абсент", "fuck"}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmd == command {
|
|
||||||
sendTo = config.ForwardChannels.Phrases
|
|
||||||
|
|
||||||
// Костыль для кармы.
|
|
||||||
if cmd == "karma" || cmd == "карма" {
|
|
||||||
j.Misc.Answer = 1
|
|
||||||
}
|
|
||||||
|
|
||||||
done = true
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Не угадали? акей, как насчёт модуля webapp?
|
|
||||||
if !done {
|
|
||||||
cmds = []string{"frog",
|
|
||||||
"лягушка", "horse", "лошадь", "лошадка", "rabbit", "bunny", "кролик", "snail", "улитка",
|
|
||||||
"owl", "сова", "сыч"}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmd == command {
|
|
||||||
sendTo = config.ForwardChannels.Webapp
|
|
||||||
done = true
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Не угадали? акей, как насчёт модуля webapp-go?
|
|
||||||
if !done {
|
|
||||||
cmds = []string{"cat", "кис", "fox", "лис", "buni", "anek", "анек", "анекдот", "xkcd", "monkeyuser", "tits",
|
|
||||||
"boobs", "tities", "boobies", "сиси", "сисечки", "butt", "booty", "ass", "попа", "попка", "drink",
|
|
||||||
"праздник"}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmd == command {
|
|
||||||
sendTo = config.ForwardChannels.WebappGo
|
|
||||||
done = true
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Не фортануло? может, повезёт с модулем games?
|
|
||||||
if !done {
|
|
||||||
cmds = []string{"dig", "копать", "fish", "fishing", "рыба", "рыбка", "рыбалка"}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmd == command {
|
|
||||||
sendTo = config.ForwardChannels.Games
|
|
||||||
done = true
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Нет? Вдруг это комплексная команда модуля webapp?
|
|
||||||
if !done {
|
|
||||||
cmdLen := len(cmd)
|
|
||||||
|
|
||||||
cmds := []string{"w ", "п ", "погода ", "погодка ", "погадка ", "weather "}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmdLen > len(command) && cmd[0:len(command)] == command {
|
|
||||||
sendTo = config.ForwardChannels.WebappGo
|
|
||||||
done = true
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Опять мимо? Давай тогда попытаем удачу в поиске комплексной команды для phrases.
|
|
||||||
if !done {
|
|
||||||
cmdLen := len(cmd)
|
|
||||||
|
|
||||||
cmds := []string{"karma ", "карма ", "rum ", "ром ", "vodka ", "водка ", "beer ", "пиво ", "tequila ",
|
|
||||||
"текила ", "whisky ", "виски ", "absinthe ", "абсент "}
|
|
||||||
|
|
||||||
for _, command := range cmds {
|
|
||||||
if cmdLen > len(command) && cmd[0:len(command)] == command {
|
|
||||||
sendTo = config.ForwardChannels.Phrases
|
|
||||||
|
|
||||||
if command == "karma " || command == "карма " {
|
|
||||||
// Костыль для кармы
|
|
||||||
j.Misc.Answer = 1
|
|
||||||
}
|
|
||||||
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// Попробуем выискать изменение кармы.
|
|
||||||
msgLen := len(j.Message)
|
|
||||||
|
|
||||||
// ++ или -- на конце фразы - это наш кандидат.
|
|
||||||
if msgLen > len("++") {
|
|
||||||
if j.Message[msgLen-len("--"):msgLen] == "--" || j.Message[msgLen-len("++"):msgLen] == "++" {
|
|
||||||
// Предполагается, что менять карму мы будем для одной фразы, то есть для 1 строки
|
|
||||||
if len(strings.Split(j.Message, "\n")) == 1 {
|
|
||||||
sendTo = config.ForwardChannels.Phrases
|
|
||||||
|
|
||||||
// Костыль для кармы
|
|
||||||
j.Misc.Answer = 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Настало время формировать json и засылать его в дальше.
|
|
||||||
var message sMsg
|
|
||||||
message.From = j.From
|
|
||||||
message.Userid = j.Userid
|
|
||||||
message.Chatid = j.Chatid
|
|
||||||
message.Threadid = j.Threadid
|
|
||||||
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
|
|
||||||
message.Misc.GoodMorning = j.Misc.GoodMorning
|
|
||||||
|
|
||||||
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))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// readConfig читает и валидирует конфиг, а также выставляет некоторые default-ы, если значений для параметров в конфиге
|
// readConfig читает и валидирует конфиг, а также выставляет некоторые default-ы, если значений для параметров в конфиге
|
||||||
// нет.
|
// нет.
|
||||||
func readConfig() {
|
func readConfig() {
|
||||||
|
|||||||
2
main.go
2
main.go
@@ -69,7 +69,7 @@ func main() {
|
|||||||
|
|
||||||
go sigHandler()
|
go sigHandler()
|
||||||
|
|
||||||
// Начнём выгребать события из редиски (длина ковеера/буфера канала по-умолчанию - 100 сообщений).
|
// Начнём выгребать события из редиски (длина конвеера/буфера канала по-умолчанию - 100 сообщений).
|
||||||
ch := subscriber.Channel()
|
ch := subscriber.Channel()
|
||||||
|
|
||||||
log.Infoln("Aleesa-misc-go started")
|
log.Infoln("Aleesa-misc-go started")
|
||||||
|
|||||||
250
msg-parsers.go
Normal file
250
msg-parsers.go
Normal file
@@ -0,0 +1,250 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
// msgParser горутинка, которая парсит json-чики прилетевшие из REDIS-ки.
|
||||||
|
func msgParser(ctx context.Context, msg string) {
|
||||||
|
var (
|
||||||
|
sendTo = "craniac"
|
||||||
|
j rMsg
|
||||||
|
)
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
// j.Threadid может быть пустым, значит либо нам его не дали, либо дали пустым. Это нормально.
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
if exist := j.Mode; exist == "" {
|
||||||
|
log.Warnf("Incorrect msg from redis, no mode field: %s", msg)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// j.Misc.Answer может и не быть, тогда ответа на такое сообщение не будет.
|
||||||
|
// 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
|
||||||
|
}
|
||||||
|
|
||||||
|
j.Misc.Fwdcnt++
|
||||||
|
|
||||||
|
// Классифицирем входящие сообщения. Первым делом, попробуем определить команды.
|
||||||
|
if j.Message[0:len(j.Misc.Csign)] == j.Misc.Csign {
|
||||||
|
// Может быть, это команда модуля phrases?
|
||||||
|
var (
|
||||||
|
done = false
|
||||||
|
cmd = j.Message[len(j.Misc.Csign):]
|
||||||
|
)
|
||||||
|
|
||||||
|
cmds := []string{"friday", "пятница", "proverb", "пословица", "пословиться", "fortune", "фортунка", "f", "ф",
|
||||||
|
"karma", "карма", "rum", "ром", "vodka", "водка", "beer", "пиво", "tequila", "текила", "whisky", "виски",
|
||||||
|
"absinthe", "абсент", "fuck"}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmd == command {
|
||||||
|
sendTo = config.ForwardChannels.Phrases
|
||||||
|
|
||||||
|
// Костыль для кармы.
|
||||||
|
if cmd == "karma" || cmd == "карма" {
|
||||||
|
j.Misc.Answer = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
done = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Не угадали? акей, как насчёт модуля webapp?
|
||||||
|
if !done {
|
||||||
|
cmds = []string{"frog",
|
||||||
|
"лягушка", "horse", "лошадь", "лошадка", "rabbit", "bunny", "кролик", "snail", "улитка",
|
||||||
|
"owl", "сова", "сыч"}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmd == command {
|
||||||
|
sendTo = config.ForwardChannels.Webapp
|
||||||
|
done = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Не угадали? акей, как насчёт модуля webapp-go?
|
||||||
|
if !done {
|
||||||
|
cmds = []string{"cat", "кис", "fox", "лис", "buni", "anek", "анек", "анекдот", "xkcd", "monkeyuser", "tits",
|
||||||
|
"boobs", "tities", "boobies", "сиси", "сисечки", "butt", "booty", "ass", "попа", "попка", "drink",
|
||||||
|
"праздник"}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmd == command {
|
||||||
|
sendTo = config.ForwardChannels.WebappGo
|
||||||
|
done = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Не фортануло? может, повезёт с модулем games?
|
||||||
|
if !done {
|
||||||
|
cmds = []string{"dig", "копать", "fish", "fishing", "рыба", "рыбка", "рыбалка"}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmd == command {
|
||||||
|
sendTo = config.ForwardChannels.Games
|
||||||
|
done = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Нет? Вдруг это комплексная команда модуля webapp?
|
||||||
|
if !done {
|
||||||
|
cmdLen := len(cmd)
|
||||||
|
|
||||||
|
cmds := []string{"w ", "п ", "погода ", "погодка ", "погадка ", "weather "}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmdLen > len(command) && cmd[0:len(command)] == command {
|
||||||
|
sendTo = config.ForwardChannels.WebappGo
|
||||||
|
done = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Опять мимо? Давай тогда попытаем удачу в поиске комплексной команды для phrases.
|
||||||
|
if !done {
|
||||||
|
cmdLen := len(cmd)
|
||||||
|
|
||||||
|
cmds := []string{"karma ", "карма ", "rum ", "ром ", "vodka ", "водка ", "beer ", "пиво ", "tequila ",
|
||||||
|
"текила ", "whisky ", "виски ", "absinthe ", "абсент "}
|
||||||
|
|
||||||
|
for _, command := range cmds {
|
||||||
|
if cmdLen > len(command) && cmd[0:len(command)] == command {
|
||||||
|
sendTo = config.ForwardChannels.Phrases
|
||||||
|
|
||||||
|
if command == "karma " || command == "карма " {
|
||||||
|
// Костыль для кармы
|
||||||
|
j.Misc.Answer = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Попробуем выискать изменение кармы.
|
||||||
|
msgLen := len(j.Message)
|
||||||
|
|
||||||
|
// ++ или -- на конце фразы - это наш кандидат.
|
||||||
|
if msgLen > len("++") {
|
||||||
|
if j.Message[msgLen-len("--"):msgLen] == "--" || j.Message[msgLen-len("++"):msgLen] == "++" {
|
||||||
|
// Предполагается, что менять карму мы будем для одной фразы, то есть для 1 строки
|
||||||
|
if len(strings.Split(j.Message, "\n")) == 1 {
|
||||||
|
sendTo = config.ForwardChannels.Phrases
|
||||||
|
|
||||||
|
// Костыль для кармы
|
||||||
|
j.Misc.Answer = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Настало время формировать json и засылать его в дальше.
|
||||||
|
var message sMsg
|
||||||
|
message.From = j.From
|
||||||
|
message.Userid = j.Userid
|
||||||
|
message.Chatid = j.Chatid
|
||||||
|
message.Threadid = j.Threadid
|
||||||
|
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
|
||||||
|
message.Misc.GoodMorning = j.Misc.GoodMorning
|
||||||
|
|
||||||
|
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))
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user