BankAccount

package BankAccount;

/**
 * Created by Bogdan on 09.02.2017.
 */
public class ATM
{
    public static void main(String[] args) {
        Menu menu = new Menu();
        menu.firstMenu();
    }
}

package BankAccount;

import java.util.Scanner;

/**
 * Created by Bogdan on 09.02.2017.
 */
public class Menu {
    private DataBase db = new DataBase();
    private boolean isTrue = false;
    private Scanner sc = new Scanner(System.in);
    private double balance = 0;
    private int id = 0;

    private void join(){ //Авторизация пользователя
        while(!isTrue){
            System.out.print("Email: ");
            String email = sc.nextLine();

            System.out.print("Пароль: ");
            String pass = sc.nextLine();

            if (!db.hasAccount(email, pass)){
                System.out.println("\nНе правильный логин или пароль.\n");
            } else {
                isTrue = true;
                userID(db.getUserID(email));
            }
        }
        user_menu();
    }

    private void userID(int id){
        this.id = id;
    }

    private void sign_up(){ // Регистрация пользователя
            System.out.print("Введите имя: ");
        String name = sc.nextLine();
            System.out.print("Введите фамилию: ");
        String lastname = sc.nextLine();
            System.out.print("Email: ");
        String email = sc.nextLine();
            System.out.print("Пароль: ");
        String pass = sc.nextLine();
            System.out.print("Повторите пароль: ");
        String pass2 = sc.nextLine();

        while(true) {
            if (pass.equals(pass2)) {
                if(!db.hasAccount(email)){
                    System.out.println("\nРегистрация окончена! Идет переход к авторизации...");
                    db.addAccount(name, lastname, email, pass);
                    join();
                    break;
                } else {
                    System.out.println("\nАккаунт с таким Email уже существует. Введите другой Email\n");
                    System.out.print("Email: ");
                    email = sc.nextLine();
                    System.out.print("Пароль: ");
                    pass = sc.nextLine();
                    System.out.print("Повторите пароль: ");
                    pass2 = sc.nextLine();
                }
            } else {
                System.out.println("\nПароли не совпадают. Повторите попытку\n");
                System.out.print("Пароль: ");
                pass = sc.nextLine();
                System.out.print("Повторите пароль: ");
                pass2 = sc.nextLine();
            }
        }
    }

    private void exit(){ // Выход из программы
        System.out.println("Закрытие программы...");
        System.exit(0);
    }

    private void back(){
        String a;
        System.out.print("Нажмите 2 чтобы вернутся в меню:");
        a = sc.next();
        while(!a.equals("2")) {
            System.out.print("\nВведите другой символ: ");
            a = sc.next();
        }
        user_menu();
    }

    private void withdraw(){ //Вывод денег
        balance = db.getBalance(id);
        System.out.print("Введите сумму которую вы хотите вывести($): ");
        double sum = sc.nextDouble();
        while(true) {
            if (balance > sum) {
                balance -= sum;
                db.setBalance(balance, id);
                break;
            } else {
                System.out.println("На вашем балансе нет столько денег");
                sum = sc.nextDouble();
            }
        }
        System.out.println("\nНа вашем балансе: " + db.getBalance(id)+"$");
    }

    private double balance() { // Пополнение баланса
        System.out.print("Введите сумму которую хотите внести($):");
        int num = sc.nextInt();
        balance += num;
        System.out.printf("\nВаш баланс пополнен на %d$\n",num);
        db.setBalance(balance,id);
        return balance;
    }

    private double getBalance(){ //Баланс пользователя
        balance = db.getBalance(id);
        return balance;
    }

    public void firstMenu(){ // Меню авторизации и регистрации
        System.out.println("|-------------------|");
        System.out.println("|  1) Вход          |");
        System.out.println("|  2) Регистрация   |");
        System.out.println("|-------------------|");

        while(true){
            System.out.println();
            System.out.print("Введите номер пункта (1-2) который вы хотите выбрать: ");
            String num = sc.nextLine();

            if (num.equals("1")){
                System.out.println("\nПереход к авторизации...\n");
                join();
                break;
            } else if (num.equals("2")){
                System.out.println("\nПереход к регистрации...\n");
                sign_up();
                break;
            } else {
                System.out.println("\n*** Ошибка, введите другой символ! ***");
            }
        }
    }

    private void user_menu(){ //Меню пользователя

        System.out.println("\n    1) Баланс");
        System.out.println("    2) Пополнить");
        System.out.println("    3) Вывести");
        System.out.println("    4) Выход");
        int num;
        while(true){
            System.out.print("\nВведите номер пункта (1-4) который вы хотите выбрать: ");
            num = sc.nextInt();
            switch (num) {
                case 1:
                    System.out.println("На вашем балансе: " + getBalance() + "$");
                    back();
                    break;
                case 2:
                    balance();
                    back();
                    break;
                case 3:
                    withdraw();
                    back();
                    break;
                case 4:
                    exit();
                default:
                    System.out.println("\n*** Ошибка, введите другой символ! ***");
            }
        }
    }
}

package BankAccount;

import java.sql.*;
/**
 * Created by Bogdan on 10.02.2017.
 */
public class DataBase {
    private final String url = "jdbc:mysql://localhost:3306/accountdb";
    private Connection connection = null;
    private ResultSet rs = null;
    private PreparedStatement ps = null;

    //Подключение к БД
    private void connect(){

        try {
            connection = DriverManager.getConnection(url,"root","root");
        }
        catch (Exception e){
            System.err.println(e.getMessage());
        }
    }
    /*
    Запись аккаунта в в БД
     */
    public void addAccount(String name, String lastname, String email, String password) {
        try {
            String query = " insert into account (name, lastname, email, password, balance)" + " values (?, ?, ?, ?, ?)";
            ps = connection.prepareStatement(query);
            ps.setString(1, name);
            ps.setString(2, lastname);
            ps.setString(3, email);
            ps.setString(4, password);
            ps.setInt(5,0);

            ps.execute();
            connection.close();
        }
        catch (Exception e){
            System.err.println(e.getMessage());
        }

    }

    //Получение ID пользователя
    //Нужен для того чтобы определить какого пользователя баланс нужно получить или обновить.
    //Принцип работы:
    //При авторизации, если она успешна, получаю ID пользователя с помощью его Email
    public int getUserID(String email){
        String query = "select id from account where email = ?";
        int id = 0;
        try {
            ps = connection.prepareStatement(query);
            ps.setString(1,email);
            rs = ps.executeQuery();
            while(rs.next()){
                id = rs.getInt("id");
                return id;
            }
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return id;
    }

    //Проверка существует ли аккаунт с таким Email
    public boolean hasAccount(String email){
        String query = "select email from account where email=?";
        try {
            connect();
            ps = connection.prepareStatement(query);
            ps.setString(1,email);
            rs = ps.executeQuery();
            while(rs.next()) {
                if (rs.getString(1).equals(email)) {
                    return false;
                } else {
                    return true;
                }
            }
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
        return false;
    }

    //Обновление баланса пользователя в БД
    public void setBalance(double balance, int userid){
        String query = "update account set balance = ? where id=? ";

        try {
            connect();
            ps = connection.prepareStatement(query);
            ps.setDouble(1, balance);
            ps.setInt(2, userid);
            ps.executeUpdate();
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
    // получение баланса пользователя из БД
    public double getBalance(int userid) {
        String query = "select balance from account where id =?";
        double balance = 0;
        try {
            connect();
            ps = connection.prepareStatement(query);
            ps.setInt(1,userid);
            rs = ps.executeQuery();
            while(rs.next()){
                balance = rs.getDouble("balance");
                return balance;
            }
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return balance;
    }

    /*
    Проверка существует ли аккаунт в БД с таким Email и паролем.
    Если такой аккаунт существует то идет авторизация.
     */
    public boolean hasAccount(String email, String password){
        String query = "select email,password from account where email=?";
        try {
            connect();
            ps = connection.prepareStatement(query);
            ps.setString(1,email);
            rs = ps.executeQuery();
                while(rs.next()) {
                    if (rs.getString(1).equals(email) && rs.getString(2).equals(password)) {
                        System.out.println("\nАвторизация прошла успешно...\n");
                        return true;
                    } else {
                        return false;
                    }
                }
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
            return false;
        }
        return false;
    }
}