removed used local var initialization
[ppastats.git] / src / lp_ws.c
index 5eca19a..ecc95cc 100644 (file)
 /*
-    Copyright (C) 2011 jeanfi@gmail.com
-
-    This program is free software; you can redistribute it and/or modify
-    it under the terms of the GNU General Public License as published by
-    the Free Software Foundation; either version 2 of the License, or
-    (at your option) any later version.
-
-    This program is distributed in the hope that it will be useful,
-    but WITHOUT ANY WARRANTY; without even the implied warranty of
-    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-    GNU General Public License for more details.
-
-    You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
-    02110-1301 USA
-*/
-
+ * Copyright (C) 2011-2012 jeanfi@gmail.com
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301 USA
+ */
+
+#include <libintl.h>
+#define _(String) gettext(String)
+
+#include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <unistd.h>
 
-#include <curl/curl.h>
 #include <json/json.h>
 
 #include "cache.h"
+#include "fcache.h"
+#include "http.h"
 #include "list.h"
+#include "log.h"
 #include "lp_ws.h"
 #include "lp_json.h"
 #include "ppastats.h"
 
-static const char *QUERY_GET_PUBLISHED_BINARIES = "?ws.op=getPublishedBinaries";
+static const char *
+QUERY_GET_PUBLISHED_BINARIES = "?ws.op=getPublishedBinaries&ws.size=150";
 static const char *QUERY_GET_DOWNLOAD_COUNT = "?ws.op=getDownloadCount";
 static const char *
 QUERY_GET_DAILY_DOWNLOAD_TOTALS = "?ws.op=getDailyDownloadTotals";
 
-static const int DEFAULT_FETCH_RETRIES = 3;
-
-static CURL *curl;
-
-struct ucontent {
-       char *data;
-       size_t len;
-};
-
-static size_t cbk_curl(void *buffer, size_t size, size_t nmemb, void *userp)
+static json_object *get_json_object(const char *url)
 {
-       size_t realsize = size * nmemb;
-       struct ucontent *mem = (struct ucontent *)userp;
+       json_object *obj = NULL;
+       char *body;
 
-       mem->data = realloc(mem->data, mem->len + realsize + 1);
+       body = get_url_content(url, 0);
 
-       memcpy(&(mem->data[mem->len]), buffer, realsize);
-       mem->len += realsize;
-       mem->data[mem->len] = 0;
+       if (body) {
+               obj = json_tokener_parse(body);
 
-       return realsize;
-}
+               free(body);
 
-static char *fetch_url(const char *url)
-{
-       struct ucontent *content = malloc(sizeof(struct ucontent));
-       char *result;
-       long code;
-       int retries;
-
-       if (debug)
-               printf("DEBUG: fetch_url %s\n", url);
-
-       if (!curl) {
-               if (debug)
-                       printf("DEBUG: initializing CURL\n");
-               curl_global_init(CURL_GLOBAL_ALL);
-               curl = curl_easy_init();
+               return obj;
        }
 
-       if (!curl)
-               exit(EXIT_FAILURE);
+       return NULL;
+}
 
-       result = NULL;
+static char *get_bpph_list_cache_key(const char *archive_url)
+{
+       char *key;
 
-       retries = DEFAULT_FETCH_RETRIES;
+       key = malloc(strlen(archive_url + 7) + strlen("/bpph") + 1);
+       sprintf(key, "%s/bpph", archive_url + 7);
 
- retrieve:
-       content->data = malloc(1);
-       content->data[0] = '\0';
-       content->len = 0;
+       return key;
+}
 
-       curl_easy_setopt(curl, CURLOPT_URL, url);
-       curl_easy_setopt(curl, CURLOPT_VERBOSE, 0);
-       curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, cbk_curl);
-       curl_easy_setopt(curl, CURLOPT_WRITEDATA, content);
-       curl_easy_setopt(curl, CURLOPT_USERAGENT, "ppastats/0.0");
+static struct bpph **get_bpph_list_from_cache(const char *key)
+{
+       char *content;
+       struct bpph **list;
+       json_object *json;
 
-       if (curl_easy_perform(curl) == CURLE_OK) {
-               curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &code);
+       content = fcache_get(key);
+       if (!content)
+               return NULL;
 
-               switch (code) {
-               case 200:
-                       result = content->data;
-                       break;
-               case 500:
-               case 502:
-               case 503:
-               case 504:
-                       if (retries) {
-                               fprintf(stderr,
-                                       "Fetch failed: with code %ld "
-                                       "for URL= %s\n",
-                                       code,
-                                       url);
-
-                               if (debug)
-                                       printf("Wait 5s before retry.\n");
-                               sleep(5);
-
-                               free(content->data);
-                               retries--;
-                               goto retrieve;
-                       }
-               default:
-                       fprintf(stderr, "Fetch failed: %ld\n", code);
-               }
-       }
+       json = json_tokener_parse(content);
+       if (!json)
+               return NULL;
 
-       if (!result)
-               free(content->data);
+       list = json_object_to_bpph_list(json);
 
+       json_object_put(json);
        free(content);
 
-       return result;
+       return list;
 }
 
-static json_object *get_json_object(const char *url)
+static char *get_last_creation_date(struct bpph **list)
 {
-       json_object *obj = NULL;
-       char *body;
+       time_t last, t;
+       struct bpph **cur;
 
-       body = fetch_url(url);
+       last = 0;
 
-       if (body) {
-               obj = json_tokener_parse(body);
+       if (list)
+               for (cur = list; *cur; cur++) {
+                       t = (*cur)->date_created;
+                       if (t > last)
+                               last = t;
+               }
 
-               free(body);
+       if (last)
+               return time_to_str(last);
+       else
+               return NULL;
+}
 
-               return obj;
-       }
+struct bpph **get_bpph_list(const char *archive_url, const char *pkg_status)
+{
+       char *url, *key, *tmp;
+       struct bpph **result;
+       struct json_object *o, *bpph_json, *o_next;
+       char *created_since_date;
+       int ok;
+
+       url = malloc(strlen(archive_url)
+                    + strlen(QUERY_GET_PUBLISHED_BINARIES)
+                    + 1);
+       strcpy(url, archive_url);
+       strcat(url, QUERY_GET_PUBLISHED_BINARIES);
 
-       return NULL;
-}
+       key = get_bpph_list_cache_key(archive_url);
 
-#define json_object_to_bpph_list \
-json_object_to_binary_package_publishing_history_list
+       result = get_bpph_list_from_cache(key);
 
-struct binary_package_publishing_history * *
-get_binary_package_publishing_history_list(const char *archive_url,
-                                          const char *pkg_status)
-{
-       struct json_object *o_next;
-       char *url;
-       json_object *o;
-       void **result = NULL;
+       if (result) {
+               created_since_date = get_last_creation_date(result);
 
-       url = malloc(strlen(archive_url)+
-                    strlen(QUERY_GET_PUBLISHED_BINARIES)+
-                    (pkg_status ? strlen("&status=")+strlen(pkg_status) : 0)+
-                    1);
+               if (created_since_date) {
+                       printf("Update package since: %s\n",
+                              created_since_date);
 
-       strcpy(url, archive_url);
-       strcat(url, QUERY_GET_PUBLISHED_BINARIES);
+                       tmp = malloc(strlen(url)
+                                    + strlen("&created_since_date=")
+                                    + strlen(created_since_date)+1);
+                       strcpy(tmp, url);
+                       strcat(tmp, "&created_since_date=");
+                       strcat(tmp, created_since_date);
+
+                       free(url);
+                       url = tmp;
 
-       if (pkg_status) {
-               strcat(url, "&status=");
-               strcat(url, pkg_status);
+                       free(created_since_date);
+               }
        }
 
+       ok = 1;
        while (url) {
                o = get_json_object(url);
                free(url);
                url = NULL;
 
-               if (!o)
+               if (!o) {
+                       ok = 0;
                        break;
+               }
 
-               result = list_append_list(result,
-                                         (void **)json_object_to_bpph_list(o));
+               result = bpph_list_append_list(result,
+                                              json_object_to_bpph_list(o));
 
                o_next = json_object_object_get(o, "next_collection_link");
 
@@ -194,7 +172,15 @@ get_binary_package_publishing_history_list(const char *archive_url,
                json_object_put(o);
        }
 
-       return (struct binary_package_publishing_history **)result;
+       if (ok) {
+               bpph_json = bpph_list_to_json(result);
+               fcache_put(key, json_object_to_json_string(bpph_json));
+               json_object_put(bpph_json);
+       }
+
+       free(key);
+
+       return result;
 }
 
 int get_download_count(const char *archive_url)
@@ -224,12 +210,20 @@ const struct distro_arch_series *get_distro_arch_series(const char *url)
 {
        json_object *obj;
        const struct distro_arch_series *distro;
+       char *content;
 
        distro = cache_get(url);
        if (distro)
                return (struct distro_arch_series *)distro;
 
-       obj = get_json_object(url);
+       content = get_url_content(url, 1);
+
+       if (!content)
+               return NULL;
+
+       obj = json_tokener_parse(content);
+
+       free(content);
 
        if (!obj)
                return NULL;
@@ -247,12 +241,20 @@ const struct distro_series *get_distro_series(const char *url)
 {
        json_object *obj;
        const struct distro_series *distro;
+       char *content;
 
        distro = cache_get(url);
        if (distro)
                return (struct distro_series *)distro;
 
-       obj = get_json_object(url);
+       content = get_url_content(url, 1);
+
+       if (!content)
+               return NULL;
+
+       obj = json_tokener_parse(content);
+
+       free(content);
 
        if (!obj)
                return NULL;
@@ -290,11 +292,3 @@ struct daily_download_total **get_daily_download_totals(const char *binary_url)
        return result;
 }
 
-void lp_ws_cleanup()
-{
-       if (debug)
-               printf("DEBUG: cleanup CURL\n");
-
-       curl_easy_cleanup(curl);
-       curl_global_cleanup();
-}