From d924ce2ef5412d59fdd1e3126b38bcd9f55bff7d Mon Sep 17 00:00:00 2001
From: Mateja <mail@matejamaric.com>
Date: Wed, 27 Jan 2021 23:03:26 +0100
Subject: Add option for non-free and contrib repos only.

---
 setup.sh | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/setup.sh b/setup.sh
index 19bbbff..6a1d82c 100755
--- a/setup.sh
+++ b/setup.sh
@@ -5,6 +5,7 @@ config_install=false;
 termite_install=false;
 neovim_install=false;
 latex_install=false;
+non_free=false;
 nvidia_install=false;
 
 read -p "Do you want to install Qtile? " -n 1 -r
@@ -21,18 +22,25 @@ echo
 
 read -p "Do you want to install Neovim? " -n 1 -r
 echo
-if [[ $REPLY =~ ^[Yy]$ ]] && $neovim_install=true;
+[[ $REPLY =~ ^[Yy]$ ]] && $neovim_install=true;
 
 read -p "Do you want to install LaTeX? " -n 1 -r
 echo
-if [[ $REPLY =~ ^[Yy]$ ]] && $latex_install=true;
+[[ $REPLY =~ ^[Yy]$ ]] && $latex_install=true;
 
+read -p "Do you want to enable non-free repos?" -n 1 -r
+echo
+[[ $REPLY =~ ^[Yy]$ ]] && $non_free=true;
+
+if [[ $non_free ]]
+then
 read -p "Do you want to install proprietary nvidia drivers? " -n 1 -r
 echo
-if [[ $REPLY =~ ^[Yy]$ ]] && $nvidia_install=true;
+[[ $REPLY =~ ^[Yy]$ ]] && $nvidia_install=true;
+fi
 
 # Adding non-free repos if necessary.
-if [[ $nvidia_install ]]
+if [[ $non_free ]]
 then
 cat > /tmp/sources.list <<EOF
 deb http://deb.debian.org/debian/ buster main contrib non-free
-- 
cgit v1.2.3