Pileus Git - grits/blob - src/data.c

   1 /*
   2  * Copyright (C) 2009 Andy Spencer <spenceal@rose-hulman.edu>
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU 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, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include <config.h>
  19 #include <stdio.h>
  20 #include <glib.h>
  21 #include <libsoup/soup.h>
  22
  23 #include "data.h"
  24
  25 typedef struct {
  26         AWeatherCacheDoneCallback callback;
  27         gpointer user_data;
  28         gchar *local;
  29         FILE  *fp;
  30 } cache_file_end_t;
  31
  32 /*
  33  * Open a file, creating parent directories if needed
  34  */
  35 static FILE *fopen_p(const gchar *path, const gchar *mode)
  36 {
  37         gchar *parent = g_path_get_dirname(path);
  38         if (!g_file_test(parent, G_FILE_TEST_EXISTS))
  39                 g_mkdir_with_parents(path, 0755);
  40         g_free(parent);
  41         return fopen(path, mode);
  42 }
  43
  44 static void cache_file_cb(SoupSession *session, SoupMessage *message, gpointer _info)
  45 {
  46         cache_file_end_t *info = _info;
  47         gchar *uri = soup_uri_to_string(soup_message_get_uri(message), FALSE);
  48         g_debug("data: cache_file_cb ([%s]->[%s])", uri, info->local);
  49
  50         if (!SOUP_STATUS_IS_SUCCESSFUL(message->status_code)) {
  51                 g_warning("data: error copying file ([%s]->[%s])", uri, info->local);
  52         } else {
  53                 gint wrote = fwrite(message->response_body->data,  1,
  54                                 message->response_body->length, info->fp);
  55                 g_debug("data: status=%u wrote=%d/%lld",
  56                                 message->status_code,
  57                                 wrote, message->response_body->length);
  58                 fclose(info->fp);
  59                 info->callback(info->local, TRUE, info->user_data);
  60         }
  61         g_free(uri);
  62         g_free(info->local);
  63         g_object_unref(session);
  64 }
  65
  66 static void do_cache(gchar *uri, gchar *local, gboolean truncate, gchar *reason,
  67                 AWeatherCacheDoneCallback callback, gpointer user_data)
  68 {
  69         char *name = g_path_get_basename(uri);
  70         g_debug("data: do_cache - Caching file %s: %s", name, reason);
  71         g_free(name);
  72
  73         cache_file_end_t *info = g_malloc0(sizeof(cache_file_end_t));
  74         info->callback  = callback;
  75         info->user_data = user_data;
  76         info->local     = local;
  77
  78         if (truncate) info->fp = fopen_p(local, "w");
  79         else          info->fp = fopen_p(local, "a");
  80         long bytes = ftell(info->fp);
  81
  82         SoupSession *session = soup_session_async_new();
  83         SoupMessage *message = soup_message_new("GET", uri);
  84         if (message == NULL)
  85                 g_error("message is null, cannot parse uri");
  86         if (bytes != 0)
  87                 soup_message_headers_set_range(message->request_headers, bytes, -1);
  88         soup_session_queue_message(session, message, cache_file_cb, info);
  89 }
  90
  91 /*
  92  * Cache a image from Ridge to the local disk
  93  * \param  path  Path to the Ridge file, starting after /ridge/
  94  * \return The local path to the cached image
  95  */
  96 void cache_file(char *base, char *path, AWeatherCacheType update,
  97                 AWeatherCacheDoneCallback callback, gpointer user_data)
  98 {
  99         gchar *uri   = g_strconcat(base, path, NULL);
 100         gchar *local = g_build_filename(g_get_user_cache_dir(), PACKAGE, path, NULL);
 101
 102         if (update == AWEATHER_REFRESH)
 103                 return do_cache(uri, local, TRUE, "cache forced",
 104                                 callback, user_data);
 105
 106         if (update == AWEATHER_UPDATE)
 107                 return do_cache(uri, local, FALSE, "attempting updating",
 108                                 callback, user_data);
 109
 110         if (update == AWEATHER_ONCE && !g_file_test(local, G_FILE_TEST_EXISTS))
 111                 return do_cache(uri, local, TRUE, "local does not exist",
 112                                 callback, user_data);
 113
 114         /* No nead to cache, run the callback now and clean up */
 115         callback(local, FALSE, user_data);
 116         g_free(local);
 117         g_free(uri);
 118         return;
 119 }