dbconnect.py

from pymongo import MongoClient
from datetime import datetime

# Connect to MongoDB
client = MongoClient("mongodb://localhost:27017/")
db = client["feasto"]  # Ensure database name is correct

# Use a fixed document for storing user details
USERS_DOC_ID = "master_user_log"

def get_restaurants():
    """Fetch restaurant names and images from MongoDB"""
    restaurants = db.restaurants.find({})

    result = {}
    for r in restaurants:
        image_path = r.get("image", "images/default.jpg")  # Use default if image is missing
        result[r["name"]] = {"image": image_path}

    return result

def get_menu_items(restaurant_name):
    """Fetch menu items dynamically from MongoDB with image paths"""
    menu_items = db.menu.find({"restaurant_name": restaurant_name})

    result = []
    for m in menu_items:
        image_path = m.get("image_path", "images/default.jpg")  # Use default if missing
        result.append((
            m.get("order_id", 0),
            m.get("dish_name", "Unnamed"),
            m.get("price", 0),
            m.get("description", "No description"),
            image_path
        ))

    return result

def insert_user_entry(name, mobile):
    """Insert a new user entry into the database"""
    new_entry = {
        "name": name,
        "mobile": mobile,
        "feasto_points": 100,  # Default points for new users
        "timestamp": datetime.utcnow()
    }

    # Insert into MongoDB collection
    db.users.insert_one(new_entry)

def get_all_user_entries():
    """Return all stored user entries"""
    return list(db.users.find({}, {"_id": 0}))  # Exclude MongoDB's default `_id` field

def add_to_cart_db(order_id, restaurant_name, dish_name, price, quantity=1):
    """Insert a menu item into the cart collection"""
    cart_item = {
        "order_id": order_id,
        "restaurant_name": restaurant_name,
        "dish_name": dish_name,
        "price": price,
        "quantity": quantity,
        "timestamp": datetime.utcnow()
    }

    db.cart.insert_one(cart_item)

def get_cart_items(restaurant_name):
    """Fetch items & quantities from the cart collection"""
    cart_items = db.cart.find({"restaurant_name": restaurant_name})
    return [(item["dish_name"], item["quantity"]) for item in cart_items]

def update_cart_quantity(restaurant_name, dish_name, change):
    """Increase or decrease item quantity in the cart."""
    cart_item = db.cart.find_one({"restaurant_name": restaurant_name, "dish_name": dish_name})

    if cart_item:
        new_quantity = max(0, cart_item["quantity"] + change)  # Ensure quantity is not negative

        if new_quantity > 0:
            db.cart.update_one(
                {"restaurant_name": restaurant_name, "dish_name": dish_name},
                {"$set": {"quantity": new_quantity}}
            )
        else:
            db.cart.delete_one({"restaurant_name": restaurant_name, "dish_name": dish_name})  # Remove if 0
        
        return new_quantity
    return 0  # Return 0 if item does not exist

# ✅ NEW FUNCTION: Clear cart collection when resto.py starts
def clear_cart():
    """Deletes all documents from the cart collection."""
    result = db.cart.delete_many({})
    print(f"🗑 Cleared {result.deleted_count} items from the cart.")

def get_latest_user():
    """Fetch the most recent user entry from the database"""
    user = db.users.find_one(
        {},
        sort=[("timestamp", -1)]  # Sort by timestamp descending
    )
    return user if user else {"name": "N/A", "table_number": "N/A"}

def verify_user(name, mobile):
    """Verify if user exists in the database using name and mobile only"""
    user = db.users.find_one({
        "name": name,
        "mobile": mobile
    })
    return user is not None

def get_user_details_by_username(username):
    """Fetch user details from the database based on username"""
    user = db.users.find_one({"name": username})
    if user:
        return {
            "name": user.get("name", "N/A"),
            "mobile": user.get("mobile", "N/A"),
            "table_number": user.get("table_number", "N/A"),
            "feasto_points": user.get("feasto_points", 0)
        }
    return {"name": "N/A", "mobile": "N/A", "table_number": "N/A", "feasto_points": 0}  

def get_user_by_credentials(name, mobile):
    """Fetch specific user's details from the database using login credentials"""
    user = db.users.find_one({
        "name": name,
        "mobile": mobile
    })
    if user:
        return {
            "name": user.get("name", "N/A"),
            "mobile": user.get("mobile", "N/A"),
            "table_number": user.get("table_number", "N/A"),
            "feasto_points": user.get("feasto_points", 0)
        }
    return {"name": "N/A", "mobile": "N/A", "table_number": "N/A", "feasto_points": 0}

# ✅ NEW FUNCTION: Deduct Feasto points from the user
def deduct_feasto_points(username, points):
    """Deduct specific number of Feasto points for a given user"""
    db.users.update_one(
        {"name": username},
        {"$inc": {"feasto_points": -points}}
    )

def add_feasto_points(username, points):
    """Add specific number of Feasto points to a given user"""
    db.users.update_one(
        {"name": username},  # Use "name" to identify the user
        {"$inc": {"feasto_points": points}}  # Increment Feasto points
    )