25 Commits

Author SHA1 Message Date
eleksir
ca982571d9 Reimplement commands parser 2022-04-22 16:28:02 +03:00
eleksir
03634958a1 Fix typo 2022-04-22 15:46:19 +03:00
eleksir
6eeb8e7b89 Tidy incomming redis message 2022-04-22 15:42:23 +03:00
eleksir
7b32737b33 Use cloned version of incomming redis message as template for outgoing message 2022-04-22 15:01:20 +03:00
eleksir
8952828886 Use command sign from config, when misc->bot_nick is not supplied, it defaults to empty string 2022-04-22 15:00:04 +03:00
eleksir
221314ae16 Userid and chatid must be strings in order to work with chats components other than Telegram 2022-04-22 14:58:26 +03:00
eleksir
27b9a4bcbd Set each async signal handler separately 2022-04-22 14:57:14 +03:00
eleksir
98bf57c5e0 In cpanfile bump Mojolicious version 9.22->9.24, zero English version - it's part of perl 2022-04-22 14:56:01 +03:00
eleksir
7565539a14 Update .gitignore 2022-04-22 14:53:38 +03:00
eleksir
0a62499302 Update README 2022-04-17 15:28:58 +03:00
eleksir
f256ad357f Add list of perlcritic modules used for linting 2022-04-03 18:19:35 +03:00
eleksir
62311dae21 Fix typo in кости command, use shorter var name for csign 2022-04-03 18:18:38 +03:00
eleksir
30ebe8aca1 Set userid and chatid explicitly to num for redis json msg 2022-03-27 16:06:43 +03:00
eleksir
bbc5173c16 Update README 2022-02-12 19:13:21 +03:00
eleksir
622b690946 Perlcritic overhaul, add missing 'use module', update cpanfile, replace Mojo::Log with Log::Any
* Use more strict policy in perlcritic (add more policies)
* Add missing Use Encode and Use File::Spec in Craniac.pm
* Replace Mojo::Log with Log::Any, as Log::Any can catch stacktraces
* Update cpanfile
2022-02-12 19:08:00 +03:00
eleksir
5df11d2437 Minor fixes for v1.0 release, mostly comments, add openrc init script 2022-01-07 16:30:53 +03:00
eleksir
f8d0c3f701 Move Lat module here, as it uses brain too 2022-01-04 17:48:13 +03:00
eleksir
b7013d6ab0 Loop detection, easter eggs, update readme, update sample config, fix telegram-crippled brain names 2022-01-04 16:38:45 +03:00
eleksir
9c56db452c Bump version Mojo::Redis in cpanfile 2022-01-01 17:45:36 +03:00
eleksir
ca0a891522 Translate comments, fix coding style 2022-01-01 17:34:45 +03:00
eleksir
0f58c30603 Implement async signal trap via Mojo::IOLoop::Signal for more robust app stop and restart 2022-01-01 17:34:07 +03:00
eleksir
637e5f8261 Pin version numbers in cpanfile 2022-01-01 17:32:57 +03:00
eleksir
240712bf42 Use eval on lazy brain init 2022-01-01 03:47:22 +03:00
eleksir
64c8300f93 add missing ping command 2022-01-01 03:44:50 +03:00
eleksir
468baf7156 Fix default loglevel notice -> info 2022-01-01 03:43:25 +03:00
13 changed files with 1835 additions and 212 deletions

2
.gitignore vendored
View File

@@ -5,5 +5,7 @@ _Deparsed_XSubs.pm
*.pid
*.log
data/cache/**
data/brains/**
data/lat/**
data/config.json
vendor_perl/**

View File

@@ -1,4 +1,19 @@
#!/usr/bin/env perlcritic
# Задействованные модули критика:
#
# Perl::Critic::More
# Perl::Critic::Pulp
# Perl::Critic::TooMuchCode
# Perl::Critic::Lax
# Perl::Critic::CognitiveComplexity
# Perl::Critic::Policy::CompileTime
# Perl::Critic::Policy::ProhibitImplicitImport
# Perl::Critic::Policy::Dynamic::NoIndirect
# Perl::Critic::StricterSubs
# Perl::Critic::Policy::ProhibitImplicitImport
# Perl::Critic::Policy::CodeLayout::RequireUseUTF8
severity = 5
verbose = 8
@@ -93,3 +108,26 @@ lib_sections = NAME | VERSION | DESCRIPTION | AUTHOR | LICENSE AND COPYRIGHT
[-ControlStructures::ProhibitDeepNests]
# Have to use it, so fuckoff
[-Variables::ProhibitPackageVars]
[-CodeLayout::RequireASCII]
[-Editor::RequireEmacsFileVariables]
[-TooMuchCode::ProhibitUnnecessaryUTF8Pragma]
[-TooMuchCode::ProhibitDuplicateLiteral]
[-ErrorHandling::RequireUseOfExceptions]
[-ValuesAndExpressions::RequireConstantOnLeftSideOfEquality]
[-Lax::RequireConstantOnLeftSideOfEquality::ExceptEq]
[-Lax::ProhibitEmptyQuotes::ExceptAsFallback]
[-CodeLayout::RequireUseUTF8]
[ValuesAndExpressions::RestrictLongStrings]
max_length = 256
[-Miscellanea::ProhibitUnrestrictedNoCritic]

View File

@@ -11,13 +11,13 @@
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
- Для публичных чатов допустимой являются нотации:
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - слово "Pong." если фраза была на английском языке или "Понг.", если фраза была на русском языке.
@@ -28,15 +28,20 @@
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + ping + опциональная точка и любое количество пробельных символов.
- Для публичных чатов допустимой являются нотации:
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - слово "Wat?" если фраза была на английском языке или "Шта?", если фраза была на русском языке.
* lat или лат
- Для приватных и публичных чатов допустимой является нотация:
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - одно произвольное предложение из предсгенерённого "мозга" с цепями Маркова.
* halp или хэлп
- Для приватных чатов допустимой являются нотации:
@@ -45,34 +50,34 @@
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
- Для публичных чатов допустимой являются нотации:
* с префиксом в виде любого количества пробельных символов + имени бота + любое количество пробельных символов +
опциональньная запятая + любое количество пробельных символов. После команды опциональная точка и любое
количество пробельных символов.
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - слово "HALP!!!11".
* kde или кде
- Для приватных и публичных чатов допустимой является нотация:
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - одно произвольное предложение из массива ответов на эту фразу.
* roll или dice или кости
- Для приватных и публичных чатов допустимой является нотация:
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - "На первой кости выпало M, а на второй — N.", где M и N произвольные целые числа от 1 до 6
включительно.
* ver или version или версия
- Для приватных и публичных чатов допустимой является нотация:
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретны чат-
* символ команды (по-умолчанию - восклицательный знак, задаётся в настройках модуля, отвечающего за конкретный чат-
протокол) + команда + опциональная точка и любое количество пробельных символов.
Ответ на эту фразу - номер версии ПО.
Ответ на эту фразу - приблизительный номер версии ПО.
* кто все эти люди?
- Для приватных и публичных чатов допустимой является нотация:

20
README
View File

@@ -6,17 +6,18 @@ aleesa-craniac
Механика работы
-----------------------------------------------------------------------------------------------------------------------
Бредогенератор подписан на redis-канал craniac. В этот канал пишет другой демоночек - aleesa-misc. Причём, особенность
такова, что если aleesa-misc не распознала фразу как команду для другого модуля, то по-умолчанию такая фраза
оправляется в канал craniac, то есть сюда, в "мозги".
Бредогенератор подписан на redis-канал craniac. В этот канал пишет другой демоночек - aleesa-misc-go. Причём,
особенность такова, что если aleesa-misc не распознала фразу, как команду для другого модуля, то по-умолчанию такая
фраза оправляется в канал craniac, то есть сюда, в "мозги".
У бредогенератора есть 2 типа входящих сообщений - ключевые фразы, на которые у него есть готовые (статические) ответы
и остальные, для которых ему надо решить нужен ли ответ и по-необходимости сгенерировать его, используя данные в цепях
Маркова.
Ответ направляется в redis-канал, указанный в поле plugin входящего сообщения.
Ответ направляется в redis-канал, указанный в поле plugin входящего сообщения, если не предполагается некоей особой
логики, связанной, наприме, с easter egg поведением.
Формат протокола - см README от aleesa-misc.
Формат протокола - см. README от aleesa-misc-go.
Как это запустить?
@@ -30,3 +31,12 @@ aleesa-craniac
Если всё хорошо, то после запуска redis-а, создания data/config.json из data/config_sample.json, можно запускать и само
приложение.
В каталоге contrib находится скрипт для alpine linux, системы инициализации openrc.
Этот скрипт достаточно положить в /etc/init.d/aleesa-craniac, сконфигурить сервис по инструкции выше и, после этого,
сервис запускается командой:
service aleesa-craniac start
В автозагрузку его можно прописать через команду:
rc-update add aleesa-craniac default

View File

@@ -1,6 +1,6 @@
#!/usr/bin/perl
use 5.018;
use 5.018; ## no critic (ProhibitImplicitImport)
use strict;
use warnings;
use utf8;
@@ -11,7 +11,7 @@ use version; our $VERSION = qw (1.0);
my $workdir;
# before we run, change working dir
# До фактического запуска логики программы, сменим рабочий каталог, чтобы подцепить либы
BEGIN {
use Cwd qw (chdir abs_path);
my @CWD = split /\//xms, abs_path ($PROGRAM_NAME);
@@ -29,8 +29,12 @@ no Cwd;
use Cwd::utf8 qw (chdir abs_path);
use File::Basename qw (basename dirname);
use File::Path qw (make_path);
use Mojo::Log;
use File::Spec ();
use Log::Any qw ($log);
use Log::Any::Adapter ();
use Log::Any::For::Std ();
use POSIX qw (setgid setuid setsid);
use Conf qw (LoadConf);
use Craniac qw (RunCraniac);
@@ -45,7 +49,6 @@ my $piddir = '/var/run/' . basename (abs_path ($PROGRAM_NAME));
my $C = LoadConf ();
my $loglevel = $C->{'loglevel'} // 'notice';
my $logfile;
my $log;
if (defined $C->{'log'}) {
$logfile = $C->{'log'};
@@ -53,7 +56,8 @@ if (defined $C->{'log'}) {
$logfile = $C->{'debug_log'};
}
# if we run under root, create pid dir and set correct owner for it, same for debug log if it is defined
# Если мы запускаемся под рутом, то попробуем создать необходимые служебные каталоги и простать на них корректных
# владельца и права
if ($UID == 0) {
my $user = $C->{user};
@@ -67,8 +71,8 @@ if ($UID == 0) {
make_path (
$piddir, {
uid => $uid,
group => $gid
}
group => $gid,
},
);
}
@@ -79,13 +83,13 @@ if ($UID == 0) {
make_path (
$logdir, {
uid => $uid,
group => $gid
}
group => $gid,
},
) or die "Unable to create $logdir: $OS_ERROR\n";
}
}
# drop privs
# Переключаемся на "рабочего" пользователя и сбрасываем привилегии.
setgid ($gid) or die "Unable to switch to group of $user($OS_ERROR)\n";
setuid ($uid) or die "Unable to switch to account $user($OS_ERROR)\n";
} else {
@@ -93,25 +97,25 @@ if ($UID == 0) {
}
if (defined $logfile) {
$log = Mojo::Log->new (path => $logfile, level => $loglevel);
Log::Any::Adapter->set ('File', $logfile, log_level => $loglevel);
} else {
$log = Mojo::Log->new (path => '/dev/null', level => 'fatal');
Log::Any::Adapter->set ('File', File::Spec->devnull);
}
# по-умолчанию, демонизируемся, если явно не указано, что этого делать не надо
# По-умолчанию, демонизируемся, если явно не указано, что этого делать не надо
if (defined ($C->{daemonize}) && $C->{daemonize}) {
daemonize ();
} elsif ( ! defined ($C->{daemonize})) {
daemonize ();
}
# write pidfile
# Запишем pid-файл
my $apath = abs_path $PROGRAM_NAME;
unless (defined $apath) {
$log->fatal ("[FATAL] Unable to get abs_path: $OS_ERROR\n");
exit 1;
};
}
my $pidfile = sprintf '%s/%s.pid', $piddir, basename $apath;
$log->info ("[INFO] Creating pidfile at $pidfile");
@@ -128,16 +132,18 @@ print {$MYPID} $PID or do {
close $MYPID; ## no critic (InputOutput::RequireCheckedSyscalls
# При наличии хака в лице Mojo::IOLoop::Signal в Misc.pm, установка хэндлера в этом месте - это фактически
# подстраховка на случай ядерной войны.
local $SIG{INT} = \&__signal_handler;
local $SIG{TERM} = \&__signal_handler;
# set correct proctitle
# Поставим правильное имя процесса в proctitle
$PROGRAM_NAME = abs_path $PROGRAM_NAME; ## no critic (Variables::RequireLocalizedPunctuationVars)
# run main app, at last
# И запустим уже наш сервис
RunCraniac ();
if (-f $pidfile) {
if (-e $pidfile) {
unlink $pidfile;
}
@@ -145,10 +151,8 @@ exit 0;
sub daemonize {
__fork ();
POSIX::setsid ();
open STDIN, '<', '/dev/null'; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
#open STDOUT, '>', '/dev/null'; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
#open STDERR, '>', '/dev/null'; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
setsid ();
open STDIN, '<', File::Spec->devnull; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
__fork ();
umask 0;
return;
@@ -172,7 +176,7 @@ sub __fork {
sub __signal_handler {
$log->info ("[INFO] Caught a signal $OS_ERROR");
if (-f $pidfile) {
if (-e $pidfile) {
unlink $pidfile;
}

27
bin/train_lat.pl Normal file
View File

@@ -0,0 +1,27 @@
#!/usr/bin/perl
# Наполняет отдельный "мозг" с бредом латинскими выражениями
use 5.018; ## no critic (ProhibitImplicitImport)
use strict;
use warnings;
use utf8;
use open qw (:std :utf8);
use English qw ( -no_match_vars );
use version; our $VERSION = qw (1.0);
my $workdir;
# До фактического запуска логики программы, сменим рабочий каталог, чтобы подцепить либы
BEGIN {
use Cwd qw (chdir abs_path);
my @CWD = split /\//xms, abs_path ($PROGRAM_NAME);
if ($#CWD > 1) { $#CWD = $#CWD - 2; }
$workdir = join '/', @CWD;
chdir $workdir;
}
use lib ("$workdir/lib", "$workdir/vendor_perl", "$workdir/vendor_perl/lib/perl5");
use Craniac::Lat qw (Train);
Train ();
exit 0;

View File

@@ -0,0 +1,11 @@
#!/sbin/openrc-run
name="$SVCNAME"
command="/var/lib/$SVCNAME/bin/$SVCNAME"
pidfile="/var/run/$SVCNAME/$SVCNAME.pid"
depend() {
need net redis
use dns
after redis
}

View File

@@ -1,17 +1,33 @@
requires 'Cwd';
requires 'Cwd::utf8';
requires 'Digest::SHA';
requires 'Encode';
requires 'English';
requires 'File::Basename';
requires 'File::Path';
requires 'Hailo';
requires 'JSON::XS';
requires 'local::lib';
requires 'Math::Random::Secure';
requires 'Mojo::IOLoop';
requires 'Mojo::Log';
requires 'Mojo::Redis';
requires 'POSIX';
requires 'Protocol::Redis::XS';
requires 'Text::Fuzzy';
## no critic
requires 'Carp', '==1.50';
requires 'Clone', '==0.45';
requires 'Cwd', '==3.75';
requires 'Cwd::utf8', '==0.011';
requires 'Data::Dumper', '==2.183';
requires 'Digest::SHA', '==6.02';
requires 'Encode', '==3.16';
# Part of perl
requires 'English', '0';
requires 'File::Basename', '==2.85';
requires 'File::Path', '==2.18';
requires 'File::Spec', '==3.75';
requires 'Hailo', '==0.75';
requires 'JSON::XS', '==4.03';
requires 'local::lib', '==2.000028';
requires 'Log::Any', '==1.710';
requires 'Log::Any::Adapter', '==1.710';
requires 'Log::Any::For::Std', '==0.05';
requires 'Math::Random::Secure', '==0.080001';
requires 'Mojolicious', '==9.24';
# Part of Mojolicious
requires 'Mojo::IOLoop', '0';
requires 'Mojo::IOLoop::Signal', '==0.003';
requires 'Mojo::Redis', '==3.27';
# Part of perl
requires 'POSIX', '0';
# Buggy, latest version works, as pub-sub, but fails tests
requires 'Protocol::Redis::XS', '0';
requires 'Text::Fuzzy', '==0.29';
# Part of perl
requires 'utf8', '0';
requires 'version', '==0.9929';

View File

@@ -1,10 +1,8 @@
{
"server" : "redis",
"port" : "6379",
# inbound redis channels, where we subscribed
"channels" : [
"craniac",
"craniac"
],
# should we fork() and become unix daemon?
@@ -22,5 +20,16 @@
"loglevel" : "info",
# ditto
"pid" : "/var/run/aleesa-craniac/aleesa-craniac.pid"
"pid" : "/var/run/aleesa-craniac/aleesa-craniac.pid",
# maximal amount of message forwardings, to avoid endless loops
"forwards_max" : 5,
# Command prefix aka command sign
"csign" : "!",
"lat" : {
"brain" : "data/lat/brain.sqlite",
"src" : "data/lat.txt"
}
}

1336
data/lat.txt Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -1,14 +1,13 @@
package Conf;
# loads config
# Загружает конфиг файл
use 5.018;
use 5.018; ## no critic (ProhibitImplicitImport)
use strict;
use warnings;
use utf8;
use open qw (:std :utf8);
use English qw ( -no_match_vars );
use Encode qw (encode decode);
use JSON::XS;
use JSON::XS ();
use version; our $VERSION = qw (1.0);
use Exporter qw (import);
@@ -16,17 +15,26 @@ our @EXPORT_OK = qw (LoadConf);
sub LoadConf {
my $c = 'data/config.json';
open my $CH, '<', $c or die "[FATAL] No conf at $c: $OS_ERROR\n";
my $json;
my $CH;
unless (open $CH, '<', $c) {
die "[FATAL] No conf at $c: $OS_ERROR\n";
}
my $len = (stat $c) [7];
my $json;
my $readlen = read $CH, $json, $len;
$json = encode 'UTF-8', $json;
unless ($readlen) {
close $CH; ## no critic (InputOutput::RequireCheckedSyscalls
die "[FATAL] Unable to read $c: $OS_ERROR\n";
}
if ($readlen != $len) {
close $CH; ## no critic (InputOutput::RequireCheckedSyscalls
die "[FATAL] File $c is $len bytes on disk, but we read only $readlen bytes\n";
}
close $CH; ## no critic (InputOutput::RequireCheckedSyscalls
my $j = JSON::XS->new->utf8->relaxed;
return $j->decode ($json);

View File

@@ -1,6 +1,6 @@
package Craniac;
# Предполагается, что craniac - это "маршрут по-умолчанию". То есть сюда попадают все фразы, которые не были распознаны
# aleesa-misc как команды.
# aleesa-misc как команды. В том числе и некоторые команды (но по идее это так и задумано).
# Основная задача мозгов - распознать, требуется ли ответ и обработать фразу.
# Соответсвенно, мы предполагаем, что фразы делятся на 2 типа:
@@ -9,7 +9,7 @@ package Craniac;
# проигнорировать.
# Общие модули - синтаксис, кодировки итд
use 5.018;
use 5.018; ## no critic (ProhibitImplicitImport)
use strict;
use warnings;
use utf8;
@@ -17,17 +17,24 @@ use open qw (:std :utf8);
use English qw ( -no_match_vars );
# Модули для работы приложения
use Clone qw (clone);
use Data::Dumper qw (Dumper);
use Digest::SHA qw (sha1_base64);
use Hailo;
use Mojo::IOLoop;
use Mojo::Log;
use Encode qw (encode_utf8);
use File::Spec ();
use Hailo ();
use Log::Any qw ($log);
use Mojo::IOLoop ();
use Mojo::IOLoop::Signal ();
# Чтобы "уж точно" использовать hiredis-биндинги, загрузим этот модуль перед Mojo::Redis
use Protocol::Redis::XS;
use 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 Craniac::Lat qw (Lat);
use version; our $VERSION = qw (1.0);
use Exporter qw (import);
our @EXPORT_OK = qw (RunCraniac);
@@ -39,24 +46,12 @@ 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 $fwd_cnt = $c->{'forward_max'} // 5;
my $hailo;
# Несколько фраз общего характера, используемые как универсальный ответ, когда ответить нечего (по тем или иным
# причинам) или ответ не будет достаточно вменяемым (слишком похож на фразу, на которую отвечаем)
sub RandomCommonPhrase () {
my @myphrase = (
'Так, блядь...',
@@ -100,32 +95,37 @@ sub fmatch (@) {
}
}
# Ленивая инициализация мозга. Создаём/открываем мозг только если нам надо "подумать" а не на каждую входящую фразу.
sub brains (@) {
my $chatid = shift;
my $telegram = shift // 0;
unless (defined $hailo->{$chatid}) {
my $cid = $chatid;
my $brainname;
# Костыль, чтобы не портировать данные из телеграммных мозгов
# Костыль, чтобы не портировать данные из телеграммных мозгов и не заниматься переименованием
unless ($telegram) {
$cid = utf2sha1 ($chatid);
$cid =~ s/\//-/xmsg;
$brainname = sprintf '%s/%s.sqlite', $c->{braindir}, $cid;
} else {
$brainname = sprintf '%s/%s.brain.sqlite', $c->{braindir}, $cid;
}
my $brainname = sprintf '%s/%s.sqlite', $c->{braindir}, $cid;
$hailo->{$chatid} = Hailo->new (
$hailo->{$chatid} = eval {
Hailo->new (
brain => $brainname,
order => 3
order => 3,
);
};
unless (defined $hailo->{$chatid}) {
$log->warn ($EVAL_ERROR);
if (defined $hailo->{$chatid}) {
$log->info (sprintf '[INFO] Lazy init brain: %s', $brainname);
} else {
$log->error ("[ERROR] $EVAL_ERROR");
return 0;
}
$log->info (sprintf 'Lazy init brain: %s', $brainname);
}
return 1;
@@ -135,68 +135,115 @@ sub brains (@) {
my $parse_message = sub {
my $self = shift;
my $m = shift;
my $answer = clone ($m);
$answer->{from} = 'craniac';
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;
if (defined $answer->{misc}) {
unless (defined $answer->{misc}->{answer}) {
$answer->{misc}->{answer} = 1;
}
unless (defined $m->{misc}->{csign}) {
$m->{misc}->{csign} = '!';
unless (defined $answer->{misc}->{bot_nick}) {
$answer->{misc}->{bot_nick} = '';
}
# TODO: не проставлять дефолтный ник!
unless (defined $m->{misc}->{bot_nick}) {
$m->{misc}->{bot_nick} = 'aleesa';
unless (defined $answer->{misc}->{csign}) {
$answer->{misc}->{csign} = $c->{csign};
}
unless (defined $answer->{misc}->{fwd_cnt}) {
$answer->{misc}->{fwd_cnt} = 1;
} else {
if ($answer->{misc}->{fwd_cnt} > $fwd_cnt) {
$log->error ('[ERROR] Forward loop detected, discarding message.');
$log->debug (Dumper $m);
return;
} else {
$answer->{misc}->{fwd_cnt}++;
}
}
unless (defined $answer->{misc}->{good_morning}) {
$answer->{misc}->{good_morning} = 0;
}
unless (defined $answer->{misc}->{msg_format}) {
$answer->{misc}->{msg_format} = 0;
}
unless (defined $answer->{misc}->{username}) {
$answer->{misc}->{username} = 'user';
}
} else {
$m->{misc}->{answer} = 1;
$m->{misc}->{csign} = '!';
$m->{misc}->{bot_nick} = 'aleesa';
$answer->{misc}->{answer} = 1;
$answer->{misc}->{bot_nick} = '';
$answer->{misc}->{csign} = $c->{csign};
$answer->{misc}->{fwd_cnt} = 1;
$answer->{misc}->{good_morning} = 0;
$answer->{misc}->{msg_format} = 0;
$answer->{misc}->{username} = 'user';
}
if (defined $m->{misc}->{bot_nick}) {
my $csign = $answer->{misc}->{csign};
$answer->{message} = '';
my $is_command = 0;
my $cmd = $m->{message};
if ($m->{message} =~ /^\Q$csign/u) {
$cmd = substr $m->{message}, length ($csign);
$is_command = 1;
}
if ($m->{misc}->{bot_nick} ne '') {
# Если нам сообщили ник бота, попробуем вымарать его из оригинального текста сообщения, в большинстве случаев он
# нам не интересн.
# нам не интересен.
$phrase =~ s/\s*\,?\s*($m->{misc}->{bot_nick})\s*\,?\s*/ /gui;
# TODO: проработать реакцию на "пьяный бред".
# В ключевых фразах можно обратиться с некоторыми командами к боту по нику, не используя символ команды
if ($m->{message} =~ /^\s*\Q;m->{misc}->{bot_nick}\E\s*;/ui) {
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+)pong\.?\s*$/i) {
} elsif ($m->{message} =~ /^(\Q$csign\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)ping\.?\s*$/i) {
$reply = 'Pong.';
} elsif ($m->{message} =~ /^(\Q$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) {
} elsif ($m->{message} =~ /^(\Q$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) {
} elsif ($m->{message} =~ /^(\Q$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) {
} elsif ($m->{message} =~ /^(\Q$csign\E|\Q$m->{misc}->{bot_nick}\E\s*\,?\s+)(хэлп|halp)\s*$/ui) {
$reply = 'HALP!!!11';
}
} else {
# Если нам не сообщили ник бота, то ищем команды, используя символ команды
if ($m->{message} =~ /^\Q$m->{misc}->{csign}\Epong\.?\s*$/i) {
if ($is_command) {
if ($cmd =~ /^ping\.?\s*$/ui) {
$reply = 'Pong.';
} elsif ($cmd =~ /^pong\.?\s*$/i) {
$reply = 'Wat?';
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\Eпин(г|х)\.?\s*$/ui) {
} elsif ($cmd=~ /^пин(г|х)\.?\s*$/ui) {
$reply = 'Понг.';
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\Eпон(г|х)\.?\s*$/ui) {
} elsif ($cmd =~ /^пон(г|х)\.?\s*$/ui) {
$reply = 'Шта?';
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(хэлп|halp)\s*$/ui) {
} elsif ($cmd=~ /^(хэлп|halp)\s*$/ui) {
$reply = 'HALP!!!11';
}
}
}
# Поищем в тексте входящего сообщения команды общего плана, с символом команды.
unless (defined $reply) {
if ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(kde|кде)\s*$/ui) {
if ($is_command) {
if ($cmd =~ /^(kde|кде)\s*$/ui) {
my @phrases = (
'Нет, я не буду поднимать вам плазму.',
'Повторяйте эту мантру по утрам не менее 5 раз: "Плазма не падает." И, возможно, она перестанет у вас падать.',
@@ -205,7 +252,8 @@ my $parse_message = sub {
);
$reply = $phrases[irand ($#phrases + 1)];
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(coin|монетка)$/ui) {
} elsif ($cmd =~ /^(coin|монетка)$/ui) {
# Нам тут говорили, что есть некая вероятность, что монета упадёт ребром... И даже цифры называли.
if (rand (101) < 0.016) {
$reply = 'ребро';
} else {
@@ -223,10 +271,13 @@ my $parse_message = sub {
}
}
}
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(roll|dice|кости')$/ui) {
} elsif ($cmd =~ /^(roll|dice|кости)$/ui) {
$reply = sprintf 'На первой кости выпало %d, а на второй — %d.', irand (6) + 1, irand (6) + 1;
} elsif ($m->{message} =~ /^\Q$m->{misc}->{csign}\E(ver|version|версия)$/ui) {
} elsif ($cmd =~ /^(ver|version|версия)$/ui) {
$reply = 'Версия: Четыре.Технологическое_превью';
} elsif ($cmd =~ /^(lat|лат)\s*$/ui) {
$reply = Lat ();
}
}
}
@@ -234,21 +285,34 @@ my $parse_message = sub {
# общения - публичного или приватного.
if (defined $reply) {
if ($m->{misc}->{answer}) {
$answer->{message} = $reply;
$self->json ($send_to)->notify (
$send_to => {
from => 'craniac',
userid => $userid,
chatid => $chatid,
plugin => $m->{plugin},
message => $reply
}
$send_to => $answer,
);
}
} else {
# Пока что нам нечего ответить, известных фраз не нашлось. Ищем дальше, на сей раз фразы, зависящие от режима
# общения - приватного ли публичного.
# общения - приватного или публичного.
if ($m->{mode} eq 'private') {
$phrase =~ s/^\s*(вот\s+)?скажи(\s*мне)?\s*//gui;
# Если указанной в unless-е строки во фразе нету, то можно поискать другие приватные фразы
unless ($phrase =~ s/^\s*(вот\s+)?скажи(\s*мне)?\s*//gui) {
if ($phrase =~ /покажи +(сиськи|сиси|титьки)\s*$/ui) {
$send_to = 'webapp';
$reply = '!boobs';
} elsif ($phrase =~ /show +(titts|tities|boobs)\s*$/i) {
$send_to = 'webapp';
$reply = '!boobs';
} elsif ($phrase =~ /покажи +(попку|попу) *$/ui) {
$send_to = 'webapp';
$reply = '!ass';
} elsif ($phrase =~ /show +(ass|butt|butty) *$/i) {
$send_to = 'webapp';
$reply = '!ass';
}
}
unless (defined $reply) {
$phrase =~ s/\s*(,)?\s*а\s*\?$/?/gui;
if ($phrase =~ /^\s*кто\s+все\s+эти\s+люди\s*\?\s*$/ui) {
@@ -258,17 +322,14 @@ my $parse_message = sub {
} elsif ($phrase =~ /^\s*кто\s+(здесь|тут)\s*\?\s*$/ui) {
$reply = 'Ты да я, да мы с тобой.';
}
}
# TODO: дописать запрос в webapp, для easter egg-ов, либо делать это на уровне aleesa-misc
if (defined ($reply) && $m->{misc}->{answer}) {
$answer->{message} = $reply;
$self->json ($send_to)->notify (
$send_to => {
from => 'craniac',
userid => $userid,
chatid => $chatid,
plugin => $m->{plugin},
message => $reply
}
$send_to => $answer,
);
return;
@@ -286,14 +347,10 @@ my $parse_message = sub {
}
if (defined ($reply) && $m->{misc}->{answer}) {
$answer->{message} = $reply;
$self->json ($send_to)->notify (
$send_to => {
from => 'craniac',
userid => $userid,
chatid => $chatid,
plugin => $m->{plugin},
message => $reply
}
$send_to => $answer,
);
return;
@@ -302,42 +359,51 @@ my $parse_message = sub {
# Попадаем сюда только если входящая фраза не распознана ранее как ключевая.
if ($m->{misc}->{answer}) {
# Пытаемся что-то генерировать, если фраза длиннее 3-х букв
# Пытаемся что-то генерировать, если фраза длиннее 3-х букв.
if (length ($m->{message}) > 3) {
my $telegram = 0;
# Костыль для совместимости со старым наименованием "мозгов".
if ($m->{plugin} eq 'telegram') {
$telegram = 1;
}
# Lazy brain init
if (brains ($chatid, 1)) {
if (brains ($chatid, $telegram)) {
$reply = $hailo->{$chatid}->learn_reply ($m->{message});
# Если из реактора вернулся undef выдаём "общую" фразу из словаря
# Если из реактора вернулся undef выдаём "общую" фразу из словаря.
unless (defined $reply) {
$reply = RandomCommonPhrase ();
} else {
# Если из реактора вернулась пустота выдаём "общую" фразу из словаря
# Если из реактора вернулась пустота выдаём "общую" фразу из словаря.
if ($reply eq '') {
$reply = RandomCommonPhrase ();
}
# Если сгенерированная фраза слишком похожа на начальную - выдаём "общую" фразу из словаря
# Если сгенерированная фраза слишком похожа на начальную - выдаём "общую" фразу из словаря.
if (fmatch (lc ($m->{message}), lc ($reply))) {
$reply = RandomCommonPhrase ();
}
}
$answer->{message} = $reply;
$self->json ($send_to)->notify (
$send_to => {
from => 'craniac',
userid => $userid,
chatid => $chatid,
plugin => $m->{plugin},
message => $reply
}
$send_to => $answer,
);
}
}
} else {
# пытаемся что-то запоминать, если фраза длиннее 3-х букв
# Пытаемся что-то запоминать, если фраза длиннее 3-х букв
if (length ($m->{message}) > 3) {
if (brains ($chatid, 1)) {
my $telegram = 0;
if ($m->{plugin} eq 'telegram') {
$telegram = 1;
}
if (brains ($chatid, $telegram)) {
$hailo->{$chatid}->learn ($m->{message});
}
}
@@ -347,12 +413,24 @@ my $parse_message = sub {
return;
};
my $__signal_handler = sub {
my ($self, $name) = @_;
$log->info ("[INFO] Caught a signal $name");
if (defined $main::pidfile && -e $main::pidfile) {
unlink $main::pidfile;
}
exit 0;
};
# 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}
sprintf 'redis://%s:%s/1', $c->{server}, $c->{port},
);
$log->info ('[INFO] Registering connection-event callback');
@@ -369,11 +447,11 @@ sub RunCraniac () {
my ($conn, $error) = @_;
$log->error ("[ERROR] Redis connection error: $error");
return;
}
},
);
return;
}
},
);
my $pubsub = $redis->pubsub;
@@ -384,10 +462,13 @@ sub RunCraniac () {
$log->debug ("[DEBUG] Subscribing to $channel");
$sub->{$channel} = $pubsub->json ($channel)->listen (
$channel => sub { $parse_message->(@_); }
$channel => sub { $parse_message->(@_); },
);
}
Mojo::IOLoop::Signal->on (TERM => $__signal_handler);
Mojo::IOLoop::Signal->on (INT => $__signal_handler);
do { Mojo::IOLoop->start } until Mojo::IOLoop->is_running;
return;
}

76
lib/Craniac/Lat.pm Normal file
View File

@@ -0,0 +1,76 @@
package Craniac::Lat;
# Крылатые латинские выражения спёрты с https://ru.wikipedia.org/wiki/Список_крылатых_латинских_выражений
use 5.018; ## no critic (ProhibitImplicitImport)
use strict;
use warnings;
use utf8;
use open qw (:std :utf8);
use English qw ( -no_match_vars );
use Carp qw (cluck);
use File::Basename qw (dirname);
use File::Path qw (make_path);
use Hailo ();
use Log::Any qw ($log);
use Conf qw (LoadConf);
use version; our $VERSION = qw (1.0);
use Exporter qw (import);
our @EXPORT_OK = qw (Train Lat);
# В нашем случае среднее количество слов в фразе равно 3, поэтому ставим 2 :)
my $brain_order = 2;
my $c = LoadConf ();
my $brain = $c->{lat}->{brain};
my $srcfile = $c->{lat}->{src};
sub Train () {
my $braindir = dirname $brain;
unless (-d $braindir) {
make_path ($braindir) or do {
cluck "Unable to create $braindir: $OS_ERROR";
exit 1;
};
}
my $lat = Hailo->new (
brain => $brain,
order => $brain_order,
);
$lat->train ($srcfile);
my ($tokens, $expressions) = ($lat->stats ())[0,1];
printf "Total tokens: %s\nTotal expressions: %s\n", $tokens, $expressions;
$lat->save ();
return;
}
# Просто вернём ответ
sub Lat () {
my $braindir = dirname $brain;
unless (-d $braindir) {
$log->error ("[ERROR] No lat module data: $braindir is absent! Train lat first.");
return '';
}
unless (-e $brain) {
$log->error ("[ERROR] No lat module data: $brain is absent! Train lat first.");
return '';
}
my $lat = Hailo->new (
brain => $brain,
order => $brain_order,
save_on_exit => 0,
);
return $lat->reply ();
}
1;
# vim: ft=perl noet ai ts=4 sw=4 sts=4: