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;
}