mirror of
https://github.com/pragma-/pbot.git
synced 2024-12-28 21:52:35 +01:00
59 lines
1.7 KiB
Perl
59 lines
1.7 KiB
Perl
# File: LWPUserAgentCached.pm
|
|
#
|
|
# Purpose: variant of LWP::UserAgent::WithCache. Instead of depending on
|
|
# the 'expires' or 'Last-Modified' attributes, we always cache for the
|
|
# specified duration.
|
|
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package PBot::Utils::LWPUserAgentCached;
|
|
|
|
use PBot::Imports;
|
|
|
|
use base qw/LWP::UserAgent/;
|
|
use Cache::FileCache;
|
|
use File::HomeDir;
|
|
use File::Spec;
|
|
|
|
our %default_cache_args = (
|
|
'namespace' => 'pbot-cached',
|
|
'cache_root' => File::Spec->catfile(File::HomeDir->my_home, '.cache'),
|
|
'default_expires_in' => 600
|
|
);
|
|
|
|
sub new {
|
|
my $class = shift;
|
|
my $cache_opt;
|
|
my %lwp_opt;
|
|
unless (scalar @_ % 2) {
|
|
%lwp_opt = @_;
|
|
$cache_opt = {};
|
|
for my $key (qw(namespace cache_root default_expires_in)) { $cache_opt->{$key} = delete $lwp_opt{$key} if exists $lwp_opt{$key}; }
|
|
} else {
|
|
$cache_opt = shift || {};
|
|
%lwp_opt = @_;
|
|
}
|
|
my $self = $class->SUPER::new(%lwp_opt);
|
|
my %cache_args = (%default_cache_args, %$cache_opt);
|
|
$self->{cache} = Cache::FileCache->new(\%cache_args);
|
|
return $self;
|
|
}
|
|
|
|
sub request {
|
|
my ($self, @args) = @_;
|
|
my $request = $args[0];
|
|
return $self->SUPER::request(@args) if $request->method ne 'GET';
|
|
|
|
my $uri = $request->uri->as_string;
|
|
my $cached = $self->{cache}->get($uri);
|
|
return HTTP::Response->parse($cached) if defined $cached;
|
|
|
|
my $res = $self->SUPER::request(@args);
|
|
$self->{cache}->set($uri, $res->as_string) if $res->code eq HTTP::Status::RC_OK;
|
|
return $res;
|
|
}
|
|
|
|
1;
|