from flask_pymongo import PyMongo

mongo = PyMongo()

def get_users_collection(user_type):
    if user_type == 'seller':
        return mongo.db.Sellers
    return mongo.db.Users

def add_user(user_data, user_type):
    user_data['user_type'] = user_type
    print(f"📝 Saving user to {user_type} collection: {user_data}")
    result = get_users_collection(user_type).insert_one(user_data)
    print("✅ Inserted document ID:", result.inserted_id)



def get_user_by_email_or_username(identifier, user_type):
    collection = get_users_collection(user_type)
    return collection.find_one({
        "$or": [
            {"email": identifier},
            {"username": identifier}
        ]
    })