feat(api): Made PDFs-enabled check earlier

.dockerignore

@@ -7,4 +7,4 @@ build
 .env
 *log
 *.md
-src/api/uv.lock
+uv.lock

Dockerfile

@@ -6,6 +6,10 @@ RUN rm -rf node_modules
 RUN rm -rf build
 RUN npm install
 COPY src/frontend/. .
+
+ARG pdfs=false
+ENV VITE_ENABLE_PDFS=$pdfs
+
 RUN npm run build
 # Thanks https://stackoverflow.com/q/76988450
 
@@ -40,6 +44,9 @@ RUN ln -s /app/src/pdf/fonts /tmp/fonts
 
 WORKDIR /app/src
 
+ARG pdfs=false
+ENV VITE_ENABLE_PDFS=$pdfs
+
 EXPOSE 80
 
 CMD [ "uv", "run", "main.py"]

README.md

@@ -53,5 +53,5 @@ My thanks to [aerkalov/ebooklib](https://github.com/aerkalov/ebooklib) for a fas
 ---
 
 <div align="center">
-    <p>TheOnlyWayUp © 2026</p>
+    <p>TheOnlyWayUp © 2024</p>
 </div>

src/api/src/main.py

@@ -2,6 +2,7 @@
 
 import asyncio
 from enum import Enum
+from os import getenv
 from pathlib import Path
 from typing import Optional
 from zipfile import ZipFile
@@ -36,6 +37,8 @@ from create_book.parser import clean_tree, fetch_tree_images
 app = FastAPI()
 BUILD_PATH = Path(__file__).parent / "build"
 
+PDFS_ENABLED = True if getenv("VITE_ENABLE_PDFS") == "true" else False
+
 
 class RequestCancelledMiddleware:
     # Thanks https://github.com/fastapi/fastapi/discussions/11360#discussion-6427734
@@ -158,6 +161,13 @@ async def handle_download(
         else:
             cookies = None
 
+        if format == DownloadFormat.pdf and not PDFS_ENABLED:
+            logger.error("PDF Downloads not enabled.")
+            return HTMLResponse(
+                status_code=403,
+                content='PDF Downloads have been disabled by the server administrator. Support is available on the <a href="https://discord.gg/P9RHC4KCwd" target="_blank">Discord</a>',
+            )
+
         match mode:
             case DownloadMode.story:
                 story_id = download_id
@@ -174,26 +184,12 @@ async def handle_download(
         story_zip = await fetch_story_content_zip(story_id, cookies)
         archive = ZipFile(story_zip, "r")
 
-        # Transform part metadata into an easily-indexable dictionary
+        part_trees: list[BeautifulSoup] = [
-        part_id_title_dictionary = {
+            clean_tree(
-            str(part["id"]): part["title"] for part in metadata["parts"]
+                part["title"], part["id"], archive.read(str(part["id"])).decode("utf-8")
-        }
-
-        part_trees: list[BeautifulSoup] = []
-
-        for id in archive.namelist():
-            if (
-                id not in part_id_title_dictionary
-            ):  # If a part is deleted and the old story_zip is cached, this is needed to avoid a KeyError exception
-                continue
-
-            part_trees.append(
-                clean_tree(
-                    part_id_title_dictionary[id],
-                    id,
-                    archive.read(id).decode("utf-8"),
-                )
             )
+            for part in metadata["parts"]
+        ]
 
         images = (
             [await fetch_tree_images(tree) for tree in part_trees]
@@ -229,7 +225,7 @@ async def handle_download(
                 yield chunk
 
         return StreamingResponse(
-            iterfile(),
+            book_buffer if PDFS_ENABLED else iterfile(),
             media_type=media_type,
             headers={
                 "Content-Disposition": f'attachment; filename="{slugify(metadata["title"])}_{story_id}{"_images" if download_images else ""}.{format.value}"',  # Thanks https://stackoverflow.com/a/72729058

src/frontend/src/routes/+error.svelte

@@ -19,7 +19,7 @@
           href="https://github.com/TheOnlyWayUp"
           class="underline"
           target="_blank">TheOnlyWayUp</a
-        > © 2026
+        > © 2025
       </p>
     </aside>
   </footer>

src/frontend/src/routes/+layout.svelte

@@ -44,7 +44,7 @@
       >
     </div>
     <p>
-      Copyright © 2026 - All rights reserved by <a
+      Copyright © 2025 - All rights reserved by <a
         href="https://rambhat.la"
         class="link"
         target="_blank"

src/frontend/src/routes/+page.svelte

@@ -1,4 +1,6 @@
 <script>
+  const PDFS_ENABLED = import.meta.env.VITE_ENABLE_PDFS === "true";
+
   let downloadImages = $state(false);
   let downloadAsPdf = $state(false); // 0 = epub, 1 = pdf
   let isPaidStory = $state(false);
@@ -110,29 +112,32 @@
           >
             WP Downloader
           </h1>
-          <div role="alert" class="alert mt-10 max-w-md break-words bg-green-200">
+          {#if !PDFS_ENABLED}
-            <svg
+            <div role="alert" class="alert mt-10 max-w-md break-words bg-green-200">
-              xmlns="http://www.w3.org/2000/svg"
+              <svg
-              fill="none"
+                xmlns="http://www.w3.org/2000/svg"
-              viewBox="0 0 24 24"
+                fill="none"
-              class="h-6 w-6 shrink-0 stroke-current"
+                viewBox="0 0 24 24"
-            >
+                class="h-6 w-6 shrink-0 stroke-current"
-              <path
-                stroke-linecap="round"
-                stroke-linejoin="round"
-                stroke-width="2"
-                d="M13 16h-1v-4h-1m1-4h.01M21 12a9 9 0 11-18 0 9 9 0 0118 0z"
-              ></path>
-            </svg>
-            <div>
-              <p>
-                Donators get access to <span class="font-semibold">high-speed PDF Downloads</span>
-              </p>
-              <a href="https://buymeacoffee.com/theonlywayup" class="link" target="_blank"
-                >Donate now</a
               >
+                <path
+                  stroke-linecap="round"
+                  stroke-linejoin="round"
+                  stroke-width="2"
+                  d="M13 16h-1v-4h-1m1-4h.01M21 12a9 9 0 11-18 0 9 9 0 0118 0z"
+                ></path>
+              </svg>
+
+              <div>
+                <p>
+                  Donators get access to <span class="font-semibold">high-speed PDF Downloads</span>
+                </p>
+                <a href="https://buymeacoffee.com/theonlywayup" class="link" target="_blank"
+                  >Donate now</a
+                >
+              </div>
             </div>
-          </div>
+          {/if}
           <!-- <div role="alert" class="alert bg-cyan-300 mt-5">
             <svg
               xmlns="http://www.w3.org/2000/svg"
@@ -154,10 +159,13 @@
           </p>
           <ul class="list list-inside pt-4 text-xl">
             <!-- TODO: 'max-lg: hidden' to hide on screen sizes smaller than lg. I'll do this when I figure out how to make this show up _below_ the card on smaller screen sizes. -->
-            <!-- <li>12/24 - ⚡ Super-fast Downloads!</li>
-            <li>12/24 - 📑 PDF Downloads!</li> -->
             <li>05/25 - ⚖️ Legal Compliance</li>
-            <li>12/24 - 📂 Less Errors, Throttled Downloads</li>
+            {#if PDFS_ENABLED}
+              <li>12/24 - ⚡ Super-fast Downloads!</li>
+              <li>12/24 - 📑 PDF Downloads!</li>
+            {:else}
+              <li>12/24 - 📂 Less Errors, Throttled Downloads</li>
+            {/if}
             <li>11/24 - 🔗 Paste Links!</li>
             <li>11/24 - 📨 Send to Kindle Support!</li>
 
@@ -249,26 +257,30 @@
                 href={url}
                 onclick={() => (afterDownloadPage = true)}>Download</a
               >
+            </div>
 
-              <!-- <label class="swap w-fit label mt-2">
+            {#if PDFS_ENABLED}
+              <label class="swap w-fit label mt-2 pb-2">
                 <input type="checkbox" bind:checked={downloadAsPdf} />
-                <div class="swap-on">
+                <div class="swap-on absolute left-0 text-gray-800">
                   Downloading as <span class=" underline text-bold">PDF</span> (Click)
                 </div>
-                <div class="swap-off">
+                <div class="swap-off absolute left-0 text-gray-800">
                   Downloading as <span class=" underline text-bold">EPUB</span> (Click)
                 </div>
-              </label> -->
-
-              <label class="label cursor-pointer">
-                <span class="label-text text-gray-800">Include Images (<strong>Slower Download</strong>)</span>
-                <input
-                  type="checkbox"
-                  class="checkbox-warning checkbox shadow-md"
-                  bind:checked={downloadImages}
-                />
               </label>
-            </div>
+            {/if}
+
+            <label class="label cursor-pointer">
+              <span class="label-text text-gray-800"
+                >Include Images (<strong>Slower Download</strong>)</span
+              >
+              <input
+                type="checkbox"
+                class="checkbox-warning checkbox shadow-md"
+                bind:checked={downloadImages}
+              />
+            </label>
           </form>
         </div>
       {:else}