# Copyright 2019 Tecnativa - Pedro M. Baeza
# License AGPL-3 - See http://www.gnu.org/licenses/agpl-3.0.html

from odoo import models, api


class AccountReconciliationWidget(models.AbstractModel):
    _inherit = 'account.reconciliation.widget'

    @api.model
    def _get_possible_payment_orders_for_statement_line(self, st_line):
        """Find orders that might be candidates for matching a statement
        line.
        """
        return self.env['account.payment.order'].search([
            ('total_company_currency', '=', st_line.amount),
            ('state', 'in', ['done', 'uploaded']),
        ])

    @api.model
    def _get_reconcile_lines_from_order(self, st_line, order,
                                        excluded_ids=None):
        """Return lines to reconcile our statement line with."""
        aml_obj = self.env['account.move.line']
        reconciled_lines = aml_obj.search([
            ('bank_payment_line_id', 'in', order.bank_line_ids.ids)
        ])
        return (
            reconciled_lines.mapped('move_id.line_ids') - reconciled_lines -
            aml_obj.browse(excluded_ids)
        ).filtered(lambda x: not x.reconciled)

    def _prepare_proposition_from_orders(self, st_line, orders,
                                         excluded_ids=None):
        """Fill with the expected format the reconciliation proposition
        for the given statement line and possible payment orders.
        """
        target_currency = (
            st_line.currency_id or st_line.journal_id.currency_id or
            st_line.journal_id.company_id.currency_id
        )
        for order in orders:
            elegible_lines = self._get_reconcile_lines_from_order(
                st_line, order, excluded_ids=excluded_ids,
            )
            if elegible_lines:
                return self._prepare_move_lines(
                    elegible_lines, target_currency=target_currency,
                    target_date=st_line.date,
                )
        return []

    def get_bank_statement_line_data(self, st_line_ids, excluded_ids=None):
        res = super().get_bank_statement_line_data(
            st_line_ids, excluded_ids=excluded_ids,
        )
        st_line_obj = self.env['account.bank.statement.line']
        for line_vals in res.get('lines', []):
            st_line = st_line_obj.browse(line_vals['st_line']['id'])
            orders = self._get_possible_payment_orders_for_statement_line(
                st_line)
            proposition_vals = self._prepare_proposition_from_orders(
                st_line, orders, excluded_ids=excluded_ids,
            )
            if proposition_vals:
                line_vals['reconciliation_proposition'] = proposition_vals
        return res