main.py

from manager import PasswordManager
import pyperclip
import asyncio
import os

async def setTimeOut(prompt, timeout):
    try:
        print(prompt, end="", flush=True)
        userInp = await asyncio.wait_for(asyncio.to_thread(input), timeout)
        return userInp
    except asyncio.TimeoutError:
        print("\nTimeout. Goodbye!")
        os._exit(0)

def validate_key_loaded(pm : PasswordManager):
    if not pm.keyloaded:
        print("Key not loaded. Please load a key first.")
        return False
    return True

async def main():
    password = {
        "gmail": "password1",
        "facebook": "password2",
        "twitter": "password3"
    }
    
    pm = PasswordManager()

    print("""What would you like to do?
          1. Create a new key
          2. Load an existing key
          3. Create a new password file
          4. Load an existing password file
          5. Add a password
          6. Get a password
          7. List all sites
          q. Quit
          """)
    
    done = False
    while not done:
        choice = await setTimeOut("Enter choice: ", 30)
        choice = choice.strip().lower()

        if choice == '1':
            path = await setTimeOut("Enter key file path: ", 30)
            path = path.strip()
            pm.create_key(path)

        elif choice == '2':
            path = await setTimeOut("Enter key file path: ", 30)
            path = path.strip()
            pm.load_key(path)

        elif choice == '3' and validate_key_loaded(pm):
            path = await setTimeOut("Enter password file path: ", 30)
            path = path.strip()
            pm.create_password_file(path, password)

        elif choice == '4' and validate_key_loaded(pm):
            path = await setTimeOut("Enter password file path: ", 30)
            path = path.strip()
            pm.load_password_file(path)

        elif choice == '5' and validate_key_loaded(pm):
            site = await setTimeOut("Enter site: ", 30)
            site = site.strip()
            password = await setTimeOut("Enter password: ", 30)
            password = password.strip()
            if pm.validate_strength(password):
                print("added successfully")
            else:
                print("WARNING: This password is weak, It is recommended to set a stronger password")
                print("- Password should be more than 8 characters long")
                print("- Password should have alphanumeric characters, capital letters and special characters")
            pm.add_password(site, password)

        elif choice == '6' and validate_key_loaded(pm):
            site = await setTimeOut("Enter site: ", 30)
            site = site.strip()
            res = pm.get_password(site)
            print(f"Password for {site}: {res}")
            if res != "Password not found.":
                pyperclip.copy(pm.get_password(site))
                print("Password copied to clipboard.")

        elif choice == '7':
            print("Saved Sites:")
            for site in pm.password_dict:
                print(site)

        elif choice == 'q':
            done = True
            print("Goodbye!")

        else:
            print("Invalid choice. Please try again.")

if __name__ == '__main__':
    asyncio.run(main())