menu.py

from flask import Blueprint, render_template, request, session, jsonify  # Added jsonify import
from salesforce import get_salesforce_connection

menu_blueprint = Blueprint('menu', __name__)

# Initialize Salesforce connection
sf = get_salesforce_connection()

@menu_blueprint.route("/menu", methods=["GET", "POST"])
def menu():
    selected_category = request.args.get("category", "All")
    user_email = session.get('user_email')

    if not user_email:
        user_email = request.args.get("email")
        user_name = request.args.get("name")

        if user_email:
            session['user_email'] = user_email
            session['user_name'] = user_name  # Store name in session
        else:
            return redirect(url_for("login"))
    else:
        user_name = session.get('user_name')  # Get name from session if it's already stored

    # Get the first letter of the user's name (make it uppercase for consistency)
    first_letter = user_name[0].upper() if user_name else "A"

    try:
        # Fetch user referral and reward points
        user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
        user_result = sf.query(user_query)

        if not user_result['records']:
            return redirect(url_for('login'))

        referral_code = user_result['records'][0].get('Referral__c', 'N/A')
        reward_points = user_result['records'][0].get('Reward_Points__c', 0)

        # Query to fetch Menu_Item__c records including Total_Ordered__c for best sellers
        menu_query = """
            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c 
            FROM Menu_Item__c
        """
        result = sf.query(menu_query)
        food_items = result['records'] if 'records' in result else []

        # Ensure Total_Ordered__c has a valid value
        for item in food_items:
            if 'Total_Ordered__c' not in item or item['Total_Ordered__c'] is None:
                item['Total_Ordered__c'] = 0  # Default value

        # Query to fetch Custom_Dish__c records created within the last 7 days with Total_Ordered__c > 10
        custom_dish_query = """
            SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c 
            FROM Custom_Dish__c
            WHERE CreatedDate >= LAST_N_DAYS:7
        """
        custom_dish_result = sf.query(custom_dish_query)
        custom_dishes = custom_dish_result['records'] if 'records' in custom_dish_result else []

        # Merge both Menu_Item__c and Custom_Dish__c records into the ordered menu
        all_items = food_items + custom_dishes

        # Define the order of sections, adding "Best Sellers" at the top
        section_order = ["Best Sellers", "Starters", "Biryanis", "Curries", "Breads", "Customized dish", "Apetizer", "Desserts", "Soft Drinks"]
        ordered_menu = {section: [] for section in section_order}

        # Sort items by Total_Ordered__c in descending order and pick top 4 as best sellers
        best_sellers = sorted(all_items, key=lambda x: x.get("Total_Ordered__c", 0), reverse=True)

        if selected_category == "Veg":
            best_sellers = [item for item in best_sellers if item.get("Veg_NonVeg__c") in ["Veg", "both"]]
        elif selected_category == "Non veg":
            best_sellers = [item for item in best_sellers if item.get("Veg_NonVeg__c") in ["Non veg", "both"]]

        # Take only the top 4 best sellers after filtering
        best_sellers = best_sellers[:4]

        # Ensure "Best Sellers" is added only if there are items after filtering
        if best_sellers:
            ordered_menu["Best Sellers"] = best_sellers

        # Create a set to track item names already added to prevent duplicates
        added_item_names = set()

        # Filter and organize menu items based on category and section (to avoid duplicates)
        for item in all_items:
            section = item.get("Section__c", "Others")  # Default to "Others" if missing
            if section not in ordered_menu:
                ordered_menu[section] = []

            # Skip item if it's already been added to avoid duplicates
            if item['Name'] in added_item_names:
                continue

            # Apply category filters
            if selected_category == "Veg" and item.get("Veg_NonVeg__c") not in ["Veg", "both"]:
                continue
            if selected_category == "Non veg" and item.get("Veg_NonVeg__c") not in ["Non veg", "both"]:
                continue

            ordered_menu[section].append(item)
            added_item_names.add(item['Name'])  # Add item to the set of added items
            print(f"Added item to {section}: {item['Name']}")  # Debugging

        # Remove empty sections
        ordered_menu = {section: items for section, items in ordered_menu.items() if items}
        print(f"Final ordered menu: {ordered_menu.keys()}")  # Debugging

        categories = ["All", "Veg", "Non veg"]

    except Exception as e:
        print(f"Error fetching menu data: {str(e)}")
        ordered_menu = {}
        categories = ["All", "Veg", "Non veg"]
        referral_code = 'N/A'
        reward_points = 0

    # Pass the user's first letter (first_letter) to the template
    return render_template(
        "menu.html",
        ordered_menu=ordered_menu,
        categories=categories,
        selected_category=selected_category,
        referral_code=referral_code,
        reward_points=reward_points,
        user_name=user_name,  # Pass name to the template
        first_letter=first_letter  # Pass first letter to the template
    )
@menu_blueprint.route('/api/addons', methods=['GET'])
def get_addons():
    item_name = request.args.get('item_name')  
    item_section = request.args.get('item_section')  

    # Check if both item_name and item_section are provided
    if not item_name or not item_section:
        return jsonify({"success": False, "error": "Item name and section are required."}), 400

    try:
        # Fetch customization options from Salesforce based on the section
        query = f"""
            SELECT Name, Customization_Type__c, Options__c, Max_Selections__c, Extra_Charge__c, Extra_Charge_Amount__c
            FROM Customization_Options__c
            WHERE Section__c = '{item_section}'
        """
        result = sf.query(query)
        addons = result.get('records', [])

        # Check if we found any addons
        if not addons:
            return jsonify({"success": False, "error": "No customization options found for the given section."}), 404

        # Format data for frontend
        formatted_addons = []
        for addon in addons:
            # Ensure 'Options__c' exists and is not None
            options = addon.get("Options__c", "")
            if options:  # If options are available, split them
                options = options.split(", ")  # Convert comma-separated options into a list
            else:
                options = []  # If no options, default to an empty list

            formatted_addons.append({
                "name": addon["Name"],
                "type": addon["Customization_Type__c"],
                "options": options,
                "max_selections": addon.get("Max_Selections__c", 1),
                "extra_charge": addon.get("Extra_Charge__c", False),
                "extra_charge_amount": addon.get("Extra_Charge_Amount__c", 0)
            })

        return jsonify({"success": True, "addons": formatted_addons})

    except Exception as e:
        # Log the exception for debugging
        app.logger.error(f"Error fetching addons: {str(e)}")
        return jsonify({"success": False, "error": "An error occurred while fetching customization options."}), 500

@menu_blueprint.route('/cart/add', methods=['POST'])
def add_to_cart():
    try:
        # Get data from request
        data = request.json
        item_name = data.get('itemName', '').strip()
        item_price = data.get('itemPrice')
        item_image = data.get('itemImage')
        addons = data.get('addons', [])
        instructions = data.get('instructions', '')
        category = data.get('category')
        section = data.get('section')
        quantity = data.get('quantity', 1)  # Get the quantity field from the request
        customer_email = session.get('user_email')

        # Basic validation for required fields
        if not item_name or not item_price:
            return jsonify({"success": False, "error": "Item name and price are required."}), 400

        if not customer_email:
            return jsonify({"success": False, "error": "User email is required."}), 400

        # Query to check if the item is already in the cart
        query = f"""
            SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c 
            FROM Cart_Item__c
            WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
        """
        result = sf.query(query)
        cart_items = result.get("records", [])

        # Calculate the total price for the addons
        addons_price = sum(addon['price'] for addon in addons)
        new_addons = "; ".join([f"{addon['name']} (${addon['price']})" for addon in addons])

        # If the item is already in the cart, update it
        if cart_items:
            cart_item_id = cart_items[0]['Id']
            existing_quantity = cart_items[0]['Quantity__c']
            existing_addons = cart_items[0].get('Add_Ons__c', "None")
            existing_addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
            existing_instructions = cart_items[0].get('Instructions__c', "")

            # Combine the new addons with the existing ones
            combined_addons = existing_addons if existing_addons != "None" else ""
            if new_addons:
                combined_addons = f"{combined_addons}; {new_addons}".strip("; ")

            # Combine existing instructions with new instructions
            combined_instructions = existing_instructions
            if instructions:
                combined_instructions = f"{combined_instructions} | {instructions}".strip(" | ")

            # Calculate total addons price
            combined_addons_list = combined_addons.split("; ")
            combined_addons_price = sum(
                float(addon.split("($")[1][:-1]) for addon in combined_addons_list if "($" in addon
            )

            # Update the cart item in Salesforce (updating quantity)
            sf.Cart_Item__c.update(cart_item_id, {
                "Quantity__c": existing_quantity + quantity,  # Add the selected quantity
                "Add_Ons__c": combined_addons,
                "Add_Ons_Price__c": combined_addons_price,
                "Instructions__c": combined_instructions,
                "Price__c": (existing_quantity + quantity) * item_price + combined_addons_price,
                "Category__c": category,
                "Section__c": section
            })
        else:
            # If the item is not already in the cart, create a new entry
            addons_string = "None"
            if addons:
                addons_string = new_addons

            total_price = item_price * quantity + addons_price  # Multiply by the quantity

            # Create new cart item in Salesforce
            sf.Cart_Item__c.create({
                "Name": item_name,
                "Price__c": total_price,
                "Base_Price__c": item_price,
                "Quantity__c": quantity,  # Use the selected quantity
                "Add_Ons_Price__c": addons_price,
                "Add_Ons__c": addons_string,
                "Image1__c": item_image,
                "Customer_Email__c": customer_email,
                "Instructions__c": instructions,
                "Category__c": category,
                "Section__c": section
            })

        return jsonify({"success": True, "message": "Item added to cart successfully."})

    except KeyError as e:
        # Handle missing expected keys in request data
        return jsonify({"success": False, "error": f"Missing required field: {str(e)}"}), 400

    except Exception as e:
        # Log the error for debugging and return a general error message
        print(f"Error adding item to cart: {str(e)}")
        return jsonify({"success": False, "error": "An error occurred while adding the item to the cart."}), 500