X-Git-Url: https://git.wpitchoune.net/gitweb/?a=blobdiff_plain;f=trunk%2Fsrc%2Flp_ws.c;fp=trunk%2Fsrc%2Flp_ws.c;h=4b51be80b31acd9f86710c86bb7529c45651cdea;hb=b19be7b6db9f737f9a9a22b3079abc774cc7583c;hp=0000000000000000000000000000000000000000;hpb=b19f912053682528d7efa15aa5641a1f055f3038;p=ppastats.git diff --git a/trunk/src/lp_ws.c b/trunk/src/lp_ws.c new file mode 100644 index 0000000..4b51be8 --- /dev/null +++ b/trunk/src/lp_ws.c @@ -0,0 +1,266 @@ +/* + 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 +*/ + +#include +#include + +#include +#include + +#include "cache.h" +#include "list.h" +#include "lp_ws.h" +#include "lp_json.h" +#include "ppastats.h" + +#define QUERY_GET_PUBLISHED_BINARIES \ + "?ws.op=getPublishedBinaries" +#define QUERY_GET_DOWNLOAD_COUNT "?ws.op=getDownloadCount" +#define QUERY_GET_DAILY_DOWNLOAD_TOTALS \ + "?ws.op=getDailyDownloadTotals" + +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) +{ + size_t realsize = size * nmemb; + struct ucontent *mem = (struct ucontent *)userp; + + mem->data = realloc(mem->data, mem->len + realsize + 1); + + memcpy(&(mem->data[mem->len]), buffer, realsize); + mem->len += realsize; + mem->data[mem->len] = 0; + + return realsize; +} + +static char *fetch_url(const char *url) +{ + struct ucontent *content = malloc(sizeof(struct ucontent)); + char *result = NULL; + long code; + + if (debug) + printf("DEBUG: fetch_url %s\n", url); + + if (!curl) { + curl_global_init(CURL_GLOBAL_ALL); + curl = curl_easy_init(); + } + + if (!curl) + exit(EXIT_FAILURE); + + content->data = malloc(1); + content->data[0] = '\0'; + content->len = 0; + + 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"); + + if (curl_easy_perform(curl) == CURLE_OK) { + + curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &code); + if (code == 200) + result = content->data; + } + + if (!result) + free(content->data); + + free(content); + + return result; +} + +static json_object *get_json_object(const char *url) +{ + json_object *obj = NULL; + char *body; + + body = fetch_url(url); + + if (body) { + obj = json_tokener_parse(body); + + free(body); + + return obj; + } + + return NULL; +} + +#define json_object_to_bpph_list \ +json_object_to_binary_package_publishing_history_list + +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; + + url = malloc(strlen(archive_url)+ + strlen(QUERY_GET_PUBLISHED_BINARIES)+ + (pkg_status ? strlen("&status=")+strlen(pkg_status) : 0)+ + 1); + + strcpy(url, archive_url); + strcat(url, QUERY_GET_PUBLISHED_BINARIES); + + if (pkg_status) { + strcat(url, "&status="); + strcat(url, pkg_status); + } + + while (url) { + o = get_json_object(url); + free(url); + url = NULL; + + if (!o) + break; + + result = list_append_list(result, + (void **)json_object_to_bpph_list(o)); + + o_next = json_object_object_get(o, "next_collection_link"); + + if (o_next) + url = strdup(json_object_get_string(o_next)); + + json_object_put(o); + } + + return (struct binary_package_publishing_history **)result; +} + +int get_download_count(const char *archive_url) +{ + int n = strlen(archive_url) + strlen(QUERY_GET_DOWNLOAD_COUNT) + 1; + char *url = malloc(n); + int result; + json_object *obj; + + strcpy(url, archive_url); + strcat(url, QUERY_GET_DOWNLOAD_COUNT); + + obj = get_json_object(url); + free(url); + + if (!obj) + return -1; + + result = json_object_get_int(obj); + + json_object_put(obj); + + return result; +} + +const struct distro_arch_series *get_distro_arch_series(const char *url) +{ + json_object *obj; + const struct distro_arch_series *distro; + + distro = cache_get(url); + if (distro) + return (struct distro_arch_series *)distro; + + obj = get_json_object(url); + + if (!obj) + return NULL; + + distro = json_object_to_distro_arch_series(obj); + + json_object_put(obj); + + cache_put(url, distro, (void (*)(void *))&distro_arch_series_free); + + return distro; +} + +const struct distro_series *get_distro_series(const char *url) +{ + json_object *obj; + const struct distro_series *distro; + + distro = cache_get(url); + if (distro) + return (struct distro_series *)distro; + + obj = get_json_object(url); + + if (!obj) + return NULL; + + distro = json_object_to_distro_series(obj); + + json_object_put(obj); + + cache_put(url, distro, (void (*)(void *))&distro_series_free); + + return distro; +} + +struct daily_download_total **get_daily_download_totals(const char *binary_url) +{ + char *url; + json_object *obj; + struct daily_download_total **result = NULL; + + url = malloc(strlen(binary_url)+ + strlen(QUERY_GET_DAILY_DOWNLOAD_TOTALS)+1); + + strcpy(url, binary_url); + strcat(url, QUERY_GET_DAILY_DOWNLOAD_TOTALS); + + obj = get_json_object(url); + + if (obj) { + result = json_object_to_daily_download_totals(obj); + json_object_put(obj); + } + + free(url); + + return result; +} + +void lp_ws_cleanup() +{ + if (debug) + printf("DEBUG: cleanup CURL\n"); + + curl_easy_cleanup(curl); + curl_global_cleanup(); +}