blob: efffd7282d89c4ff92e174a9f49d56c140715ba3 [file] [log] [blame]
/*
* Copyright (C) 2008 Carlos Garcia Campos <carlosgc@gnome.org>
*
* 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, 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 <gtk/gtk.h>
#include "attachments.h"
#include "utils.h"
enum {
ATTACHMENTS_NAME_COLUMN,
ATTACHMENTS_DESCRIPTION_COLUMN,
ATTACHMENTS_SIZE_COLUMN,
ATTACHMENTS_CTIME_COLUMN,
ATTACHMENTS_MTIME_COLUMN,
ATTACHMENTS_ATTACHMENT_COLUMN,
N_COLUMNS
};
static void
pgd_attachments_fill_model (GtkListStore *model,
PopplerDocument *document)
{
GList *list, *l;
list = poppler_document_get_attachments (document);
for (l = list; l && l->data; l = g_list_next (l)) {
PopplerAttachment *attachment = POPPLER_ATTACHMENT (l->data);
GtkTreeIter iter;
gchar *size;
gchar *ctime, *mtime;
size = g_strdup_printf ("%" G_GSIZE_FORMAT, attachment->size);
ctime = pgd_format_date (attachment->ctime);
mtime = pgd_format_date (attachment->mtime);
gtk_list_store_append (model, &iter);
gtk_list_store_set (model, &iter,
ATTACHMENTS_NAME_COLUMN,
attachment->name ? attachment->name : "Unknown",
ATTACHMENTS_DESCRIPTION_COLUMN,
attachment->description ? attachment->description : "Unknown",
ATTACHMENTS_SIZE_COLUMN,
size ? size : "Unknown",
ATTACHMENTS_CTIME_COLUMN,
ctime ? ctime : "Unknown",
ATTACHMENTS_MTIME_COLUMN,
mtime ? mtime : "Unknown",
ATTACHMENTS_ATTACHMENT_COLUMN, attachment,
-1);
g_free (size);
g_free (ctime);
g_free (mtime);
g_object_unref (attachment);
}
g_list_free (list);
}
static GtkWidget *
pgd_attachments_create_list (GtkTreeModel *model)
{
GtkWidget *treeview;
GtkCellRenderer *renderer;
treeview = gtk_tree_view_new_with_model (GTK_TREE_MODEL (model));
gtk_tree_view_set_headers_visible (GTK_TREE_VIEW (treeview), TRUE);
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
0, "Name",
renderer,
"text", ATTACHMENTS_NAME_COLUMN,
NULL);
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
1, "Description",
renderer,
"text", ATTACHMENTS_DESCRIPTION_COLUMN,
NULL);
g_object_set (G_OBJECT (renderer), "ellipsize", PANGO_ELLIPSIZE_END, NULL);
g_object_set (G_OBJECT (gtk_tree_view_get_column (GTK_TREE_VIEW (treeview), 1)),
"expand", TRUE, NULL);
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
2, "Size",
renderer,
"text", ATTACHMENTS_SIZE_COLUMN,
NULL);
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
3, "Creation Date",
renderer,
"text", ATTACHMENTS_CTIME_COLUMN,
NULL);
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
4, "Modification Date",
renderer,
"text", ATTACHMENTS_MTIME_COLUMN,
NULL);
return treeview;
}
static void
pgd_attachments_save_dialog_response (GtkFileChooser *file_chooser,
gint response,
PopplerAttachment *attachment)
{
gchar *filename;
GError *error = NULL;
if (response != GTK_RESPONSE_ACCEPT) {
g_object_unref (attachment);
gtk_widget_destroy (GTK_WIDGET (file_chooser));
return;
}
filename = gtk_file_chooser_get_filename (file_chooser);
if (!poppler_attachment_save (attachment, filename, &error)) {
g_warning ("%s", error->message);
g_error_free (error);
}
g_free (filename);
g_object_unref (attachment);
gtk_widget_destroy (GTK_WIDGET (file_chooser));
}
static void
pgd_attachments_save_button_clicked (GtkButton *button,
GtkTreeView *treeview)
{
GtkTreeSelection *selection;
GtkTreeModel *model;
GtkTreeIter iter;
GtkWidget *file_chooser;
PopplerAttachment *attachment;
selection = gtk_tree_view_get_selection (treeview);
if (!gtk_tree_selection_get_selected (selection, &model, &iter))
return;
gtk_tree_model_get (model, &iter,
ATTACHMENTS_ATTACHMENT_COLUMN, &attachment,
-1);
if (!attachment)
return;
file_chooser = gtk_file_chooser_dialog_new ("Save attachment",
GTK_WINDOW (gtk_widget_get_toplevel (GTK_WIDGET (treeview))),
GTK_FILE_CHOOSER_ACTION_SAVE,
"_Cancel", GTK_RESPONSE_CANCEL,
"_Save", GTK_RESPONSE_ACCEPT,
NULL);
gtk_file_chooser_set_current_name (GTK_FILE_CHOOSER (file_chooser), attachment->name);
g_signal_connect (G_OBJECT (file_chooser), "response",
G_CALLBACK (pgd_attachments_save_dialog_response),
(gpointer) attachment);
gtk_widget_show (file_chooser);
}
static gboolean
attachment_save_callback (const gchar *buf,
gsize count,
gpointer data,
GError **error)
{
GChecksum *cs = (GChecksum *)data;
g_checksum_update (cs, (guchar *) buf, count);
return TRUE;
}
static void
message_dialog_run (GtkWindow *parent,
const gchar *message)
{
GtkWidget *dialog;
dialog = gtk_message_dialog_new (parent,
GTK_DIALOG_DESTROY_WITH_PARENT,
GTK_MESSAGE_INFO,
GTK_BUTTONS_CLOSE,
"%s", message);
gtk_dialog_run (GTK_DIALOG (dialog));
gtk_widget_destroy (dialog);
}
static void
pgd_attachments_validate_button_clicked (GtkButton *button,
GtkTreeView *treeview)
{
GtkTreeSelection *selection;
GtkTreeModel *model;
GtkTreeIter iter;
GChecksum *cs;
guint8 *digest;
gsize digest_len;
PopplerAttachment *attachment;
gboolean valid = TRUE;
selection = gtk_tree_view_get_selection (treeview);
if (!gtk_tree_selection_get_selected (selection, &model, &iter))
return;
gtk_tree_model_get (model, &iter,
ATTACHMENTS_ATTACHMENT_COLUMN, &attachment,
-1);
if (!attachment)
return;
if (attachment->checksum->len == 0) {
message_dialog_run (GTK_WINDOW (gtk_widget_get_toplevel (GTK_WIDGET (treeview))),
"Impossible to validate attachment: checksum is not available");
g_object_unref (attachment);
return;
}
cs = g_checksum_new (G_CHECKSUM_MD5);
poppler_attachment_save_to_callback (attachment, attachment_save_callback,
(gpointer)cs, NULL);
digest_len = g_checksum_type_get_length (G_CHECKSUM_MD5);
digest = (guint8 *) g_malloc (digest_len);
g_checksum_get_digest (cs, digest, &digest_len);
g_checksum_free (cs);
if (attachment->checksum->len == digest_len) {
gint i;
for (i = 0; i < digest_len; i++) {
if ((guint8)attachment->checksum->str[i] != digest[i]) {
valid = FALSE;
break;
}
}
}
if (valid) {
message_dialog_run (GTK_WINDOW (gtk_widget_get_toplevel (GTK_WIDGET (treeview))),
"Attachment is valid");
} else {
message_dialog_run (GTK_WINDOW (gtk_widget_get_toplevel (GTK_WIDGET (treeview))),
"Attachment is not valid: the checksum does not match");
}
g_free (digest);
g_object_unref (attachment);
}
GtkWidget *
pgd_attachments_create_widget (PopplerDocument *document)
{
GtkWidget *vbox;
GtkWidget *treeview;
GtkListStore *model;
GtkWidget *swindow;
GtkWidget *hbox, *button;
gboolean has_attachments;
vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 12);
swindow = gtk_scrolled_window_new (NULL, NULL);
gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (swindow),
GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
has_attachments = poppler_document_has_attachments (document);
if (has_attachments) {
model = gtk_list_store_new (N_COLUMNS,
G_TYPE_STRING, G_TYPE_STRING,
G_TYPE_STRING,
G_TYPE_STRING, G_TYPE_STRING,
G_TYPE_OBJECT);
pgd_attachments_fill_model (model, document);
treeview = pgd_attachments_create_list (GTK_TREE_MODEL (model));
} else {
GtkCellRenderer *renderer;
GtkTreeIter iter;
gchar *markup;
model = gtk_list_store_new (1, G_TYPE_STRING);
gtk_list_store_append (model, &iter);
markup = g_strdup_printf ("<span size=\"larger\" style=\"italic\">%s</span>",
"The document doesn't contain attachments");
gtk_list_store_set (model, &iter, 0, markup, -1);
g_free (markup);
treeview = gtk_tree_view_new_with_model (GTK_TREE_MODEL (model));
renderer = gtk_cell_renderer_text_new ();
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
0, "Name",
renderer,
"markup", 0,
NULL);
}
g_object_unref (model);
gtk_container_add (GTK_CONTAINER (swindow), treeview);
gtk_widget_show (treeview);
gtk_box_pack_start (GTK_BOX (vbox), swindow, TRUE, TRUE, 0);
gtk_widget_show (swindow);
if (!has_attachments)
return vbox;
hbox = gtk_button_box_new (GTK_ORIENTATION_HORIZONTAL);
gtk_button_box_set_layout (GTK_BUTTON_BOX (hbox), GTK_BUTTONBOX_SPREAD);
button = gtk_button_new_with_label ("Save");
g_signal_connect (G_OBJECT (button), "clicked",
G_CALLBACK (pgd_attachments_save_button_clicked),
(gpointer)treeview);
gtk_box_pack_start (GTK_BOX (hbox), button, FALSE, FALSE, 0);
gtk_widget_show (button);
button = gtk_button_new_with_label ("Validate");
g_signal_connect (G_OBJECT (button), "clicked",
G_CALLBACK (pgd_attachments_validate_button_clicked),
(gpointer)treeview);
gtk_box_pack_start (GTK_BOX (hbox), button, FALSE, FALSE, 0);
gtk_widget_show (button);
gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, TRUE, 6);
gtk_widget_show (hbox);
return vbox;
}