lib/PXElator/upstream.pm

package upstream;

use warnings;
use strict;
use autodie;

use Data::Dump qw/dump/;
use LWP::Simple qw/mirror RC_NOT_MODIFIED/;
use Storable;
use server;

sub mirror_file {
        my ( $url, $file ) = @_;
        warn "mirror_file $url -> $file\n";
        mirror( $url, $file )
                == RC_NOT_MODIFIED
                && warn(" not modified\n")
                || warn(" done ", -s $file, " bytes\n")
                ;
}

my $once_path = '/tmp/pxelator.once';
our $just_once = retrieve $once_path if -e $once_path;
sub first_time {
        my $what = shift;
        return if $just_once->{$what}++;
        store $just_once, $once_path;
        return 1;
}

sub iso {
        my $url = shift;

        my $name = (caller(1))[3];
        $name =~ s{config::}{} || die "caller isn't package config !";

        warn "$name $url";

        my $dir = "$server::base_dir/iso";
        mkdir $dir unless -e $dir;

        my $file = $1 if $url =~ m{/([^/]+\.iso$)}i;
        die "can't find iso file in $url" unless $file;

        my $iso = "$dir/$file";

        mirror_file( $url, $iso ) if first_time $url;

        $file =~ s{\.iso$}{}i;
        my $mnt = "$server::base_dir/tftp/$name";
        mkdir $mnt unless -d $mnt;
        $mnt .= '/iso';
        mkdir $mnt unless -d $mnt;

        system("mount -t iso9660 | grep $name/iso || sudo mount $iso $mnt -o loop -t iso9660 -v") == 0;
}

sub files {
        my $url = shift;
        my @files = @_;

        if ( ! @files ) {
                push @files, $1 if $url =~ s{/([^/]+)$}{};
        }

        my $name = (caller(1))[3];
        $name =~ s{config::}{} || die "caller isn't package config !";

        my $path = "$server::base_dir/tftp/$name";
        mkdir $path unless -d $path;

        warn "# files $url ",dump( @files ), " -> $path\n";

        foreach my $file ( @files ) {
                mirror_file( "$url/$file", "$path/$file" )
                        if ! -e "$path/$file" || first_time( "$url/$file" );
        }
}

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