# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import api, fields, models, _
from odoo.tools.sql import column_exists, create_column
from odoo.exceptions import UserError


class ResPartner(models.Model):
    _inherit = 'res.partner'

    l10n_ke_branch_code = fields.Char('eTIMS Branch Code', compute="_compute_l10n_ke_branch_code", store=True, readonly=False)

    def _auto_init(self):
        if not column_exists(self.env.cr, "res_partner", "l10n_ke_branch_code"):
            create_column(self.env.cr, "res_partner", "l10n_ke_branch_code", "varchar")
            self.env.cr.execute("""
                UPDATE res_partner AS p
                   SET l10n_ke_branch_code = '00'
                  FROM res_country cc
                 WHERE p.country_id = cc.id
                   AND cc.code = 'KE'
            """)
        super()._auto_init()

    @api.depends('country_id')
    def _compute_l10n_ke_branch_code(self):
        ke_partners_wo_branch_code = self.filtered(
            lambda partner: not partner.l10n_ke_branch_code and partner.country_id.code == 'KE'
        )
        # set default branch code for Kenyan partners without a branch code
        ke_partners_wo_branch_code.l10n_ke_branch_code = '00'

    def _l10n_ke_oscu_partner_content(self):
        """Returns a dict with the commonly required fields on partner for requests to the OSCU """
        self.ensure_one()
        return {
            'custNo':  self.id,                              # Customer Number
            'custTin': self.vat,                             # Customer PIN
            'custNm':  self.name,                            # Customer Name
            'adrs':    self.contact_address_inline or None,  # Address
            'email':   self.email or None,                   # Email
            'useYn':   'Y' if self.active else 'N',          # Used (Y/N)
        }

    def action_l10n_ke_oscu_register_bhf_customer(self):
        """Save the partner information on the OSCU."""
        for partner in self:
            content = {
                **self.env.company._l10n_ke_get_user_dict(partner.create_uid, partner.write_uid),
                **partner._l10n_ke_oscu_partner_content()    # Partner details
            }
            company = partner.company_id or self.env.company
            error, _data, _dummy = company._l10n_ke_call_etims('saveBhfCustomer', content)
            if error:
                raise UserError(self.env._("[%(code)s] %(message)s", code=error["code"], message=error["message"]))
        return {
            'type': 'ir.actions.client',
            'tag': 'display_notification',
            'params': {
                'type': 'success',
                'sticky': False,
                'message': _("Partner successfully registered"),
                'next': {'type': 'ir.actions.act_window_close'},
            }
        }

    def action_l10n_ke_oscu_fetch_bhf_customer(self):
        """ Fetch saved customer information from eTIMS.
            We don't use this method but must have it to demonstrate we are able to call their API.
        """
        company = self.company_id or self.env.company
        error, data, _dummy = company._l10n_ke_call_etims('selectCustomer', {'custmTin': self.vat})
        raise UserError(data or error)