add zero is allowed flag to users/nodes
[spider.git] / cmd / mrtg.pl
index dba856e5b4495d429e78c6750255a4a9b8d8c791..05d426e26266bf2174a08582b4a98b58154e0cee 100644 (file)
 #
 # Your mrtg binary must live in one of the standard places
 #
-# you will need perl 5.6 (probably) to be able to run this command
+# The arguments (keywords) to the mrtg command are these
+#
+# a) content          (you always get the node users and nodes)
+#    totalspots       - all spots
+#    hfvhf            - all spots split into HF and VHF
+#    wwv              - two graphs of WWV, one SFI and R other A and K
+#    wcy              - WCY A and K 
+#    all              - all of the above 
+#    
+# b) actions          
+#    test             - do everything except check for and run mrtg
+#    nomrtg           - ditto (better name)
+#    dataonly         - only generate the data files for mrtg
+#    cfgonly          - only generate the mrtg.cfg file (like cfgmaker)
+#    runmrtg          - run mrtg, this is probably used with dataonly
+#                     - together with a home rolled mrtg.cfg 
+#
+# Copyright (c) 2002 Dirk Koopman G1TLH
+#
+# $Id$
 #
-
-sub cfgprint($$@$$$$$);
 
 my ($self, $line) = @_;
 
 # create the arg list
 my %want;
 for (split /\s+/, $line) { $want{lc $_} = 1};
-                        
-return (1, "MRTG not installed") unless $want{test} || -e '/usr/bin/mrtg' || -e '/usr/local/bin/mrtg';
-
-my $dir = "$main::root/mrtg";
-my $html = "$main::root/html/mrtg";
-my $cfg = "$dir/mrtg.cfg";
-
-my $mc = new IO::File ">$cfg" or return(1, "cannot open $cfg for mrtg writing");
-
-# print out the header
-print $mc <<"EOF";
-### Global Defaults
-
-#  to get bits instead of bytes and graphs growing to the right
-# Options[_]: growright, bits
-
-Htmldir: $html
-Imagedir: $html
-Logdir: $dir
-Options[_]: growright
-
-##
-##
-
-EOF
-
+$want{nomrtg} = 1 if $want{cfgonly} || $want{test};
+                        
+return (1, "MRTG not installed") unless $want{nomrtg} || -e '/usr/bin/mrtg' || -e '/usr/local/bin/mrtg';
 
-#dbg "$dir\n$html\n";
+my $mc = new Mrtg or return (1, "cannot initialise Mrtg $!");
                         
 # do the users and nodes
 my $users = DXChannel::get_all_users();
 my $nodes = DXChannel::get_all_nodes();
-my $uptime = main::uptime();
-#dbg "$users $nodes $uptime\n";
-if (my $m = new IO::File ">$dir/users") {
-       print $m "$users\n$nodes\n$uptime\nUsers and Nodes\n";
-       close $m;
-}
-cfgprint($mc, 'users', [qw(gauge)], 500, 
+
+$mc->cfgprint('users', [qw(unknaszero gauge)], 500, 
                 "Users and Nodes on $main::mycall",
-                'Users / Nodes', 'Users', 'Nodes');
+                'Users / Nodes', 'Users', 'Nodes') unless $want{dataonly};
+$mc->data('users', $users, $nodes, 'Users / Nodes') unless $want{cfgonly};
 
 # do the  total users and nodes
 if ($want{totalusers} || $want{all}) {
        $nodes = Route::Node::count();
        $users = Route::User::count();
-       #dbg "$users $nodes $uptime\n";
-       if (my $m = new IO::File ">$dir/totalusers") {
-               print $m "$users\n$nodes\n$uptime\nTotal Users and Nodes\n";
-               close $m;
-       }
-       cfgprint($mc, 'totalusers', [qw(gauge)], 10000, 
+       $mc->cfgprint('totalusers', [qw(unknaszero gauge)], 10000, 
                        'Total Users and Nodes in the Visible Cluster Network',
-                        'Users / Nodes', 'Users', 'Nodes');
+                        'Users / Nodes', 'Users', 'Nodes') unless $want{dataonly};
+       $mc->data('totalusers', $users, $nodes, 'Total Users and Nodes in the Visible Cluster Network') unless $want{cfgonly};
 }
 
 # do the total spots
 if ($want{totalspots} || $want{all}) {
-       if (my $m = new IO::File ">$dir/totalspots") {
-               print $m "$Spot::totalspots\n$Spot::totalspots\n$uptime\nTotal Spots\n";
-               close $m;
-       }
+       $mc->cfgprint('totalspots',  [qw(unknaszero gauge noi)], 1000, 'Total Spots',
+                        'Spots', 'Spots', 'Spots') unless $want{dataonly};
+       $mc->data('totalspots', $Spot::totalspots, $Spot::totalspots, 'Total Spots') unless $want{cfgonly};
        $Spot::totalspots = 0;
-       cfgprint($mc, 'totalspots', [qw(unknaszero gauge noi)], 1000, 'Total Spots',
-                        'Spots', 'Spots', 'Spots');
 }
 
 # do the HF and VHF spots
 if ($want{hfvhf} || $want{all}) {
-       if (my $m = new IO::File ">$dir/hfspots") {
-               print $m "$Spot::hfspots\n$Spot::vhfspots\n$uptime\nHF and VHF+ Spots\n";
-               close $m;
-       }
+       $mc->cfgprint('hfspots', [qw(unknaszero gauge)], 1000, 'HF and VHF+ Spots',
+                        'Spots', 'HF', 'VHF') unless $want{dataonly};
+       $mc->data('hfspots', $Spot::hfspots, $Spot::vhfspots, 'HF and VHF+ Spots') unless $want{cfgonly};
        $Spot::hfspots = $Spot::vhfspots = 0;
-       cfgprint($mc, 'hfspots', [qw(unknaszero gauge)], 1000, 'HF and VHF+ Spots',
-                        'Spots', 'HF', 'VHF');
+}
+
+# wwv stuff
+if ($want{wwv} || $want{all}) {
+       $mc->cfgprint('wwvsfi', [qw(gauge)], 1000, 'WWV SFI and R',
+                        'SFI / R', 'SFI', 'R') unless $want{dataonly};
+       $mc->data('wwvsfi', ($Geomag::r || $WCY::r), ($Geomag::sfi || $WCY::sfi), 'WWV SFI and R') unless $want{cfgonly};
+       $mc->cfgprint('wwvka', [qw(gauge)], 1000, 'WWV A and K',
+                        'A / K', 'A', 'K') unless $want{dataonly};
+       $mc->data('wwvka', $Geomag::a, $Geomag::k, 'WWV A and K') unless $want{cfgonly};
+}
+
+# WCY stuff
+if ($want{wcy} || $want{all}) {
+       $mc->cfgprint('wcyka', [qw(gauge)], 1000, 'WCY A and K',
+                        'A / K', 'A', 'K') unless $want{dataonly};
+       $mc->data('wcyka', $WCY::a, $WCY::k, 'WCY A and K') unless $want{cfgonly};
 }
 
 # 
 # do the mrtg thing
 #
-close $mc;
-my @out = `mrtg $cfg`;
+my @out = $mc->run unless $want{nomrtg};
 return (1, @out);
-
-sub cfgprint
-{
-       my ($mc, $name, $options, $max, $title, $legend, $iname, $oname) = @_;
-       my $opt = join ', ', @$options, qw(withzeroes gauge growright nopercent integer);
-               
-       print $mc <<"EOF";
-
-#
-# $title
-#
-
-Target[$name]: `cat /spider/mrtg/$name`
-MaxBytes[$name]: $max
-Title[$name]: $title
-Options[$name]: $opt
-YLegend[$name]: $legend
-YTicsFactor[$name]: 1
-ShortLegend[$name]: \&nbsp;
-Legend1[$name]:Maximum No of $iname
-Legend2[$name]:Maximum No of $oname
-LegendI[$name]:$iname
-LegendO[$name]:$oname
-PageTop[$name]: <H1>$title</H1>
- <TABLE>
-   <TR><TD>System:</TD>     <TD>$main::mycall</TD></TR>
-   <TR><TD>Maintainer:</TD> <TD>$main::myemail</TD></TR>
-   <TR><TD>Description:</TD><TD>$title</TD></TR>
- </TABLE>
-EOF
-
-}