From 5426f6c75f967e5cf9ea8344cbff49e483f933f0 Mon Sep 17 00:00:00 2001
From: Matt Theissen <theissenm@skyfalltech.net>
Date: Mon, 19 Jul 2021 01:19:15 -0500
Subject: [PATCH] Basic withdraw is working

---
 ddbank.sh | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/ddbank.sh b/ddbank.sh
index 29e8f79..f83bf3f 100755
--- a/ddbank.sh
+++ b/ddbank.sh
@@ -47,7 +47,9 @@ new_account() {
 }
 
 list_user() {
+    echo ""
     $db_ssh "psql -U ddbank -d ddbank -c \"select * from account;\""
+    echo ""
 }
 
 list_balance() {
@@ -68,7 +70,7 @@ list_balance() {
             1 ) clear
                 $db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id);\"" ;;
             2 ) read -p "Enter the user account name: " ex_mat
-                $db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id) where username='$ex_mat';\""
+                $db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id) where username='$ex_mat';\"" ;;
             3 ) read -p "Enter search term: " bal_user
                     bal_user1=$($db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id);\"")
                     echo "$bal_user1" | awk 'FNR == 1 {print}'
@@ -94,14 +96,14 @@ do
 
     case $ii_bank in
         [0q] ) break ;;
-        1 ) read -p "Withdraw from user: " bank_user
+        1 ) read -p "Withdraw from user: " bank_user 
             read -p "Amount to withdraw: " bank_wth_amt
-            echo "$bal_user2" | awk 'FNR == 1 {print}'
-            bank_1_amt=$($db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id) where username='$bank_user';\"")
-            new_amt="$bank_wth_amt - $bank_1_amt"
-            $db_ssh "psql -U ddbank -d ddbank -c \"update accounts set balance = '$new_amt' where username = '$bank_user';\"" ;;
-            
-
+            echo "$bank_user" | awk 'FNR == 1 {print}' 
+            bank_1_amt=$($db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id) where username='$bank_user';\""| awk 'FNR == 3' | awk -F\| '{print $3}') 
+            bank_user_id=$($db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id) where username='$bank_user';\"" | awk 'FNR == 3' | awk -F\| '{print $1}') 
+            new_amt=$($db_ssh "psql -U ddbank -d ddbank -c \"select balance - '$bank_wth_amt' from balance where id = '$bank_user_id';\"" | awk 'FNR ==3')
+            $db_ssh "psql -U ddbank -d ddbank -c \"update balance set balance = '$new_amt' where id = '$bank_user_id';\"" ;;
+        * ) clear ;;
     esac
 done
 }
@@ -115,7 +117,7 @@ do
     echo "1) Create new user account."
     echo "2) List user accounts."
     echo "3) List user(s) balance."
-    echo "4) Deposit, withdraw, transfer"
+    echo "4) Banking (Deposit, withdraw, transfer)"
     echo "9) Clear contents of screen"
     echo "0) Exit"
     echo ""