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


class Wishlist(controller.Controller):
    prefix = '/api/v1/'
    
    @http.route(prefix + 'wishlist', auth='public', methods=['GET'])
    def get_wishlist_by_user_id(self, **kw):
        if not self.authenticate():
            return self.response(code=401, description='Unauthorized')
        
        user_id = int(kw.get('user_id', 0))
        product_id = kw.get('product_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))]
        wishlists = request.env['website.user.wishlist'].search(query, limit=limit, offset=offset, order='create_date desc')
        data = {
            'product_total': request.env['website.user.wishlist'].search_count(query),
            'products': [request.env['product.template'].api_single_response(wishlist.product_id) for wishlist in wishlists]
        }
        return self.response(data)
    
    @http.route(prefix + 'wishlist/create-or-delete', auth='public', methods=['POST'], csrf=False)
    def create_or_delete_wishlist(self, **kw):
        if not self.authenticate():
            return self.response(code=401, description='Unauthorized')
        
        user_id = int(kw.get('user_id', 0))
        product_id = kw.get('product_id', 0)
        if not product_id:
            return self.response(code=400, description='product_id is required')
        
        query = [
            ('user_id', '=', user_id),
            ('product_id', '=', int(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({
                'user_id': user_id,
                'product_id': product_id
            })
            result['id'] = create.id
        return self.response(result)