Merge remote-tracking branch 'origin/github/pr/45'

[grml.org.git] / gen_website

diff --git a/gen_website b/gen_website
index 0a92b8a..7c26751 100755 (executable)
--- a/gen_website
+++ b/gen_website
@@ -10,8 +10,62 @@ use File::Basename qw (fileparse dirname);
 use File::Copy::Recursive qw(fcopy dirmove);
 use File::Temp qw (tempdir);
 
+use lib '/usr/share/perl5/';
+
+my $have_mirmon = 0;
+my $masterlist = '/usr/local/src/grml-mirrors/Mirrors.masterlist';
+my $mirmonconf = '/etc/mirmon.conf';
+
+my $fh;
+my $m;
+my $conf;
+my $state;
+if (eval {require Mirmon; 1 } && -f $masterlist && -f $mirmonconf) {
+       $have_mirmon = 1;
+       open ($fh, '<', $masterlist) or die "Could not open $masterlist: $!";
+       $m = Mirmon ->new($mirmonconf);
+       $conf  = $m->conf  ; # a Mirmon::Conf object
+       $state = $m->state ; # the mirmon state
+} else {
+       print "Skipping mirmon, it's either not installed or not configured.\n";
+}
+
+
 my $out_dir = shift || "out/";
 
+my $mirrors; 
+
+sub get_last_state ($) {
+       my $url = shift;
+       my $mirror_state = $state->{ $url } ; # a Mirmon::Mirror object
+       my ($time, $history) = split('-', $mirror_state->{state_history});
+       my $last_state = substr($history,-1,1);
+       return $last_state;
+}
+
+
+my $data;
+while (($have_mirmon) && (my $line = <$fh>)) {
+    chomp $line;
+    if ($line =~ /([^:]+): (.*)/) {
+        my $key = lc($1);
+        my $value = $2;
+        $data->{$key} = $value;
+    } elsif ($line eq '') {
+       my $url = sprintf ("https://%s%s", $data->{'site'}, $data->{'grml-http'});
+               next if get_last_state($url) eq 'f';
+       $mirrors->{ $data->{'country'} }->{ $data->{'site'} } = $data;
+        $data = undef;
+    } else {
+        print "Malformed line: $line\n";
+    }
+}
+
+if ($data) {
+       my $url = sprintf ("https://%s%s", $data->{'site'}, $data->{'grml-http'});
+       $mirrors->{ $data->{'country'} }->{ $data->{'site'} } = $data if get_last_state($url) eq 'f';
+}
+
 #find all files
 
 #rule to match git directorys
@@ -23,12 +77,17 @@ my $git = File::Find::Rule->directory
 #matches all files
 my $file_rule = File::Find::Rule->file();
 
-#combine both
-my @files = File::Find::Rule->or( $git, $file_rule )
+#match symlinks
+my $symlink_rule =  File::Find::Rule->symlink();
+
+#combine all rules
+my @files = File::Find::Rule->or( $git, $file_rule, $symlink_rule )
                             ->in('.');
 
 #create a tempdir
 my $tempdir = tempdir( CLEANUP => 1 );
+make_path("$tempdir/out") or die "Could not create $tempdir/out: $!";
+
 
 #initialize template toolkit
 
@@ -39,17 +98,17 @@ foreach my $file (@files) {
        next if $file =~ /$0$/;
        if ($file =~ /\.tt2$/) {
                my $output;
-               $template->process($file, undef, \$output)
+               $template->process($file, { mirrors => $mirrors }, \$output)
                        || die "Could not process file \"$file\": $!";
 
                my ($name,$path,$suffix) = fileparse($file,qw (.tt2));
-        make_path("$tempdir/$path") unless -d "$tempdir/$path";
-               open (my $fh, '>', "$tempdir/$path/$name")
+               make_path("$tempdir/out/$path") unless -d "$tempdir/out/$path";
+               open (my $fh, '>', "$tempdir/out/$path/$name")
                        or die "Could not write to $file: $!";
                print $fh $output;
                close($fh);
        } else {
-               fcopy ($file, "$tempdir/$file") or die "Could not copy $file to $tempdir/$file: $!";
+               fcopy ($file, "$tempdir/out/$file") or die "Could not copy $file to $tempdir/out/$file: $!";
        }
 }
 
@@ -58,8 +117,8 @@ if (-d $out_dir) {
     dirmove ($out_dir, $out_dir . ".bak")
         or die "Could not move $out_dir to $out_dir.bak: $!";
 }
-if (! dirmove ($tempdir, "$out_dir")) {
-       warn "Could not move $tempdir to $out_dir: $!";
+if (! dirmove ("$tempdir/out", "$out_dir")) {
+       warn "Could not move $tempdir/out to $out_dir: $!";
        warn "Rollback";
        remove_tree($out_dir);
        dirmove ($out_dir . ".bak", $out_dir);