from odoo import http
from odoo.http import request
from odoo.addons.web.controllers.home import Home

class CustomHome(Home):
    @http.route('/web/login', type='http', auth="public", website=True)
    def web_login(self, *args, **kw):
        response = super(CustomHome, self).web_login(*args, **kw)
        if request.params.get('login_success') and request.session.uid:
            return request.redirect('/')
        return response

from odoo.addons.website.controllers.main import Website

class ImageHome(Website):
    @http.route('/', type='http', auth='public', website=True, sitemap=True)
    def index(self, **kwargs):
        if not request.session.uid:
            return request.render('website.homepage')
            
        # Remove sudo() to respect Odoo's standard menu group restrictions
        menus = request.env['ir.ui.menu'].search([
            ('parent_id', '=', False)
        ], order='sequence')
        
        # User role checks
        is_admin = request.env.user.has_group('base.group_system') or request.env.user.has_group('dine360_restaurant.group_restaurant_admin')
        is_kitchen = request.env.user.has_group('dine360_restaurant.group_restaurant_kitchen')

        filtered_menus = []
        seen_names = set()
        for menu in menus:
            # 1. Hide "Apps" for non-admins (usually name is 'Apps' or xmlid has base.menu_management)
            if (menu.name == 'Apps' or (menu.web_icon and menu.web_icon.startswith('base,'))) and not is_admin:
                continue
                
            # 2. Hide "Kitchen (KDS)" for Waiters (only show for Kitchen Staff or Admin)
            if 'Kitchen' in menu.name or 'KDS' in menu.name:
                if not (is_kitchen or is_admin):
                    continue
            
            # 3. De-duplicate by name to prevent double icons (like Kitchen showing twice)
            if menu.name in seen_names:
                continue
            seen_names.add(menu.name)
            
            filtered_menus.append(menu)
        
        return request.render('dine360_dashboard.image_home_template', {
            'menus': filtered_menus,
            'user_id': request.env.user
        })


    @http.route('/home', type='http', auth="public", website=True, sitemap=True)
    def website_home(self, **kw):
        # Explicit route for standard Website Homepage
        return request.render('website.homepage')