diff options
Diffstat (limited to '')
-rw-r--r-- | plug-ins/common/web-page.c | 551 |
1 files changed, 551 insertions, 0 deletions
diff --git a/plug-ins/common/web-page.c b/plug-ins/common/web-page.c new file mode 100644 index 0000000..3ceef8a --- /dev/null +++ b/plug-ins/common/web-page.c @@ -0,0 +1,551 @@ +/* GIMP - The GNU Image Manipulation Program + * Copyright (C) 1995 Spencer Kimball and Peter Mattis + * + * 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 3 of the License, 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, see <https://www.gnu.org/licenses/>. + */ + +/* Webpage plug-in. + * Copyright (C) 2011 Mukund Sivaraman <muks@banu.com>. + * Portions are copyright of the author of the + * file-open-location-dialog.c code. + */ + +#include "config.h" + +#include <libgimp/gimp.h> +#include <libgimp/gimpui.h> + +#include <webkit/webkit.h> + +#include "libgimp/stdplugins-intl.h" + +/* Defines */ +#define PLUG_IN_PROC "plug-in-web-page" +#define PLUG_IN_BINARY "web-page" +#define PLUG_IN_ROLE "gimp-web-page" +#define MAX_URL_LEN 2048 + +typedef struct +{ + char *url; + gint32 width; + gint font_size; + GdkPixbuf *pixbuf; + GError *error; +} WebpageVals; + +static WebpageVals webpagevals; + +typedef struct +{ + char url[MAX_URL_LEN]; + gint32 width; + gint font_size; +} WebpageSaveVals; + +static void query (void); +static void run (const gchar *name, + gint nparams, + const GimpParam *param, + gint *nreturn_vals, + GimpParam **return_vals); +static gboolean webpage_dialog (void); +static gint32 webpage_capture (void); + + +/* Global Variables */ +const GimpPlugInInfo PLUG_IN_INFO = +{ + NULL, /* init_proc */ + NULL, /* quit_proc */ + query, /* query_proc */ + run /* run_proc */ +}; + + +/* Functions */ + +MAIN () + +static void +query (void) +{ + static const GimpParamDef args[] = + { + { GIMP_PDB_INT32, "run-mode", "The run mode { RUN-INTERACTIVE (0), RUN-NONINTERACTIVE (1) }" }, + { GIMP_PDB_STRING, "url", "URL of the webpage to screenshot" }, + { GIMP_PDB_INT32, "width", "The width of the screenshot (in pixels)" }, + { GIMP_PDB_INT32, "font-size", "The font size to use in the page (in pt)" } + }; + + static const GimpParamDef return_vals[] = + { + { GIMP_PDB_IMAGE, "image", "Output image" } + }; + + gimp_install_procedure (PLUG_IN_PROC, + N_("Create an image of a webpage"), + "The plug-in allows you to take a screenshot " + "of a webpage.", + "Mukund Sivaraman <muks@banu.com>", + "2011", + "2011", + N_("From _Webpage..."), + NULL, + GIMP_PLUGIN, + G_N_ELEMENTS (args), + G_N_ELEMENTS (return_vals), + args, return_vals); + + gimp_plugin_menu_register (PLUG_IN_PROC, "<Image>/File/Create/Acquire"); +} + +static void +run (const gchar *name, + gint nparams, + const GimpParam *param, + gint *nreturn_vals, + GimpParam **return_vals) +{ + GimpRunMode run_mode = param[0].data.d_int32; + GimpPDBStatusType status = GIMP_PDB_EXECUTION_ERROR; + gint32 image_id = -1; + static GimpParam values[2]; + WebpageSaveVals save = {"https://www.gimp.org/", 1024, 12}; + + INIT_I18N (); + + /* initialize the return of the status */ + *nreturn_vals = 1; + *return_vals = values; + values[0].type = GIMP_PDB_STATUS; + + gimp_get_data (PLUG_IN_PROC, &save); + + webpagevals.url = g_strdup (save.url); + webpagevals.width = save.width; + webpagevals.font_size = save.font_size; + + /* how are we running today? */ + switch (run_mode) + { + case GIMP_RUN_INTERACTIVE: + if (webpage_dialog ()) + status = GIMP_PDB_SUCCESS; + else + status = GIMP_PDB_CANCEL; + break; + + case GIMP_RUN_WITH_LAST_VALS: + /* This is currently not supported. */ + break; + + case GIMP_RUN_NONINTERACTIVE: + webpagevals.url = param[1].data.d_string; + webpagevals.width = param[2].data.d_int32; + webpagevals.font_size = param[3].data.d_int32; + status = GIMP_PDB_SUCCESS; + break; + + default: + break; + } + + if (status == GIMP_PDB_SUCCESS) + { + image_id = webpage_capture (); + + if (image_id == -1) + { + status = GIMP_PDB_EXECUTION_ERROR; + + if (webpagevals.error) + { + *nreturn_vals = 2; + + values[1].type = GIMP_PDB_STRING; + values[1].data.d_string = webpagevals.error->message; + } + } + else + { + save.width = webpagevals.width; + save.font_size = webpagevals.font_size; + + if (strlen (webpagevals.url) < MAX_URL_LEN) + { + strncpy (save.url, webpagevals.url, MAX_URL_LEN); + save.url[MAX_URL_LEN - 1] = 0; + } + else + { + memset (save.url, 0, MAX_URL_LEN); + } + + gimp_set_data (PLUG_IN_PROC, &save, sizeof save); + + if (run_mode == GIMP_RUN_INTERACTIVE) + gimp_display_new (image_id); + + *nreturn_vals = 2; + + values[1].type = GIMP_PDB_IMAGE; + values[1].data.d_image = image_id; + } + } + + values[0].data.d_status = status; +} + +static gboolean +webpage_dialog (void) +{ + GtkWidget *dialog; + GtkWidget *hbox; + GtkWidget *vbox; + GtkWidget *image; + GtkWidget *label; + GtkWidget *entry; + GtkSizeGroup *sizegroup; + GtkAdjustment *adjustment; + GtkWidget *spinbutton; + GtkWidget *combo; + gint active; + gint status; + gboolean ret = FALSE; + + gimp_ui_init (PLUG_IN_BINARY, FALSE); + + dialog = gimp_dialog_new (_("Create from webpage"), PLUG_IN_ROLE, + NULL, 0, + gimp_standard_help_func, PLUG_IN_PROC, + + _("_Cancel"), GTK_RESPONSE_CANCEL, + _("Cre_ate"), GTK_RESPONSE_OK, + + NULL); + + gtk_dialog_set_alternative_button_order (GTK_DIALOG (dialog), + GTK_RESPONSE_OK, + GTK_RESPONSE_CANCEL, + -1); + gimp_window_set_transient (GTK_WINDOW (dialog)); + + hbox = gtk_box_new (GTK_ORIENTATION_HORIZONTAL, 12); + gtk_container_set_border_width (GTK_CONTAINER (hbox), 12); + gtk_box_pack_start (GTK_BOX (gtk_dialog_get_content_area (GTK_DIALOG (dialog))), + hbox, FALSE, FALSE, 0); + gtk_widget_show (hbox); + + vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 0); + gtk_box_pack_start (GTK_BOX (hbox), vbox, FALSE, FALSE, 0); + gtk_widget_show (vbox); + + image = gtk_image_new_from_icon_name (GIMP_ICON_WEB, + GTK_ICON_SIZE_BUTTON); + gtk_box_pack_start (GTK_BOX (vbox), image, FALSE, FALSE, 0); + gtk_widget_show (image); + + vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 6); + gtk_box_pack_start (GTK_BOX (hbox), vbox, TRUE, TRUE, 0); + gtk_widget_show (vbox); + + label = gtk_label_new (_("Enter location (URI):")); + gtk_label_set_xalign (GTK_LABEL (label), 0.0); + gtk_box_pack_start (GTK_BOX (vbox), label, FALSE, FALSE, 0); + gtk_widget_show (label); + + entry = gtk_entry_new (); + gtk_entry_set_activates_default (GTK_ENTRY (entry), TRUE); + gtk_widget_set_size_request (entry, 400, -1); + gtk_box_pack_start (GTK_BOX (vbox), entry, FALSE, FALSE, 0); + + if (webpagevals.url) + gtk_entry_set_text (GTK_ENTRY (entry), + webpagevals.url); + gtk_widget_show (entry); + + sizegroup = gtk_size_group_new (GTK_SIZE_GROUP_HORIZONTAL); + + /* Width */ + hbox = gtk_box_new (GTK_ORIENTATION_HORIZONTAL, 6); + gtk_box_pack_start (GTK_BOX (vbox), + hbox, FALSE, FALSE, 0); + gtk_widget_show (hbox); + + label = gtk_label_new (_("Width (pixels):")); + gtk_size_group_add_widget (sizegroup, label); + + gtk_label_set_xalign (GTK_LABEL (label), 0.0); + gtk_box_pack_start (GTK_BOX (hbox), label, FALSE, FALSE, 0); + gtk_widget_show (label); + + adjustment = (GtkAdjustment *) + gtk_adjustment_new (webpagevals.width, + 1, 8192, 1, 10, 0); + spinbutton = gimp_spin_button_new (adjustment, 1.0, 0); + gtk_spin_button_set_numeric (GTK_SPIN_BUTTON (spinbutton), TRUE); + gtk_box_pack_start (GTK_BOX (hbox), spinbutton, FALSE, FALSE, 0); + gtk_widget_show (spinbutton); + + /* Font size */ + hbox = gtk_box_new (GTK_ORIENTATION_HORIZONTAL, 6); + gtk_box_pack_start (GTK_BOX (vbox), + hbox, FALSE, FALSE, 0); + gtk_widget_show (hbox); + + label = gtk_label_new (_("Font size:")); + gtk_size_group_add_widget (sizegroup, label); + + gtk_label_set_xalign (GTK_LABEL (label), 0.0); + gtk_box_pack_start (GTK_BOX (hbox), label, FALSE, FALSE, 0); + gtk_widget_show (label); + + combo = gimp_int_combo_box_new (_("Huge"), 16, + _("Large"), 14, + C_("web-page", "Default"), 12, + _("Small"), 10, + _("Tiny"), 8, + NULL); + + switch (webpagevals.font_size) + { + case 16: + case 14: + case 12: + case 10: + case 8: + active = webpagevals.font_size; + break; + default: + active = 12; + } + + gimp_int_combo_box_set_active (GIMP_INT_COMBO_BOX (combo), active); + + gtk_box_pack_start (GTK_BOX (hbox), combo, FALSE, FALSE, 0); + gtk_widget_show (combo); + + g_object_unref (sizegroup); + + status = gimp_dialog_run (GIMP_DIALOG (dialog)); + if (status == GTK_RESPONSE_OK) + { + g_free (webpagevals.url); + webpagevals.url = g_strdup (gtk_entry_get_text (GTK_ENTRY (entry))); + + webpagevals.width = (gint) gtk_adjustment_get_value + (GTK_ADJUSTMENT (adjustment)); + + gimp_int_combo_box_get_active (GIMP_INT_COMBO_BOX (combo), + &webpagevals.font_size); + + ret = TRUE; + } + + gtk_widget_destroy (dialog); + + return ret; +} + +static void +notify_progress_cb (WebKitWebView *view, + GParamSpec *pspec, + gpointer user_data) +{ + static gdouble old_progress = 0.0; + gdouble progress; + + g_object_get (view, + "progress", &progress, + NULL); + + if ((progress - old_progress) > 0.01) + { + gimp_progress_update (progress); + old_progress = progress; + } +} + +static gboolean +load_error_cb (WebKitWebView *view, + WebKitWebFrame *web_frame, + gchar *uri, + gpointer web_error, + gpointer user_data) +{ + webpagevals.error = g_error_copy ((GError *) web_error); + + gtk_main_quit (); + + return TRUE; +} + +static void +notify_load_status_cb (WebKitWebView *view, + GParamSpec *pspec, + gpointer user_data) +{ + WebKitLoadStatus status; + + g_object_get (view, + "load-status", &status, + NULL); + + if (status == WEBKIT_LOAD_FINISHED) + { + if (!webpagevals.error) + { + webpagevals.pixbuf = gtk_offscreen_window_get_pixbuf + (GTK_OFFSCREEN_WINDOW (user_data)); + } + + gtk_main_quit (); + } +} + +static gint32 +webpage_capture (void) +{ + gint32 image = -1; + gchar *scheme; + GtkWidget *window; + GtkWidget *view; + WebKitWebSettings *settings; + char *ua_old; + char *ua; + + if (webpagevals.pixbuf) + { + g_object_unref (webpagevals.pixbuf); + webpagevals.pixbuf = NULL; + } + if (webpagevals.error) + { + g_error_free (webpagevals.error); + webpagevals.error = NULL; + } + + if ((!webpagevals.url) || + (strlen (webpagevals.url) == 0)) + { + g_set_error (&webpagevals.error, 0, 0, _("No URL was specified")); + return -1; + } + + scheme = g_uri_parse_scheme (webpagevals.url); + if (!scheme) + { + char *url; + + /* If we were not given a well-formed URL, make one. */ + + url = g_strconcat ("http://", webpagevals.url, NULL); + g_free (webpagevals.url); + webpagevals.url = url; + + g_free (scheme); + } + + if (webpagevals.width < 32) + { + g_warning ("Width '%d' is too small. Clamped to 32.", + webpagevals.width); + webpagevals.width = 32; + } + else if (webpagevals.width > 8192) + { + g_warning ("Width '%d' is too large. Clamped to 8192.", + webpagevals.width); + webpagevals.width = 8192; + } + + window = gtk_offscreen_window_new (); + gtk_widget_show (window); + + view = webkit_web_view_new (); + gtk_widget_show (view); + + gtk_widget_set_size_request (view, webpagevals.width, -1); + gtk_container_add (GTK_CONTAINER (window), view); + + /* Append "GIMP/<GIMP_VERSION>" to the user agent string */ + settings = webkit_web_view_get_settings (WEBKIT_WEB_VIEW (view)); + g_object_get (settings, + "user-agent", &ua_old, + NULL); + ua = g_strdup_printf ("%s GIMP/%s", ua_old, GIMP_VERSION); + g_object_set (settings, + "user-agent", ua, + NULL); + g_free (ua_old); + g_free (ua); + + /* Set font size */ + g_object_set (settings, + "default-font-size", webpagevals.font_size, + NULL); + + g_signal_connect (view, "notify::progress", + G_CALLBACK (notify_progress_cb), + window); + g_signal_connect (view, "load-error", + G_CALLBACK (load_error_cb), + window); + g_signal_connect (view, "notify::load-status", + G_CALLBACK (notify_load_status_cb), + window); + + gimp_progress_init_printf (_("Downloading webpage '%s'"), webpagevals.url); + + webkit_web_view_open (WEBKIT_WEB_VIEW (view), + webpagevals.url); + + gtk_main (); + + gtk_widget_destroy (window); + + gimp_progress_update (1.0); + + if (webpagevals.pixbuf) + { + gint width; + gint height; + gint32 layer; + + gimp_progress_init_printf (_("Transferring webpage image for '%s'"), + webpagevals.url); + + width = gdk_pixbuf_get_width (webpagevals.pixbuf); + height = gdk_pixbuf_get_height (webpagevals.pixbuf); + + image = gimp_image_new (width, height, GIMP_RGB); + + gimp_image_undo_disable (image); + layer = gimp_layer_new_from_pixbuf (image, _("Webpage"), + webpagevals.pixbuf, + 100, + gimp_image_get_default_new_layer_mode (image), + 0.0, 1.0); + gimp_image_insert_layer (image, layer, -1, 0); + gimp_image_undo_enable (image); + + g_object_unref (webpagevals.pixbuf); + webpagevals.pixbuf = NULL; + + gimp_progress_update (1.0); + } + + return image; +} |