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	dpavlin	138	package upstream;
2
3			use warnings;
4			use strict;
5			use autodie;
6
7	dpavlin	266	use Data::Dump qw/dump/;
8	dpavlin	138	use LWP::Simple qw/mirror RC_NOT_MODIFIED/;
9	dpavlin	233	use Storable;
10	dpavlin	138	use server;
11
12	dpavlin	147	sub mirror_file {
13			my ( $url, $file ) = @_;
14	dpavlin	266	warn "mirror_file $url -> $file\n";
15	dpavlin	147	mirror( $url, $file )
16			== RC_NOT_MODIFIED
17			&& warn(" not modified\n")
18			\|\| warn(" done ", -s $file, " bytes\n")
19			;
20			}
21
22	dpavlin	233	my $once_path = '/tmp/pxelator.once';
23	dpavlin	242	our $just_once = retrieve $once_path if -e $once_path;
24	dpavlin	233	sub first_time {
25			my $what = shift;
26			return if $just_once->{$what}++;
27			store $just_once, $once_path;
28			return 1;
29			}
30	dpavlin	138
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	dpavlin	272	mirror_file( $url, $iso ) if first_time $url;
48	dpavlin	138
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	dpavlin	147	system("mount -t iso9660 \| grep $name/iso \|\| sudo mount $iso $mnt -o loop -t iso9660 -v") == 0;
56	dpavlin	138	}
57
58	dpavlin	233	sub files {
59			my $url = shift;
60	dpavlin	266	my @files = @_;
61	dpavlin	233
62	dpavlin	266	if ( ! @files ) {
63			push @files, $1 if $url =~ s{/([^/]+)$}{};
64			}
65
66	dpavlin	233	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	dpavlin	266	mkdir $path unless -d $path;
71	dpavlin	233
72	dpavlin	266	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	dpavlin	233	}
78			}
79
80	dpavlin	138	1;