From a3e7c9d33ab9816d251b7332fb1af914a6adb8bb Mon Sep 17 00:00:00 2001
From: Renat <renat.sakenov@googlemail.com>
Date: Mon, 22 Jan 2018 22:20:04 +0100
Subject: [PATCH] dialog for partitionsize

---
 src/arch.sh | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/arch.sh b/src/arch.sh
index 112fcb0..e9cc08e 100755
--- a/src/arch.sh
+++ b/src/arch.sh
@@ -16,6 +16,15 @@ dialog --defaultno --title "DON'T BE A BRAINLET!" --yesno "Do you think I'm memi
 
 dialog --no-cancel --inputbox "Enter a name for your computer." 10 60 2> comp
 
+dialog --no-cancel --inputbox "Enter partitionsize in gb, separated by space (swap & root)." 10 60 2>psize
+
+IFS=' ' read -ra SIZE <<< $(cat psize)
+
+re='^[0-9]+$'
+if ! [ ${#SIZE[@]} -eq 2 ] || ! [[ ${SIZE[0]} =~ $re ]] || ! [[ ${SIZE[1]} =~ $re ]] ; then
+    SIZE=(12 25);
+fi
+
 timedatectl set-ntp true
 
 cat <<EOF | fdisk /dev/sda
@@ -29,12 +38,12 @@ n
 p
 
 
-+12G
++${SIZE[0]}G
 n
 p
 
 
-+25G
++${SIZE[1]}G
 n
 p