Added initial bot module.
This commit is contained in:
Stefan Härter
285
TelegramBot.pm
Normal file
285
TelegramBot.pm
Normal file
@@ -0,0 +1,285 @@
|
||||
# Important for switch feature
|
||||
use v5.32;
|
||||
|
||||
use utf8;
|
||||
use strict;
|
||||
use warnings;
|
||||
|
||||
no warnings qw(experimental);
|
||||
|
||||
# core packages
|
||||
use Encode;
|
||||
use FindBin;
|
||||
|
||||
# CPAN packages
|
||||
use Cache::FastMmap;
|
||||
use JSON;
|
||||
use Log::Log4perl;
|
||||
use YAML;
|
||||
|
||||
# Package name
|
||||
package TelegramBot;
|
||||
|
||||
# Constants and initalisations
|
||||
Log::Log4perl->init("$FindBin::Bin/log.conf");
|
||||
my $CACHE = Cache::FastMmap->new(
|
||||
share_file => '/tmp/telegram_bot',
|
||||
serializer => 'json',
|
||||
unlink_on_exit => 0,
|
||||
expire_time => 86400,
|
||||
);
|
||||
|
||||
sub new {
|
||||
my ( $Type, %Param ) = @_;
|
||||
|
||||
# allocate new hash for object
|
||||
my $Self = {};
|
||||
bless( $Self, $Type );
|
||||
|
||||
$Self->{LogLevel} = $Param{LogLevel} || 'info';
|
||||
$Self->{LogObject} = Log::Log4perl->get_logger('TelegramBot');
|
||||
$Self->{Token} = 'bot5868933096:AAE8Oe-AxU6m_yCWfpqTqwwjERqnRpBGJtE';
|
||||
$Self->{URL} = {
|
||||
Telegram => 'https://api.telegram.org',
|
||||
};
|
||||
|
||||
# load remembered update ids
|
||||
$Self->{MessageIDs} = YAML::LoadFile("$FindBin::Bin/message_ids.yml");
|
||||
|
||||
# whitelist
|
||||
$Self->{Whitelist} = $Param{Whitelist};
|
||||
|
||||
return $Self;
|
||||
}
|
||||
|
||||
=head1 greet
|
||||
|
||||
Merely a dummy routine to test the bot's functionallity. Maybe using it for easter eggs or some kind of fun later.
|
||||
|
||||
=cut
|
||||
|
||||
sub greet {
|
||||
|
||||
my ( $Self, %Param ) = @_;
|
||||
|
||||
$Self->{LogObject}->info('greet: Initiating greet routine');
|
||||
|
||||
if ( !defined $Param{Message} ) {
|
||||
$Self->{LogObject}->error('greet: Message not defined!');
|
||||
return;
|
||||
}
|
||||
|
||||
my @Greetings =
|
||||
qw(Hallo Gruezi Hello Holá Bonjour Konichiwa Shalom Godmorgen);
|
||||
my $Greet = $Greetings[ int( rand(7) ) ] || '';
|
||||
$Self->{LogObject}->debug( 'Random Greet is ' . $Greet );
|
||||
my $ReturnContent =
|
||||
$Greet . ', '
|
||||
. ( $Param{Message}->{chat}->{first_name}
|
||||
? $Param{Message}->{chat}->{first_name}
|
||||
: $Param{Message}->{chat}->{username} )
|
||||
. '!';
|
||||
$Self->{LogObject}->debug( 'ReturnContent is ' . $ReturnContent );
|
||||
return { text => $ReturnContent, };
|
||||
}
|
||||
|
||||
=head1 build
|
||||
|
||||
Sub which is used to offer custom query building to the user by offering a decision tree of options via inline keyboard.
|
||||
|
||||
=cut
|
||||
|
||||
sub build {
|
||||
|
||||
my ( $Self, %Param ) = @_;
|
||||
use Data::Dumper;
|
||||
use JSON;
|
||||
use LWP::Simple::REST qw(POST plain);
|
||||
|
||||
my $KeyboardData;
|
||||
my $KeyboardMessage;
|
||||
if ( $Param{QueryStep} ) {
|
||||
if ( $Param{QueryStep} eq 'hist' ) {
|
||||
# show next selection
|
||||
}
|
||||
else {
|
||||
# for now, testing fallback
|
||||
my $ResponseResult = plain POST(
|
||||
join( '/', ( $Self->{URL}{Telegram}, $Self->{Token}, 'sendMessage' ) ),
|
||||
{
|
||||
'chat_id' => $Param{Message}->{callback_query}->{from}->{id},
|
||||
'reply_to_message_id' => $Param{Message}->{callback_query}->{message}->{message_id},
|
||||
'text' => "Kommando $Param{QueryStep} erkannt",
|
||||
}
|
||||
);
|
||||
$Self->{LogObject}->info( 'build: Sending result is ' . $ResponseResult );
|
||||
return {};
|
||||
}
|
||||
|
||||
}
|
||||
else {
|
||||
$KeyboardData = {
|
||||
'inline_keyboard' => [
|
||||
[
|
||||
{
|
||||
'text' => 'Grüßen',
|
||||
'callback_data' => 'greet'
|
||||
},
|
||||
{
|
||||
'text' => 'Statistik',
|
||||
'callback_data' => 'statistics'
|
||||
},
|
||||
],
|
||||
],
|
||||
'resize' => 1,
|
||||
'single_use' => 1,
|
||||
'placeholder' => 'test',
|
||||
};
|
||||
$KeyboardMessage = "Hallo $Param{Message}->{chat}->{first_name}, über die folgenden Fragen kannst du auswählen, welche Interaktion du ausführen möchtest. Was möchtest du tun?";
|
||||
}
|
||||
my $EncodedKeyboard = JSON::encode_json(
|
||||
$KeyboardData,
|
||||
);
|
||||
my $ResponseResult = plain POST(
|
||||
join( '/', ( $Self->{URL}{Telegram}, $Self->{Token}, 'sendMessage' ) ),
|
||||
{
|
||||
'chat_id' => $Param{Message}->{chat}->{id},
|
||||
'reply_to_message_id' => $Param{Message}->{id},
|
||||
'text' => $KeyboardMessage,
|
||||
'reply_markup' => $EncodedKeyboard,
|
||||
}
|
||||
);
|
||||
$Self->{LogObject}->info( 'build: Sending result is ' . $ResponseResult );
|
||||
|
||||
return {};
|
||||
|
||||
}
|
||||
|
||||
=head1 processMessage
|
||||
|
||||
Function which receives a single message and decides what to to based on message content and attributes.
|
||||
|
||||
=cut
|
||||
|
||||
sub processMessage {
|
||||
|
||||
my ( $Self, %Param ) = @_;
|
||||
use Data::Dumper;
|
||||
use LWP::Simple::REST qw(POST plain json);
|
||||
|
||||
if ( !defined $Param{Message} ) {
|
||||
$Self->{LogObject}->error('processMessage: Message not defined!');
|
||||
return;
|
||||
}
|
||||
|
||||
$Self->{LogObject}->info('processMessage: Starting to process message');
|
||||
$Self->{LogObject}
|
||||
->info( 'processMessage: Message is ' . Dumper( $Param{Message} ) );
|
||||
|
||||
|
||||
my $ResponseData = {};
|
||||
if ( defined $Param{Message}->{message} && $Param{Message}->{message}->{text} =~
|
||||
/\/(?<command>greet|statistics|build)\s?(?<arguments>.*)?/ )
|
||||
{
|
||||
my $Message = $Param{Message}->{message};
|
||||
my $Command = $+{command};
|
||||
my $ArgumentsString = $+{arguments};
|
||||
$ResponseData = $Self->$Command(
|
||||
Message => $Message,
|
||||
Arguments => $ArgumentsString,
|
||||
);
|
||||
if ( !keys $ResponseData->%* ) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
elsif ( defined $Param{Message}->{callback_query} ) {
|
||||
$Self->build(
|
||||
Message => $Param{Message},
|
||||
QueryStep => $Param{Message}->{callback_query}->{data},
|
||||
);
|
||||
}
|
||||
else {
|
||||
$Self->{LogObject}
|
||||
->debug( 'Command not recognized. Data: ' . $Param{Message}->{message}->{text} );
|
||||
if ( $Self->{Whitelist}->{$Param{Message}->{message}->{from}->{id}} eq 'Sarah'
|
||||
|| $Self->{Whitelist}->{$Param{Message}->{message}->{from}->{id}} eq 'Stefan' ) {
|
||||
$ResponseData = $Self->replyLoveQuote( Message => $Param{Message}->{message} );
|
||||
}
|
||||
else {
|
||||
$ResponseData->{text} =
|
||||
"I'm sorry, "
|
||||
. ( $Param{Message}->{message}->{chat}->{first_name}
|
||||
? $Param{Message}->{message}->{chat}->{first_name}
|
||||
: $Param{Message}->{message}->{chat}->{username} )
|
||||
. ", I couldn't understand your request. Currently I can process the commands:\n\n\t\/greet\n\t\/statistics driver standings\n\t\/statistics constructor standings";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
my $ResponseResult = plain POST(
|
||||
join( '/', ( $Self->{URL}{Telegram}, $Self->{Token}, 'sendMessage' ) ),
|
||||
{
|
||||
chat_id => $Param{Message}->{message}->{chat}->{id},
|
||||
$ResponseData->%*,
|
||||
}
|
||||
);
|
||||
my $Response = JSON::decode_json($ResponseResult);
|
||||
$Self->{LogObject}
|
||||
->info( 'processMessage: Answering result is ' . Dumper($Response) );
|
||||
|
||||
# mark message as read
|
||||
my $SeenResult = plain POST(
|
||||
join( '/', ( $Self->{URL}{Telegram}, $Self->{Token}, 'readMessageContents' ) ),
|
||||
{
|
||||
id => $Param{Message}->{message}->{id},
|
||||
}
|
||||
);
|
||||
|
||||
}
|
||||
|
||||
=head1 fetchMessages
|
||||
|
||||
Requesting messages from Telegram API and passing them one by one to processMessage.
|
||||
|
||||
=cut
|
||||
|
||||
sub fetchMessages {
|
||||
|
||||
my ( $Self, %Param ) = @_;
|
||||
use Data::Dumper;
|
||||
use LWP::Simple::REST qw(GET json);
|
||||
|
||||
my $Method = 'getUpdates';
|
||||
$Self->{LogObject}->info('fetchMessages: Initiating getUpdates');
|
||||
|
||||
my $MessageDataRaw =
|
||||
json GET( join( '/', ( $Self->{URL}{Telegram}, $Self->{Token}, $Method ) ), {} );
|
||||
$Self->{LogObject}
|
||||
->info( 'fetchMessages: Messages raw are ' . Dumper($MessageDataRaw) );
|
||||
my @Messages = $MessageDataRaw->{result}->@*;
|
||||
$Self->{LogObject}
|
||||
->info( 'fetchMessages: Messages returned are ' . Dumper( \@Messages ) );
|
||||
|
||||
MESSAGE:
|
||||
for my $Message (@Messages) {
|
||||
if ( !$Self->{Whitelist}{$Message->{message}{from}{id}} ) {
|
||||
$Self->{LogObject}
|
||||
->info( 'fetchMessages: User not whitelisted, skipping message ' . Dumper($MessageDataRaw) );
|
||||
|
||||
$Self->{MessageIDs}{$Message->{message}{message_id}} = 1;
|
||||
next MESSAGE;
|
||||
}
|
||||
if ($Self->{MessageIDs}{$Message->{message}{message_id}}) {
|
||||
$Self->{LogObject}->info('fetchMessages: Skipping known message_id');
|
||||
next MESSAGE;
|
||||
}
|
||||
else {
|
||||
$Self->{LogObject}->info('fetchMessages: Calling processMessage');
|
||||
$Self->{MessageIDs}{$Message->{message}{message_id}} = 1;
|
||||
$Self->processMessage( Message => $Message, );
|
||||
}
|
||||
}
|
||||
YAML::DumpFile("$FindBin::Bin/message_ids.yml", $Self->{MessageIDs});
|
||||
}
|
||||
|
||||
1;
|
||||
Reference in New Issue
Block a user