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

class VPSPortal(CustomerPortal):

    def _prepare_home_portal_values(self, counters):
        values = super()._prepare_home_portal_values(counters)
        partner = request.env.user.partner_id
        VPSServer = request.env['vps.server']
        if 'vps_server_count' in counters:
            values['vps_server_count'] = VPSServer.search_count([('customer_id', '=', partner.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()
        partner = request.env.user.partner_id
        VPSServer = request.env['vps.server']

        domain = [('customer_id', '=', partner.id)]

        searchbar_sortings = {
            'name': {'label': _('Name'), 'order': 'name'},
            'state': {'label': _('State'), 'order': 'state'},
        }

        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_id>'], type='http', auth="user", website=True)
    def portal_my_vps_server(self, vps_id, **kw):
        try:
            vps_sudo = self._document_check_access('vps.server', vps_id)
        except AccessError:
            return request.redirect('/my')

        if vps_sudo.customer_id.id != request.env.user.partner_id.id:
            return request.redirect('/my')

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

    def _vps_server_get_page_view_values(self, vps_server, access_token=None, **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)

    @http.route(['/my/vps/<int:vps_id>/restart'], type='http', auth="user", website=True)
    def portal_restart_vps(self, vps_id, **kw):
        try:
            vps_sudo = self._document_check_access('vps.server', vps_id)
            if vps_sudo.customer_id.id != request.env.user.partner_id.id:
                return request.redirect('/my')
            vps_sudo.action_restart_from_portal()
            return request.redirect(vps_sudo.get_portal_url())
        except AccessError:
            return request.redirect('/my')

    def _document_check_access(self, model_name, document_id, access_token=None):
        document = request.env[model_name].sudo().browse(document_id)
        document_sudo = document.with_user(request.env.user).sudo()
        try:
            document_sudo.check_access_rights('read')
            document_sudo.check_access_rule('read')
        except AccessError:
            raise
        return document_sudo