From 16cd42e6fbfbf6fff06fd50926ac374b771e3f97 Mon Sep 17 00:00:00 2001 From: Guillermo Ramos Date: Tue, 25 Jun 2019 12:19:59 +0200 Subject: Move Telegram comm logic to TgLib::Api --- tgrecv | 52 +++++++++++++++++++--------------------------------- 1 file changed, 19 insertions(+), 33 deletions(-) (limited to 'tgrecv') diff --git a/tgrecv b/tgrecv index 4578428..01b2d9e 100755 --- a/tgrecv +++ b/tgrecv @@ -7,23 +7,20 @@ # Dependencies: HTTP-Message, JSON ################################################################################ -$main::VERSION = "0.1.0"; +$main::VERSION = "0.1.1"; use Getopt::Long qw(:config auto_version); use Pod::Usage qw; -use LWP::UserAgent; use Data::Dumper; - -use HTTP::Request; -use JSON qw; +use JSON qw; # Local modules use FindBin; use lib "$FindBin::Bin/lib"; use TgLib qw; -use TgLib::Env qw<$HOME $CONFIG_HOME $CACHE_HOME>; require TgLib::Cache; require TgLib::Logger; +require TgLib::Api; my $TOKEN; my $OUTPUT; @@ -44,7 +41,7 @@ pod2usage(-verbose => $VERBOSE+1) if $HELP; my $logger = TgLib::Logger->new($VERBOSE); -# If token was not specified in CLI, get it from ENV/file +# Fetch token: CLI || env || file $TOKEN ||= fetch_token() or pod2usage(-message => "ERROR: Unable to get bot token ($!).\n", -verbose => 99, -sections => "AUTHENTICATION"); @@ -57,34 +54,23 @@ if ($AUTO_OFFSET && ! $OFFSET) { $OFFSET |= $cache->offset; } -my $ua = LWP::UserAgent->new; -my $uri = "https://api.telegram.org/bot$TOKEN/getUpdates?timeout=$TIMEOUT"; -$uri = $uri . "&offset=$OFFSET" if $OFFSET; -my $req = HTTP::Request->new("GET", $uri); -$logger->debug("Request: " . Dumper($req)); - -my $resp = $ua->request($req); -$logger->debug("Response: " . Dumper($resp)); -if ($resp->is_error()) { - die $resp->message; -} else { - my $out = STDOUT; - if ($OUTPUT) { - open($out, ">", $OUTPUT) or die "Cannot open $OUTPUT for writing: $!"; - } - my $results = decode_json($resp->content)->{'result'}; - - # Store new offset in cache - if ($AUTO_OFFSET && @$results) { - # Update cache: set offset to last update id +1 - $cache->offset($results->[-1]{'update_id'}+1); - $logger->debug(sprintf "Cached offset %s (--auto-offset)\n", $cache->offset); - $cache->save; - } - - print $out encode_json($results); +my $api = TgLib::Api->new($TOKEN, $logger); +my $updates = $api->get_updates($TIMEOUT, $OFFSET); + +my $out = STDOUT; +if ($OUTPUT) { + open($out, ">", $OUTPUT) or die "Cannot open $OUTPUT for writing: $!"; } +# Store new offset in cache (last update id +1) +if ($AUTO_OFFSET && @$updates) { + $cache->offset($updates->[-1]{'update_id'}+1); + $logger->debug(sprintf "Cached offset %s (--auto-offset)\n", $cache->offset); + $cache->save; +} + +print $out encode_json($updates); + __END__ -- cgit v1.2.3