import bcrypt
from pymongo import MongoClient

# --- CONFIGURACIÓN ---
# PEGA AQUÍ tu cadena de conexión que copiaste de Atlas
# IMPORTANTE: Reemplaza <password> por 04092013 si no aparece ya puesto
URI = "mongodb+srv://kim2007:04092013@proyecto.feguffv.mongodb.net/?retryWrites=true&w=majority"

client = MongoClient(URI)
db = client['ayuda']
usuarios = db['usuarios']

# --- REGISTRO DE USUARIO ---
def registrar_usuario():
    print("\n--- Registro de Usuario ---")
    user = input("Usuario: ")
    password = input("Contraseña: ").encode('utf-8')

    # Ciframos la contraseña con bcrypt
    hashed = bcrypt.hashpw(password, bcrypt.gensalt())
    usuarios.insert_one({"user": user, "pass": hashed})
    print(" Usuario registrado con éxito.")

# --- INICIO DE SESIÓN ---
def iniciar_sesion():
    print("\n--- Iniciar Sesión ---")
    user = input("Usuario: ")
    password = input("Contraseña: ").encode('utf-8')

    datos = usuarios.find_one({"user": user})
    if datos and bcrypt.checkpw(password, datos['pass']):
        print("¡Bienvenido! Sesión iniciada.")
    else:
        print(" Usuario o contraseña incorrectos.")

# --- MENÚ PRINCIPAL ---
def menu():
    while True:
        print("\n=== PROYECTO DE AUTENTICACIÓN ===")
        print("1. Registrar Usuario\n2. Iniciar Sesión\n3. Salir")
        opcion = input("Elige una opción: ")
        if opcion == "1":
            registrar_usuario()
        elif opcion == "2":
            iniciar_sesion()
        elif opcion == "3":
            break

if __name__ == "__main__":
    menu()