Use sha256_hex of plugin name and chatid as filename for brains in commin case

This commit is contained in:
eleksir
2022-05-31 02:54:13 +03:00
parent ca982571d9
commit e6d8e4c1a2
2 changed files with 22 additions and 43 deletions

View File

@@ -5,7 +5,6 @@ requires 'Cwd', '==3.75';
requires 'Cwd::utf8', '==0.011'; requires 'Cwd::utf8', '==0.011';
requires 'Data::Dumper', '==2.183'; requires 'Data::Dumper', '==2.183';
requires 'Digest::SHA', '==6.02'; requires 'Digest::SHA', '==6.02';
requires 'Encode', '==3.16';
# Part of perl # Part of perl
requires 'English', '0'; requires 'English', '0';
requires 'File::Basename', '==2.85'; requires 'File::Basename', '==2.85';

View File

@@ -19,8 +19,7 @@ use English qw ( -no_match_vars );
# Модули для работы приложения # Модули для работы приложения
use Clone qw (clone); use Clone qw (clone);
use Data::Dumper qw (Dumper); use Data::Dumper qw (Dumper);
use Digest::SHA qw (sha1_base64); use Digest::SHA qw (sha256_hex);
use Encode qw (encode_utf8);
use File::Spec (); use File::Spec ();
use Hailo (); use Hailo ();
use Log::Any qw ($log); use Log::Any qw ($log);
@@ -40,7 +39,6 @@ use Exporter qw (import);
our @EXPORT_OK = qw (RunCraniac); our @EXPORT_OK = qw (RunCraniac);
sub RandomCommonPhrase (); sub RandomCommonPhrase ();
sub utf2sha1 ($);
sub fmatch (@); sub fmatch (@);
sub brains (@); sub brains (@);
sub RunCraniac (); sub RunCraniac ();
@@ -69,17 +67,6 @@ sub RandomCommonPhrase () {
return $myphrase[irand ($#myphrase + 1)]; 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 (@) { sub fmatch (@) {
my $srcphrase = shift; my $srcphrase = shift;
my $answer = shift; my $answer = shift;
@@ -97,30 +84,27 @@ sub fmatch (@) {
# Ленивая инициализация мозга. Создаём/открываем мозг только если нам надо "подумать" а не на каждую входящую фразу. # Ленивая инициализация мозга. Создаём/открываем мозг только если нам надо "подумать" а не на каждую входящую фразу.
sub brains (@) { sub brains (@) {
my $chatid = shift; my $plugin = shift;
my $telegram = shift // 0; my $cid = shift;
unless (defined $hailo->{$chatid}) { unless (defined $hailo->{$cid}) {
my $cid = $chatid;
my $brainname; my $brainname;
# Костыль, чтобы не портировать данные из телеграммных мозгов и не заниматься переименованием # Костыль, чтобы не портировать данные из телеграммных мозгов и не заниматься переименованием
unless ($telegram) { if ($plugin eq '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; $brainname = sprintf '%s/%s.brain.sqlite', $c->{braindir}, $cid;
} else {
$brainname = sprintf '%s/%s.sqlite', $c->{braindir}, $cid;
} }
$hailo->{$chatid} = eval { $hailo->{$cid} = eval {
Hailo->new ( Hailo->new (
brain => $brainname, brain => $brainname,
order => 3, order => 3,
); );
}; };
if (defined $hailo->{$chatid}) { if (defined $hailo->{$cid}) {
$log->info (sprintf '[INFO] Lazy init brain: %s', $brainname); $log->info (sprintf '[INFO] Lazy init brain: %s', $brainname);
} else { } else {
$log->error ("[ERROR] $EVAL_ERROR"); $log->error ("[ERROR] $EVAL_ERROR");
@@ -358,19 +342,20 @@ my $parse_message = sub {
} }
# Попадаем сюда только если входящая фраза не распознана ранее как ключевая. # Попадаем сюда только если входящая фраза не распознана ранее как ключевая.
my $cid;
if ($m->{plugin} eq 'telegram') {
$cid = $chatid;
} else {
$cid = sha256_hex (sprintf '%s/%s', $m->{plugin}, $chatid);
}
if ($m->{misc}->{answer}) { if ($m->{misc}->{answer}) {
# Пытаемся что-то генерировать, если фраза длиннее 3-х букв. # Пытаемся что-то генерировать, если фраза длиннее 3-х букв.
if (length ($m->{message}) > 3) { if (length ($m->{message}) > 3) {
my $telegram = 0;
# Костыль для совместимости со старым наименованием "мозгов".
if ($m->{plugin} eq 'telegram') {
$telegram = 1;
}
# Lazy brain init # Lazy brain init
if (brains ($chatid, $telegram)) { if (brains ($m->{plugin}, $cid)) {
$reply = $hailo->{$chatid}->learn_reply ($m->{message}); $reply = $hailo->{$cid}->learn_reply ($m->{message});
# Если из реактора вернулся undef выдаём "общую" фразу из словаря. # Если из реактора вернулся undef выдаём "общую" фразу из словаря.
unless (defined $reply) { unless (defined $reply) {
@@ -397,14 +382,9 @@ my $parse_message = sub {
} else { } else {
# Пытаемся что-то запоминать, если фраза длиннее 3-х букв # Пытаемся что-то запоминать, если фраза длиннее 3-х букв
if (length ($m->{message}) > 3) { if (length ($m->{message}) > 3) {
my $telegram = 0; # Lazy brain init
if (brains ($m->{plugin}, $cid)) {
if ($m->{plugin} eq 'telegram') { $hailo->{$cid}->learn ($m->{message});
$telegram = 1;
}
if (brains ($chatid, $telegram)) {
$hailo->{$chatid}->learn ($m->{message});
} }
} }
} }