From d896a5831b29b9dafe84a006b72aa88413d919f9 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 26 Nov 2021 00:06:18 +0000
Subject: [PATCH] Fill out apt cache config file

---
 src/run.sh                | 31 +++++++++++++++++++++++++++++++
 src/steps/10-apt-cache.sh | 31 +++++++++++++++++++++++++------
 2 files changed, 56 insertions(+), 6 deletions(-)

diff --git a/src/run.sh b/src/run.sh
index 7ea18c5..318bdf0 100644
--- a/src/run.sh
+++ b/src/run.sh
@@ -7,6 +7,11 @@ fi
 
 ###############################################################################
 
+step_current="0";
+step_max="1";
+
+###############################################################################
+
 ###
 # Load the lantern build engine
 ###
@@ -22,4 +27,30 @@ if [ ! -f "${lantern_path}/lantern.sh" ]; then git submodule update --init "${la
 #shellcheck disable=SC1090
 source "${lantern_path}/lantern.sh";
 
+#shellcheck disable=SC1090
+source "/etc/os-release";
+
 ###############################################################################
+
+
+###############################################################################
+
+ask_yesno() {
+	local question="$1";
+	
+	whiptail --title "Step ${step_current} / ${step_max}" --yesno "${question}" 40 8;
+	return "$?"; # Not actually needed, but best to be explicit
+}
+
+###############################################################################
+
+# ███████ ████████ ███████ ██████       ██
+# ██         ██    ██      ██   ██     ███
+# ███████    ██    █████   ██████       ██
+#      ██    ██    ██      ██           ██
+# ███████    ██    ███████ ██           ██
+step_current="1";
+
+if ask_yesno "Use apt cache?"; then
+	source ./steps/10-apt-cache.sh;
+fi
diff --git a/src/steps/10-apt-cache.sh b/src/steps/10-apt-cache.sh
index 686b19c..a2cc754 100644
--- a/src/steps/10-apt-cache.sh
+++ b/src/steps/10-apt-cache.sh
@@ -1,9 +1,28 @@
 #!/usr/bin/env bash
 
+enable_apt_cache() {
+	local apt_caching_server="http://172.16.230.100:3142";
+	
+	task_begin "Configuring ${apt_caching_server} as apt cache";
+	
+	subtask_begin "Updating apt configuration";
+	echo 'Acquire::http { Proxy \"${apt_caching_server}\"; }' >/etc/apt/apt.conf.d/proxy
+	subtask_end "$?";
+	
+	# From /etc/os-release - see run.sh
+	if [[ "${ID}" == "raspbian" ]]; then
+		subtask_begin "Patching for Raspbian [ 1 / 2 ]";
+		echo 'deb http://archive.raspberrypi.org/debian/ buster main ui' >/etc/apt/sources.list.d/raspi.list;
+		subtask_end "$?";
+		
+		subtask_begin "Patching for Raspbian [ 2 / 2 ]";
+		if [[ -e "/etc/apt/sources.list.d/raspi" ]]; then
+			rm /etc/apt/sources.list.d/raspi;
+		fi
+		subtask_end "$?";
+	fi
+	
+	task_end "$?" "Failed to configure apt cache.";
+}
 
-echo 'Acquire::http { Proxy \"http://172.16.230.100:3142\"; }' | tee /etc/apt/apt.conf.d/proxy
-
-
-echo 'deb http://archive.raspberrypi.org/debian/ buster main ui' | tee /etc/apt/sources.list.d/raspi.list
-
-rm /etc/apt/sources.list.d/raspi;
+enable_apt_cache;