2 * Copyright (C) 2009 Andy Spencer <spenceal@rose-hulman.edu>
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.
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.
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/>.
21 #include <libsoup/soup.h>
29 AWeatherCacheDoneCallback user_done_cb;
30 AWeatherCacheChunkCallback user_chunk_cb;
35 * Open a file, creating parent directories if needed
37 static FILE *fopen_p(const gchar *path, const gchar *mode)
39 gchar *parent = g_path_get_dirname(path);
40 if (!g_file_test(parent, G_FILE_TEST_EXISTS))
41 g_mkdir_with_parents(parent, 0755);
43 return fopen(path, mode);
46 static void done_cb(SoupSession *session, SoupMessage *message, gpointer _info)
48 cache_file_end_t *info = _info;
49 g_debug("data: done_cb");
51 if (message->status_code == 416)
52 /* Range unsatisfiable, file already complete */
53 info->user_done_cb(info->local, FALSE, info->user_data);
54 else if (SOUP_STATUS_IS_SUCCESSFUL(message->status_code))
55 info->user_done_cb(info->local, TRUE, info->user_data);
57 g_warning("data: done_cb - error copying file, status=%d\n"
60 message->status_code, info->uri, info->local);
65 //g_object_unref(session); This is probably leaking
68 void chunk_cb(SoupMessage *message, SoupBuffer *chunk, gpointer _info)
70 cache_file_end_t *info = _info;
71 if (!SOUP_STATUS_IS_SUCCESSFUL(message->status_code))
74 fwrite(chunk->data, chunk->length, 1, info->fp);
75 goffset cur = ftell(info->fp);
76 //goffset total = soup_message_headers_get_range(message->response_headers);
77 goffset start=0, end=0, total=0;
78 soup_message_headers_get_content_range(message->response_headers,
79 &start, &end, &total);
81 if (info->user_chunk_cb)
82 info->user_chunk_cb(info->local, cur, total, info->user_data);
85 static SoupSession *do_cache(cache_file_end_t *info, gboolean truncate, gchar *reason)
87 char *name = g_path_get_basename(info->uri);
88 g_debug("data: do_cache - Caching file %s: %s", name, reason);
91 /* TODO: move this to callback so we don't end up with 0 byte files
92 * Then change back to check for valid file after download */
93 if (truncate) info->fp = fopen_p(info->local, "w");
94 else info->fp = fopen_p(info->local, "a");
95 long bytes = ftell(info->fp);
97 SoupSession *session = soup_session_async_new();
98 SoupMessage *message = soup_message_new("GET", info->uri);
100 g_error("message is null, cannot parse uri");
101 g_signal_connect(message, "got-chunk", G_CALLBACK(chunk_cb), info);
102 soup_message_headers_set_range(message->request_headers, bytes, -1);
103 soup_session_queue_message(session, message, done_cb, info);
108 * Cache a image from Ridge to the local disk
109 * \param path Path to the Ridge file, starting after /ridge/
110 * \return The local path to the cached image
112 SoupSession *cache_file(char *base, char *path, AWeatherCacheType update,
113 AWeatherCacheChunkCallback user_chunk_cb,
114 AWeatherCacheDoneCallback user_done_cb,
118 cache_file_end_t *info = g_malloc0(sizeof(cache_file_end_t));
119 info->uri = g_strconcat(base, path, NULL);
120 info->local = g_build_filename(g_get_user_cache_dir(), PACKAGE, path, NULL);
122 info->user_chunk_cb = user_chunk_cb;
123 info->user_done_cb = user_done_cb;
124 info->user_data = user_data;
126 if (update == AWEATHER_REFRESH)
127 return do_cache(info, TRUE, "cache forced");
129 if (update == AWEATHER_UPDATE)
130 return do_cache(info, FALSE, "attempting updating");
132 if (update == AWEATHER_ONCE && !g_file_test(info->local, G_FILE_TEST_EXISTS))
133 return do_cache(info, TRUE, "local does not exist");
135 /* No nead to cache, run the callback now and clean up */
136 user_done_cb(info->local, FALSE, user_data);