git.rot13.org Git - pxelator/blob - lib/PXElator/upstream.pm

   1 package upstream;
   2
   3 use warnings;
   4 use strict;
   5 use autodie;
   6
   7 use Data::Dump qw/dump/;
   8 use LWP::Simple qw/mirror RC_NOT_MODIFIED/;
   9 use Storable;
  10 use server;
  11
  12 sub mirror_file {
  13         my ( $url, $file ) = @_;
  14         warn "mirror_file $url -> $file\n";
  15         mirror( $url, $file )
  16                 == RC_NOT_MODIFIED
  17                 && warn(" not modified\n")
  18                 || warn(" done ", -s $file, " bytes\n")
  19                 ;
  20 }
  21
  22 my $once_path = '/tmp/pxelator.once';
  23 our $just_once = retrieve $once_path if -e $once_path;
  24 sub first_time {
  25         my $what = shift;
  26         return if $just_once->{$what}++;
  27         store $just_once, $once_path;
  28         return 1;
  29 }
  30
  31 sub iso {
  32         my $url = shift;
  33
  34         my $name = (caller(1))[3];
  35         $name =~ s{config::}{} || die "caller isn't package config !";
  36
  37         warn "$name $url";
  38
  39         my $dir = "$server::base_dir/iso";
  40         mkdir $dir unless -e $dir;
  41
  42         my $file = $1 if $url =~ m{/([^/]+\.iso$)}i;
  43         die "can't find iso file in $url" unless $file;
  44
  45         my $iso = "$dir/$file";
  46
  47         mirror_file( $url, $iso ) if first_time $url;
  48
  49         $file =~ s{\.iso$}{}i;
  50         my $mnt = "$server::base_dir/tftp/$name";
  51         mkdir $mnt unless -d $mnt;
  52         $mnt .= '/iso';
  53         mkdir $mnt unless -d $mnt;
  54
  55         system("mount -t iso9660 | grep $name/iso || sudo mount $iso $mnt -o loop -t iso9660 -v") == 0;
  56 }
  57
  58 sub files {
  59         my $url = shift;
  60         my @files = @_;
  61
  62         if ( ! @files ) {
  63                 push @files, $1 if $url =~ s{/([^/]+)$}{};
  64         }
  65
  66         my $name = (caller(1))[3];
  67         $name =~ s{config::}{} || die "caller isn't package config !";
  68
  69         my $path = "$server::base_dir/tftp/$name";
  70         mkdir $path unless -d $path;
  71
  72         warn "# files $url ",dump( @files ), " -> $path\n";
  73
  74         foreach my $file ( @files ) {
  75                 mirror_file( "$url/$file", "$path/$file" )
  76                         if ! -e "$path/$file" || first_time( "$url/$file" );
  77         }
  78 }
  79
  80 1;