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(parent, 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");
  49
  50         if (!SOUP_STATUS_IS_SUCCESSFUL(message->status_code)) {
  51                 g_warning("data: cache_file_cb - error copying file, status=%d\n"
  52                                 "\tsrc=%s\n"
  53                                 "\tdst=%s",
  54                                 message->status_code, uri, info->local);
  55         } else {
  56                 gint wrote = fwrite(message->response_body->data,  1,
  57                                 message->response_body->length, info->fp);
  58                 g_debug("data: status=%u wrote=%d/%lld",
  59                                 message->status_code,
  60                                 wrote, message->response_body->length);
  61                 fclose(info->fp);
  62                 info->callback(info->local, TRUE, info->user_data);
  63         }
  64         g_free(uri);
  65         g_free(info->local);
  66         g_object_unref(session);
  67 }
  68
  69 static void do_cache(gchar *uri, gchar *local, gboolean truncate, gchar *reason,
  70                 AWeatherCacheDoneCallback callback, gpointer user_data)
  71 {
  72         char *name = g_path_get_basename(uri);
  73         g_debug("data: do_cache - Caching file %s: %s", name, reason);
  74         g_free(name);
  75
  76         cache_file_end_t *info = g_malloc0(sizeof(cache_file_end_t));
  77         info->callback  = callback;
  78         info->user_data = user_data;
  79         info->local     = local;
  80
  81         if (truncate) info->fp = fopen_p(local, "w");
  82         else          info->fp = fopen_p(local, "a");
  83         long bytes = ftell(info->fp);
  84
  85         SoupSession *session = soup_session_async_new();
  86         SoupMessage *message = soup_message_new("GET", uri);
  87         if (message == NULL)
  88                 g_error("message is null, cannot parse uri");
  89         if (bytes != 0)
  90                 soup_message_headers_set_range(message->request_headers, bytes, -1);
  91         soup_session_queue_message(session, message, cache_file_cb, info);
  92 }
  93
  94 /*
  95  * Cache a image from Ridge to the local disk
  96  * \param  path  Path to the Ridge file, starting after /ridge/
  97  * \return The local path to the cached image
  98  */
  99 void cache_file(char *base, char *path, AWeatherCacheType update,
 100                 AWeatherCacheDoneCallback callback, gpointer user_data)
 101 {
 102         gchar *uri   = g_strconcat(base, path, NULL);
 103         gchar *local = g_build_filename(g_get_user_cache_dir(), PACKAGE, path, NULL);
 104
 105         if (update == AWEATHER_REFRESH)
 106                 return do_cache(uri, local, TRUE, "cache forced",
 107                                 callback, user_data);
 108
 109         if (update == AWEATHER_UPDATE)
 110                 return do_cache(uri, local, FALSE, "attempting updating",
 111                                 callback, user_data);
 112
 113         if (update == AWEATHER_ONCE && !g_file_test(local, G_FILE_TEST_EXISTS))
 114                 return do_cache(uri, local, TRUE, "local does not exist",
 115                                 callback, user_data);
 116
 117         /* No nead to cache, run the callback now and clean up */
 118         callback(local, FALSE, user_data);
 119         g_free(local);
 120         g_free(uri);
 121         return;
 122 }