new update-master-info seems to work

[ypp-sc-tools.db-test.git] / yarrg / update-master-info

diff --git a/yarrg/update-master-info b/yarrg/update-master-info
index 4c1547bd23cdfef10a0c15d701a51b47c28ddf1f..1dd432ff895584365fbc7c2aebfbe05193466562 100755 (executable)
--- a/yarrg/update-master-info
+++ b/yarrg/update-master-info
@@ -1,49 +1,104 @@
 #!/usr/bin/perl -w
 
 use strict (qw(vars));
-
 use DBI;
-
 use Commods;
 
-@ARGV==1 or die;
-my ($rsyncdir) = @ARGV;
 
-parse_info_serverside();
+sub full ($) {
+    my ($ocean) = @_;
+    quick($ocean);
+    print "## updating topology of $ocean\n";
+    system('./yppedia-chart-parser',$ocean); die $? if $?;
+}
 
-foreach my $oceanname (sort keys %oceans) {
-    print STDERR "updating ocean $oceanname...\n";
-    system('./db-idempotent-populate',$oceanname); die $? if $?;
+sub quick ($) {
+    my ($ocean) = @_;
+    print STDERR "## updating schema and commodities for $ocean\n";
+    system('./db-idempotent-populate',$ocean); die $? if $?;
 }
 
-my $sfn= 'source-info.txt';
+my $rsyncdir;
 
-foreach my $v (1..$masterinfoversion) {
-    my $dfnl= sprintf "master-info%s.txt", ($v>1 ? "-v$v" : '');
-    print STDERR "installing new $dfnl...\n";
+sub update_master_info () {
+    my $sfn= 'source-info.txt';
+
+    foreach my $v (1..$masterinfoversion) {
+       my $dfnl= sprintf "master-info%s.txt", ($v>1 ? "-v$v" : '');
+       print STDERR "installing new $dfnl...\n";
     
-    my $dfn= "$rsyncdir/$dfnl";
-    my $sf= new IO::File $sfn or die "$sfn $!";
-    my $df= new IO::File "$dfn.tmp", 'w' or die "$dfn.tmp $!";
-
-    my $h;
-    while (<$sf>) {
-       chomp; s/\s+$//;
-       next if m/^\s*\#/ || !m/\S/;
-       if (m/^\S.*/) {
-           $h= $&;
-       }
-       die "$_ ?" unless defined $h;
-       if ($h =~ m/^commods|^\%[a-z]\b/) {
-           s/\t.*//;
-       }
-       if ($v<2) {
-           next if $h =~ m/^nocommods/;
+       my $dfn= "$rsyncdir/$dfnl";
+       my $sf= new IO::File $sfn or die "$sfn $!";
+       my $df= new IO::File "$dfn.tmp", 'w' or die "$dfn.tmp $!";
+
+       my $h;
+       while (<$sf>) {
+           chomp; s/\s+$//;
+           next if m/^\s*\#/ || !m/\S/;
+           if (m/^\S.*/) {
+               $h= $&;
+           }
+           die "$_ ?" unless defined $h;
+           if ($h =~ m/^commods|^\%[a-z]\b/) {
+               s/\t.*//;
+           }
+           if ($v<2) {
+               next if $h =~ m/^nocommods/;
+           }
+           print $df $_, "\n" or die $!;
        }
-       print $df $_, "\n" or die $!;
+
+       $sf->error and die $!;
+       close $df or die $!;
+       rename "$dfn.tmp", "$dfn" or die $!;
+    }
+
+}
+
+
+my @specoceans;
+my $alloceans;
+
+sub optarg () {
+    return $_ if length;
+    die unless @ARGV;
+    return scalar shift @ARGV;
+}
+
+while (@ARGV && $ARGV[0] =~ m/^-/) {
+    $_= shift @ARGV;
+    last if m/^--?$/;
+    while (m/^-./) {
+       if (s/^-d//) { $rsyncdir= optarg(); }
+       elsif (s/^-O//) { push @specoceans, optarg(); }
+       elsif (s/^-a//) { $alloceans=1; }
+       else { die "$_ ?"; }
     }
+}
+
+die if @specoceans && $alloceans;
+
+parse_info_serverside();
+
+if (@specoceans) {
+    print "### full update of specified oceans ...\n";
+    foreach my $ocean (@specoceans) {
+       die "$ocean ?" unless defined $oceans{$ocean};
+       full($ocean);
+    }
+} elsif ($alloceans) {
+    print "### full (inc.topology) update of all oceans ...\n";
+    foreach my $ocean (sort keys %oceans) {
+       full($ocean);
+    }
+} else {
+    print "### quick (no topology) update only (of all oceans) ...\n";
+    foreach my $ocean (sort keys %oceans) {
+       quick($ocean);
+    }
+}
 
-    $sf->error and die $!;
-    close $df or die $!;
-    rename "$dfn.tmp", "$dfn" or die $!;
+if (defined $rsyncdir) {
+    print "### master-info update ...\n";
+    update_master_info();
 }