From dd70bba3d93caf500cf2aca172f8cebb2930ca21 Mon Sep 17 00:00:00 2001
From: Alan Janis <alanjanis@icloud.com>
Date: Tue, 20 Aug 2024 22:54:09 +0000
Subject: [PATCH] add iso build script

---
 .../harmonic-install-scripts/rebuild-iso.sh   | 138 ++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 harmonic/scripts/harmonic-install-scripts/rebuild-iso.sh

diff --git a/harmonic/scripts/harmonic-install-scripts/rebuild-iso.sh b/harmonic/scripts/harmonic-install-scripts/rebuild-iso.sh
new file mode 100644
index 0000000..971b0b6
--- /dev/null
+++ b/harmonic/scripts/harmonic-install-scripts/rebuild-iso.sh
@@ -0,0 +1,138 @@
+#!/bin/bash
+##############################################################################
+#
+#   rebuild-iso.sh
+#
+#   This script is executed by a user to rebuild the Apollo ISO using
+#   a modified squashfs root filesystem.
+#
+##############################################################################
+
+export buildTemp="${pwd}/buildtmp"
+export newSquashfs="${buildTemp}/rootfs.squashfs"
+export chrootMounts=("proc" "sys" "dev")
+
+
+runPrint() {
+cat << EOF
+===========================================================
+  $@
+===========================================================
+EOF
+}
+
+# Script Help Function
+showHelp() {
+cat << EOH
+Usage: $0 [-c <chroot directory>] [-r <rootfs source directory> ] [-i <Original Unmodified ISO>] [-h]
+
+Rebuild Apollo ISO
+
+-c|     Setup Chroot Environment in provided directory path
+
+-r|     Use provided rootfs directory to create new  rootfs.squashfs squashfile
+
+-i|     Create a modified ISO file from the original ISO and the modified rootfs.squashfs squashfile.
+
+-h|     Display help
+
+EOH
+}
+
+
+
+function setupChroot() {
+  for mount in "${chrootMounts[@]}"; do
+    runPrint "Bind mounting /${mount} at ${chrootPath}/${mount}"
+    mount --bind "/${mount}" "${chrootPath}/${mount}"
+    done
+  return
+}
+
+function cleanupChroot() {
+  for mount in "${chrootMounts[@]}"; do
+    runPrint "Unmounting ${chrootPath}/${mount}"
+    umount "${chrootPath}/${mount}"
+    done
+    return
+}
+
+function resquashRootfs() {
+  if [[ ! -d ${rootfsSourcePath} ]]; then 
+  runPrint "No root source found!!"
+  exit 1
+  fi
+  runPrint "Creating new squashfs at ${newSquashfs} from ${rootfsSourcePath} ..."
+  mksquashfs "${rootfsSourcePath}" "${newSquashfs}" -noappend
+  return
+}
+
+function rebuildIso() {
+  if [[ ! -f ${originalIsoPath} ]]; then
+  runPrint "No source iso found!!"
+  exit 1
+  fi
+  if [[ ! -e ${newSquashFs} ]]; then
+  runprint "No squashFS found!!"
+  exit 1
+  fi
+
+  export newIso="${buildTemp}/${isoFile}"
+  xorriso -overwrite on -indev "${originalIsoPath}" -outdev "${newIso}" -pathspecs on -add rootfs.squashfs="${newSquashFS}"
+  return
+}
+
+
+while getopts ":hc:r:i:" o; do
+    case "${o}" in
+        h)
+            showHelp
+            exit 0
+            ;;
+        c)
+            chrootPath=${OPTARG}
+            export doChroot=1
+            ;;
+        r)
+            rootfsSourcePath=${OPTARG}
+            export doBuildRootfs=1
+            ;;
+        i)
+            originalIsoPath=${OPTARG}
+            isoFile=$(basename "${originalIsoPath}")
+            export doBuildIso=1
+            ;;
+        :)
+            runPrint "Invalid option: -${OPTARG} requires an argument" 1>&2
+            showHelp
+            exit 1
+            ;;
+        \?|*)
+            runPrint "Invalid option: -${OPTARG}" 1>&2
+            showHelp
+            exit 1
+            ;;
+    esac
+done
+shift $((OPTIND-1))
+
+
+
+# Main Runners
+
+if [[ ${doChroot} == 1 ]]; then
+  # shellcheck disable=SC2312
+  setupChroot  >(tee -a "${buildlog}" >&2) > >(tee -a "${buildlog}")
+  trap cleanupChroot EXIT
+  chroot "${chrootPath}"
+fi
+
+if [[ "${doBuildRootfs}" == 1 ]]; then
+  # shellcheck disable=SC2312
+  resquashRootfs >(tee -a  "${buildlog}" >&2) > >(tee -a  "${buildlog}")
+fi
+
+if [[ "${doBuildIso}" == 1 ]]; then
+  # shellcheck disable=SC2312
+  rebuildIso >(tee -a "${buildlog}">&2) > >(tee -a "${buildlog}")
+fi
\ No newline at end of file