#!/bin/bash

if [[ $1 == "Debian" ]]; then

    echo "$1 Sudo Setup"
    while [[ true ]]; do
        if [[ ! -z $(grep $2 /etc/passwd) ]]; then
            echo "usermod -aG sudo $2"
            if [[ -f /home/$2/ ]]; then
                echo "Home exists"
            else
                echo "Home doesn't Exist"
            fi
            break
        else
            read -p "Invalid Username. Use an valid existing username." username
        fi
    done

elif [[ $1 == "Arch" ]]; then

    echo "$1 Sudo Setup"
    while [[ true ]]; do
        if [[ ! -z $(grep $2 /etc/passwd) ]]; then
            echo "usermod -aG wheel $2"
            if [[ -f /home/$2/ ]]; then
                echo "Home exists"
            else
                echo "Home doesn't Exist"
            fi
            break
        else
            read -p "Invalid Username. Use a valid existing username:" username
        fi
    done

fi