1/*
2 * Copyright (C) 2011 Igalia S.L.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "BrowserDownloadsBar.h"
27
28#include <glib/gi18n.h>
29
30#define BROWSER_TYPE_DOWNLOAD (browser_download_get_type())
31#define BROWSER_DOWNLOAD(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), BROWSER_TYPE_DOWNLOAD, BrowserDownload))
32
33typedef struct _BrowserDownload BrowserDownload;
34typedef struct _BrowserDownloadClass BrowserDownloadClass;
35
36static GType browser_download_get_type();
37
38struct _BrowserDownloadsBar {
39 GtkInfoBar parent;
40};
41
42struct _BrowserDownloadsBarClass {
43 GtkInfoBarClass parentClass;
44};
45
46G_DEFINE_TYPE(BrowserDownloadsBar, browser_downloads_bar, GTK_TYPE_INFO_BAR)
47
48static void
49browserDownloadsBarChildRemoved(GtkContainer *infoBar, GtkWidget *widget, BrowserDownloadsBar *downloadsBar)
50{
51 GList *children = gtk_container_get_children(infoBar);
52 if (g_list_length(children) == 1)
53 gtk_info_bar_response(GTK_INFO_BAR(downloadsBar), GTK_RESPONSE_CLOSE);
54 g_list_free(children);
55}
56
57static void browserDownloadsBarResponse(GtkInfoBar *infoBar, gint responseId)
58{
59 gtk_widget_destroy(GTK_WIDGET(infoBar));
60}
61
62static void browser_downloads_bar_init(BrowserDownloadsBar *downloadsBar)
63{
64 GtkWidget *contentBox = gtk_info_bar_get_content_area(GTK_INFO_BAR(downloadsBar));
65 g_signal_connect_after(contentBox, "remove", G_CALLBACK(browserDownloadsBarChildRemoved), downloadsBar);
66 gtk_orientable_set_orientation(GTK_ORIENTABLE(contentBox), GTK_ORIENTATION_VERTICAL);
67
68 GtkWidget *title = gtk_label_new(NULL);
69 gtk_label_set_markup(GTK_LABEL(title), "<span size='xx-large' weight='bold'>Downloads</span>");
70 gtk_misc_set_alignment(GTK_MISC(title), 0., 0.5);
71 gtk_box_pack_start(GTK_BOX(contentBox), title, FALSE, FALSE, 12);
72 gtk_widget_show(title);
73}
74
75static void browser_downloads_bar_class_init(BrowserDownloadsBarClass *klass)
76{
77 GtkInfoBarClass *infoBarClass = GTK_INFO_BAR_CLASS(klass);
78 infoBarClass->response = browserDownloadsBarResponse;
79}
80
81GtkWidget *browser_downloads_bar_new()
82{
83 GtkInfoBar *downloadsBar = GTK_INFO_BAR(g_object_new(BROWSER_TYPE_DOWNLOADS_BAR, NULL));
84 gtk_info_bar_add_buttons(downloadsBar, GTK_STOCK_CLOSE, GTK_RESPONSE_CLOSE, NULL);
85 return GTK_WIDGET(downloadsBar);
86}
87
88struct _BrowserDownload {
89 GtkBox parent;
90
91 WebKitDownload *download;
92 guint64 contentLength;
93 guint64 downloadedSize;
94 gboolean finished;
95
96 GtkWidget *statusLabel;
97 GtkWidget *remainingLabel;
98 GtkWidget *progressBar;
99 GtkWidget *actionButton;
100};
101
102struct _BrowserDownloadClass {
103 GtkBoxClass parentClass;
104};
105
106G_DEFINE_TYPE(BrowserDownload, browser_download, GTK_TYPE_BOX)
107
108static void actionButtonClicked(GtkButton *button, BrowserDownload *browserDownload)
109{
110 if (!browserDownload->finished) {
111 webkit_download_cancel(browserDownload->download);
112 return;
113 }
114
115 gtk_show_uri(gtk_widget_get_screen(GTK_WIDGET(browserDownload)),
116 webkit_download_get_destination(browserDownload->download),
117 gtk_get_current_event_time(), NULL);
118 gtk_widget_destroy(GTK_WIDGET(browserDownload));
119}
120
121static void browserDownloadFinalize(GObject *object)
122{
123 BrowserDownload *browserDownload = BROWSER_DOWNLOAD(object);
124
125 if (browserDownload->download) {
126 g_object_unref(browserDownload->download);
127 browserDownload->download = NULL;
128 }
129
130 G_OBJECT_CLASS(browser_download_parent_class)->finalize(object);
131}
132
133static void browser_download_init(BrowserDownload *download)
134{
135 GtkWidget *mainBox = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 6);
136 gtk_box_pack_start(GTK_BOX(download), mainBox, FALSE, FALSE, 0);
137 gtk_widget_show(mainBox);
138
139 GtkWidget *vbox = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0);
140 gtk_box_pack_start(GTK_BOX(mainBox), vbox, TRUE, TRUE, 0);
141 gtk_widget_show(vbox);
142
143 GtkWidget *statusBox = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 0);
144 gtk_box_pack_start(GTK_BOX(vbox), statusBox, TRUE, TRUE, 0);
145 gtk_widget_show(statusBox);
146
147 download->statusLabel = gtk_label_new("Starting Download");
148 gtk_label_set_ellipsize(GTK_LABEL(download->statusLabel), PANGO_ELLIPSIZE_END);
149 gtk_misc_set_alignment(GTK_MISC(download->statusLabel), 0., 0.5);
150 gtk_box_pack_start(GTK_BOX(statusBox), download->statusLabel, TRUE, TRUE, 0);
151 gtk_widget_show(download->statusLabel);
152
153 download->remainingLabel = gtk_label_new(NULL);
154 gtk_misc_set_alignment(GTK_MISC(download->remainingLabel), 1., 0.5);
155 gtk_box_pack_end(GTK_BOX(statusBox), download->remainingLabel, TRUE, TRUE, 0);
156 gtk_widget_show(download->remainingLabel);
157
158 download->progressBar = gtk_progress_bar_new();
159 gtk_box_pack_start(GTK_BOX(vbox), download->progressBar, FALSE, FALSE, 0);
160 gtk_widget_show(download->progressBar);
161
162 download->actionButton = gtk_button_new_from_stock(GTK_STOCK_CANCEL);
163 g_signal_connect(download->actionButton, "clicked", G_CALLBACK(actionButtonClicked), download);
164 gtk_box_pack_end(GTK_BOX(mainBox), download->actionButton, FALSE, FALSE, 0);
165 gtk_widget_show(download->actionButton);
166}
167
168static void browser_download_class_init(BrowserDownloadClass *klass)
169{
170 GObjectClass *objectClass = G_OBJECT_CLASS(klass);
171
172 objectClass->finalize = browserDownloadFinalize;
173}
174
175static void downloadReceivedResponse(WebKitDownload *download, GParamSpec *paramSpec, BrowserDownload *browserDownload)
176{
177 WebKitURIResponse *response = webkit_download_get_response(download);
178 browserDownload->contentLength = webkit_uri_response_get_content_length(response);
179 char *text = g_strdup_printf("Downloading %s", webkit_uri_response_get_uri(response));
180 gtk_label_set_text(GTK_LABEL(browserDownload->statusLabel), text);
181 g_free(text);
182}
183
184static gchar *remainingTime(BrowserDownload *browserDownload)
185{
186 guint64 total = browserDownload->contentLength;
187 guint64 current = browserDownload->downloadedSize;
188 gdouble elapsedTime = webkit_download_get_elapsed_time(browserDownload->download);
189
190 if (current <= 0)
191 return NULL;
192
193 gdouble perByteTime = elapsedTime / current;
194 gdouble interval = perByteTime * (total - current);
195
196 int hours = (int) (interval / 3600);
197 interval -= hours * 3600;
198 int mins = (int) (interval / 60);
199 interval -= mins * 60;
200 int secs = (int) interval;
201
202 if (hours > 0) {
203 if (mins > 0)
204 return g_strdup_printf (ngettext ("%u:%02u hour left", "%u:%02u hours left", hours), hours, mins);
205 return g_strdup_printf (ngettext ("%u hour left", "%u hours left", hours), hours);
206 }
207
208 if (mins > 0)
209 return g_strdup_printf (ngettext ("%u:%02u minute left", "%u:%02u minutes left", mins), mins, secs);
210 return g_strdup_printf (ngettext ("%u second left", "%u seconds left", secs), secs);
211}
212
213static void downloadProgress(WebKitDownload *download, GParamSpec *paramSpec, BrowserDownload *browserDownload)
214{
215 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(browserDownload->progressBar),
216 webkit_download_get_estimated_progress(download));
217 char *remaining = remainingTime(browserDownload);
218 gtk_label_set_text(GTK_LABEL(browserDownload->remainingLabel), remaining);
219 g_free(remaining);
220}
221
222static void downloadReceivedData(WebKitDownload *download, guint64 dataLength, BrowserDownload *browserDownload)
223{
224 browserDownload->downloadedSize += dataLength;
225}
226
227static void downloadFinished(WebKitDownload *download, BrowserDownload *browserDownload)
228{
229 gchar *text = g_strdup_printf("Download completed: %s", webkit_download_get_destination(download));
230 gtk_label_set_text(GTK_LABEL(browserDownload->statusLabel), text);
231 g_free(text);
232 gtk_label_set_text(GTK_LABEL(browserDownload->remainingLabel), NULL);
233 gtk_button_set_image(GTK_BUTTON(browserDownload->actionButton),
234 gtk_image_new_from_stock(GTK_STOCK_OPEN, GTK_ICON_SIZE_BUTTON));
235 gtk_button_set_label(GTK_BUTTON(browserDownload->actionButton), "Open ...");
236 browserDownload->finished = TRUE;
237}
238
239static void downloadFailed(WebKitDownload *download, GError *error, BrowserDownload *browserDownload)
240{
241 g_signal_handlers_disconnect_by_func(browserDownload->download, downloadFinished, browserDownload);
242 if (g_error_matches(error, WEBKIT_DOWNLOAD_ERROR, WEBKIT_DOWNLOAD_ERROR_CANCELLED_BY_USER)) {
243 gtk_widget_destroy(GTK_WIDGET(browserDownload));
244 return;
245 }
246
247 char *errorMessage = g_strdup_printf("Download failed: %s", error->message);
248 gtk_label_set_text(GTK_LABEL(browserDownload->statusLabel), errorMessage);
249 g_free(errorMessage);
250 gtk_label_set_text(GTK_LABEL(browserDownload->remainingLabel), NULL);
251 gtk_widget_set_sensitive(browserDownload->actionButton, FALSE);
252}
253
254GtkWidget *browserDownloadNew(WebKitDownload *download)
255{
256 BrowserDownload *browserDownload = BROWSER_DOWNLOAD(g_object_new(BROWSER_TYPE_DOWNLOAD,
257 "orientation", GTK_ORIENTATION_VERTICAL,
258 NULL));
259
260 browserDownload->download = g_object_ref(download);
261 g_signal_connect(browserDownload->download, "notify::response", G_CALLBACK(downloadReceivedResponse), browserDownload);
262 g_signal_connect(browserDownload->download, "notify::estimated-progress", G_CALLBACK(downloadProgress), browserDownload);
263 g_signal_connect(browserDownload->download, "received-data", G_CALLBACK(downloadReceivedData), browserDownload);
264 g_signal_connect(browserDownload->download, "finished", G_CALLBACK(downloadFinished), browserDownload);
265 g_signal_connect(browserDownload->download, "failed", G_CALLBACK(downloadFailed), browserDownload);
266
267 return GTK_WIDGET(browserDownload);
268}
269
270void browser_downloads_bar_add_download(BrowserDownloadsBar *downloadsBar, WebKitDownload *download)
271{
272 GtkWidget *browserDownload = browserDownloadNew(download);
273 GtkWidget *contentBox = gtk_info_bar_get_content_area(GTK_INFO_BAR(downloadsBar));
274 gtk_box_pack_start(GTK_BOX(contentBox), browserDownload, FALSE, TRUE, 0);
275 gtk_widget_show(browserDownload);
276}