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;

        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 @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" );
        }
}

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 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	return if ! first_time($url);
35
36	my $name = (caller(1))[3];
37	$name =~ s{config::}{} \|\| die "caller isn't package config !";
38
39	warn "$name $url";
40
41	my $dir = "$server::base_dir/iso";
42	mkdir $dir unless -e $dir;
43
44	my $file = $1 if $url =~ m{/([^/]+\.iso$)}i;
45	die "can't find iso file in $url" unless $file;
46
47	my $iso = "$dir/$file";
48
49	mirror_file( $url, $iso );
50
51	$file =~ s{\.iso$}{}i;
52	my $mnt = "$server::base_dir/tftp/$name";
53	mkdir $mnt unless -d $mnt;
54	$mnt .= '/iso';
55	mkdir $mnt unless -d $mnt;
56
57	system("mount -t iso9660 \| grep $name/iso \|\| sudo mount $iso $mnt -o loop -t iso9660 -v") == 0;
58	}
59
60	sub files {
61	my $url = shift;
62	my @files = @_;
63
64	if ( ! @files ) {
65	push @files, $1 if $url =~ s{/([^/]+)$}{};
66	}
67
68	my $name = (caller(1))[3];
69	$name =~ s{config::}{} \|\| die "caller isn't package config !";
70
71	my $path = "$server::base_dir/tftp/$name";
72	mkdir $path unless -d $path;
73
74	warn "# files $url ",dump( @files ), " -> $path\n";
75
76	foreach my $file ( @files ) {
77	mirror_file( "$url/$file", "$path/$file" )
78	if ! -e "$path/$file" \|\| first_time( "$url/$file" );
79	}
80	}
81
82	sub mkpath {
83	my $file = shift;
84
85	my @file_parts = split m{/}, $file;
86	foreach ( 1 .. $#file_parts - 1 ) {
87	my $path = splice @file_parts, 0, $_;
88	warn "? $path\n";
89	mkdir $path unless -e $path;
90	}
91	}
92
93	1;