aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--lib/ManaTools/Module/PkgDownloader.pm207
-rwxr-xr-xscripts/urpm-downloader27
2 files changed, 234 insertions, 0 deletions
diff --git a/lib/ManaTools/Module/PkgDownloader.pm b/lib/ManaTools/Module/PkgDownloader.pm
new file mode 100644
index 00000000..e78c9eb4
--- /dev/null
+++ b/lib/ManaTools/Module/PkgDownloader.pm
@@ -0,0 +1,207 @@
+#!/usr/bin/perl -w
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+# Copyright: 2012 - 2016 by Matteo Pasotti <matteo.pasotti@gmail.com>
+
+package ManaTools::Module::PkgDownloader;
+
+use strict;
+use warnings;
+# use diagnostics;
+use Term::ANSIColor qw(:constants);
+# use Getopt::Long qw(:config permute);
+# use urpmex::Urpmex;
+use Moose;
+use Moose::Autobox;
+use ManaTools::Shared::urpmi_backend::DB;
+
+
+with 'MooseX::Getopt';
+
+extends qw( ManaTools::Module );
+
+my $PKG_QUERYMAKER = "urpmq";
+my $QUERY_LISTMEDIA_PARM = "--list-media";
+my $QUERY_LISTURL_PARM = "--list-url";
+my $QUERY_LOOKFORSRPM_PARM = "--sourcerpm";
+my $QUERY_PKG_FULL = "-f";
+my $DLDER = "--wget";
+
+
+has 'use_wget' => (
+ is => 'rw',
+ isa => 'Bool',
+ default => sub { return 0; },
+);
+
+has 'use_axel' => (
+ is => 'rw',
+ isa => 'Bool',
+ default => sub { return 0; },
+);
+
+has 'srpm' => (
+ is => 'rw',
+ isa => 'Bool',
+ default => sub { return 0; },
+);
+
+has 'use_major' => (
+ is => 'rw',
+ isa => 'Bool',
+ default => sub { return 0; },
+);
+
+has 'packagelist' => (
+ is => 'rw',
+ isa => 'Str',
+ required => 1,
+);
+
+has 'packages' => (
+ is => 'rw',
+ isa => 'ArrayRef',
+ default => sub { [] },
+);
+
+sub process_args {
+ my ($self, $pkglist) = @_;
+ for(split(/\s/,$pkglist)){
+ push @{$self->packages()}, $_;
+ }
+}
+
+sub download {
+ my $self = shift();
+ my $url = shift();
+ my $rpm = shift();
+ if($self->use_wget()){
+ `wget "$url/$rpm" -O $rpm`;
+ }elsif($self->use_axel()){
+ `axel -a "$url/$rpm" -o $rpm`;
+ }else{
+ `curl -s "$url/$rpm" -o $rpm`;
+ }
+}
+
+
+# IMPORTANT!
+# THOSE TWO ROUTINES MUST BE PORTED TO THE URPMI_BACKEND FROM MANATOOLS:
+# * retrieve_brpm_pkgname
+# * retrieve_srpm_pkgname
+
+# ----------------------------------------------------------------------
+# retrieve the binary rpm's pkg name - array
+# ----------------------------------------------------------------------
+sub retrieve_brpm_pkgname {
+ my $pkg = shift();
+ my @lista_brpms = `$PKG_QUERYMAKER -a $QUERY_PKG_FULL $pkg | grep "^$pkg" | sort -u`;
+ return @lista_brpms;
+}
+
+# ----------------------------------------------------------------------
+# retrieve the srpm's pkg name - array
+# ----------------------------------------------------------------------
+sub retrieve_srpm_pkgname {
+ my $pkg = shift();
+ my @lista_srpms = `$PKG_QUERYMAKER $QUERY_LOOKFORSRPM_PARM $pkg | sort -u | grep "$pkg:" | awk -F':' '{print \$2}'`;
+ return @lista_srpms;
+}
+
+sub start {
+ my $self = shift;
+ my $pkg = "";
+ my $rpmbackend = ManaTools::Shared::urpmi_backend::DB->new();
+ my $urpm = $rpmbackend->fast_open_urpmi_db();
+ my @media_urls=map { if(!$self->srpm()) {
+ $_->{url}
+ }else{
+ my @a = split(/\//,$_->{url});
+ my $newurl = "";
+ my $i = 0;
+ for($i=0;$i<scalar(@a)-4;$i++)
+ {
+ $newurl .= $a[$i]."/";
+ }
+ $newurl."SRPMS/".$a[scalar(@a)-2]."/".$a[scalar(@a)-1];
+ }
+ } $rpmbackend->get_active_media($urpm,$self->srpm());
+
+ $self->process_args($self->packagelist());
+
+
+ if(scalar(@media_urls) lt 1)
+ {
+ print BOLD, RED, "==", RESET, " no active media found\n";
+ return 3;
+ }
+
+ if(scalar(@{$self->packages()}) gt 0){
+ for $pkg(@{$self->packages()}){
+ my @lista_srpms;
+ if($self->srpm()){
+ @lista_srpms = retrieve_srpm_pkgname($pkg);
+ }else{
+ @lista_srpms = retrieve_brpm_pkgname($pkg);
+ }
+ #print "@lista_srpms\n";
+ if($self->use_major()){
+ #print "Using only major version\n";
+ for(my $i=0;$i<scalar(@lista_srpms)-1;$i++){
+ shift @lista_srpms;
+ #print "@lista_srpms\n";
+ }
+ }
+ for my $srpm(@lista_srpms){
+ $srpm =~s/^\s+//g;
+ chomp $srpm;
+ $srpm = $srpm.".rpm" if(!$self->srpm());
+ print BOLD, WHITE, "== Processing $srpm\n", RESET;
+ for my $url(@media_urls){
+ chomp $url;
+ my @protocol = split(':',$url);
+ if($protocol[0] eq "http"){
+ print BOLD, WHITE, "== ", RESET, "protocol in use: ", BOLD, WHITE, $protocol[0], RESET, "\n";
+ print BOLD, WHITE, "== ", RESET, "trying with $url/$srpm\n";
+ my $check = `curl -s --head "$url/$srpm" | head -n 1 | grep "200 OK" > /dev/null ; echo \$?`;
+ chomp $check;
+ if($check eq "0"){
+ $self->download($url,$srpm);
+ last;
+ }
+ }elsif($protocol[0] eq "ftp"){
+ print BOLD, WHITE, "== ", "protocol in use: ", BOLD, WHITE, $protocol[0], RESET, "\n";
+ my $check = `curl -s --head "$url/$srpm"`;
+ $check =~s/\n/ /g;
+ $check =~s/^\s+//g;
+ $check =~s/\s+$//g;
+ if($check ne ""){
+ $self->download($url,$srpm);
+ last;
+ }
+ }elsif($protocol[0] eq "rsync"){
+ print BOLD, WHITE, "== ", "protocol in use: ", BOLD, WHITE, $protocol[0], RESET, "\n";
+ }
+ }
+ print BOLD, WHITE, "== $srpm", RESET, " downloaded successfully\n";
+ }
+ }
+ return 0;
+ }else{
+ print BOLD, RED, "== no packages passed as argument\n", RESET;
+ return 2;
+ }
+}
+
+1;
diff --git a/scripts/urpm-downloader b/scripts/urpm-downloader
new file mode 100755
index 00000000..6497554c
--- /dev/null
+++ b/scripts/urpm-downloader
@@ -0,0 +1,27 @@
+#!/usr/bin/perl
+# vim: set et ts=4 sw=4:
+# Copyright 2013-2016 Matteo Pasotti
+# 2015-2016 Angelo Naselli
+#
+# This file is part of manatools
+#
+# manaproxy is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 2 of the License, or
+# (at your option) any later version.
+#
+# manaproxy is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with ManaTools. If not, see <http://www.gnu.org/licenses/>.
+use Modern::Perl '2011';
+use autodie;
+use ManaTools::Module::PkgDownloader;
+
+my $pkgdlManager = ManaTools::Module::PkgDownloader->new_with_options();
+$pkgdlManager->start();
+
+1;