#include <pdf-document.h>

#include <libgnomevfs/gnome-vfs-mime-utils.h>
#include <libgnomevfs/gnome-vfs-uri.h>
#include <libgnomevfs/gnome-vfs-utils.h>
#include <libgnomevfs/gnome-vfs-init.h>

#include <ev-document.h>
#include <ev-document-thumbnails.h>

#include <string.h>

static gboolean
evince_thumbnail_pngenc_get (const char *uri, const char *thumbnail)
{
	EvDocument *document = NULL;
	char *mime_type;
	GError *error;
	GdkPixbuf *pixbuf;

	mime_type = gnome_vfs_get_mime_type (uri);
	if (mime_type == NULL)
		return FALSE;

	if (!strcmp (mime_type, "application/pdf"))
		document = g_object_new (PDF_TYPE_DOCUMENT, NULL);
	else
		return FALSE;

	if (!ev_document_load (document, uri, &error)) {
		if (error->domain == EV_DOCUMENT_ERROR &&
            	    error->code == EV_DOCUMENT_ERROR_ENCRYPTED) {
			/* FIXME: Create a thumb for cryp docs */
		}
		g_error_free (error);
		return FALSE;
	}

	pixbuf = ev_document_thumbnails_get_thumbnail
			(EV_DOCUMENT_THUMBNAILS (document), 1, 100);
	
	if (pixbuf != NULL) {
		GdkPixbuf *pdflogo;

		pdflogo = gdk_pixbuf_new_from_file (DATADIR"/pdf-icon.png", NULL);
		if (pdflogo != NULL) {
			int delta_height, delta_width;

			delta_width = gdk_pixbuf_get_width (pixbuf) -
			              gdk_pixbuf_get_width (pdflogo);
			delta_height = gdk_pixbuf_get_height (pixbuf) -
			               gdk_pixbuf_get_height (pdflogo);

			gdk_pixbuf_composite (pdflogo, pixbuf,
					      delta_width, delta_height,
					      gdk_pixbuf_get_width (pdflogo),
					      gdk_pixbuf_get_height (pdflogo),
					      delta_width, delta_height,
					      1, 1,
					      GDK_INTERP_NEAREST, 100);

			gdk_pixbuf_unref  (pdflogo);
		}
		if (gdk_pixbuf_save (pixbuf, thumbnail, "png", NULL, NULL)) {
			gdk_pixbuf_unref  (pixbuf);
			g_object_unref (document);
			return TRUE;
		} else {
			gdk_pixbuf_unref  (pixbuf);
			g_object_unref (document);
		}
	}
	return FALSE;
}

int
main (int argc, char *argv[])
{
	int res;
	char *uri;

	if (argc != 3) {
		g_print ("%s: thumbnailer for Nautilus\n", argv[0]);
		g_print ("usage: %s <input-filename> <output-filename>\n", argv[0]);
		return -1;
	}

	res = gnome_vfs_init ();

	uri = gnome_vfs_make_uri_from_shell_arg (argv[1]);

	if (evince_thumbnail_pngenc_get (uri, argv[2])) {
		g_free (uri);
		return 0;
	} else {
		g_free (uri);
		return -2;
	}
}