402 lines
13 KiB
Perl
402 lines
13 KiB
Perl
package Craniac;
|
|
# Предполагается, что craniac - это "маршрут по-умолчанию". То есть сюда попадают все фразы, которые не были распознаны
|
|
# aleesa-misc как команды.
|
|
|
|
# Основная задача мозгов - распознать, требуется ли ответ и обработать фразу.
|
|
# Соответсвенно, мы предполагаем, что фразы делятся на 2 типа:
|
|
# * ключевая фраза, на которую у нас есть один или несколько готовых ответов.
|
|
# * просто фраза в чате, на которую надо напрячь мозги и (в зависимости от необходимости) ответить на неё или
|
|
# проигнорировать.
|
|
|
|
# Общие модули - синтаксис, кодировки итд
|
|
use 5.018;
|
|
use strict;
|
|
use warnings;
|
|
use utf8;
|
|
use open qw (:std :utf8);
|
|
use English qw ( -no_match_vars );
|
|
|
|
# Модули для работы приложения
|
|
use Digest::SHA qw (sha1_base64);
|
|
use Hailo;
|
|
use Mojo::IOLoop;
|
|
use Mojo::Log;
|
|
# Чтобы "уж точно" использовать hiredis-биндинги, загрузим этот модуль перед Mojo::Redis
|
|
use Protocol::Redis::XS;
|
|
use Mojo::Redis;
|
|
use Math::Random::Secure qw (irand);
|
|
use Text::Fuzzy qw (distance_edits);
|
|
|
|
use Conf qw (LoadConf);
|
|
use version; our $VERSION = qw (1.0);
|
|
use Exporter qw (import);
|
|
our @EXPORT_OK = qw (RunCraniac);
|
|
|
|
sub RandomCommonPhrase ();
|
|
sub utf2sha1 ($);
|
|
sub fmatch (@);
|
|
sub brains (@);
|
|
sub RunCraniac ();
|
|
|
|
my $c = LoadConf ();
|
|
my $loglevel = $c->{'loglevel'} // 'info';
|
|
my $logfile;
|
|
my $log;
|
|
|
|
if (defined $c->{'log'}) {
|
|
$logfile = $c->{'log'};
|
|
} elsif (defined $c->{'debug_log'}) {
|
|
$logfile = $c->{'debug_log'};
|
|
}
|
|
|
|
if (defined $logfile) {
|
|
$log = Mojo::Log->new (path => $logfile, level => $loglevel);
|
|
} else {
|
|
$log = Mojo::Log->new (path => '/dev/null', level => 'fatal');
|
|
}
|
|
|
|
my $hailo;
|
|
|
|
sub RandomCommonPhrase () {
|
|
my @myphrase = (
|
|
'Так, блядь...',
|
|
'*Закатывает рукава* И ради этого ты меня позвал?',
|
|
'Ну чего ты начинаешь, нормально же общались',
|
|
'Повтори свой вопрос, не поняла',
|
|
'Выйди и зайди нормально',
|
|
'Я подумаю',
|
|
'Даже не знаю что на это ответить',
|
|
'Ты упал такие вопросы девочке задавать?',
|
|
'Можно и так, но не уверена',
|
|
'А как ты думаешь?',
|
|
);
|
|
|
|
return $myphrase[irand ($#myphrase + 1)];
|
|
}
|
|
|
|
sub utf2sha1 ($) {
|
|
my $string = shift;
|
|
|
|
if ($string eq '') {
|
|
return sha1_base64 '';
|
|
}
|
|
|
|
my $bytes = encode_utf8 $string;
|
|
return sha1_base64 $bytes;
|
|
}
|
|
|
|
sub fmatch (@) {
|
|
my $srcphrase = shift;
|
|
my $answer = shift;
|
|
|
|
my ($distance, undef) = distance_edits ($srcphrase, $answer);
|
|
my $srcphraselen = length $srcphrase;
|
|
my $distance_max = int ($srcphraselen - ($srcphraselen * (100 - (90 / ($srcphraselen ** 0.5))) * 0.01));
|
|
|
|
if ($distance >= $distance_max) {
|
|
return 0;
|
|
} else {
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
sub brains (@) {
|
|
my $chatid = shift;
|
|
my $telegram = shift // 0;
|
|
|
|
unless (defined $hailo->{$chatid}) {
|
|
my $cid = $chatid;
|
|
|
|
# Костыль, чтобы не портировать данные из телеграммных мозгов
|
|
unless ($telegram) {
|
|
$cid = utf2sha1 ($chatid);
|
|
$cid =~ s/\//-/xmsg;
|
|
}
|
|
|
|
my $brainname = sprintf '%s/%s.sqlite', $c->{braindir}, $cid;
|
|
|
|
$hailo->{$chatid} = eval {
|
|
Hailo->new (
|
|
brain => $brainname,
|
|
order => 3
|
|
);
|
|
};
|
|
|
|
if (defined $hailo->{$chatid}) {
|
|
$log->info (sprintf 'Lazy init brain: %s', $brainname);
|
|
} else {
|
|
$log->warn ($EVAL_ERROR);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
# Основной парсер
|
|
my $parse_message = sub {
|
|
my $self = shift;
|
|
my $m = shift;
|
|
my $send_to = $m->{plugin};
|
|
my $chatid = $m->{chatid};
|
|
my $userid = $m->{userid};
|
|
my $phrase = $m->{message};
|
|
my $reply;
|
|
|
|
# Если $m->{misc}->{answer} не существует, то проставим его как 1, предполагаем, что по-умолчанию ответ от нас
|
|
# всё-таки ожидают. Если что - уточним ниже.
|
|
if (defined $m->{misc}) {
|
|
unless (defined $m->{misc}->{answer}) {
|
|
$m->{misc}->{answer} = 1;
|
|
}
|
|
|
|
unless (defined $m->{misc}->{csign}) {
|
|
$m->{misc}->{csign} = '!';
|
|
}
|
|
|
|
# TODO: не проставлять дефолтный ник!
|
|
unless (defined $m->{misc}->{bot_nick}) {
|
|
$m->{misc}->{bot_nick} = 'aleesa';
|
|
}
|
|
} else {
|
|
$m->{misc}->{answer} = 1;
|
|
$m->{misc}->{csign} = '!';
|
|
$m->{misc}->{bot_nick} = 'aleesa';
|
|
}
|
|
|
|
if (defined $m->{misc}->{bot_nick}) {
|
|
# Если нам сообщили ник бота, попробуем вымарать его из оригинального текста сообщения, в большинстве случаев он
|
|
# нам не интересн.
|
|
$phrase =~ s/\s*\,?\s*($m->{misc}->{bot_nick})\s*\,?\s*/ /gui;
|
|
|
|
# В ключевых фразах можно обратиться с некоторыми командами к боту по нику, не используя символ команды
|
|
if ($m->{message} =~ /^\s*\Q;m->{misc}->{bot_nick}\E\s*;/ui) {
|
|
$reply = 'Чего?';
|
|
} elsif ($m->{message} =~ /^(\Q$m->{misc}->{csign}\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)ping\.?\s*$/i) {
|
|
$reply = 'Pong.';
|
|
} elsif ($m->{message} =~ /^(\Q$m->{misc}->{csign}\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)pong\.?\s*$/i) {
|
|
$reply = 'Wat?';
|
|
} elsif ($m->{message} =~ /^(\Q$m->{misc}->{csign}\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)пин(г|х)\.?\s*$/ui) {
|
|
$reply = 'Понг.';
|
|
} elsif ($m->{message} =~ /^(\Q$m->{misc}->{csign}\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)пон(г|х)\.?\s*$/ui) {
|
|
$reply = 'Шта?';
|
|
} elsif ($m->{message} =~ /^(\Q$m->{misc}->{csign}\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)(хэлп|halp)\s*$/ui) {
|
|
$reply = 'HALP!!!11';
|
|
}
|
|
} else {
|
|
# Если нам не сообщили ник бота, то ищем команды, используя символ команды
|
|
if ($m->{message} =~ /^\Q$m->{misc}->{csign}\Eping\.?\s*$/ui) {
|
|
$reply = 'Pong.';
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\Epong\.?\s*$/i) {
|
|
$reply = 'Wat?';
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\Eпин(г|х)\.?\s*$/ui) {
|
|
$reply = 'Понг.';
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\Eпон(г|х)\.?\s*$/ui) {
|
|
$reply = 'Шта?';
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(хэлп|halp)\s*$/ui) {
|
|
$reply = 'HALP!!!11';
|
|
}
|
|
}
|
|
|
|
# Поищем в тексте входящего сообщения команды общего плана, с символом команды.
|
|
unless (defined $reply) {
|
|
if ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(kde|кде)\s*$/ui) {
|
|
my @phrases = (
|
|
'Нет, я не буду поднимать вам плазму.',
|
|
'Повторяйте эту мантру по утрам не менее 5 раз: "Плазма не падает." И, возможно, она перестанет у вас падать.',
|
|
'Плазма не падает, она просто выходит отдохнуть.',
|
|
'Плазма это не агрегатное состояние, это суперпозиция. Во время работы она находится сразу в жидком, мягком и тёплом состояниях.',
|
|
);
|
|
|
|
$reply = $phrases[irand ($#phrases + 1)];
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(coin|монетка)$/ui) {
|
|
if (rand (101) < 0.016) {
|
|
$reply = 'ребро';
|
|
} else {
|
|
if (irand (2) == 0) {
|
|
if (irand (2) == 0) {
|
|
$reply = 'орёл';
|
|
} else {
|
|
$reply = 'аверс';
|
|
}
|
|
} else {
|
|
if (irand (2) == 0) {
|
|
$reply = 'решка';
|
|
} else {
|
|
$reply = 'реверс';
|
|
}
|
|
}
|
|
}
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(roll|dice|кости')$/ui) {
|
|
$reply = sprintf 'На первой кости выпало %d, а на второй — %d.', irand (6) + 1, irand (6) + 1;
|
|
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(ver|version|версия)$/ui) {
|
|
$reply = 'Версия: Четыре.Технологическое_превью';
|
|
}
|
|
}
|
|
|
|
# У нас уже есть чего ответить, нам попалась известная фраза, на которую одинаковый ответ, независимо от типа
|
|
# общения - публичного или приватного.
|
|
if (defined $reply) {
|
|
if ($m->{misc}->{answer}) {
|
|
$self->json ($send_to)->notify (
|
|
$send_to => {
|
|
from => 'craniac',
|
|
userid => $userid,
|
|
chatid => $chatid,
|
|
plugin => $m->{plugin},
|
|
message => $reply
|
|
}
|
|
);
|
|
}
|
|
} else {
|
|
# Пока что нам нечего ответить, известных фраз не нашлось. Ищем дальше, на сей раз фразы, зависящие от режима
|
|
# общения - приватного ли публичного.
|
|
if ($m->{mode} eq 'private') {
|
|
$phrase =~ s/^\s*(вот\s+)?скажи(\s*мне)?\s*//gui;
|
|
$phrase =~ s/\s*(,)?\s*а\s*\?$/?/gui;
|
|
|
|
if ($phrase =~ /^\s*кто\s+все\s+эти\s+люди\s*\?\s*$/ui) {
|
|
$reply = 'Какие "люди"? Мы здесь вдвоём, только ты и я.';
|
|
} elsif ($phrase =~ /^\s*кто\s+я\s*\?\s*$/ui) {
|
|
$reply = 'Где?';
|
|
} elsif ($phrase =~ /^\s*кто\s+(здесь|тут)\s*\?\s*$/ui) {
|
|
$reply = 'Ты да я, да мы с тобой.';
|
|
}
|
|
|
|
# TODO: дописать запрос в webapp, для easter egg-ов, либо делать это на уровне aleesa-misc
|
|
if (defined ($reply) && $m->{misc}->{answer}) {
|
|
$self->json ($send_to)->notify (
|
|
$send_to => {
|
|
from => 'craniac',
|
|
userid => $userid,
|
|
chatid => $chatid,
|
|
plugin => $m->{plugin},
|
|
message => $reply
|
|
}
|
|
);
|
|
|
|
return;
|
|
}
|
|
} elsif ($m->{mode} eq 'public') {
|
|
$phrase =~ s/^\s*(вот\s+)?скажи(\s*мне)?\s*//gui;
|
|
$phrase =~ s/\s*(,)?\s*а\s*\?$/?/gui;
|
|
|
|
if ($phrase =~ /^\s*кто\s+все\s+эти\s+люди\s*\?\s*$/ui) {
|
|
$reply = 'Кто здесь?';
|
|
} elsif ($phrase =~ /^\s*кто\s+я\s*?\s*$/ui){
|
|
$reply = 'Где?';
|
|
} elsif ($phrase =~ /^\s*кто\s+(здесь|тут)\s*\?\s*$/ui) {
|
|
$reply = 'Здесь все: Никита, Стас, Гена, Турбо и Дюша Метёлкин.';
|
|
}
|
|
|
|
if (defined ($reply) && $m->{misc}->{answer}) {
|
|
$self->json ($send_to)->notify (
|
|
$send_to => {
|
|
from => 'craniac',
|
|
userid => $userid,
|
|
chatid => $chatid,
|
|
plugin => $m->{plugin},
|
|
message => $reply
|
|
}
|
|
);
|
|
|
|
return;
|
|
}
|
|
}
|
|
|
|
# Попадаем сюда только если входящая фраза не распознана ранее как ключевая.
|
|
if ($m->{misc}->{answer}) {
|
|
# Пытаемся что-то генерировать, если фраза длиннее 3-х букв
|
|
if (length ($m->{message}) > 3) {
|
|
# Lazy brain init
|
|
if (brains ($chatid, 1)) {
|
|
$reply = $hailo->{$chatid}->learn_reply ($m->{message});
|
|
|
|
# Если из реактора вернулся undef выдаём "общую" фразу из словаря
|
|
unless (defined $reply) {
|
|
$reply = RandomCommonPhrase ();
|
|
} else {
|
|
# Если из реактора вернулась пустота выдаём "общую" фразу из словаря
|
|
if ($reply eq '') {
|
|
$reply = RandomCommonPhrase ();
|
|
}
|
|
|
|
# Если сгенерированная фраза слишком похожа на начальную - выдаём "общую" фразу из словаря
|
|
if (fmatch (lc ($m->{message}), lc ($reply))) {
|
|
$reply = RandomCommonPhrase ();
|
|
}
|
|
}
|
|
|
|
$self->json ($send_to)->notify (
|
|
$send_to => {
|
|
from => 'craniac',
|
|
userid => $userid,
|
|
chatid => $chatid,
|
|
plugin => $m->{plugin},
|
|
message => $reply
|
|
}
|
|
);
|
|
}
|
|
}
|
|
} else {
|
|
# пытаемся что-то запоминать, если фраза длиннее 3-х букв
|
|
if (length ($m->{message}) > 3) {
|
|
if (brains ($chatid, 1)) {
|
|
$hailo->{$chatid}->learn ($m->{message});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return;
|
|
};
|
|
|
|
# Main loop, он же event loop
|
|
sub RunCraniac () {
|
|
$log->info ("[INFO] Connecting to $c->{server}, $c->{port}");
|
|
|
|
my $redis = Mojo::Redis->new (
|
|
sprintf 'redis://%s:%s/1', $c->{server}, $c->{port}
|
|
);
|
|
|
|
$log->info ('[INFO] Registering connection-event callback');
|
|
|
|
$redis->on (
|
|
connection => sub {
|
|
my ($r, $connection) = @_;
|
|
|
|
$log->info ('[INFO] Triggering callback on new client connection');
|
|
|
|
# Пишем ошибку в лог, если соединение с редиской внезапно порвалось.
|
|
$connection->on (
|
|
error => sub {
|
|
my ($conn, $error) = @_;
|
|
$log->error ("[ERROR] Redis connection error: $error");
|
|
return;
|
|
}
|
|
);
|
|
|
|
return;
|
|
}
|
|
);
|
|
|
|
my $pubsub = $redis->pubsub;
|
|
my $sub;
|
|
$log->info ('[INFO] Subscribing to redis channels');
|
|
|
|
foreach my $channel (@{$c->{channels}}) {
|
|
$log->debug ("[DEBUG] Subscribing to $channel");
|
|
|
|
$sub->{$channel} = $pubsub->json ($channel)->listen (
|
|
$channel => sub { $parse_message->(@_); }
|
|
);
|
|
}
|
|
|
|
do { Mojo::IOLoop->start } until Mojo::IOLoop->is_running;
|
|
return;
|
|
}
|
|
|
|
1;
|
|
|
|
# vim: set ft=perl noet ai ts=4 sw=4 sts=4:
|