WIP Adding paste and image download modules

* Adding link shortener/paste module * Adding image downloder * Added TODO/Wish list * add Exporter to modules
2019-09-03 22:43:05 +03:00
parent 5b17cfaa74
commit 183dee668d
8 changed files with 212 additions and 16 deletions
--- a/lib/image_dl.pm
+++ b/lib/image_dl.pm
@@ -0,0 +1,135 @@
+package image_dl;
+
+use strict;
+use warnings "all";
+use HTTP::Tiny;
+use Net::SSLeay;
+use IO::Socket::SSL;
+use Image::Magick;
+
+use Exporter qw(import);
+use vars qw/$VERSION/;
+$VERSION = "1.0";
+our @EXPORT = qw(queue_image dl_thread);
+
+
+
+sub queue_image (@) {
+	my $url = shift;
+	my $images = shift;
+	push @{$images}, $url;
+	return('200', 'text/plain', "Download queued\n");
+}
+
+sub __urlencode($) {
+	my $url = shift;
+	my $urlobj = url $url;
+	$url = $urlobj->as_string;
+	undef $urlobj;
+	return $url;
+}
+
+sub __is_picture($) {
+	my $url = shift;
+	$url = __urlencode($url);
+	my $r = undef;
+
+	eval {
+		my $http = HTTP::Tiny->new();
+		$r = $http->request('HEAD', $url);
+		undef $http;
+	};
+
+	if ($r->{'success'} and defined($r->{'headers'}->{'content-type'})) {
+		if    ($r->{'headers'}->{'content-type'} =~ /^image\/gif/)  { $r = 'gif'; }
+		elsif ($r->{'headers'}->{'content-type'} =~ /^image\/jpe?g/){ $r = 'jpeg';}
+		elsif ($r->{'headers'}->{'content-type'} =~ /^image\/png/)  { $r = 'png'; }
+		elsif ($r->{'headers'}->{'content-type'} =~ /^video\/webm/) { $r = 'webm';}
+		elsif ($r->{'headers'}->{'content-type'} =~ /^video\/mp4/)  { $r = 'mp4'; }
+		else  { $r = undef; }
+	} else {
+		$r = undef;
+	}
+
+	undef $url;
+	return $r;
+}
+
+sub __dlfunc(@) {
+	my $url = shift;
+	my $file = shift;
+	$url = __urlencode($url);
+
+	eval {
+		my $http1 = HTTP::Tiny->new();
+		$http1->mirror($url, $file);
+		undef $http1;
+	};
+
+	eval {
+		if ($file =~ /(png|jpe?g|gif)$/i) {
+			my $im = Image::Magick->new();
+			my $rename = 1;
+			my (undef, undef, undef, $format) = $im->Ping($file);
+
+			if (defined($format)) {
+				$rename = 0 if (($format eq 'JPEG') and ($file =~ /jpe?g$/i));
+				$rename = 0 if (($format eq 'GIF') and ($file =~ /gif$/i));
+				$rename = 0 if (($format =~ /^PNG/) and ($file =~ /png$/i));
+				rename $file, sprintf("%s.%s", $file, lc($format)) if ($rename == 1);
+			}
+
+			undef $im;
+			undef $rename;
+			undef $format;
+		}
+	}
+
+	#undef $url;
+	#undef $file;
+	#return;
+}
+
+
+sub dl_thread (@) {
+	my $dir = shift;
+	my $sleep = shift;
+	threads->detach();
+
+	while ( 1 ) {
+		if (@{main::images} > 0) {
+			my @pic = @{main::images};
+			@{main::images} = -1;
+
+			foreach my $url (@pic) {
+				my $extension = __is_picture($url);
+
+				if (defined($extension)) {
+					my $savepath = $dir;
+					mkdir ($savepath) unless (-d $savepath);
+					my $fname = $url;
+					$fname =~ s/[^\w!., -#]/_/g;
+					$savepath = sprintf("%s/%s.%s", $savepath, $fname, $extension);
+
+					if ( (lc($url) =~ /\.(gif|jpe?g|png|webm|mp4)$/) and ($1 eq $extension) ) {
+						$savepath = $dir . "/" . $fname;
+					}
+
+					__dlfunc($url, $savepath);
+					undef $savepath;
+					undef $fname;
+				}
+
+				undef $url;
+				undef $extension;
+			}
+			@pic = -1; undef @pic;
+		} else {
+			sleep ($sleep);
+		}
+	}
+
+	return;
+}
+
+1;