From 169b364c79259f98ec0cabb0c7dda1edd707b2dc Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Sun, 14 Jan 2024 15:19:58 +0100
Subject: [PATCH] feat: Add installer (#5)

---
 Dockerfile                |  2 +-
 src/{init.sh => entry.sh} |  5 ++-
 src/install.sh            | 64 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 69 insertions(+), 2 deletions(-)
 rename src/{init.sh => entry.sh} (88%)
 create mode 100644 src/install.sh

diff --git a/Dockerfile b/Dockerfile
index ccdefc7..b9839e4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -23,4 +23,4 @@ ENV DISK_SIZE "64G"
 ARG VERSION_ARG "0.0"
 RUN echo "$VERSION_ARG" > /run/version
 
-ENTRYPOINT ["/usr/bin/tini", "-s", "/run/init.sh"]
+ENTRYPOINT ["/usr/bin/tini", "-s", "/run/entry.sh"]
diff --git a/src/init.sh b/src/entry.sh
similarity index 88%
rename from src/init.sh
rename to src/entry.sh
index 6a0ebc0..e8a0594 100644
--- a/src/init.sh
+++ b/src/entry.sh
@@ -4,6 +4,9 @@ set -Eeuo pipefail
 echo "❯ Starting Windows for Docker v$(</run/version)..."
 echo "❯ For support visit https://github.com/dockur/windows"
 
+export DISPLAY=web
+export BOOT_MODE=windows
+
 cd /run
 
 . reset.sh      # Initialize system
@@ -21,7 +24,7 @@ if [[ "${DISPLAY,,}" == "web" ]]; then
   websockify -D --web /usr/share/novnc/ 8006 localhost:5900 2>/dev/null
 fi
 
-info "Booting image using $VERS..."
+info "Booting Windows using $VERS..."
 
 [[ "$DEBUG" == [Yy1]* ]] && set -x
 exec qemu-system-x86_64 ${ARGS:+ $ARGS}
diff --git a/src/install.sh b/src/install.sh
new file mode 100644
index 0000000..f18f69e
--- /dev/null
+++ b/src/install.sh
@@ -0,0 +1,64 @@
+#!/usr/bin/env bash
+set -Eeuo pipefail
+
+# Check if running with interactive TTY or redirected to docker log
+if [ -t 1 ]; then
+  PROGRESS="--progress=bar:noscroll"
+else
+  PROGRESS="--progress=dot:giga"
+fi
+
+info "Downloading installer..."
+
+URL="https://raw.githubusercontent.com/ElliotKillick/Mido/main/Mido.sh"
+{ wget "$URL" -O /run/Mido.sh -q --no-check-certificate; rc=$?; } || :
+
+(( rc != 0 )) && error "Failed to download $URL, reason: $rc" && exit 65
+
+chmod +x /run/Mido.sh
+bash /run/Mido.sh
+
+exit 99
+
+BASE="boot.img"
+[ -f "$STORAGE/$BASE" ] && return 0
+
+if [ -z "$BOOT" ]; then
+  error "No boot disk specified, set BOOT= to the URL of an ISO file." && exit 64
+fi
+
+BASE=$(basename "$BOOT")
+[ -f "$STORAGE/$BASE" ] && return 0
+
+TMP="$STORAGE/${BASE%.*}.tmp"
+rm -f "$TMP"
+
+info "Downloading $BASE as boot image..."
+
+{ wget "$BOOT" -O "$TMP" -q --no-check-certificate --show-progress "$PROGRESS"; rc=$?; } || :
+
+(( rc != 0 )) && error "Failed to download $BOOT, reason: $rc" && exit 60
+[ ! -f "$TMP" ] && error "Failed to download $BOOT" && exit 61
+
+SIZE=$(stat -c%s "$TMP")
+
+if ((SIZE<100000)); then
+  error "Invalid ISO file: Size is smaller than 100 KB" && exit 62
+fi
+
+DEST="$STORAGE/drivers.img"
+
+if [ ! -f "$DEST" ]; then
+
+  info "Downloading VirtIO drivers for Windows..."
+  DRIVERS="https://fedorapeople.org/groups/virt/virtio-win/direct-downloads/stable-virtio/virtio-win.iso"
+
+  { wget "$DRIVERS" -O "$DEST" -q --no-check-certificate --show-progress "$PROGRESS"; rc=$?; } || :
+
+  (( rc != 0 )) && info "Failed to download $DRIVERS, reason: $rc" && rm -f "$DEST"
+
+fi
+
+mv -f "$TMP" "$STORAGE/$BASE"
+
+return 0