from odoo import http, _
from odoo.exceptions import AccessError, MissingError
from odoo.http import request
from odoo.addons.portal.controllers.portal import CustomerPortal, pager as portal_pager
from odoo.osv.expression import OR

class CustomerPortalVPS(CustomerPortal):

    def _prepare_home_portal_values(self, counters):
        values = super()._prepare_home_portal_values(counters)
        if 'vps_server_count' in counters:
            values['vps_server_count'] = request.env['vps.server'].search_count([('customer_id', '=', request.env.user.partner_id.id)])
        return values

    @http.route(['/my/vps-servers', '/my/vps-servers/page/<int:page>'], type='http', auth="user", website=True)
    def portal_my_vps_servers(self, page=1, date_begin=None, date_end=None, sortby=None, **kw):
        values = self._prepare_portal_layout_values()
        VPSServer = request.env['vps.server']

        domain = [('customer_id', '=', request.env.user.partner_id.id)]

        searchbar_sortings = {
            'name': {'label': _('Name'), 'order': 'name'},
            'ip_address': {'label': _('IP Address'), 'order': 'ip_address'},
        }

        if not sortby:
            sortby = 'name'
        sort_order = searchbar_sortings[sortby]['order']

        if date_begin and date_end:
            domain += [('create_date', '>', date_begin), ('create_date', '<=', date_end)]

        # count for pager
        vps_server_count = VPSServer.search_count(domain)

        # make pager
        pager = portal_pager(
            url="/my/vps-servers",
            url_args={'date_begin': date_begin, 'date_end': date_end, 'sortby': sortby},
            total=vps_server_count,
            page=page,
            step=self._items_per_page
        )

        # search the count to display, according to the pager data
        vps_servers = VPSServer.search(
            domain,
            order=sort_order,
            limit=self._items_per_page,
            offset=pager['offset']
        )

        values.update({
            'date': date_begin,
            'vps_servers': vps_servers,
            'page_name': 'vps_server',
            'pager': pager,
            'default_url': '/my/vps-servers',
            'searchbar_sortings': searchbar_sortings,
            'sortby': sortby,
        })
        return request.render("ow_vm_management.portal_my_vps_servers", values)

    @http.route(['/my/vps-servers/<int:vps_server_id>'], type='http', auth="user", website=True)
    def portal_my_vps_server(self, vps_server_id=None, access_token=None, **kw):
        try:
            vps_server_sudo = self._document_check_access('vps.server', vps_server_id, access_token)
        except (AccessError, MissingError):
            return request.redirect('/my')

        values = self._vps_server_get_page_view_values(vps_server_sudo, access_token, **kw)
        return request.render("ow_vm_management.portal_vps_server_page", values)

    def _vps_server_get_page_view_values(self, vps_server, access_token, **kwargs):
        values = {
            'page_name': 'vps_server',
            'vps_server': vps_server,
        }
        return self._get_page_view_values(vps_server, access_token, values, 'my_vps_servers_history', False, **kwargs)