from .. import controller
from odoo import http
from odoo.http import request


class Wishlist(controller.Controller):
    prefix = '/api/v1/'
    PREFIX_USER = prefix + 'user/<user_id>/'
    
    @http.route(prefix + 'wishlist', auth='public', methods=['GET'])
    @controller.Controller.must_authorized()
    def get_wishlist_by_user_id(self, **kw):
        user_id = int(kw.get('user_id', 0))
        product_id = kw.get('product_id', 0)
        variant_id = kw.get('variant_id', 0)
        limit = int(kw.get('limit', 0))
        offset = int(kw.get('offset', 0))
        query = [('user_id', '=', user_id)]
        if product_id:
            query += [('product_id', '=', int(product_id))]
        elif variant_id :
            query += [('variant_id', '=', int(variant_id))]
        wishlists = request.env['website.user.wishlist'].search(query, limit=limit, offset=offset, order='create_date desc')

        product = []
        for wishlist in wishlists:
            if wishlist.product_id:
                product.append(request.env['product.template'].api_single_response(wishlist.product_id))
            else:
                product.append(request.env['product.product'].api_single_response(wishlist.variant_id))
        data = {
            'product_total': request.env['website.user.wishlist'].search_count(query),
            'products': product 
        }
        return self.response(data)
    
    @http.route(prefix + 'wishlist/create-or-delete', auth='public', methods=['POST'], csrf=False)
    @controller.Controller.must_authorized()
    def create_or_delete_wishlist(self, **kw):
        user_id = int(kw.get('user_id', 0))
        product_id = kw.get('product_id', 0)
        variant_id = kw.get('variant_id', 0)
        if product_id == 0 and variant_id == 0:
            return self.response(code=400, description='product_id or variant_id is required')
        
        query = [
            ('user_id', '=', user_id),
        ]
        if product_id :
            query +=  [('product_id', '=', int(product_id))]
            params = { 
                'user_id': user_id,
                'product_id': product_id
            }
        else:
            query +=  [('variant_id', '=', int(variant_id))]
            params = { 
                'user_id': user_id,
                'variant_id': variant_id
            }
        wishlist = request.env['website.user.wishlist'].search(query, limit=1)
        result = {}
        if wishlist:
            wishlist.unlink()
            result['id'] = wishlist.id
        else:
            create = request.env['website.user.wishlist'].create(params)
            result['id'] = create.id
        return self.response(result)

    @http.route(PREFIX_USER + 'wishlist', auth='public', methods=['GET', 'OPTIONS'])
    @controller.Controller.must_authorized(private=True, private_key='user_id')
    def get_user_wishlist(self, **kw):
        params = self.get_request_params(kw, {
            'user_id': ['number'],
            'product_id': ['number'],
            'limit': ['default:0', 'number'],
            'offset': ['default:0', 'number'],
        })
        limit = params['value']['limit']
        offset = params['value']['offset']
        
        if not params['valid']:
            return self.response(code=400, description=params) 

        domain = [
            ('user_id', '=', params['value']['user_id']),
            ('variant_id', '=', False)
        ]
        if params['value']['product_id']:
            domain.append(('product_id', '=', params['value']['product_id']))
        wishlists = request.env['website.user.wishlist'].search(domain, limit=limit, offset=offset, order='create_date DESC')
        product = []
        for wishlist in wishlists:
            product.append(request.env['product.template'].api_single_response(wishlist.product_id))
        data = {
            'product_total': request.env['website.user.wishlist'].search_count(domain),
            'products': product 
        }
        return self.response(data)

    @http.route(PREFIX_USER + 'wishlist/create-or-delete', auth='public', methods=['POST', 'OPTIONS'], csrf=False)
    @controller.Controller.must_authorized(private=True, private_key='user_id')
    def create_or_delete_user_wishlist(self, **kw):
        params = self.get_request_params(kw, {
            'user_id': ['number'],
            'product_id': ['required', 'number'],
        })
        if not params['valid']:
            return self.response(code=400, description=params) 
        
        query = [
            ('user_id', '=', params['value']['user_id']),
            ('product_id', '=', params['value']['product_id'])
        ]
        wishlist = request.env['website.user.wishlist'].search(query, limit=1)
        result = {}
        if wishlist:
            wishlist.unlink()
            result['id'] = wishlist.id
        else:
            create = request.env['website.user.wishlist'].create(params['value'])
            result['id'] = create.id
        return self.response(result)