epub: Add EPUB document type.

3 files changed, 307 insertions, 0 deletions

diff --git a/source/fitz/document-all.c b/source/fitz/document-all.c
index 1b8c52c1..bfe57592 100644
--- a/source/fitz/document-all.c
+++ b/source/fitz/document-all.c
@@ -8,4 +8,5 @@ void fz_register_document_handlers(fz_context *ctx)
 	fz_register_document_handler(ctx, &img_document_handler);
 	fz_register_document_handler(ctx, &tiff_document_handler);
 	fz_register_document_handler(ctx, &html_document_handler);
+	fz_register_document_handler(ctx, &epub_document_handler);
 }
diff --git a/source/html/epub-doc.c b/source/html/epub-doc.c
new file mode 100644
index 00000000..11b43e0a
--- /dev/null
+++ b/source/html/epub-doc.c
@@ -0,0 +1,305 @@
+#include "mupdf/html.h"
+
+#define DEFW (450)
+#define DEFH (600)
+#define DEFEM (12)
+
+typedef struct epub_document_s epub_document;
+typedef struct epub_chapter_s epub_chapter;
+typedef struct epub_page_s epub_page;
+
+struct epub_document_s
+{
+	fz_document super;
+	fz_context *ctx;
+	fz_archive *zip;
+	fz_html_font_set *set;
+	float page_w, page_h, em;
+	int count;
+	epub_chapter *spine;
+};
+
+struct epub_chapter_s
+{
+	int start;
+	struct box *box;
+	epub_chapter *next;
+};
+
+static void
+epub_layout(epub_document *doc, float w, float h, float em)
+{
+	fz_context *ctx = doc->ctx;
+	epub_chapter *ch;
+
+	doc->page_w = w;
+	doc->page_h = h;
+	doc->em = em;
+
+	printf("epub: laying out chapters.\n");
+	for (ch = doc->spine; ch; ch = ch->next)
+		fz_layout_html(ctx, ch->box, w, h, em);
+	printf("epub: done.\n");
+}
+
+static int
+epub_count_pages(epub_document *doc)
+{
+	epub_chapter *ch;
+	int count = 0;
+	for (ch = doc->spine; ch; ch = ch->next)
+		count += ceilf(ch->box->h / doc->page_h);
+	return count;
+}
+
+static epub_page *
+epub_load_page(epub_document *doc, int number)
+{
+	return (void*)((intptr_t)number + 1);
+}
+
+static void
+epub_free_page(epub_document *doc, epub_page *page)
+{
+}
+
+static fz_rect *
+epub_bound_page(epub_document *doc, epub_page *page, fz_rect *bbox)
+{
+	bbox->x0 = 0;
+	bbox->y0 = 0;
+	bbox->x1 = doc->page_w;
+	bbox->y1 = doc->page_h;
+	return bbox;
+}
+
+static void
+epub_run_page(epub_document *doc, epub_page *page, fz_device *dev, const fz_matrix *ctm, fz_cookie *cookie)
+{
+	fz_context *ctx = doc->ctx;
+	epub_chapter *ch;
+	int n = ((intptr_t)page) - 1;
+	int count = 0;
+	for (ch = doc->spine; ch; ch = ch->next)
+	{
+		int cn = ceilf(ch->box->h / doc->page_h);
+		if (n < count + cn)
+		{
+			fz_draw_html(ctx, ch->box, (n-count) * doc->page_h, (n-count+1) * doc->page_h, dev, ctm);
+			break;
+		}
+		count += cn;
+	}
+}
+
+static void
+epub_close_document(epub_document *doc)
+{
+	/* TODO:free all chapters */
+	fz_close_archive(doc->ctx, doc->zip);
+	fz_free_html_font_set(doc->ctx, doc->set);
+	fz_free(doc->ctx, doc);
+}
+
+static const char *
+rel_path_from_idref(fz_xml *manifest, const char *idref)
+{
+	fz_xml *item;
+	if (!idref)
+		return NULL;
+	item = fz_xml_find_down(manifest, "item");
+	while (item)
+	{
+		const char *id = fz_xml_att(item, "id");
+		if (id && !strcmp(id, idref))
+			return fz_xml_att(item, "href");
+		item = fz_xml_find_next(item, "item");
+	}
+	return NULL;
+}
+
+static const char *
+path_from_idref(char *path, fz_xml *manifest, const char *base_uri, const char *idref, int n)
+{
+	const char *rel_path = rel_path_from_idref(manifest, idref);
+	if (!rel_path)
+	{
+		path[0] = 0;
+		return NULL;
+	}
+	fz_strlcpy(path, base_uri, n);
+	fz_strlcat(path, "/", n);
+	fz_strlcat(path, rel_path, n);
+	return fz_cleanname(path);
+}
+
+static epub_chapter *
+epub_parse_chapter(epub_document *doc, const char *path)
+{
+	fz_context *ctx = doc->ctx;
+	fz_archive *zip = doc->zip;
+	fz_buffer *buf;
+	epub_chapter *ch;
+	char base_uri[2048];
+
+	fz_dirname(base_uri, path, sizeof base_uri);
+
+	buf = fz_read_archive_entry(ctx, zip, path);
+	fz_write_buffer_byte(ctx, buf, 0);
+
+	ch = fz_malloc_struct(ctx, epub_chapter);
+	ch->box = fz_generate_html(ctx, doc->set, zip, base_uri, buf);
+	ch->next = NULL;
+
+	fz_drop_buffer(ctx, buf);
+
+	return ch;
+}
+
+static void
+epub_parse_header(epub_document *doc)
+{
+	fz_context *ctx = doc->ctx;
+	fz_archive *zip = doc->zip;
+	fz_buffer *buf;
+	fz_xml *container_xml, *content_opf;
+	fz_xml *container, *rootfiles, *rootfile;
+	fz_xml *package, *manifest, *spine, *itemref;
+	char base_uri[2048];
+	const char *full_path;
+	char ncx[2048], s[2048];
+	epub_chapter *head, *tail;
+
+	/* parse META-INF/container.xml to find OPF */
+
+	buf = fz_read_archive_entry(ctx, zip, "META-INF/container.xml");
+	fz_write_buffer_byte(ctx, buf, 0);
+	container_xml = fz_parse_xml(ctx, buf->data, buf->len, 0);
+	fz_drop_buffer(ctx, buf);
+
+	container = fz_xml_find(container_xml, "container");
+	rootfiles = fz_xml_find_down(container, "rootfiles");
+	rootfile = fz_xml_find_down(rootfiles, "rootfile");
+	full_path = fz_xml_att(rootfile, "full-path");
+	if (!full_path)
+		fz_throw(ctx, FZ_ERROR_GENERIC, "cannot find root file in EPUB");
+
+	printf("epub: found root: %s\n", full_path);
+
+	fz_dirname(base_uri, full_path, sizeof base_uri);
+
+	/* parse OPF to find NCX and spine */
+
+	buf = fz_read_archive_entry(ctx, zip, full_path);
+	fz_write_buffer_byte(ctx, buf, 0);
+	content_opf = fz_parse_xml(ctx, buf->data, buf->len, 0);
+	fz_drop_buffer(ctx, buf);
+
+	package = fz_xml_find(content_opf, "package");
+	manifest = fz_xml_find_down(package, "manifest");
+	spine = fz_xml_find_down(package, "spine");
+
+	if (path_from_idref(ncx, manifest, base_uri, fz_xml_att(spine, "toc"), sizeof ncx))
+	{
+		/* TODO: parse NCX to create fz_outline */
+		printf("epub: found outline: %s\n", ncx);
+	}
+
+	head = tail = NULL;
+	itemref = fz_xml_find_down(spine, "itemref");
+	while (itemref)
+	{
+		if (path_from_idref(s, manifest, base_uri, fz_xml_att(itemref, "idref"), sizeof s))
+		{
+			printf("epub: found spine %s\n", s);
+			if (!head)
+				head = tail = epub_parse_chapter(doc, s);
+			else
+				tail = tail->next = epub_parse_chapter(doc, s);
+		}
+		itemref = fz_xml_find_next(itemref, "itemref");
+	}
+
+	doc->spine = head;
+
+	printf("epub: done.\n");
+
+	fz_free_xml(ctx, container_xml);
+	fz_free_xml(ctx, content_opf);
+}
+
+static epub_document *
+epub_init(fz_context *ctx, fz_archive *zip)
+{
+	epub_document *doc;
+
+	doc = fz_malloc_struct(ctx, epub_document);
+	doc->ctx = ctx;
+	doc->zip = zip;
+	doc->set = fz_new_html_font_set(ctx);
+
+	doc->super.close = (void*)epub_close_document;
+	doc->super.layout = (void*)epub_layout;
+	doc->super.count_pages = (void*)epub_count_pages;
+	doc->super.load_page = (void*)epub_load_page;
+	doc->super.bound_page = (void*)epub_bound_page;
+	doc->super.run_page_contents = (void*)epub_run_page;
+	doc->super.free_page = (void*)epub_free_page;
+
+	fz_try(ctx)
+	{
+		epub_parse_header(doc);
+		epub_layout(doc, DEFW, DEFH, DEFEM);
+	}
+	fz_catch(ctx)
+	{
+		epub_close_document(doc);
+		fz_rethrow(ctx);
+	}
+
+	return doc;
+}
+
+static epub_document *
+epub_open_document_with_stream(fz_context *ctx, fz_stream *file)
+{
+	return epub_init(ctx, fz_open_archive_with_stream(ctx, file));
+}
+
+static epub_document *
+epub_open_document(fz_context *ctx, const char *filename)
+{
+	if (strstr(filename, "META-INF/container.xml") || strstr(filename, "META-INF\\container.xml"))
+	{
+		char dirname[2048], *p;
+		fz_strlcpy(dirname, filename, sizeof dirname);
+		p = strstr(dirname, "META-INF");
+		*p = 0;
+		if (!dirname[0])
+			fz_strlcpy(dirname, ".", sizeof dirname);
+		return epub_init(ctx, fz_open_directory(ctx, dirname));
+	}
+
+	return epub_init(ctx, fz_open_archive(ctx, filename));
+}
+
+static int
+epub_recognize(fz_context *doc, const char *magic)
+{
+	char *ext = strrchr(magic, '.');
+	if (ext)
+		if (!fz_strcasecmp(ext, ".epub"))
+			return 100;
+	if (strstr(magic, "META-INF/container.xml") || strstr(magic, "META-INF\\container.xml"))
+		return 200;
+	if (!strcmp(magic, "application/epub+zip"))
+		return 100;
+	return 0;
+}
+
+fz_document_handler epub_document_handler =
+{
+	(fz_document_recognize_fn *)&epub_recognize,
+	(fz_document_open_fn *)&epub_open_document,
+	(fz_document_open_with_stream_fn *)&epub_open_document_with_stream
+};
diff --git a/source/html/layout.c b/source/html/layout.c
index 3243a6ae..31c41a71 100644
--- a/source/html/layout.c
+++ b/source/html/layout.c
@@ -31,6 +31,7 @@ static const char *default_css =
 "u,ins{text-decoration:underline}"
 "center{text-align:center}"
 "svg{display:none}"
+"a{color:blue}"
 ;
 
 static int iswhite(int c)