From 6ccf3f533c8e6e19ebd192039de8e5be6c961e42 Mon Sep 17 00:00:00 2001
From: Matt Theissen <theissenm@skyfalltech.net>
Date: Sun, 18 Jul 2021 13:30:01 -0500
Subject: [PATCH] Added listing users and balances

---
 ddbank.sh | 57 +++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 45 insertions(+), 12 deletions(-)

diff --git a/ddbank.sh b/ddbank.sh
index 982ffc0..f738855 100755
--- a/ddbank.sh
+++ b/ddbank.sh
@@ -43,20 +43,53 @@ new_account() {
         ;;
 
     esac
-
 }
 
+list_user() {
+    ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"select * from account;\""
+}
 
+list_balance() {
+    # Bal Menu
+    clear
+    echo "1) List all users balances."
+    echo "2) List a specific user's balance." 
+    echo ""
+    read -n1 -p "Select from the menu above: " ii
+    echo ""
 
+    case $ii in
+        1 ) ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"select id,username from account join balance using (id);\"" ;;
+        2 ) read -p "Enter the user account name: " bal_user
+                bal_user1=$(ssh $ssh_user@$ssh_host "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id);\"")
+                echo "$bal_user1" | awk 'FNR == 1 {print}'
+                echo "$bal_user1" | grep "$bal_user"     ;;
+    esac
+}
 
-# Menu
-echo "1) Create new user account."
-echo "2) List user accounts."
-echo "3) List user(s) balance."
-echo " "
-read -p "Select from the menu above: " i
-if [ $i == 1 ]; then {
-    new_account
-} else {
-    echo "Some other thing..."
-} fi
+while true
+do
+    # Menu
+    echo ""
+    echo "DDBank Menu: "
+    echo "1) Create new user account."
+    echo "2) List user accounts."
+    echo "3) List user(s) balance."
+    echo "9) Clear contents of screen"
+    echo "0) Exit"
+    echo ""
+    read -p "Select from the menu above: " -n1 i
+    case $i in 
+        1 ) new_account     ;;
+        2 ) list_user       ;;
+        3 ) list_balance    ;;
+        9 ) clear           ;;
+        0 ) exit
+    esac
+done
+
+#if [ $i == 1 ]; then {
+#    new_account
+#} else {
+#    echo "Some other thing..."
+#} fi