collectd2html.pl: Added --recursive command line option.
[collectd.git] / contrib / collectd2html.pl
1 #!/usr/bin/perl
2
3 ################################################################################
4 #
5 # collectd2html.pl
6 #
7 # Description:
8 #   Generate an html page with all rrd data gathered by collectd.
9 #
10 # Usage:
11 #   collectd2html.pl
12 #
13 #   When run on <host>, it generated <host>.html and <host>.dir, the latter
14 #   containing all necessary images.
15 #
16 #
17 # Copyright 2006 Vincent StehlĂ© <vincent.stehle@free.fr>
18 #
19 # Patch to configure the data directory and hostname by Eddy Petrisor
20 # <eddy.petrisor@gmail.com>.
21 #
22 # This program is free software; you can redistribute it and/or modify
23 # it under the terms of the GNU General Public License as published by
24 # the Free Software Foundation; either version 2 of the License, or
25 # (at your option) any later version.
26 #
27 # This program is distributed in the hope that it will be useful,
28 # but WITHOUT ANY WARRANTY; without even the implied warranty of
29 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
30 # GNU General Public License for more details.
31 #
32 # You should have received a copy of the GNU General Public License
33 # along with this program; if not, write to the Free Software
34 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
35 #
36 ################################################################################
37
38 use warnings;
39 use strict;
40 use Fatal qw(open close);
41 use File::Basename;
42 use Getopt::Long qw(:config no_ignore_case bundling pass_through);
43
44 my $DIR       = "/var/lib/collectd";
45 my $HOST      = undef;
46 my $IMG_FMT   = "PNG";
47 my $RECURSIVE = 0;
48
49 GetOptions (
50     "host=s"         => \$HOST,
51     "data-dir=s"     => \$DIR,
52     "image-format=s" => \$IMG_FMT,
53     "recursive"      => \$RECURSIVE
54 );
55
56 if (($DIR !~ m/\/rrd\/?$/) && (-d "$DIR/rrd")) {
57         $DIR .= "/rrd";
58 }
59
60 if (defined($HOST) && ($DIR !~ m/\/$HOST\/?$/) && (-d "$DIR/$HOST")) {
61         $DIR .= "/$HOST";
62 }
63
64 my @COLORS = (0xff7777, 0x7777ff, 0x55ff55, 0xffcc77, 0xff77ff, 0x77ffff,
65         0xffff77, 0x55aaff);
66 my @tmp = `/bin/hostname`; chomp(@tmp);
67 $HOST = $tmp[0] if (! defined $HOST);
68 my $svg_p = ($IMG_FMT eq "SVG");
69 my $IMG_SFX = $svg_p ? ".svg" : ".png";
70 my $IMG_DIR = "${HOST}.dir";
71 my $HTML = "${HOST}.html";
72
73 ################################################################################
74 #
75 # fade_component
76 #
77 # Description:
78 #   Fade a color's component to the white.
79 #
80 ################################################################################
81 sub fade_component($)
82 {
83         my($component) = @_;
84         return (($component + 255 * 5) / 6);
85 }
86
87 ################################################################################
88 #
89 # fade_color
90 #
91 # Description:
92 #   Fade a color to the white.
93 #
94 ################################################################################
95 sub fade_color($)
96 {
97         my($color) = @_;
98         my $r = 0;
99
100         for my $i (0 .. 2){
101                 my $shft = ($i * 8);
102                 my $component = (($color >> $shft) & 255);
103                 $r |= (fade_component($component) << $shft);
104         }
105
106         return $r;
107 }
108
109 ################################################################################
110 #
111 # main
112 #
113 ################################################################################
114 system("rm -fR $IMG_DIR");
115 system("mkdir -p $IMG_DIR");
116 local *OUT;
117 open(OUT, ">$HTML");
118 my $title="Rrd plot for $HOST";
119
120 print OUT <<END;
121 <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"
122    "http://www.w3.org/TR/html4/loose.dtd">
123 <html>
124 <head>
125 <title>$title</title>
126 </head>
127 <body>
128 <center>
129 END
130
131 # list interesting rrd
132 my @rrds;
133 my @list;
134
135 if ($RECURSIVE) {
136         @list = `find $DIR -type f -name '*.rrd'`;
137 }
138 else {
139         @list = `ls $DIR/*.rrd`;
140 }
141 chomp(@list);
142
143 foreach my $rrd (sort @list){
144         $rrd =~ m/^$DIR\/(.*)\.rrd$/;
145         push(@rrds, $1);
146 }
147
148 # table of contents
149 print OUT <<END;
150 <A name="top"></A><H1>$title</H1>
151 <P>
152 END
153
154 foreach my $bn (@rrds){
155         my $cleaned_bn = $bn;
156         $cleaned_bn =~ tr/%\//__/;
157         print OUT <<END;
158 <A href="#$cleaned_bn">$bn</A>
159 END
160 }
161
162 print OUT <<END;
163 </P>
164 END
165
166 # graph interesting rrd
167 for (my $i = 0; $i < scalar(@rrds); ++$i) {
168         my $bn = $rrds[$i];
169         print "$bn\n";
170
171         my $rrd = $list[$i];
172         my $cmd = "rrdtool info $rrd |grep 'ds\\[' |sed 's/^ds\\[//'" 
173                 ." |sed 's/\\].*//' |sort |uniq";
174         my @dss = `$cmd`; chomp(@dss);
175
176         # all DEF
177         my $j = 0;
178         my $defs = "";
179
180         foreach my $ds (@dss){
181                 $defs .= " DEF:${ds}_avg=$rrd:$ds:AVERAGE"
182                         ." DEF:${ds}_max=$rrd:$ds:MAX ";
183         }
184
185         # all AREA
186         $j = 0;
187
188         foreach my $ds (@dss){
189                 my $color = $COLORS[$j % scalar(@COLORS)]; $j++;
190                 my $faded_color = fade_color($color);
191                 $defs .= sprintf(" AREA:${ds}_max#%06x ", $faded_color);
192         }
193
194         # all LINE      
195         $j = 0;
196
197         foreach my $ds (@dss){
198                 my $color = $COLORS[$j % scalar(@COLORS)]; $j++;
199                 $defs .= sprintf(" LINE2:${ds}_avg#%06x:$ds"
200                         ." GPRINT:${ds}_avg:AVERAGE:%%5.1lf%%sAvg"
201                         ." GPRINT:${ds}_max:MAX:%%5.1lf%%sMax"
202                         , $color);
203         }
204
205         my $cleaned_bn = $bn;
206         $cleaned_bn =~ tr/%\//__/;
207         print OUT <<END;
208 <A name="$cleaned_bn"></A><H1>$bn</H1>
209 END
210
211         # graph various ranges
212         foreach my $span qw(1hour 1day 1week 1month){
213                 system("mkdir -p $IMG_DIR/" . dirname($bn));
214                 my $img = "$IMG_DIR/${bn}-$span$IMG_SFX";
215
216                 my $cmd = "rrdtool graph $img"
217                         ." -t \"$bn $span\" --imgformat $IMG_FMT --width 600 --height 100"
218                         ." --start now-$span --end now --interlaced"
219                         ." $defs >/dev/null 2>&1";
220                 system($cmd);
221
222                 my $cleaned_img = $img; $cleaned_img =~ s/%/%25/g;
223                 if (! $svg_p) {
224                         print OUT <<END;
225 <P><IMG src="$cleaned_img" alt="${bn} $span"></P>
226 END
227                 } else {
228                         print OUT <<END;
229 <P><object data="$cleaned_img" type="image/svg+xml"
230            width="670" height="179">
231   ${bn} $span</object></P>
232 END
233                 }
234         }
235
236         print OUT <<END;
237 <A href="#top">[top]</A>
238 END
239 }
240
241 print OUT <<END;
242 </center>
243 </body>
244 </html>
245 END
246
247 close(OUT);