1 |
dpavlin |
138 |
package upstream; |
2 |
|
|
|
3 |
|
|
use warnings; |
4 |
|
|
use strict; |
5 |
|
|
use autodie; |
6 |
|
|
|
7 |
|
|
use LWP::Simple qw/mirror RC_NOT_MODIFIED/; |
8 |
dpavlin |
233 |
use Storable; |
9 |
dpavlin |
138 |
use server; |
10 |
|
|
|
11 |
dpavlin |
147 |
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 |
dpavlin |
233 |
my $once_path = '/tmp/pxelator.once'; |
22 |
dpavlin |
242 |
our $just_once = retrieve $once_path if -e $once_path; |
23 |
dpavlin |
233 |
sub first_time { |
24 |
|
|
my $what = shift; |
25 |
|
|
return if $just_once->{$what}++; |
26 |
|
|
store $just_once, $once_path; |
27 |
|
|
return 1; |
28 |
|
|
} |
29 |
dpavlin |
138 |
|
30 |
|
|
sub iso { |
31 |
|
|
my $url = shift; |
32 |
|
|
|
33 |
dpavlin |
233 |
return if ! first_time($url); |
34 |
dpavlin |
138 |
|
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 |
dpavlin |
147 |
mirror_file( $url, $iso ); |
49 |
dpavlin |
138 |
|
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 |
dpavlin |
147 |
system("mount -t iso9660 | grep $name/iso || sudo mount $iso $mnt -o loop -t iso9660 -v") == 0; |
57 |
dpavlin |
138 |
} |
58 |
|
|
|
59 |
dpavlin |
233 |
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 |
dpavlin |
147 |
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 |
dpavlin |
138 |
1; |