src/graph_list.c: Sort the instances of all graphs.
[collection4.git] / src / action_graph.c
1 /**
2  * collection4 - action_graph.c
3  * Copyright (C) 2010  Florian octo Forster
4  * 
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  * 
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor,
18  * Boston, MA  02110-1301  USA
19  *
20  * Authors:
21  *   Florian octo Forster <ff at octo.it>
22  **/
23
24 #include "config.h"
25
26 #include <stdlib.h>
27 #include <stdio.h>
28 #include <string.h>
29 #include <errno.h>
30 #include <stdint.h>
31 #include <inttypes.h>
32 #include <dirent.h> /* for PATH_MAX */
33 #include <assert.h>
34 #include <math.h>
35
36 #include <rrd.h>
37
38 #include "common.h"
39 #include "action_graph.h"
40 #include "graph.h"
41 #include "graph_instance.h"
42 #include "graph_list.h"
43 #include "utils_cgi.h"
44 #include "utils_array.h"
45
46 #include <fcgiapp.h>
47 #include <fcgi_stdio.h>
48
49 struct graph_data_s
50 {
51   rrd_args_t *args;
52   rrd_info_t *info;
53   time_t mtime;
54   time_t expires;
55   long now;
56   long begin;
57   long end;
58 };
59 typedef struct graph_data_s graph_data_t;
60
61 static void emulate_graph (int argc, char **argv) /* {{{ */
62 {
63   int i;
64
65   printf ("rrdtool \\\n");
66   for (i = 0; i < argc; i++)
67   {
68     if (i < (argc - 1))
69       printf ("  \"%s\" \\\n", argv[i]);
70     else
71       printf ("  \"%s\"\n", argv[i]);
72   }
73 } /* }}} void emulate_graph */
74
75 static int ag_info_print (rrd_info_t *info) /* {{{ */
76 {
77   if (info->type == RD_I_VAL)
78     printf ("[info] %s = %g;\n", info->key, info->value.u_val);
79   else if (info->type == RD_I_CNT)
80     printf ("[info] %s = %lu;\n", info->key, info->value.u_cnt);
81   else if (info->type == RD_I_STR)
82     printf ("[info] %s = %s;\n", info->key, info->value.u_str);
83   else if (info->type == RD_I_INT)
84     printf ("[info] %s = %i;\n", info->key, info->value.u_int);
85   else if (info->type == RD_I_BLO)
86     printf ("[info] %s = [blob, %lu bytes];\n", info->key, info->value.u_blo.size);
87   else
88     printf ("[info] %s = [unknown type %#x];\n", info->key, info->type);
89
90   return (0);
91 } /* }}} int ag_info_print */
92
93 static int output_graph (graph_data_t *data) /* {{{ */
94 {
95   rrd_info_t *img;
96   char time_buffer[256];
97   time_t expires;
98   int status;
99
100   for (img = data->info; img != NULL; img = img->next)
101     if ((strcmp ("image", img->key) == 0)
102         && (img->type == RD_I_BLO))
103       break;
104
105   if (img == NULL)
106     return (ENOENT);
107
108   printf ("Content-Type: image/png\n"
109       "Content-Length: %lu\n",
110       img->value.u_blo.size);
111   if (data->mtime > 0)
112   {
113     int status;
114     
115     status = time_to_rfc1123 (data->mtime, time_buffer, sizeof (time_buffer));
116     if (status == 0)
117       printf ("Last-Modified: %s\n", time_buffer);
118   }
119
120   /* Print Expires header. */
121   if (data->end >= data->now)
122   {
123     /* The end of the timespan can be seen. */
124     long secs_per_pixel;
125
126     /* FIXME: Handle graphs with width != 400. */
127     secs_per_pixel = (data->end - data->begin) / 400;
128
129     expires = (time_t) (data->now + secs_per_pixel);
130   }
131   else /* if (data->end < data->now) */
132   {
133     expires = (time_t) (data->now + 86400);
134   }
135   status = time_to_rfc1123 (expires, time_buffer, sizeof (time_buffer));
136   if (status == 0)
137     printf ("Expires: %s\n", time_buffer);
138
139   printf ("X-Generator: "PACKAGE_STRING"\n");
140   printf ("\n");
141
142   fwrite (img->value.u_blo.ptr, img->value.u_blo.size,
143       /* nmemb = */ 1, stdout);
144
145   return (0);
146 } /* }}} int output_graph */
147
148 #define OUTPUT_ERROR(...) do {             \
149   printf ("Content-Type: text/plain\n\n"); \
150   printf (__VA_ARGS__);                    \
151   return (0);                              \
152 } while (0)
153
154 int action_graph (void) /* {{{ */
155 {
156   graph_data_t data;
157   graph_config_t *cfg;
158   graph_instance_t *inst;
159   int status;
160
161   int argc;
162   char **argv;
163
164   cfg = gl_graph_get_selected ();
165   if (cfg == NULL)
166     OUTPUT_ERROR ("gl_graph_get_selected () failed.\n");
167
168   inst = inst_get_selected (cfg);
169   if (inst == NULL)
170     OUTPUT_ERROR ("inst_get_selected (%p) failed.\n", (void *) cfg);
171
172   data.args = ra_create ();
173   if (data.args == NULL)
174     return (ENOMEM);
175
176   array_append (data.args->options, "graph");
177   array_append (data.args->options, "-");
178   array_append (data.args->options, "--imgformat");
179   array_append (data.args->options, "PNG");
180
181   status = get_time_args (&data.begin, &data.end, &data.now);
182   if (status == 0)
183   {
184     array_append (data.args->options, "-s");
185     array_append_format (data.args->options, "%li", data.begin);
186     array_append (data.args->options, "-e");
187     array_append_format (data.args->options, "%li", data.end);
188   }
189
190   status = inst_get_rrdargs (cfg, inst, data.args);
191   if (status != 0)
192   {
193     ra_destroy (data.args);
194     OUTPUT_ERROR ("inst_get_rrdargs failed with status %i.\n", status);
195   }
196
197   argc = ra_argc (data.args);
198   argv = ra_argv (data.args);
199   if ((argc < 0) || (argv == NULL))
200   {
201     ra_destroy (data.args);
202     return (-1);
203   }
204
205   rrd_clear_error ();
206   data.info = rrd_graph_v (argc, argv);
207   if ((data.info == NULL) || rrd_test_error ())
208   {
209     printf ("Content-Type: text/plain\n\n");
210     printf ("rrd_graph_v failed: %s\n", rrd_get_error ());
211     emulate_graph (argc, argv);
212   }
213   else
214   {
215     int status;
216
217     data.mtime = inst_get_mtime (inst);
218
219     status = output_graph (&data);
220     if (status != 0)
221     {
222       rrd_info_t *ptr;
223
224       printf ("Content-Type: text/plain\n\n");
225       printf ("output_graph failed. Maybe the \"image\" info was not found?\n\n");
226
227       for (ptr = data.info; ptr != NULL; ptr = ptr->next)
228       {
229         ag_info_print (ptr);
230       }
231     }
232   }
233
234   if (data.info != NULL)
235     rrd_info_free (data.info);
236
237   ra_argv_free (argv);
238   ra_destroy (data.args);
239   data.args = NULL;
240
241   return (0);
242 } /* }}} int action_graph */
243
244 /* vim: set sw=2 sts=2 et fdm=marker : */