lib/PXElator/upstream.pm

package upstream;

use warnings;
use strict;
use autodie;

use LWP::Simple qw/mirror RC_NOT_MODIFIED/;
use Storable;
use server;

sub mirror_file {
        my ( $url, $file ) = @_;
        print STDERR "mirror $url";
        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;

        return if ! first_time($url);

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

        $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 $name = (caller(1))[3];
        $name =~ s{config::}{} || die "caller isn't package config !";

        my $path = "$server::base_dir/tftp/$name";

        foreach my $file ( @_ ) {
                mirror_file "$url/$file", "$path/$file"
                        if first_time( "$url/$file" );
        }
}

sub mkpath {
        my $file = shift;

        my @file_parts = split m{/}, $file;
        foreach ( 1 .. $#file_parts - 1 ) {
                my $path = splice @file_parts, 0, $_;
                warn "? $path\n";
                mkdir $path unless -e $path;
        }
}

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