From 889df881ba5a4662284cc60e1fe783729ba95d58 Mon Sep 17 00:00:00 2001
From: "Aaron D. Lee" <aaron.daniel.lee@gmail.com>
Date: Sat, 10 Jan 2026 22:20:40 -0500
Subject: [PATCH] RPi: Fix jpeglib ARM64 build (missing headers)

jpeglib has no pre-built ARM64 wheel and the source tarball is missing
libjpeg header files. This adds a workaround that downloads the official
libjpeg headers before building.

- Add rpi/patches/jpeglib/install-jpeglib-arm64.sh helper script
- Update setup.sh to download headers when building from source
- Downloads headers for libjpeg 6b, 7-9f, turbo, and mozjpeg versions

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
---
 rpi/patches/jpeglib/install-jpeglib-arm64.sh | 57 ++++++++++++++++++++
 rpi/setup.sh                                 | 46 ++++++++++++++--
 2 files changed, 100 insertions(+), 3 deletions(-)
 create mode 100755 rpi/patches/jpeglib/install-jpeglib-arm64.sh

diff --git a/rpi/patches/jpeglib/install-jpeglib-arm64.sh b/rpi/patches/jpeglib/install-jpeglib-arm64.sh
new file mode 100755
index 0000000..582673a
--- /dev/null
+++ b/rpi/patches/jpeglib/install-jpeglib-arm64.sh
@@ -0,0 +1,57 @@
+#!/bin/bash
+#
+# Install jpeglib on ARM64 Linux (Raspberry Pi)
+# Works around missing headers in the source tarball
+#
+# Usage: ./install-jpeglib-arm64.sh
+#
+
+set -e
+
+echo "Installing jpeglib for ARM64..."
+
+# Create temp directory
+WORKDIR=$(mktemp -d)
+cd "$WORKDIR"
+
+# Download jpeglib source
+echo "  Downloading jpeglib source..."
+pip download jpeglib==1.0.2 --no-binary :all: --no-deps -d . -q
+tar -xzf jpeglib-1.0.2.tar.gz
+cd jpeglib-1.0.2
+
+# Download official libjpeg sources and copy headers
+echo "  Downloading libjpeg headers..."
+CJPEGLIB="src/jpeglib/cjpeglib"
+
+# libjpeg 6b
+curl -sL "https://www.ijg.org/files/jpegsrc.v6b.tar.gz" | tar -xzf -
+cp jpeg-6b/*.h "$CJPEGLIB/6b/"
+
+# libjpeg 7-9f (all use similar headers from 9e)
+curl -sL "https://www.ijg.org/files/jpegsrc.v9f.tar.gz" | tar -xzf -
+for v in 7 8 8a 8b 8c 8d 9 9a 9b 9c 9d 9e 9f; do
+    cp jpeg-9f/*.h "$CJPEGLIB/$v/"
+done
+
+# libjpeg-turbo versions
+curl -sL "https://github.com/libjpeg-turbo/libjpeg-turbo/archive/refs/tags/2.1.0.tar.gz" | tar -xzf -
+for v in turbo120 turbo130 turbo140 turbo150 turbo200 turbo210; do
+    cp libjpeg-turbo-2.1.0/*.h "$CJPEGLIB/$v/" 2>/dev/null || true
+done
+
+# mozjpeg versions
+curl -sL "https://github.com/mozilla/mozjpeg/archive/refs/tags/v4.0.3.tar.gz" | tar -xzf -
+for v in mozjpeg101 mozjpeg201 mozjpeg300 mozjpeg403; do
+    cp mozjpeg-4.0.3/*.h "$CJPEGLIB/$v/" 2>/dev/null || true
+done
+
+# Build and install
+echo "  Building jpeglib..."
+pip install . -q
+
+# Cleanup
+cd /
+rm -rf "$WORKDIR"
+
+echo "  Done! jpeglib installed successfully."
diff --git a/rpi/setup.sh b/rpi/setup.sh
index 3a7349b..9e787ba 100755
--- a/rpi/setup.sh
+++ b/rpi/setup.sh
@@ -340,9 +340,49 @@ else
     # Upgrade pip and install build tools
     pip install --upgrade pip setuptools wheel
 
-    # Install stegasoo with all dependencies
-    # jpeglib installs cleanly via pip (no patching needed like jpegio)
-    echo "  Installing dependencies (jpeglib, scipy, etc.)..."
+    # Install jpeglib (has no ARM64 wheel, needs headers fix)
+    echo "  Installing jpeglib for ARM64..."
+    if [ -f "$INSTALL_DIR/rpi/patches/jpeglib/install-jpeglib-arm64.sh" ]; then
+        bash "$INSTALL_DIR/rpi/patches/jpeglib/install-jpeglib-arm64.sh"
+    else
+        # Inline fix: download headers and build
+        JPEGLIB_WORKDIR=$(mktemp -d)
+        cd "$JPEGLIB_WORKDIR"
+        pip download jpeglib==1.0.2 --no-binary :all: --no-deps -d . -q
+        tar -xzf jpeglib-1.0.2.tar.gz
+        cd jpeglib-1.0.2
+        CJPEGLIB="src/jpeglib/cjpeglib"
+
+        # Download and copy libjpeg 6b headers
+        curl -sL "https://www.ijg.org/files/jpegsrc.v6b.tar.gz" | tar -xzf -
+        cp jpeg-6b/*.h "$CJPEGLIB/6b/"
+
+        # Download and copy libjpeg 9f headers (works for 7-9f)
+        curl -sL "https://www.ijg.org/files/jpegsrc.v9f.tar.gz" | tar -xzf -
+        for v in 7 8 8a 8b 8c 8d 9 9a 9b 9c 9d 9e 9f; do
+            cp jpeg-9f/*.h "$CJPEGLIB/$v/"
+        done
+
+        # Download and copy libjpeg-turbo headers
+        curl -sL "https://github.com/libjpeg-turbo/libjpeg-turbo/archive/refs/tags/2.1.0.tar.gz" | tar -xzf -
+        for v in turbo120 turbo130 turbo140 turbo150 turbo200 turbo210; do
+            cp libjpeg-turbo-2.1.0/*.h "$CJPEGLIB/$v/" 2>/dev/null || true
+        done
+
+        # Download and copy mozjpeg headers
+        curl -sL "https://github.com/mozilla/mozjpeg/archive/refs/tags/v4.0.3.tar.gz" | tar -xzf -
+        for v in mozjpeg101 mozjpeg201 mozjpeg300 mozjpeg403; do
+            cp mozjpeg-4.0.3/*.h "$CJPEGLIB/$v/" 2>/dev/null || true
+        done
+
+        # Build and install
+        pip install .
+        cd "$INSTALL_DIR"
+        rm -rf "$JPEGLIB_WORKDIR"
+    fi
+
+    # Install remaining dependencies
+    echo "  Installing remaining dependencies..."
     pip install -e ".[web]"
 fi