1 files changed, 87 insertions, 0 deletions
diff --git a/CPP/BABS/BankSystem.cpp b/CPP/BABS/BankSystem.cpp
new file mode 100644
index 0000000..1493451
--- /dev/null
+++ b/CPP/BABS/BankSystem.cpp
@@ -0,0 +1,87 @@
+#include <iostream>
+#include <vector>
+
+#include "BankSystem.h"
+
+using namespace std;
+
+std::vector<BankAccount> BADatabase;
+
+int getIndexFromID(int id) {
+    for(int i = 0; i < BADatabase.size(); i++) {
+        if (BADatabase[i].getId() == id) {
+            return i;
+        }
+    }
+    return -1;
+}
+
+void listAllAccounts() {
+    for(BankAccount ba: BADatabase) { 
+        cout << ba.toString() << endl; 
+    }
+}
+
+void openBankAccount() {
+    cout << "Initial balance: ";
+    int initBalance;
+    cin >> initBalance;
+    
+    BankAccount newBA(initBalance);
+    BADatabase.push_back(newBA);
+    cout << "Created new bank account with ID: " << newBA.getId() << endl;
+}
+
+void closeBankAccount() {
+    cout << "Bank account ID: ";
+    int id;
+    cin >> id;
+    
+    cout << "Are you really sure you want to close bank account with ID: " << id << "? [y/n]";
+    char answer;
+    cin >> answer;
+    
+    if (answer == 'y' || answer == 'Y') {
+        BADatabase[getIndexFromID(id)].close();
+    }
+    
+    cout << "Bank account with ID: " << id << " is closed!" << endl;
+}
+
+void bankerMode() {
+    while(true) {
+        cout << "What do you want to do: [0] Exit, [1] List bank accounts, [2] Open bank account, [3] Close bank account" << endl;
+        
+        int value;
+        cin >> value;
+        
+        switch(value) {
+            case 0: return;
+            case 1: listAllAccounts(); break;
+            case 2: openBankAccount(); break;
+            case 3: closeBankAccount(); break;
+        }
+    }
+}
+
+void atmMode() {
+    cout << "Bank accont ID: ";
+    int userId;
+    cin >> userId;
+    cout << endl;
+    
+    BankAccount ba = BADatabase[userId];
+    while(true) {
+        cout << "What do you want to do: [0] Exit, [1] View balance, [2] Deposit money, [3] Withdraw money?" << endl;
+        
+        int value;
+        cin >> value;
+        
+        switch(value) {
+            case 0: return;
+            case 1: cout << ba.getBalance() << endl; break;
+            case 2: cout << "Amount to deposit: "; cin >> value; ba.deposit(value); break;
+            case 3: cout << "Amount to withdraw: "; cin >> value; ba.deposit(value); break;
+        }
+    }
+}
+\ No newline at end of file