improved preseed fetch script

1 files changed, 24 insertions, 3 deletions

diff --git a/preseed/fetch-example.sh b/preseed/fetch-example.sh
index 52f087db..fa89bdba 100755
--- a/preseed/fetch-example.sh
+++ b/preseed/fetch-example.sh
@@ -1,8 +1,29 @@
 #!/bin/bash
 
-if [ -z "$1" ]; then
-  echo "Usage: $0 <release>"
+if [ -z "$1" ] && [ -z "$2" ]; then
+  echo "Usage: $0 <distro> <codename>"
   exit 1
 fi
 
-wget "https://www.debian.org/releases/$1/example-preseed.txt" -O "$1-example.cfg"
+distro="$1"
+codename="$2"
+
+case "$distro" in
+  debian)
+     wget "https://www.debian.org/releases/$codename/example-preseed.txt" -O "$codename-example.cfg"
+     ;;
+  ubuntu)
+     rel="$codename"
+     case "$codename" in
+       xenial)
+         rel="16.04"
+         ;;
+       bionic)
+         rel="18.04"
+         ;;
+     esac
+     wget "https://help.ubuntu.com/$rel/installation-guide/example-preseed.txt" -O "$codename-example.cfg"
+     ;;
+esac
+
+exit 0