# Lucas Mathews - Fontys Student ID: 5023572
# Banking System CLI Utility

import json
import argparse
import sys
from config import CONFIG
from getpass import getpass
from connection import login, logout, get_client, add_client, promote, demote, initialise
from test_database_generator import generate_test_database 


def show_menu():
    print("\nAvailable options:")
    print("1. Logout and exit")
    print("2. New client")
    print("3. Add test data to database (3 clients, 2 accounts each, 3 transactions each)")
    print("4. Promote to Admin")
    print("5. Demote from Admin")
    #print("6. Delete user")
    print("\n")

def main():
    parser = argparse.ArgumentParser(description='Banking System CLI Utility')
    parser.add_argument('-u', '--username', type=str, default=CONFIG["client"]["default_id"], help='Username for login')
    parser.add_argument('-p', '--password', type=str, default=CONFIG["client"]["default_password"], help='Password for login')
    parser.add_argument('-e', '--email', type=str, help='Email for initialisation')
    parser.add_argument('-init', '--initialise', action='store_true', help='Initialise the system')

    subparsers = parser.add_subparsers(dest='command')

    login_parser = subparsers.add_parser('login', help='Login to the system')
    logout_parser = subparsers.add_parser('logout', help='Logout from the system')

    args = parser.parse_args()

    if args.initialise:
        if not args.email or not args.password:
            print("Email and password are required for initialisation.")
            sys.exit(1)
        response = initialise(args.password, args.email)
        if response['success']:
            print(f"Initialisation successful: {response['message']}")
        else:
            print(f"Initialisation failed: {response['message']}")
        return
    
    if not args.command:
        while True:
            if not args.username:
                args.username = input("Enter username: ")
            if not args.password:
                args.password = getpass("Enter password: ")

            response = login(args.username, args.password)
            if response['success']: 
                print(f"Login successful: {response['message']}")  
                try:
                    with open('session_data.json', 'r') as f:
                        session_data = json.load(f)
                    client_id = session_data['client_id']
                    client_info = get_client(client_id)
                    if client_info.get('success'):
                        client = client_info['data']
                        name = client['name']
                    print(f"Welcome, {name}!")
                except Exception as e:
                    print(f"Error loading client data: {str(e)}")

                while True:
                    show_menu()
                    option = input("Choose an option: ")
                    if option == "1": # Menu option 1 - Logout and exit
                        response = logout()
                        json_response = response.json()
                        if json_response['success']:  
                            print(f"Logout successful: {json_response['message']}")  
                        else:
                            print(f"Logout failed: {json_response['message']}")
                        args.username = None
                        args.password = None

                    elif option == "2": # Menu option 2 - New client
                        print("New user option selected.")
                        name = input("Enter name: ")
                        birthdate = input("Enter birthdate (YYYY-MM-DD): ")
                        address = input("Enter address: ")
                        phone_number = input("Enter phone number: ")
                        email = input("Enter email: ")
                        password = input("Enter password: ")
                        notes = input("Enter notes: ")
                        response = add_client(name, birthdate, address, phone_number, email, password, notes)
                        if response['success']:
                            print(f"Client added successfully: {response['message']}")
                        else:
                            print(f"Client addition failed: {response['message']}")

                    elif option == "3": # Menu option 3 - Add test data to database
                        print("Add test users option selected.")
                        generate_test_database(args.username, args.password)
                        
                    elif option == "4": # Menu option 4 - Promote to Admin
                        print("Enter the client ID to promote to Admin:")
                        client_id = input("Enter client ID: ")
                        response = promote(client_id)
                        if response['success']:
                            print(f"Promotion successful: {response['message']}")
                        else:
                            print(f"Promotion failed: {response['message']}")

                    elif option == "5": # Menu option 5 - Demote from Admin
                        print("Enter the client ID to demote from Admin:")
                        client_id = input("Enter client ID: ")
                        response = demote(client_id)
                        if response['success']:
                            print(f"Demotion successful: {response['message']}")
                        else:
                            print(f"Demotion failed: {response['message']}") 

                    elif option == "6": # Menu option 6 - Delete user
                        print("Not implemented yet, exiting...")
                        break
                    else:
                        print("Invalid option. Please try again.")
                break
            else:
                print(f"Login failed: {response['message']}. Please try again.")
                args.username = None
                args.password = None
    elif args.command == 'login':
        if not args.username or not args.password:
            print("Username and password are required for login.")
            sys.exit(1)
        response = login(args.username, args.password)
        if response['success']:
            print(f"Login successful: {response['message']}")
        else:
            print(f"Login failed: {response['message']}. Please try again.")
    elif args.command == 'logout':
        response = logout()
        if response['success']:
            print(f"Logout successful: {response['message']}")
        else:
            print(f"Logout failed: {response['message']}")
    else:
        print("Invalid command. Use 'login' or 'logout'.")

if __name__ == "__main__":
    main()