Merge pull request #47 from fkamau1/withdraw-function

Add withdraw function
fkamau1 · Jan 21, 2024 · a5fb8ed · a5fb8ed
2 parents fb5325e + a697373
commit a5fb8ed
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 33 deletions.
diff --git a/.idea/modules.xml b/.idea/modules.xml
diff --git a/.idea/modules/bankingApp.iml b/.idea/modules/bankingApp.iml
diff --git a/account.db b/account.db
diff --git a/app/account.db b/app/account.db
diff --git a/app/src/main/java/bankingapp/LogIn.java b/app/src/main/java/bankingapp/LogIn.java
@@ -36,9 +36,10 @@ public void checkAccount(DBTransactions db, Connection con, String accountNumber
                 do {
                     System.out.println("\n1. Balance\n" +
                             "2. Add income\n" +
-                            "3. Do transfer\n" +
-                            "4. Close Account\n" +
-                            "5. Log Out\n" +
+                            "3. Withdraw\n" +
+                            "4. Do transfer\n" +
+                            "5. Close Account\n" +
+                            "6. Log Out\n" +
                             "0. Exit");
 
                     answer = input.nextInt();
@@ -57,10 +58,18 @@ public void checkAccount(DBTransactions db, Connection con, String accountNumber
                             break;
 
                         case 3:
-                            App.transferFunds(db, con, accountNumber);
+                            System.out.println("Enter amount to withdraw:");
+                            int withdraw = input.nextInt();
+                            db.deductBalance(withdraw, accountNumber);
+                            System.out.println("Amount withdrawn: " + withdraw);
+                            System.out.println("New balance: " + db.checkBalance(accountNumber));
                             break;
 
                         case 4:
+                            App.transferFunds(db, con, accountNumber);
+                            break;
+
+                        case 5:
                             if (db.checkBalance(accountNumber) > 0) {
                                 System.out.println("Transfer account balance to a different account before closing");
                             } else {
@@ -70,12 +79,14 @@ public void checkAccount(DBTransactions db, Connection con, String accountNumber
                             }
                             break;
 
-                        case 5:
+                        case 6:
                             System.out.println("\nYou have successfully logged out");
                             break subMenu;
 
                         case 0:
+                            System.out.println("Bye");
                             System.exit(0);
+
                     }
                 } while (answer != 0);