src/ppastats.c

   1 /*
   2   Copyright (C) 2011-2014 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
  27 #include <list.h>
  28 #include <lp_ws.h>
  29 #include <plog.h>
  30 #include <ppastats.h>
  31
  32 static void arch_stats_free(struct arch_stats *arch)
  33 {
  34         free(arch->name);
  35         free(arch);
  36 }
  37
  38 static struct distro_stats *distro_stats_new(const char *name)
  39 {
  40         struct distro_stats *d;
  41
  42         d = malloc(sizeof(struct distro_stats));
  43         d->name = strdup(name);
  44         d->archs = NULL;
  45         d->download_count = 0;
  46         d->ddts = NULL;
  47
  48         return d;
  49 }
  50
  51 static void distro_stats_free(struct distro_stats *distro)
  52 {
  53         struct arch_stats **archs;
  54
  55         archs = distro->archs;
  56         if (archs) {
  57                 while (*archs) {
  58                         arch_stats_free(*archs);
  59                         archs++;
  60                 }
  61                 free(distro->archs);
  62         }
  63
  64         daily_download_total_list_free(distro->ddts);
  65
  66         free(distro->name);
  67         free(distro);
  68 }
  69
  70 static void distro_stats_list_free(struct distro_stats **distros)
  71 {
  72         struct distro_stats **cur;
  73
  74         if (distros) {
  75                 cur = distros;
  76                 while (*cur) {
  77                         distro_stats_free(*cur);
  78                         cur++;
  79                 }
  80                 free(distros);
  81         }
  82 }
  83
  84 static void version_stats_free(struct version_stats *version)
  85 {
  86         distro_stats_list_free(version->distros);
  87         daily_download_total_list_free(version->daily_download_totals);
  88
  89         free(version->version);
  90         free(version);
  91 }
  92
  93 static void package_stats_free(struct package_stats *package)
  94 {
  95         struct version_stats **versions;
  96
  97         versions = package->versions;
  98         if (versions) {
  99                 while (*versions) {
 100                         version_stats_free(*versions);
 101                         versions++;
 102                 }
 103                 free(package->versions);
 104         }
 105         distro_stats_list_free(package->distros);
 106         daily_download_total_list_free(package->daily_download_totals);
 107         free(package->name);
 108         free(package);
 109 }
 110
 111 static struct package_stats *package_stats_new(const char *name)
 112 {
 113         struct package_stats *p;
 114
 115         p = malloc(sizeof(struct package_stats));
 116         p->name = strdup(name);
 117         p->versions = NULL;
 118         p->download_count = 0;
 119         p->daily_download_totals = NULL;
 120         p->distros = NULL;
 121
 122         return p;
 123 }
 124
 125 static struct package_stats *get_package_stats(struct ppa_stats *stats,
 126                                                const char *name)
 127
 128 {
 129         struct package_stats *p, **p_cur, **tmp;
 130
 131         p_cur = stats->packages;
 132         while (p_cur && *p_cur) {
 133                 struct package_stats *p = *p_cur;
 134
 135                 if (!strcmp(p->name, name))
 136                         return p;
 137
 138                 p_cur++;
 139         }
 140
 141         p = package_stats_new(name);
 142
 143         tmp = (struct package_stats **)list_add((void **)stats->packages, p);
 144         free(stats->packages);
 145         stats->packages = tmp;
 146
 147         return p;
 148 }
 149
 150 static struct version_stats *version_stats_new(const char *version)
 151 {
 152         struct version_stats *v;
 153
 154         v = malloc(sizeof(struct version_stats));
 155         v->version = strdup(version);
 156         v->distros = NULL;
 157         v->download_count = 0;
 158         v->daily_download_totals = NULL;
 159         v->date_created = 0;
 160
 161         return v;
 162 }
 163
 164 static struct version_stats *get_version_stats(struct package_stats *package,
 165                                                const char *version)
 166 {
 167         struct version_stats *v, **cur, **tmp;
 168
 169         cur = package->versions;
 170         while (cur && *cur) {
 171                 struct version_stats *v = *cur;
 172
 173                 if (!strcmp(v->version, version))
 174                         return v;
 175
 176                 cur++;
 177         }
 178
 179         v = version_stats_new(version);
 180
 181         tmp = (struct version_stats **)list_add((void **)package->versions,
 182                                                 v);
 183         free((void **)package->versions);
 184         package->versions = tmp;
 185
 186         return v;
 187 }
 188
 189 static struct distro_stats *get_distro_stats(struct version_stats *version,
 190                                              const char *name)
 191 {
 192         struct distro_stats **cur, *d, **tmp;
 193
 194         cur = version->distros;
 195
 196         while (cur && *cur) {
 197                 d = *cur;
 198
 199                 if (!strcmp(d->name, name))
 200                         return d;
 201
 202                 cur++;
 203         }
 204
 205         d = distro_stats_new(name);
 206
 207         tmp = (struct distro_stats **)list_add((void **)version->distros,
 208                                                d);
 209         free(version->distros);
 210         version->distros = tmp;
 211
 212         return d;
 213 }
 214
 215 static struct arch_stats *get_arch_stats(struct distro_stats *distro,
 216                                          const char *name)
 217 {
 218         struct arch_stats **cur, *a, **tmp;
 219
 220         cur = distro->archs;
 221         while (cur && *cur) {
 222                 a = *cur;
 223
 224                 if (!strcmp(a->name, name))
 225                         return a;
 226
 227                 cur++;
 228         }
 229
 230         a = malloc(sizeof(struct arch_stats));
 231         a->name = strdup(name);
 232         a->download_count = 0;
 233
 234         tmp = (struct arch_stats **)list_add((void **)distro->archs,
 235                                              a);
 236         free((void **)distro->archs);
 237         distro->archs = tmp;
 238
 239         return a;
 240 }
 241
 242
 243 static struct daily_download_total **add_total
 244 (struct daily_download_total **totals, struct daily_download_total *total)
 245 {
 246         struct daily_download_total **cur, **result, *item;
 247
 248         if (totals) {
 249                 cur = totals;
 250                 while (*cur) {
 251                         item = *cur;
 252
 253                         if (item->date.tm_year == total->date.tm_year &&
 254                             item->date.tm_mon == total->date.tm_mon &&
 255                             item->date.tm_mday == total->date.tm_mday) {
 256                                 item->count += total->count;
 257                                 return totals;
 258                         }
 259
 260                         cur++;
 261                 }
 262         }
 263
 264         result = (struct daily_download_total **)list_add((void **)totals,
 265                                                           ddt_clone(total));
 266         return result;
 267 }
 268
 269 static struct daily_download_total **add_totals
 270 (struct daily_download_total **total1, struct daily_download_total **total2)
 271 {
 272         struct daily_download_total **cur, **result, **tmp;
 273
 274         result = total1;
 275         cur = total2;
 276         while (*cur) {
 277                 tmp = add_total(result, *cur);
 278                 if (result != total1 && result != tmp)
 279                         free(result);
 280                 result = tmp;
 281                 cur++;
 282         }
 283
 284         return result;
 285 }
 286
 287 static void
 288 pkg_add_distro(struct package_stats *pkg,
 289                const char *distro_name,
 290                int distro_count,
 291                struct daily_download_total **ddts)
 292 {
 293         struct distro_stats **pkg_distros, *pkg_distro, **tmp;
 294         struct daily_download_total **tmp_ddts;
 295
 296         pkg_distros = pkg->distros;
 297         pkg_distro = NULL;
 298
 299         if (pkg_distros)
 300                 while (*pkg_distros)  {
 301                         if (!strcmp((*pkg_distros)->name, distro_name)) {
 302                                 pkg_distro = *pkg_distros;
 303                                 break;
 304                         }
 305
 306                         pkg_distros++;
 307                 }
 308
 309         if (!pkg_distro) {
 310                 pkg_distro = distro_stats_new(distro_name);
 311                 tmp = (struct distro_stats **)list_add((void **)pkg->distros,
 312                                                        (void *)pkg_distro);
 313                 if (pkg->distros != tmp)
 314                         free(pkg->distros);
 315                 pkg->distros = tmp;
 316         }
 317
 318         pkg_distro->download_count += distro_count;
 319
 320         tmp_ddts = add_totals(pkg_distro->ddts, ddts);
 321         if (pkg_distro->ddts && pkg_distro->ddts != tmp_ddts)
 322                 free(pkg_distro->ddts);
 323         pkg_distro->ddts = tmp_ddts;
 324 }
 325
 326 static struct ppa_stats *ppa_stats_new(const char *owner, const char *ppa_name)
 327 {
 328         struct ppa_stats *ppa;
 329
 330         ppa = malloc(sizeof(struct ppa_stats));
 331         ppa->name = strdup(ppa_name);
 332         ppa->owner = strdup(owner);
 333         ppa->packages = NULL;
 334         ppa->daily_download_totals = NULL;
 335         ppa->download_count = 0;
 336
 337         return ppa;
 338 }
 339
 340 struct ppa_stats *
 341 create_ppa_stats(const char *owner,
 342                  const char *ppa_name,
 343                  const char *package_status,
 344                  int ws_size)
 345 {
 346         struct ppa_stats *ppa;
 347         struct bpph **history, **h_cur, *h;
 348         char *ppa_url, *pkg_name, *pkg_version;
 349         struct package_stats *pkg;
 350         struct version_stats *version;
 351         const struct distro_series *distro_series;
 352         const struct distro_arch_series *arch_series;
 353         struct distro_stats *distro;
 354         struct arch_stats *arch;
 355         int count;
 356         struct daily_download_total **totals, **tmp;
 357
 358         ppa_url = get_archive_url(owner, ppa_name);
 359         history = get_bpph_list(ppa_url, package_status, ws_size);
 360         free(ppa_url);
 361
 362         if (!history) {
 363                 log_err(_("Failed to retrieve PPA information"));
 364                 exit(EXIT_FAILURE);
 365         }
 366
 367         ppa = ppa_stats_new(owner, ppa_name);
 368
 369         for (h_cur = history; *h_cur; ++h_cur) {
 370                 h = *h_cur;
 371                 totals = get_daily_download_totals(h->self_link,
 372                                                    h->date_created);
 373                 if (!totals) {
 374                         log_err(_("Failed to retrieve download totals for %s"),
 375                                 h->self_link);
 376                         continue;
 377                 }
 378                 count = ddts_get_count(totals);
 379                 pkg_name = h->binary_package_name;
 380                 pkg_version = h->binary_package_version;
 381                 arch_series
 382                         = get_distro_arch_series(h->distro_arch_series_link);
 383                 distro_series
 384                         = get_distro_series(arch_series->distroseries_link);
 385
 386                 ppa->download_count += count;
 387                 tmp = add_totals(ppa->daily_download_totals, totals);
 388                 if (ppa->daily_download_totals != tmp)
 389                         free(ppa->daily_download_totals);
 390                 ppa->daily_download_totals = tmp;
 391
 392                 pkg = get_package_stats(ppa, pkg_name);
 393                 pkg->download_count += count;
 394                 tmp = add_totals(pkg->daily_download_totals, totals);
 395                 if (pkg->daily_download_totals != tmp)
 396                         free(pkg->daily_download_totals);
 397                 pkg->daily_download_totals = tmp;
 398
 399                 version = get_version_stats(pkg, pkg_version);
 400                 version->date_created = h->date_created;
 401
 402                 version->download_count += count;
 403                 tmp = add_totals(version->daily_download_totals, totals);
 404                 if (version->daily_download_totals != tmp)
 405                         free(version->daily_download_totals);
 406                 version->daily_download_totals = tmp;
 407
 408                 distro = get_distro_stats(version, distro_series->name);
 409                 distro->download_count += count;
 410
 411                 arch = get_arch_stats(distro, arch_series->architecture_tag);
 412                 arch->download_count += count;
 413
 414                 pkg_add_distro(pkg, distro_series->name, count, totals);
 415
 416                 daily_download_total_list_free(totals);
 417         }
 418
 419         bpph_list_free(history);
 420
 421         return ppa;
 422 }
 423
 424 void ppa_stats_free(struct ppa_stats *ppastats)
 425 {
 426         struct package_stats **packages;
 427
 428         packages = ppastats->packages;
 429         if (packages) {
 430                 while (*packages) {
 431                         package_stats_free(*packages);
 432                         packages++;
 433                 }
 434                 free(ppastats->packages);
 435         }
 436
 437         free(ppastats->owner);
 438         free(ppastats->name);
 439
 440         daily_download_total_list_free(ppastats->daily_download_totals);
 441
 442         free(ppastats);
 443 }