DocumentWidget: Retain Gempub data; cleanup

Keep hold of the `Gempub` data while the page is open so it can be used for other purposes.

4 files changed, 65 insertions, 40 deletions

diff --git a/src/gempub.c b/src/gempub.c
index 45a4c1b3..448349c8 100644
--- a/src/gempub.c
+++ b/src/gempub.c
@@ -24,6 +24,7 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
 #include "gmutil.h"
 #include "lang.h"
 #include "defs.h"
+#include "gmdocument.h"
 #include "ui/util.h"
 
 #include <the_Foundation/archive.h>
@@ -162,7 +163,7 @@ static iBool isRemote_Gempub_(const iGempub *d) {
     return !equalCase_Rangecc(urlScheme_String(&d->baseUrl), "file");
 }
 
-iBlock *coverPageSource_Gempub(const iGempub *d) {
+iString *coverPageSource_Gempub(const iGempub *d) {
     iAssert(!isEmpty_String(&d->baseUrl));
     const iString *baseUrl = withSpacesEncoded_String(&d->baseUrl);
     iString *out = new_String();
@@ -204,7 +205,27 @@ iBlock *coverPageSource_Gempub(const iGempub *d) {
     appendProperty_Gempub_(d, "Language:", language_GempubProperty, out);
     appendProperty_Gempub_(d, "License:", license_GempubProperty, out);
     appendProperty_Gempub_(d, "\u00a9", copyright_GempubProperty, out);
-    iBlock *output = copy_Block(utf8_String(out));
-    delete_String(out);
-    return output;
+    return out;
+}
+
+iBool preloadCoverImage_Gempub(const iGempub *d, iGmDocument *doc) {
+    iBool haveImage = iFalse;
+    for (size_t linkId = 1; ; linkId++) {
+        const iString *linkUrl = linkUrl_GmDocument(doc, linkId);
+        if (!linkUrl) break;
+        if (findLinkImage_Media(media_GmDocument(doc), linkId)) {
+            continue; /* got this already */
+        }
+        if (linkFlags_GmDocument(doc, linkId) & imageFileExtension_GmLinkFlag) {
+            iString *imgEntryPath = collect_String(copy_String(linkUrl));
+            remove_Block(&imgEntryPath->chars, 0, size_String(&d->baseUrl) + 1 /* slash, too */);
+            setData_Media(media_GmDocument(doc),
+                          linkId,
+                          collectNewCStr_String(mediaType_Path(linkUrl)),
+                          data_Archive(d->arch, imgEntryPath),
+                          0);
+            haveImage = iTrue;
+        }
+    }
+    return haveImage;
 }
diff --git a/src/gempub.h b/src/gempub.h
index 41d14824..6c1103de 100644
--- a/src/gempub.h
+++ b/src/gempub.h
@@ -24,6 +24,8 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
 
 #include <the_Foundation/string.h>
 
+iDeclareType(GmDocument)
+
 iDeclareType(Gempub)
 iDeclareTypeConstruction(Gempub)
         
@@ -51,7 +53,8 @@ void        close_Gempub            (iGempub *);
 void        setBaseUrl_Gempub       (iGempub *, const iString *baseUrl);
 
 iBool       isOpen_Gempub           (const iGempub *);
-iBlock *    coverPageSource_Gempub  (const iGempub *);
+iString *   coverPageSource_Gempub  (const iGempub *);
+iBool       preloadCoverImage_Gempub(const iGempub *, iGmDocument *doc);
 
 const iString *property_Gempub      (const iGempub *, enum iGempubProperty);
 
diff --git a/src/mimehooks.c b/src/mimehooks.c
index 364449ec..4788d534 100644
--- a/src/mimehooks.c
+++ b/src/mimehooks.c
@@ -209,7 +209,9 @@ iBlock *translateGemPubCoverPage_(const iBlock *source, const iString *requestUr
     if (open_Gempub(gempub, source)) {
         setBaseUrl_Gempub(gempub, requestUrl);
         output = newCStr_Block("20 text/gemini; charset=utf-8\r\n");
-        append_Block(output, collect_Block(coverPageSource_Gempub(gempub)));
+        iString *src = coverPageSource_Gempub(gempub);
+        append_Block(output, utf8_String(src));
+        delete_String(src);
     }
     delete_Gempub(gempub);
     return output;
diff --git a/src/ui/documentwidget.c b/src/ui/documentwidget.c
index b3dd2e83..290bf95e 100644
--- a/src/ui/documentwidget.c
+++ b/src/ui/documentwidget.c
@@ -244,6 +244,7 @@ struct Impl_DocumentWidget {
     iString        sourceMime;
     iBlock         sourceContent; /* original content as received, for saving */
     iTime          sourceTime;
+    iGempub *      sourceGempub; /* NULL unless the page is Gempub content */
     iGmDocument *  doc;
     int            certFlags;
     iBlock *       certFingerprint;
@@ -346,6 +347,7 @@ void init_DocumentWidget(iDocumentWidget *d) {
     init_String(&d->sourceMime);
     init_Block(&d->sourceContent, 0);
     iZap(d->sourceTime);
+    d->sourceGempub = NULL;
     init_PtrArray(&d->visibleLinks);
     init_PtrArray(&d->visiblePre);
     init_PtrArray(&d->visibleWideRuns);
@@ -387,6 +389,7 @@ void deinit_DocumentWidget(iDocumentWidget *d) {
     delete_PtrSet(d->invalidRuns);
     iRelease(d->media);
     iRelease(d->request);
+    delete_Gempub(d->sourceGempub);
     deinit_String(&d->pendingGotoHeading);
     deinit_Block(&d->sourceContent);
     deinit_String(&d->sourceMime);
@@ -1067,51 +1070,43 @@ static const char *zipPageHeading_(const iRangecc mime) {
 }
 
 static void postProcessRequestContent_DocumentWidget_(iDocumentWidget *d) {
+    delete_Gempub(d->sourceGempub);
+    d->sourceGempub = NULL;
     if (!cmpCase_String(&d->sourceMime, "application/octet-stream") ||
         !cmpCase_String(&d->sourceMime, mimeType_Gempub) ||
         endsWithCase_String(d->mod.url, ".gpub")) {
         iGempub *gempub = new_Gempub();
         if (open_Gempub(gempub, &d->sourceContent)) {
             setBaseUrl_Gempub(gempub, d->mod.url);
-            /* TODO: just return a String from coverPageSource_Gempub... */
-            setSource_DocumentWidget(d, collect_String(newBlock_String(collect_Block(coverPageSource_Gempub(gempub)))));
+            setSource_DocumentWidget(d, collect_String(coverPageSource_Gempub(gempub)));
             setCStr_String(&d->sourceMime, mimeType_Gempub);
+            d->sourceGempub = gempub;
+        }
+        else {
+            delete_Gempub(gempub);
         }
-        delete_Gempub(gempub);
     }
-    /* Gempub: Preload cover image. */ {
-        /* TODO: move to gempub.c along with other related code */
+    if (!d->sourceGempub) {
         iString *localPath = localFilePathFromUrl_String(d->mod.url);
-        if (localPath) {
-            if (!iCmpStr(mediaType_Path(localPath), "application/gpub+zip")) {
-                iArchive *arch = iClob(new_Archive());
-                if (openFile_Archive(arch, localPath)) {
-                    iBool haveImage = iFalse;
-                    for (size_t linkId = 1; ; linkId++) {
-                        const iString *linkUrl = linkUrl_GmDocument(d->doc, linkId);
-                        if (!linkUrl) break;
-                        if (findLinkImage_Media(media_GmDocument(d->doc), linkId)) {
-                            continue; /* got this already */
-                        }
-                        if (linkFlags_GmDocument(d->doc, linkId) & imageFileExtension_GmLinkFlag) {
-                            iString *imgEntryPath = collect_String(localFilePathFromUrl_String(linkUrl));
-                            remove_Block(&imgEntryPath->chars, 0, size_String(localPath) + 1 /* slash, too */);
-                            setData_Media(media_GmDocument(d->doc),
-                                          linkId,
-                                          collectNewCStr_String(mediaType_Path(linkUrl)),
-                                          data_Archive(arch, imgEntryPath),
-                                          0);
-                            haveImage = iTrue;
-                        }
-                    }
-                    if (haveImage) {
-                        redoLayout_GmDocument(d->doc);
-                        updateVisible_DocumentWidget_(d);
-                        invalidate_DocumentWidget_(d);
-                    }
-                }
+        if (localPath && equal_CStr(mediaType_Path(localPath), "application/gpub+zip")) {
+            iGempub *gempub = new_Gempub();
+            if (openFile_Gempub(gempub, localPath)) {
+                setBaseUrl_Gempub(gempub, d->mod.url);
+                setSource_DocumentWidget(d, collect_String(coverPageSource_Gempub(gempub)));
+                setCStr_String(&d->sourceMime, mimeType_Gempub);
+                d->sourceGempub = gempub;
             }
-            delete_String(localPath);
+            else {
+                delete_Gempub(gempub);
+            }
+        }
+        delete_String(localPath);
+    }
+    if (d->sourceGempub) {
+        if (preloadCoverImage_Gempub(d->sourceGempub, d->doc)) {
+            redoLayout_GmDocument(d->doc);
+            updateVisible_DocumentWidget_(d);
+            invalidate_DocumentWidget_(d);
         }
     }
 }
@@ -1325,6 +1320,8 @@ static iBool updateFromHistory_DocumentWidget_(iDocumentWidget *d) {
     if (recent && recent->cachedResponse) {
         const iGmResponse *resp = recent->cachedResponse;
         clear_ObjectList(d->media);
+        delete_Gempub(d->sourceGempub);
+        d->sourceGempub = NULL;
         reset_GmDocument(d->doc);
         resetWideRuns_DocumentWidget_(d);
         d->state = fetching_RequestState;
@@ -1527,6 +1524,8 @@ static void checkResponse_DocumentWidget_(iDocumentWidget *d) {
             case categorySuccess_GmStatusCode:
                 reset_SmoothScroll(&d->scrollY);
                 reset_GmDocument(d->doc); /* new content incoming */
+                delete_Gempub(d->sourceGempub);
+                d->sourceGempub = NULL;
                 resetWideRuns_DocumentWidget_(d);
                 updateDocument_DocumentWidget_(d, resp, iTrue);
                 break;