|
|
@ -1,12 +1,23 @@
|
|
|
|
#!/usr/bin/env sh
|
|
|
|
#!/usr/bin/env sh
|
|
|
|
|
|
|
|
|
|
|
|
#Config
|
|
|
|
## **Config file should have the following:
|
|
|
|
## Starting balance
|
|
|
|
## #Config
|
|
|
|
bal="00.00"
|
|
|
|
## ## Starting balance
|
|
|
|
## SSH Config
|
|
|
|
## bal="00.00"
|
|
|
|
ssh_user="theissenm"
|
|
|
|
##
|
|
|
|
ssh_host="knd-db01.kniod.corp"
|
|
|
|
## ## Postgres
|
|
|
|
db_ssh="ssh $ssh_user@$ssh_host"
|
|
|
|
## ### Database user
|
|
|
|
|
|
|
|
## dbuser="ddbank"
|
|
|
|
|
|
|
|
## ### Database Port
|
|
|
|
|
|
|
|
## dbport="5432"
|
|
|
|
|
|
|
|
## ### Datavase host
|
|
|
|
|
|
|
|
## dbhost="knd-db01.kniod.corp"
|
|
|
|
|
|
|
|
## ### Database name
|
|
|
|
|
|
|
|
## datab="ddbank"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## db_conn="psql -h $dbhost -p $dbport -U $dbuser -d $datab -c"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
source $HOME/git/knd-ddbank/ddbank_config
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Functions
|
|
|
|
# Functions
|
|
|
@ -21,16 +32,16 @@ new_account() {
|
|
|
|
case $yn in
|
|
|
|
case $yn in
|
|
|
|
|
|
|
|
|
|
|
|
# If user selects yes. Insert their username and a timestamp into the database.
|
|
|
|
# If user selects yes. Insert their username and a timestamp into the database.
|
|
|
|
[Yy] ) $db_ssh "psql -U ddbank -d ddbank -c \"insert into account (username,timestamp) values ('$user1','now');\"" &>/dev/null || echo "Username already taken.";
|
|
|
|
[Yy] ) $db_conn "insert into account (username,timestamp) values ('$user1','now');" &>/dev/null || echo "Username already taken.";
|
|
|
|
|
|
|
|
|
|
|
|
# Sets the user's id as the variable.
|
|
|
|
# Sets the user's id as the variable.
|
|
|
|
user1_id=$($db_ssh "psql -U ddbank -d ddbank -c \"select id from account where username='$user1';\"" | awk 'FNR == 3 {print}')
|
|
|
|
user1_id=$($db_conn "select id from account where username='$user1';" | awk 'FNR == 3 {print}')
|
|
|
|
|
|
|
|
|
|
|
|
# Gives the user the starting balance.
|
|
|
|
# Gives the user the starting balance.
|
|
|
|
$db_ssh "psql -U ddbank -d ddbank -c \"insert into balance (id,balance) values ('$user1_id',$bal);\"" &>/dev/null
|
|
|
|
$db_conn "insert into balance (id,balance) values ('$user1_id',$bal);" &>/dev/null
|
|
|
|
|
|
|
|
|
|
|
|
# Pulls the user's account and ID for awk'ing
|
|
|
|
# Pulls the user's account and ID for awk'ing
|
|
|
|
user_account=$($db_ssh "psql -U ddbank -d ddbank -c \"select id,username from account;\"" | grep $user1_id)
|
|
|
|
user_account=$($db_conn "select id,username from account;" | grep $user1_id)
|
|
|
|
echo ""
|
|
|
|
echo ""
|
|
|
|
acc_id=$(echo "$user_account" | awk -F\| '{print $1}')
|
|
|
|
acc_id=$(echo "$user_account" | awk -F\| '{print $1}')
|
|
|
|
acc_name=$(echo "$user_account" | awk -F\| '{print $2}')
|
|
|
|
acc_name=$(echo "$user_account" | awk -F\| '{print $2}')
|
|
|
@ -48,7 +59,7 @@ new_account() {
|
|
|
|
|
|
|
|
|
|
|
|
list_user() {
|
|
|
|
list_user() {
|
|
|
|
echo ""
|
|
|
|
echo ""
|
|
|
|
$db_ssh "psql -U ddbank -d ddbank -c \"select * from account;\""
|
|
|
|
$db_conn "select * from account;"
|
|
|
|
echo ""
|
|
|
|
echo ""
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
@ -68,11 +79,11 @@ list_balance() {
|
|
|
|
|
|
|
|
|
|
|
|
case $ii in
|
|
|
|
case $ii in
|
|
|
|
1 ) clear
|
|
|
|
1 ) clear
|
|
|
|
$db_ssh "psql -U ddbank -d ddbank -c \"select id,username,balance from account join balance using (id);\"" ;;
|
|
|
|
$db_conn "select id,username,balance from account join balance using (id);" ;;
|
|
|
|
2 ) read -p "Enter the user account name: " ex_mat
|
|
|
|
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_conn "select id,username,balance from account join balance using (id) where username='$ex_mat';" ;;
|
|
|
|
3 ) read -p "Enter search term: " bal_user
|
|
|
|
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);\"")
|
|
|
|
bal_user1=$($db_conn "select id,username,balance from account join balance using (id);")
|
|
|
|
echo "$bal_user1" | awk 'FNR == 1 {print}'
|
|
|
|
echo "$bal_user1" | awk 'FNR == 1 {print}'
|
|
|
|
echo "$bal_user1" | grep -iE "$bal_user"
|
|
|
|
echo "$bal_user1" | grep -iE "$bal_user"
|
|
|
|
echo "" ;;
|
|
|
|
echo "" ;;
|
|
|
@ -99,10 +110,10 @@ do
|
|
|
|
1 ) read -p "Withdraw from user: " bank_user
|
|
|
|
1 ) read -p "Withdraw from user: " bank_user
|
|
|
|
read -p "Amount to withdraw: " bank_wth_amt
|
|
|
|
read -p "Amount to withdraw: " bank_wth_amt
|
|
|
|
echo "$bank_user" | awk 'FNR == 1 {print}'
|
|
|
|
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_1_amt=$($db_conn "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}')
|
|
|
|
bank_user_id=$($db_conn "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')
|
|
|
|
new_amt=$($db_conn "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';\"" ;;
|
|
|
|
$db_conn "update balance set balance = '$new_amt' where id = '$bank_user_id';" ;;
|
|
|
|
* ) clear ;;
|
|
|
|
* ) clear ;;
|
|
|
|
esac
|
|
|
|
esac
|
|
|
|
done
|
|
|
|
done
|
|
|
|