screenshot.c

#include <gtk/gtk.h>
#include <webkit/webkit.h>
#include <libsoup/soup.h>
#include <cairo-pdf.h>
#include <glib/gprintf.h>
#include <stdlib.h>


typedef struct IdleData_ {
    WebKitWebView *web_view;
    const char    *filename;
} IdleData;


static void
save_as_pdf (GtkWidget *widget, const char *filename) {
    GtkAllocation allocation;

    g_printf("Saving PDF to file %s\n", filename);
    gtk_widget_get_allocation(widget, &allocation);
    cairo_surface_t *surface = cairo_pdf_surface_create(
        filename,
        1.0 * allocation.width,
        1.0 * allocation.height
    );

    cairo_t *cr = cairo_create(surface);
    gtk_widget_draw(widget, cr);
    cairo_destroy(cr);
    cairo_surface_destroy(surface);
}


static gboolean
idle_cb (gpointer data) {
    IdleData *idle_data = (IdleData *) data;
    save_as_pdf(GTK_WIDGET(idle_data->web_view), idle_data->filename);
    g_free(idle_data);
    gtk_main_quit();
    return FALSE;
}


static void
load_status_cb (GObject* object, GParamSpec* pspec, gpointer data) {
    WebKitWebView *web_view = WEBKIT_WEB_VIEW(object);
    WebKitLoadStatus status = webkit_web_view_get_load_status(web_view);
    if (status != WEBKIT_LOAD_FINISHED) {
        return;
    }
    g_printf("Downloaded\n");

    IdleData *idle_data = g_new0(IdleData, 1);
    idle_data->web_view = web_view;
    idle_data->filename = (const char*) data;
    g_timeout_add(2000, idle_cb, (gpointer) idle_data);
}


int
main (int argc, gchar* argv[]) {
    gtk_init(&argc, &argv);

    if (argc < 2) {
        printf("Usage: URI [filename]\n");
        return 1;
    }
    const gchar *uri = argv[1];
    const gchar *filename = argc > 2 ? argv[2] : "a.pdf";

    if (!g_thread_supported()) {g_thread_init(NULL);}

    g_printf("Running against webkit %d.%d.%d\n",
        webkit_major_version(),
        webkit_minor_version(),
        webkit_micro_version()
    );


    gchar *proxy = getenv("http_proxy");
    if (proxy == NULL) proxy = getenv("HTTP_PROXY");
    if (proxy != NULL) {
        SoupSession *session = webkit_get_default_session();
        SoupURI *proxy_uri = soup_uri_new(proxy);
        g_printf("Using proxy: %s", proxy);
        g_object_set(
            session,
            SOUP_SESSION_PROXY_URI, proxy_uri,
            NULL
        );
    }

    WebKitWebView *web_view = WEBKIT_WEB_VIEW(webkit_web_view_new());

    g_signal_connect(web_view, "notify::load-status", G_CALLBACK(load_status_cb), (gpointer) filename);

    GtkWidget *window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
    gtk_window_set_default_size(GTK_WINDOW(window), 800, 640);
    gtk_container_add(GTK_CONTAINER(window), GTK_WIDGET(web_view));
    gtk_widget_show_all(window);

    webkit_web_view_load_uri(web_view, uri);
    gtk_main();

    return 0;
}