diff --git a/ddbank.sh b/ddbank.sh index 3e803f6..982ffc0 100755 --- a/ddbank.sh +++ b/ddbank.sh @@ -12,19 +12,21 @@ ssh_host="knd-db01.kniod.corp" # Functions new_account() { read -p "Enter the name of your account: " user1 + echo "" echo "Your account name will be $user1." - read -p "Is that correct? [y/N] " yn + echo "" + read -n1 -p "Is that correct? [y/N] " yn case $yn in - # If user selects yes. Insert their username and a timestamp into the database. - [Yy]* ) ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"insert into account (username,timestamp) values ('$user1','now');\"" 2>/dev/null || echo "Username already taken."; + # If user selects yes. Insert their username and a timestamp into the database. + [Yy] ) ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"insert into account (username,timestamp) values ('$user1','now');\"" &>/dev/null || echo "Username already taken."; # Sets the user's id as the variable. user1_id=$(ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"select id from account where username='$user1';\"" | awk 'FNR == 3 {print}') # Gives the user the starting balance. - ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"insert into balance (id,balance) values ('$user1_id',$bal);\"" + ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"insert into balance (id,balance) values ('$user1_id',$bal);\"" &>/dev/null # Pulls the user's account and ID for awk'ing user_account=$(ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"select id,username from account;\"" | grep $user1_id) @@ -36,8 +38,8 @@ new_account() { ;; - # If user selects no - [Nn]* ) echo "Aborted" + # If the user selects anything other than [Yy] + * ) echo "Aborted" ;; esac @@ -53,9 +55,8 @@ echo "2) List user accounts." echo "3) List user(s) balance." echo " " read -p "Select from the menu above: " i -if [ $i == 1 ] -then +if [ $i == 1 ]; then { new_account -else +} else { echo "Some other thing..." -fi +} fi