# -*- coding: utf-8 -*-

from odoo import fields, models, api


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

    gpt_id = fields.Many2one('ai.robot', string='Bind to Ai', ondelete='set null')

    is_chat_private = fields.Boolean('Allow Chat Private', default=False)

    @api.model
    def im_search(self, name, limit=20):
        users = self.env['res.users'].search([
            ('id', '!=', self.env.user.id),
            ('name', 'ilike', name),
            ('active', '=', True),
            ('share', '=', False),
            ('is_chat_private', '=', True)
        ], order='gpt_id, name, id', limit=limit)
        return list(users.partner_id.mail_partner_format().values())

    def mail_partner_format(self, fields=None):
        # 直接覆盖原生，增加 gpt_id 字段
        partners_format = dict()
        if not fields:
            fields = {'id': True, 'name': True, 'email': True, 'active': True, 'im_status': True, 'gpt_id': 0, 'user': {}}
        for partner in self:
            data = {}
            if 'id' in fields:
                data['id'] = partner.id
            if 'name' in fields:
                name = partner.name
                # 英文不好分，暂时不隐名
                # if not partner.related_user_id.gpt_id:
                #     name = partner.name[0] + '*' * (len(partner.name) - 1)
                data['name'] = name
            if 'email' in fields:
                data['email'] = partner.email
            if 'active' in fields:
                data['active'] = partner.active
            if 'im_status' in fields:
                data['im_status'] = partner.im_status
            if 'gpt_id' in fields:
                data['gpt_id'] = partner.gpt_id.id if partner.gpt_id else 0
            if 'user' in fields:
                internal_users = partner.user_ids - partner.user_ids.filtered('share')
                main_user = internal_users[0] if len(internal_users) > 0 else partner.user_ids[0] if len(partner.user_ids) > 0 else self.env['res.users']
                data['user'] = {
                    "id": main_user.id,
                    "isInternalUser": not main_user.share,
                } if main_user else [('clear',)]
            # if 'guest' in self.env.context or not self.env.user.has_group('base.group_erp_manager'):
            # 完全不显示 邮箱
            data.pop('email', None)
            partners_format[partner] = data
        return partners_format