158 lines
3.6 KiB
Perl
158 lines
3.6 KiB
Perl
use HexChat qw(:all);
|
|
use threads;
|
|
use threads::shared;
|
|
|
|
use strict;
|
|
use warnings "all";
|
|
|
|
use URI::URL;
|
|
use Image::Magick;
|
|
|
|
sub dlfunc($); # thread, that checks and downloads stuff
|
|
sub is_picture($);
|
|
sub urlencode($);
|
|
sub hookfn;
|
|
sub freehooks;
|
|
|
|
my $wgetpath; share($wgetpath);
|
|
|
|
my $script_name = "Image URL Auto Grabber and Downloader, wget flavour";
|
|
HexChat::register($script_name, '0.8.6', 'Automatically grabs and downloads image URLs via wget', \&freehooks);
|
|
|
|
HexChat::print("$script_name loaded\n");
|
|
my @hooks;
|
|
push @hooks, HexChat::hook_print('Channel Message', \&hookfn);
|
|
push @hooks, HexChat::hook_print('Channel Msg Hilight', \&hookfn);
|
|
push @hooks, HexChat::hook_print('Channel Action', \&hookfn);
|
|
push @hooks, HexChat::hook_print('Channel Action Hilight', \&hookfn);
|
|
|
|
my $active = 0;
|
|
share($active);
|
|
|
|
sub hookfn {
|
|
my ($nick, $text, $modechar) = @{$_[0]};
|
|
my @words = split(/\s+/, $text);
|
|
|
|
foreach (@words) {
|
|
my $str = $_;
|
|
next unless(substr($str, 0, 4) eq 'http');
|
|
|
|
if ($str =~ m{https?://([a-zA-Z0-9.-]+\.[a-zA-Z]+)/(?:.*)}) {
|
|
$active = 1;
|
|
my $t = undef;
|
|
|
|
do {
|
|
$t = threads->create('dlfunc', $str);
|
|
sleep 1 unless(defined($t));
|
|
} unless (defined($t));
|
|
|
|
$t->detach;
|
|
undef $t;
|
|
|
|
if ($active == 1) {
|
|
HexChat::hook_timer( 100, sub { return REMOVE if ($active == 0); return KEEP; } );
|
|
}
|
|
}
|
|
}
|
|
|
|
@words = -1; undef @words;
|
|
undef $nick; undef $text; undef $modechar;
|
|
|
|
return HexChat::EAT_NONE;
|
|
}
|
|
|
|
sub dlfunc($) {
|
|
my $url = shift;
|
|
|
|
if (-f "/bin/wget") {
|
|
$wgetpath = '/bin/wget';
|
|
} elsif (-f "/usr/bin/wget") {
|
|
$wgetpath = '/usr/bin/wget';
|
|
} elsif (-f "/usr/local/bin/wget") {
|
|
$wgetpath = '/usr/local/bin/wget';
|
|
}
|
|
|
|
my $extension = is_picture($url);
|
|
|
|
if (defined($extension)) {
|
|
my $savepath = sprintf("%s/imgsave", HexChat::get_info("configdir"));
|
|
mkdir ($savepath) unless (-d $savepath);
|
|
$savepath = $savepath . "/" . s/[^\w!., -#]/_/gr . ".$extension";
|
|
|
|
if ( (lc($url) =~ /\.(gif|jpeg|png|webm|mp4)$/) and ($1 eq $extension) ){
|
|
$savepath = HexChat::get_info("configdir") . "/imgsave/" . s/[^\w!., -#]/_/gr;
|
|
}
|
|
|
|
$url = urlencode($url);
|
|
system($wgetpath, '--no-check-certificate', '-q', '-T', '20', '-O', $savepath, '-o', '/dev/null', $url);
|
|
|
|
if ($savepath =~ /(png|jpe?g|gif)$/i){
|
|
my $im = Image::Magick->new();
|
|
my $rename = 1;
|
|
my (undef, undef, undef, $format) = $im->Ping($savepath);
|
|
|
|
if (defined($format)) {
|
|
$rename = 0 if (($format eq 'JPEG') and ($savepath =~ /jpe?g$/i));
|
|
$rename = 0 if (($format eq 'GIF') and ($savepath =~ /gif$/i));
|
|
$rename = 0 if (($format =~ /^PNG/) and ($savepath =~ /png$/i));
|
|
rename $savepath, sprintf("%s.%s", $savepath, lc($format)) if ($rename == 1);
|
|
}
|
|
|
|
undef $im; undef $rename;
|
|
}
|
|
|
|
undef $savepath;
|
|
}
|
|
|
|
$active = 0;
|
|
undef $url; undef $extension;
|
|
}
|
|
|
|
sub is_picture($) {
|
|
my $url = shift;
|
|
$url = urlencode($url);
|
|
my $r = undef;
|
|
|
|
$r = `$wgetpath --no-check-certificate -q --method=HEAD -S --timeout=5 "$url" 2>&1`;
|
|
undef $url;
|
|
|
|
if ($? == 0) {
|
|
foreach (split(/\n/, $r)) {
|
|
next unless($_ =~ /Content\-Type: (.+)/);
|
|
$r = ($1); chomp($r);
|
|
last;
|
|
}
|
|
|
|
if (defined($r)) {
|
|
if ($r =~ /^image\/gif/) { $r = 'gif'; }
|
|
elsif ($r =~ /^image\/jpe?g/){ $r = 'jpeg';}
|
|
elsif ($r =~ /^image\/png/) { $r = 'png'; }
|
|
elsif ($r =~ /^video\/webm/) { $r = 'webm';}
|
|
elsif ($r =~ /^video\/mp4/) { $r = 'mp4'; }
|
|
else { $r = undef; }
|
|
} else {
|
|
$r = undef;
|
|
}
|
|
} else {
|
|
$r = undef;
|
|
}
|
|
|
|
return $r;
|
|
}
|
|
|
|
sub urlencode($) {
|
|
my $url = shift;
|
|
my $urlobj = url $url;
|
|
$url = $urlobj->as_string;
|
|
undef $urlobj;
|
|
return $url;
|
|
}
|
|
|
|
sub freehooks {
|
|
foreach (@hooks) {
|
|
HexChat::unhook($_);
|
|
}
|
|
|
|
return HexChat::EAT_ALL;
|
|
}
|