# Copyright 2019 Ecosoft Co., Ltd (http://ecosoft.co.th/)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html)
from odoo import _, models
from odoo.exceptions import UserError


class AccountPayment(models.Model):
    _name = "account.bank.statement.line"
    _inherit = ["account.bank.statement.line", "account.document.reversal"]

    def button_cancel_reconciliation(self):
        """ If cancel method is to reverse, use document reversal wizard """
        cancel_reversal = all(
            self.mapped("journal_entry_ids.move_id.journal_id.is_cancel_reversal")
        )
        states = self.mapped("statement_id.state")
        if cancel_reversal:
            if not all(st == "open" for st in states):
                raise UserError(_("Only new bank statement can be cancelled"))
            return self.reverse_document_wizard()
        return super().button_cancel_reconciliation()

    def action_document_reversal(self, date=None, journal_id=None):
        """ Reverse all moves related to this statement + delete payment """
        # This part is from button_cancel_reconciliation()
        aml_to_unbind = self.env['account.move.line']
        aml_to_cancel = self.env['account.move.line']
        payment_to_unreconcile = self.env['account.payment']
        payment_to_cancel = self.env['account.payment']
        for st_line in self:
            aml_to_unbind |= st_line.journal_entry_ids
            for line in st_line.journal_entry_ids:
                payment_to_unreconcile |= line.payment_id
                if st_line.move_name and line.payment_id.payment_reference == st_line.move_name:
                    #there can be several moves linked to a statement line but maximum one created by the line itself
                    aml_to_cancel |= line
                    payment_to_cancel |= line.payment_id
        aml_to_unbind = aml_to_unbind - aml_to_cancel

        if aml_to_unbind:
            aml_to_unbind.write({'statement_line_id': False})

        payment_to_unreconcile = payment_to_unreconcile - payment_to_cancel
        if payment_to_unreconcile:
            payment_to_unreconcile.unreconcile()
        # --

        # Find account moves to cancel reversal
        moves = aml_to_cancel.mapped("move_id")
        # Create reverse entries
        moves._cancel_reversal(journal_id)
        # Set cancel related payments
        payment_to_cancel.write({"state": "cancelled"})
        return True