# -*- coding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2017 Alda Hotels <informatica@aldahotels.com>
#                       Jose Luis Algara <osotranquilo@gmail.com>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
import functools
import itertools
import logging
import psycopg2

from odoo import api, fields, models, _
from odoo.osv.expression import get_unaccent_wrapper
from odoo.exceptions import ValidationError, UserError
from odoo.tools import mute_logger
_logger = logging.getLogger(__name__)


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

    document_type = fields.Selection([
        ('D', 'DNI'),
        ('P', 'Pasaporte'),
        ('C', 'Permiso de Conducir'),
        ('I', 'Carta o Doc. de Identidad'),
        ('N', 'Permiso Residencia Español'),
        ('X', 'Permiso Residencia Europeo')],
        help=_('Select a valid document type'),
        string='Doc. type',
        )
    document_number = fields.Char('Document number', index=True)
    document_expedition_date = fields.Date('Document expedition date')
    code_ine_id = fields.Many2one('code.ine',
        help=_('Country or province of origin. Used for INE statistics.'))
    unconfirmed = fields.Boolean('Unconfirmed', default=True)
    main_partner_id = fields.Many2one('res.partner')

    @api.model
    def name_search(self, name, args=None, operator='ilike', limit=100):
        if not args:
            args = []
        domain = ['|',
                  ('document_number', operator, name),
                  ('vat', operator, name),
                  ]
        partners = self.search(domain + args, limit=limit,)
        res = partners.name_get()
        if limit:
            limit_rest = limit - len(partners)
        else:
            limit_rest = limit
        if limit_rest or not limit:
            args += [('id', 'not in', partners.ids)]
            res += super(ResPartner, self).name_search(
                name, args=args, operator=operator, limit=limit_rest)
        return res

    @api.model
    def _get_duplicated_ids(self, partner):
        partner_ids = []
        if partner.vat:
            partner_ids += self.env['res.partner'].search([
                ('vat', '=', partner.vat),
                ('parent_id', '=', False)
                ]).ids
        if partner.document_number:
            partner_ids += self.env['res.partner'].search([
                ('document_number', '=', partner.document_number),
                ('child_ids', '=', False)
                ]).ids
        if partner_ids:
            return partner_ids

    def _merge_fields(self):
        duplicated_fields = ['vat', 'document_number']
        return duplicated_fields

    @api.multi
    def write(self, vals):
        if vals.get('vat') and not self._context.get(
                "ignore_vat_update", False):
            for record in self:
                vat = vals.get('vat')
                if self.env.context.get('company_id'):
                    company = self.env['res.company'].browse(self.env.context['company_id'])
                else:
                    company = self.env.user.company_id
                if company.vat_check_vies:
                    # force full VIES online check
                    check_func = self.vies_vat_check
                else:
                    # quick and partial off-line checksum validation
                    check_func = self.simple_vat_check
                vat_country, vat_number = self._split_vat(vat)
                #check with country code as prefix of the TIN
                if not check_func(vat_country, vat_number):
                    country_id = vals.get('country_id') or record.country_id.id
                    vals['vat'] = record.fix_eu_vat_number(country_id, vat)
        return super(ResPartner, self).write(vals)

    @api.constrains('country_id')
    def update_vat_code_country(self):
        for record in self:
            country_id = record.country_id.id
            vat = record.vat
            #check with country code as prefix of the TIN
            if vat:
                if self.env.context.get('company_id'):
                    company = self.env['res.company'].browse(self.env.context['company_id'])
                else:
                    company = self.env.user.company_id
                if company.vat_check_vies:
                    # force full VIES online check
                    check_func = self.vies_vat_check
                else:
                    # quick and partial off-line checksum validation
                    check_func = self.simple_vat_check
                vat_country, vat_number = self._split_vat(vat)
                if not check_func(vat_country, vat_number):
                    vat_with_code = record.fix_eu_vat_number(country_id, vat)
                    if country_id and vat != vat_with_code:
                        record.with_context({'ignore_vat_update': True}).write({
                            'vat': vat_with_code
                            })

    @api.constrains('vat', 'commercial_partner_country_id')
    def check_vat(self):
        spain = self.env['res.country'].search([
            ('code', '=', 'ES')
        ])
        from_spain = False
        for partner in self:
            if partner.country_id == spain:
                from_spain = True
        if from_spain:
            return super(ResPartner, self).check_vat()

    @api.constrains('vat')
    def _check_vat_unique(self):
        for record in self:
            if record.unconfirmed:
                if record.vat:
                    record.update({'unconfirmed': False})
                    partner_ids = self.env['res.partner'].search([
                        ('vat', '=', record.vat),
                        ('parent_id', '=', False)
                        ]).ids
                    if len(partner_ids) > 1:
                        partners = self.env['res.partner'].browse(partner_ids)
                        record._merge(partners._ids)
            else:
                return super(ResPartner, self)._check_vat_unique()
        return True

    @api.constrains('document_number', 'document_type')
    def _check_document_number_unique(self):
        for record in self:
            if not record.document_number:
                continue
            if record.unconfirmed:
                if record.document_number:
                    record.update({'unconfirmed': False})
                    partner_ids = self.env['res.partner'].search([
                        ('document_number', '=', record.document_number),
                        ]).ids
                    if len(partner_ids) > 1:
                        partners = self.env['res.partner'].browse(partner_ids)
                        record._merge(partners._ids)
                    if not record.parent_id and record.document_type == 'D' \
                            and not record.vat:
                        record.update({
                            'vat': record.document_number,
                            })
            else:
                results = self.env['res.partner'].search_count([
                    ('document_type', '=', record.document_type),
                    ('document_number', '=', record.document_number),
                    ('id', '!=', record.id)
                ])
                if results:
                    raise ValidationError(_(
                        "The Document Number %s already exists in another "
                        "partner.") % record.document_number)

    @api.multi
    def open_main_partner(self):
        self.ensure_one()
        action = self.env.ref('base.action_partner_form').read()[0]
        if self.main_partner_id:
            action['views'] = [(self.env.ref('base.view_partner_form').id, 'form')]
            action['res_id'] = self.main_partner_id.id
        else:
            action = {'type': 'ir.actions.act_window_close'}
        return action

    def _get_fk_on(self, table):
        """ return a list of many2one relation with the given table.
            :param table : the name of the sql table to return relations
            :returns a list of tuple 'table name', 'column name'.
        """
        query = """
            SELECT cl1.relname as table, att1.attname as column
            FROM pg_constraint as con, pg_class as cl1, pg_class as cl2, pg_attribute as att1, pg_attribute as att2
            WHERE con.conrelid = cl1.oid
                AND con.confrelid = cl2.oid
                AND array_lower(con.conkey, 1) = 1
                AND con.conkey[1] = att1.attnum
                AND att1.attrelid = cl1.oid
                AND cl2.relname = %s
                AND att2.attname = 'id'
                AND array_lower(con.confkey, 1) = 1
                AND con.confkey[1] = att2.attnum
                AND att2.attrelid = cl2.oid
                AND con.contype = 'f'
        """
        self._cr.execute(query, (table,))
        return self._cr.fetchall()

    def _merge(self, partner_ids, dst_partner=None):
        """ private implementation of merge partner
            :param partner_ids : ids of partner to merge
            :param dst_partner : record of destination res.partner
        """
        partner = self.env['res.partner']
        partner_ids = partner.browse(partner_ids).exists()
        if len(partner_ids) < 2:
            return

        if len(partner_ids) > 3:
            raise UserError(_("For safety reasons, you cannot merge more than 3 contacts together. You can re-open the wizard several times if needed."))

        # check if the list of partners to merge contains child/parent relation
        child_ids = self.env['res.partner']
        for partner_id in partner_ids:
            child_ids |= partner.search([('id', 'child_of', [partner_id.id])]) - partner_id
        if partner_ids & child_ids:
            raise UserError(_("You cannot merge a contact with one of his parent."))

        # remove dst_partner from partners to merge
        if dst_partner and dst_partner in partner_ids:
            src_partners = partner_ids - dst_partner
        else:
            ordered_partners = self._get_ordered_partner(partner_ids.ids)
            dst_partner = ordered_partners[-1]
            src_partners = ordered_partners[:-1]
        _logger.info("dst_partner: %s", dst_partner.id)

        # call sub methods to do the merge
        self._update_foreign_keys(src_partners, dst_partner)
        self._update_reference_fields(src_partners, dst_partner)
        self._update_values(src_partners, dst_partner)

        _logger.info('(uid = %s) merged the partners %r with %s', self._uid, src_partners.ids, dst_partner.id)
        dst_partner.message_post(body='%s %s' % (_("Merged with the following partners:"), ", ".join('%s <%s> (ID %s)' % (p.name, p.email or 'n/a', p.id) for p in src_partners)))

        return dst_partner

    @api.model
    def _update_foreign_keys(self, src_partners, dst_partner):
        """ Update all foreign key from the src_partner to dst_partner. All many2one fields will be updated.
            :param src_partners : merge source res.partner recordset (does not include destination one)
            :param dst_partner : record of destination res.partner
        """
        _logger.debug('_update_foreign_keys for dst_partner: %s for src_partners: %s', dst_partner.id, str(src_partners.ids))

        # find the many2one relation to a partner
        partner = self.env['res.partner']
        relations = self._get_fk_on('res_partner')

        for table, column in relations:
            # get list of columns of current table (exept the current fk column)
            query = "SELECT column_name FROM information_schema.columns WHERE table_name LIKE '%s'" % (table)
            self._cr.execute(query, ())
            columns = []
            for data in self._cr.fetchall():
                if data[0] != column:
                    columns.append(data[0])

            # do the update for the current table/column in SQL
            query_dic = {
                'table': table,
                'column': column,
                'value': columns[0],
            }
            if len(columns) <= 1:
                # unique key treated
                query = """
                    UPDATE "%(table)s" as ___tu
                    SET %(column)s = %%s
                    WHERE
                        %(column)s = %%s AND
                        NOT EXISTS (
                            SELECT 1
                            FROM "%(table)s" as ___tw
                            WHERE
                                %(column)s = %%s AND
                                ___tu.%(value)s = ___tw.%(value)s
                        )""" % query_dic
                for partner in src_partners:
                    self._cr.execute(query, (dst_partner.id, partner.id, dst_partner.id))
            else:
                try:
                    with mute_logger('odoo.sql_db'), self._cr.savepoint():
                        query = 'UPDATE "%(table)s" SET %(column)s = %%s WHERE %(column)s IN %%s' % query_dic
                        self._cr.execute(query, (dst_partner.id, tuple(src_partners.ids),))

                        # handle the recursivity with parent relation
                        if column == partner._parent_name and table == 'res_partner':
                            query = """
                                WITH RECURSIVE cycle(id, parent_id) AS (
                                        SELECT id, parent_id FROM res_partner
                                    UNION
                                        SELECT  cycle.id, res_partner.parent_id
                                        FROM    res_partner, cycle
                                        WHERE   res_partner.id = cycle.parent_id AND
                                                cycle.id != cycle.parent_id
                                )
                                SELECT id FROM cycle WHERE id = parent_id AND id = %s
                            """
                            self._cr.execute(query, (dst_partner.id,))
                            # NOTE JEM : shouldn't we fetch the data ?
                except psycopg2.Error:
                    # updating fails, most likely due to a violated unique constraint
                    # keeping record with nonexistent partner_id is useless, better delete it
                    query = 'DELETE FROM "%(table)s" WHERE "%(column)s" IN %%s' % query_dic
                    self._cr.execute(query, (tuple(src_partners.ids),))

    @api.model
    def _update_reference_fields(self, src_partners, dst_partner):
        """ Update all reference fields from the src_partner to dst_partner.
            :param src_partners : merge source res.partner recordset (does not include destination one)
            :param dst_partner : record of destination res.partner
        """
        _logger.debug('_update_reference_fields for dst_partner: %s for src_partners: %r', dst_partner.id, src_partners.ids)

        def update_records(model, src, field_model='model', field_id='res_id'):
            Model = self.env[model] if model in self.env else None
            if Model is None:
                return
            records = Model.sudo().search([(field_model, '=', 'res.partner'), (field_id, '=', src.id)])
            try:
                with mute_logger('odoo.sql_db'), self._cr.savepoint():
                    return records.sudo().write({field_id: dst_partner.id})
            except psycopg2.Error:
                # updating fails, most likely due to a violated unique constraint
                # keeping record with nonexistent partner_id is useless, better delete it
                return records.sudo().unlink()

        update_records = functools.partial(update_records)

        for partner in src_partners:
            update_records('calendar', src=partner, field_model='model_id.model')
            update_records('ir.attachment', src=partner, field_model='res_model')
            update_records('mail.followers', src=partner, field_model='res_model')
            update_records('mail.message', src=partner)
            update_records('ir.model.data', src=partner)

        records = self.env['ir.model.fields'].search([('ttype', '=', 'reference')])
        for record in records.sudo():
            try:
                Model = self.env[record.model]
                field = Model._fields[record.name]
            except KeyError:
                # unknown model or field => skip
                continue

            if field.compute is not None:
                continue

            for partner in src_partners:
                records_ref = Model.sudo().search([(record.name, '=', 'res.partner,%d' % partner.id)])
                values = {
                    record.name: 'res.partner,%d' % dst_partner.id,
                }
                records_ref.sudo().write(values)

    @api.model
    def _update_values(self, src_partners, dst_partner):
        """ Update values of dst_partner with the ones from the src_partners.
            :param src_partners : recordset of source res.partner
            :param dst_partner : record of destination res.partner
        """
        _logger.debug('_update_values for dst_partner: %s for src_partners: %r', dst_partner.id, src_partners.ids)

        model_fields = dst_partner.fields_get().keys()

        def write_serializer(item):
            if isinstance(item, models.BaseModel):
                return item.id
            else:
                return item
        # get all fields that are not computed or x2many
        values = dict()
        for column in model_fields:
            field = dst_partner._fields[column]
            if field.type not in ('many2many', 'one2many') and field.compute is None:
                for item in itertools.chain(src_partners, [dst_partner]):
                    if item[column]:
                        values[column] = write_serializer(item[column])
        # remove fields that can not be updated (id and parent_id)
        values.pop('id', None)
        parent_id = values.pop('parent_id', None)
        src_partners.update({
            'active': False,
            'document_number': False,
            'vat': False,
            'main_partner_id': dst_partner.id})
        dst_partner.write(values)
        # try to update the parent_id
        if parent_id and parent_id != dst_partner.id:
            try:
                dst_partner.write({'parent_id': parent_id})
            except ValidationError:
                _logger.info('Skip recursive partner hierarchies for parent_id %s of partner: %s', parent_id, dst_partner.id)

    @api.model
    def _get_ordered_partner(self, partner_ids):
        """ Helper : returns a `res.partner` recordset ordered by create_date/active fields
            :param partner_ids : list of partner ids to sort
        """
        return self.env['res.partner'].browse(partner_ids).sorted(
            key=lambda p: (p.active, (p.create_date or '')),
            reverse=True,
        )

    @api.multi
    def _compute_models(self):
        """ Compute the different models needed by the system if you want to exclude some partners. """
        model_mapping = {}
        if self.exclude_contact:
            model_mapping['res.users'] = 'partner_id'
        if 'account.move.line' in self.env and self.exclude_journal_item:
            model_mapping['account.move.line'] = 'partner_id'
        return model_mapping