fixed too large malloc
[ppastats.git] / src / html.c
1 /*
2  * Copyright (C) 2011-2012 jeanfi@gmail.com
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17  * 02110-1301 USA
18  */
19
20 #include <libintl.h>
21 #define _(String) gettext(String)
22
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <sys/time.h>
27 #include <sys/stat.h>
28 #include <sys/types.h>
29
30 #include <json/json.h>
31
32 #include "html.h"
33 #include "io.h"
34 #include "log.h"
35 #include "lp.h"
36 #include "lp_ws.h"
37 #include "ppastats.h"
38
39 #define HTML_FOOTER \
40 " <div id=\"footer\">Generated by \
41 <a href='http://wpitchoune.net/ppastats'>ppastats</a></div>\n\
42   </body>\n\
43 </html>"
44
45 #define HTML_PKG_TEMPLATE \
46 "       <h1><span id=\"pkg_name\">N/A</span></h1>\n\
47         <div id=\"charts\">\n\
48                 <div id=\"chart\"></div>\n\
49                 <div id=\"chart_distro\"></div>\n\
50         </div>\n\
51         <div id=\"details\">\n\
52                 <em>PPA</em>: \n\
53                 <a href=\"index.html\">\n\
54                 <span id=\"ppa_owner\">N/A</span>/\n\
55                 <span id=\"ppa_name\">N/A</span>\n\
56                 </a>\n\
57 \n\
58                 <div id=\"distros\"><em>Distros:</em></div>\n\
59                 <div id=\"versions\"><em>Versions:</em></div>\n\
60         </div>"
61
62 #define HTML_VERSION_TEMPLATE \
63 "<html>\n\
64   <head>\n\
65     <link type=\"text/css\"\n\
66           rel=\"stylesheet\"\n\
67           href=\n\
68 \"http://fonts.googleapis.com/css?family=Ubuntu:regular,bold&subset=Latin\">\n\
69     <link type=\"text/css\" href=\"css/ppastats.css\" rel=\"stylesheet\" />\n\
70     <link type=\"text/css\" href=\"css/jquery.jqplot.min.css\"\n\
71           rel=\"stylesheet\" />\n\
72 <!--[if lt IE 9]><script \
73 language=\"javascript\" \
74 type=\"text/javascript\" \
75 src=\"js/excanvas.js\"></script><![endif]-->\n\
76     <script type=\"text/javascript\" src=\"js/jquery.min.js\"></script>\n\
77     <script type=\"text/javascript\"\n\
78             src=\"js/jquery.jqplot.min.js\"></script>\n\
79     <script type=\"text/javascript\"\n\
80             src=\"js/jqplot.dateAxisRenderer.min.js\"></script>\n\
81     <script type=\"text/javascript\" src=\"js/ppastats.js\"></script>\n\
82     <script>var data = %s;\n\
83             ppastats_ver();\n\
84     </script>\n\
85   </head>\n\
86   <body>\n\
87     <h1><span id=\"pkg_name\">N/A</span></h1>\n\
88     <div id=\"version\"><em>Version:</em></div>\n\
89     <p><em>PPA</em>: \n\
90        <a href=\"index.html\">\n\
91            <span id=\"ppa_owner\">N/A</span>/<span id=\"ppa_name\">N/A</span>\n\
92        </a></p>\n\
93     <div id=\"charts\">\n\
94         <div id=\"chart\"></div>\n\
95     </div>\n\
96     <div id=\"details\">\n\
97       <div class=\"distros\">\n\
98         <em>Distros</em>:\n\
99         <ul id=\"distros\"></ul>\n\
100       </div>\n\
101     </div>\n\
102 %s"
103
104 #define HTML_HEADER \
105 "<html>\n\
106         <head>\n\
107                 <title>%s</title>\n\
108                 <link type=\"text/css\"\n\
109                       rel=\"stylesheet\"\n\
110                       href=\
111 \"http://fonts.googleapis.com/css?family=Ubuntu:regular,bold&subset=Latin\">\n\
112                 <link type=\"text/css\" href=\"css/ppastats.css\"\n\
113                       rel=\"stylesheet\"/>\n\
114                 <link type=\"text/css\" href=\"css/jquery.jqplot.min.css\"\n\
115                       rel=\"stylesheet\" />\n\
116 <!--[if lt IE 9]><script \
117 language=\"javascript\" \
118 type=\"text/javascript\" \
119 src=\"js/excanvas.js\"></script><![endif]-->\n\
120                 <script type=\"text/javascript\"\n\
121                         src=\"js/jquery.min.js\"></script>\n\
122                 <script type=\"text/javascript\"\n\
123                         src=\"js/jquery.jqplot.min.js\"></script>\n\
124                 <script type=\"text/javascript\"\n\
125                         src=\"js/jqplot.dateAxisRenderer.min.js\"></script>\n\
126                 <script type=\"text/javascript\"\n\
127                         src=\"js/ppastats.js\"></script>\n\
128                 <script>%s</script>\n\
129         </head>\n\
130         <body>\n"
131
132 #define HTML_INDEX_TEMPLATE \
133 "       <h1><span id=\"ppa_name\">N/A</span></h1>\n\
134         <div id=\"details\">\n\
135                 <div class=\"pkgs\">\n\
136                         <em>Packages</em>:\n\
137                         <ul id=\"pkgs\"></ul>\n\
138                 </div>\n\
139         </div>\n\
140         <div  id=\"charts\">\n\
141                 <div id=\"chart\"></div>\n\
142         </div>\n"
143
144 static char *path_new(const char *dir, const char *file, const char *suffixe)
145 {
146         char *path = malloc(strlen(dir)+1+
147                             strlen(file)+
148                             (suffixe ? strlen(suffixe) : 0) +
149                             1);
150
151         strcpy(path, dir);
152         strcat(path, "/");
153         strcat(path, file);
154         strcat(path, suffixe);
155
156         return path;
157 }
158
159 static struct json_object *date_to_json(struct tm *tm)
160 {
161         json_object *json;
162
163         json = json_object_new_array();
164         json_object_array_add(json, json_object_new_int(tm->tm_year+1900));
165         json_object_array_add(json, json_object_new_int(tm->tm_mon+1));
166         json_object_array_add(json, json_object_new_int(tm->tm_mday));
167
168         return json;
169 }
170
171 static void json_add_ddts(json_object *json,
172                           struct daily_download_total **ddts)
173 {
174         json_object *json_ddt, *json_ddts;
175         struct daily_download_total *ddt;
176
177         json_ddts = json_object_new_array();
178         json_object_object_add(json, "ddts", json_ddts);
179
180         if (!ddts)
181                 return ;
182
183         while (*ddts) {
184                 ddt = *ddts;
185
186                 json_ddt = json_object_new_object();
187                 json_object_object_add(json_ddt,
188                                        "value",
189                                        json_object_new_int(ddt->count));
190                 json_object_object_add(json_ddt,
191                                        "time",
192                                        date_to_json(&ddt->date));
193
194                 json_object_array_add(json_ddts, json_ddt);
195
196                 ddts++;
197         }
198 }
199
200 static json_object *distro_to_json(struct distro_stats *d)
201 {
202         json_object *json;
203
204         json = json_object_new_object();
205
206         json_object_object_add(json,
207                                "name",
208                                json_object_new_string(d->name));
209
210         json_object_object_add(json,
211                                "count",
212                                json_object_new_int(d->download_count));
213
214         json_add_ddts(json, d->ddts);
215
216         return json;
217 }
218
219 static json_object *
220 pkg_to_json(struct ppa_stats *ppa, struct package_stats *pkg)
221 {
222         json_object *json, *json_versions, *json_distros, *json_distro;
223         struct version_stats **versions;
224         struct distro_stats **distros, *d;
225
226         json = json_object_new_object();
227
228         json_object_object_add(json,
229                                "ppa_name", json_object_new_string(ppa->name));
230         json_object_object_add(json,
231                                "ppa_owner",
232                                json_object_new_string(ppa->owner));
233
234         json_object_object_add(json,
235                                "name", json_object_new_string(pkg->name));
236
237         json_versions = json_object_new_array();
238         json_object_object_add(json, "versions", json_versions);
239         versions = pkg->versions;
240         while (*versions) {
241                 json_object_array_add
242                         (json_versions,
243                          json_object_new_string((*versions)->version));
244
245                 versions++;
246         }
247
248         distros = pkg->distros;
249         if (distros) {
250                 json_distros = json_object_new_array();
251                 json_object_object_add(json, "distros", json_distros);
252
253                 while (*distros) {
254                         d = *distros;
255
256                         if (d->download_count) {
257                                 json_distro = distro_to_json(d);
258
259                                 json_object_array_add(json_distros,
260                                                       json_distro);
261                         }
262
263                         distros++;
264                 }
265         }
266
267         json_add_ddts(json, pkg->daily_download_totals);
268
269         return json;
270 }
271
272 static char *version_to_json(struct ppa_stats *ppa,
273                              struct package_stats *pkg,
274                              struct version_stats *ver)
275 {
276         char *ret;
277         struct distro_stats **distros, *distro;
278         json_object *json, *json_distros, *json_distro, *json_archs, *json_arch;
279         struct arch_stats **archs;
280
281         json = json_object_new_object();
282
283         json_object_object_add(json,
284                                "ppa_name", json_object_new_string(ppa->name));
285         json_object_object_add(json,
286                                "ppa_owner",
287                                json_object_new_string(ppa->owner));
288
289         json_object_object_add(json,
290                                "pkg_name", json_object_new_string(pkg->name));
291
292         json_object_object_add(json,
293                                "name", json_object_new_string(ver->version));
294
295         json_add_ddts(json, ver->daily_download_totals);
296
297         distros = ver->distros;
298         json_distros = json_object_new_array();
299         json_object_object_add(json, "distros", json_distros);
300         while (*distros) {
301                 distro = *distros;
302                 json_distro = json_object_new_object();
303
304                 json_object_array_add(json_distros, json_distro);
305
306                 json_object_object_add(json_distro,
307                                        "name",
308                                        json_object_new_string(distro->name));
309
310                 archs = distro->archs;
311                 json_archs = json_object_new_array();
312                 json_object_object_add(json_distro, "archs", json_archs);
313                 while (*archs) {
314                         json_arch = json_object_new_object();
315
316                         json_object_object_add
317                                 (json_arch,
318                                  "name",
319                                  json_object_new_string((*archs)->name));
320
321                         json_object_object_add
322                                 (json_arch,
323                                  "count",
324                                  json_object_new_int((*archs)->download_count));
325
326                         json_object_array_add(json_archs, json_arch);
327                         archs++;
328                 }
329
330                 distros++;
331         }
332
333         ret = strdup(json_object_to_json_string(json));
334
335         json_object_put(json);
336
337         return ret;
338 }
339
340 static json_object *ppa_to_json(struct ppa_stats *ppa)
341 {
342         json_object *json, *json_pkgs, *json_pkg;
343         struct package_stats **pkgs;
344
345         json = json_object_new_object();
346
347         json_object_object_add(json,
348                                "ppa_name", json_object_new_string(ppa->name));
349         json_object_object_add(json,
350                                "ppa_owner",
351                                json_object_new_string(ppa->owner));
352
353         json_add_ddts(json, ppa->daily_download_totals);
354
355         pkgs = ppa->packages;
356         json_pkgs = json_object_new_array();
357         json_object_object_add(json, "packages", json_pkgs);
358         while (*pkgs) {
359                 json_pkg = json_object_new_object();
360                 json_object_array_add(json_pkgs, json_pkg);
361
362                 json_object_object_add(json_pkg, "name",
363                                        json_object_new_string((*pkgs)->name));
364
365                 json_object_object_add
366                         (json_pkg, "count",
367                          json_object_new_int((*pkgs)->download_count));
368
369                 pkgs++;
370         }
371
372         return json;
373 }
374
375 static void
376 version_to_html(struct ppa_stats *ppa,
377                 struct package_stats *pkg,
378                 struct version_stats *version,
379                 const char *dir)
380 {
381         char *f_name, *path;
382         FILE *f;
383
384         f_name = malloc(strlen(pkg->name)+1+strlen(version->version)+1);
385         sprintf(f_name, "%s_%s", pkg->name, version->version);
386
387         path = path_new(dir, f_name, ".html");
388         f = fopen(path, "w");
389
390         if (!f) {
391                 log_err(_("failed to open: %s"), path);
392                 return ;
393         }
394
395         fprintf(f, HTML_VERSION_TEMPLATE,
396                 version_to_json(ppa, pkg, version),
397                 HTML_FOOTER);
398
399         fclose(f);
400
401         free(path);
402         free(f_name);
403 }
404
405 static void
406 create_html(const char *path,
407             const char *title,
408             const char *body_template,
409             const char *script)
410 {
411         FILE *f;
412
413         f = fopen(path, "w");
414
415         if (!f) {
416                 log_err(_("failed to open: %s"), path);
417                 return ;
418         }
419
420         fprintf(f, HTML_HEADER, title, script);
421         fputs(body_template, f);
422         fputs(HTML_FOOTER, f);
423
424         fclose(f);
425 }
426
427 static char *ppa_display_name(const struct ppa_stats *ppa)
428 {
429         char *ret;
430
431         ret = malloc(4+strlen(ppa->name)+1+strlen(ppa->owner)+1);
432
433         sprintf(ret, "ppa:%s/%s", ppa->owner, ppa->name);
434
435         return ret;
436 }
437
438 static void
439 index_to_html(struct ppa_stats *ppa, const char *dir)
440 {
441         char *path, *json_path, *dname;
442         json_object *json;
443
444         json = ppa_to_json(ppa);
445         json_path = path_new(dir, "index", ".json");
446
447         log_debug(_("generating %s"), json_path);
448         json_object_to_file(json_path, json);
449         json_object_put(json);
450         free(json_path);
451
452         path = path_new(dir, "index", ".html");
453         dname = ppa_display_name(ppa);
454         create_html(path, dname, HTML_INDEX_TEMPLATE, "ppastats_ppa();");
455         free(path);
456         free(dname);
457 }
458
459 static void
460 pkg_to_html(struct ppa_stats *ppa, struct package_stats *pkg, const char *dir)
461 {
462         char *path, *json_path, *script;
463         json_object *json;
464
465         json_path = path_new(dir, pkg->name, ".json");
466         json = pkg_to_json(ppa, pkg);
467         log_debug(_("generating %s"), json_path);
468
469         json_object_to_file(json_path, json);
470         json_object_put(json);
471         free(json_path);
472
473         path = path_new(dir, pkg->name, ".html");
474         script = malloc(strlen("ppastats_pkg(\"\");")+
475                         strlen(pkg->name)+
476                         strlen(".json")+
477                         1);
478         sprintf(script, "ppastats_pkg(\"%s%s\");", pkg->name, ".json");
479
480         log_debug(_("generating %s"), path);
481
482         create_html(path, pkg->name, HTML_PKG_TEMPLATE, script);
483         free(path);
484         free(script);
485 }
486
487 static void
488 pkgs_to_html(struct ppa_stats *ppa,
489              struct package_stats **pkgs,
490              const char *dir)
491 {
492         struct version_stats **versions;
493
494         while (*pkgs) {
495                 pkg_to_html(ppa, *pkgs, dir);
496
497                 versions = (*pkgs)->versions;
498                 while (*versions) {
499                         version_to_html(ppa, *pkgs, *versions, dir);
500
501                         versions++;
502                 }
503
504                 pkgs++;
505         }
506 }
507
508 void
509 ppa_to_html(const char *owner,
510             const char *ppa,
511             const char *package_status,
512             const char *output_dir,
513             const int install_static_files)
514 {
515         struct ppa_stats *ppastats;
516         char *path, *f_dst;
517         char *css_dir, *js_dir;
518         int i;
519         static char *www_files[]
520                 = { DEFAULT_WWW_DIR"/jquery.min.js", "js/jquery.min.js",
521                     DEFAULT_WWW_DIR"/ppastats.js", "js/ppastats.js",
522                     DEFAULT_WWW_DIR"/jqplot.dateAxisRenderer.min.js",
523                     "js/jqplot.dateAxisRenderer.min.js",
524                     DEFAULT_WWW_DIR"/jquery.jqplot.min.js",
525                     "js/jquery.jqplot.min.js",
526                     DEFAULT_WWW_DIR"/excanvas.js", "js/excanvas.js",
527                     DEFAULT_WWW_DIR"/ppastats.css", "css/ppastats.css",
528                     DEFAULT_WWW_DIR"/jquery.jqplot.min.css",
529                     "css/jquery.jqplot.min.css" };
530
531         mkdirs(output_dir, 0777);
532
533         if (install_static_files) {
534                 css_dir = path_append(output_dir, "css");
535                 js_dir = path_append(output_dir, "js");
536
537                 mkdir(css_dir, 0777);
538                 mkdir(js_dir, 0777);
539
540                 for (i = 0; i < 7; i++) {
541                         f_dst = path_append(output_dir, www_files[2*i+1]);
542
543                         log_debug(_("copying %s %s"), www_files[2*i], f_dst);
544                         fcopy(www_files[2*i], f_dst);
545
546                         free(f_dst);
547                 }
548                 free(css_dir);
549                 free(js_dir);
550         }
551
552         ppastats = create_ppa_stats(owner, ppa, package_status);
553
554         path = path_new(output_dir, "ppa", ".html");
555
556         pkgs_to_html(ppastats, ppastats->packages, output_dir);
557
558         index_to_html(ppastats, output_dir);
559
560         ppa_stats_free(ppastats);
561
562         free(path);
563 }