* move plugins out of botlib.pm * use make_path instead of mkpath * squash some bugs * update .gitignore * use more correct randomizer, rather than perl own rand() * coding style fixes
184 lines
4.0 KiB
Perl
184 lines
4.0 KiB
Perl
package fortune;
|
|
# Any fortune_mod sources are suitable. Text file with "\n%\n" sentence delimeter.
|
|
|
|
use 5.018;
|
|
use strict;
|
|
use warnings;
|
|
use utf8;
|
|
use open qw (:std :utf8);
|
|
use English qw ( -no_match_vars );
|
|
use Carp qw (cluck croak);
|
|
use File::Path qw (make_path);
|
|
use Math::Random::Secure qw (irand);
|
|
use SQLite_File;
|
|
use conf qw (loadConf);
|
|
|
|
use version; our $VERSION = qw (1.0);
|
|
use Exporter qw(import);
|
|
our @EXPORT_OK = qw(seed fortune fortune_toggle fortune_toggle_list fortune_status);
|
|
|
|
my $c = loadConf();
|
|
my $dir = $c->{fortune}->{dir};
|
|
my $srcdir = $c->{fortune}->{srcdir};
|
|
|
|
sub seed () {
|
|
unless (-d $dir) {
|
|
make_path ($dir) || croak "Unable to create $dir: $OS_ERROR";
|
|
}
|
|
|
|
my $backingfile = sprintf '%s/fortune.sqlite', $dir;
|
|
|
|
if (-f $backingfile) {
|
|
unlink $backingfile || croak "Unable to remove $backingfile: $OS_ERROR";
|
|
}
|
|
|
|
tie my @fortune, 'SQLite_File', $backingfile || croak "Unable to tie to $backingfile: $OS_ERROR";
|
|
|
|
opendir (my $srcdirhandle, $srcdir) || croak "Unable to open $srcdir: $OS_ERROR";
|
|
|
|
while (my $fortunefile = readdir ($srcdirhandle)) {
|
|
my $srcfile = sprintf '%s/%s', $srcdir, $fortunefile;
|
|
|
|
unless (-f $srcfile) {
|
|
next;
|
|
}
|
|
|
|
if ($fortunefile =~ m/\./) {
|
|
next;
|
|
}
|
|
|
|
open (my $fh, '<', $srcfile) || croak "Unable to open $srcfile, $OS_ERROR";
|
|
# set correct phrase delimiter
|
|
local $INPUT_RECORD_SEPARATOR = "\n%\n";
|
|
|
|
while (readline $fh) {
|
|
my $phrase = substr $_, 0, -3;
|
|
push @fortune, $phrase;
|
|
}
|
|
|
|
close $fh; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
|
|
}
|
|
|
|
closedir $srcdirhandle;
|
|
untie @fortune;
|
|
return;
|
|
}
|
|
|
|
# just return answer
|
|
sub fortune () {
|
|
my $backingfile = sprintf '%s/fortune.sqlite', $dir;
|
|
|
|
tie my @array, 'SQLite_File', $backingfile || do {
|
|
cluck "[ERROR] Unable to tie to $backingfile: $OS_ERROR";
|
|
return '';
|
|
};
|
|
|
|
my $phrase = $array[irand ($#array + 1)];
|
|
# decode?
|
|
utf8::decode $phrase;
|
|
untie @array;
|
|
return $phrase;
|
|
}
|
|
|
|
sub fortune_toggle (@) {
|
|
my $chatid = shift;
|
|
my $action = shift // undef;
|
|
my $phrase = 'Фортунка с утра ';
|
|
|
|
unless (-d $dir) {
|
|
make_path ($dir) || do {
|
|
cluck "[ERROR] Unable to create $dir: $OS_ERROR";
|
|
return;
|
|
};
|
|
}
|
|
|
|
my $backingfile = sprintf '%s/chats.sqlite', $dir;
|
|
|
|
tie my %toggle, 'SQLite_File', $backingfile || do {
|
|
cluck "[ERROR] Unable to tie to $backingfile: $OS_ERROR";
|
|
return;
|
|
};
|
|
|
|
my $state = $toggle{$chatid};
|
|
|
|
if (defined $action) {
|
|
if ($action) {
|
|
$toggle{$chatid} = 1;
|
|
$phrase .= 'будет показываться.';
|
|
} else {
|
|
if (defined $state) {
|
|
delete $toggle{$chatid};
|
|
}
|
|
|
|
$phrase .= 'не будет показываться.';
|
|
}
|
|
} else {
|
|
if (defined $state && $state) {
|
|
delete $toggle{$chatid};
|
|
$phrase .= 'не будет показываться.';
|
|
} else {
|
|
$toggle{$chatid} = 1;
|
|
$phrase .= 'будет показываться.';
|
|
}
|
|
}
|
|
|
|
untie %toggle;
|
|
return $phrase;
|
|
}
|
|
|
|
sub fortune_status ($) {
|
|
my $chatid = shift;
|
|
my $phrase = 'Фортунка с утра ';
|
|
|
|
unless (-d $dir) {
|
|
make_path ($dir) || do {
|
|
cluck "[ERROR] Unable to create $dir: $OS_ERROR";
|
|
return;
|
|
};
|
|
}
|
|
|
|
my $backingfile = sprintf '%s/chats.sqlite', $dir;
|
|
|
|
tie my %toggle, 'SQLite_File', $backingfile || do {
|
|
cluck "[ERROR] Unable to tie to $backingfile: $OS_ERROR\n";
|
|
return;
|
|
};
|
|
|
|
my $state = $toggle{$chatid};
|
|
|
|
if (defined $state && $state) {
|
|
$phrase .= 'показывается.';
|
|
} else {
|
|
$phrase .= 'не показывается.';
|
|
}
|
|
|
|
untie %toggle;
|
|
return $phrase;
|
|
}
|
|
|
|
sub fortune_toggle_list () {
|
|
unless (-d $dir) {
|
|
make_path ($dir) || do {
|
|
cluck "[ERROR] Unable to create $dir: $OS_ERROR";
|
|
my @empty = ();
|
|
return @empty;
|
|
}
|
|
}
|
|
|
|
my $backingfile = sprintf '%s/chats.sqlite', $dir;
|
|
|
|
tie my %toggle, 'SQLite_File', $backingfile || do {
|
|
cluck "[ERROR] Unable to tie to $backingfile: $OS_ERROR";
|
|
my @empty = ();
|
|
return @empty;
|
|
};
|
|
|
|
my @list = keys %toggle;
|
|
untie %toggle;
|
|
return @list;
|
|
}
|
|
|
|
1;
|
|
|
|
# vim: ft=perl noet ai ts=4 sw=4 sts=4:
|