# 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 _, api, models
from odoo.exceptions import UserError, ValidationError


class AccountInvoice(models.Model):
    _name = "account.invoice"
    _inherit = ["account.invoice", "account.document.reversal"]

    @api.multi
    def action_invoice_cancel(self):
        """ If cancel method is to reverse, use document reversal wizard
        * Draft invoice, fall back to standard invoice cancel
        * Non draft, must be fully open (not even partial reconciled) to cancel
        """
        cancel_reversal = all(self.mapped("journal_id.is_cancel_reversal"))
        states = self.mapped("state")
        if cancel_reversal and "draft" not in states:
            if not all(st == "open" for st in states) or (
                self.mapped("move_id.line_ids.matched_debit_ids")
                | self.mapped("move_id.line_ids.matched_credit_ids")
            ):
                raise UserError(
                    _(
                        "Only fully unpaid invoice can be cancelled.\n"
                        "To cancel this invoice, make sure all payment(s) "
                        "are also cancelled."
                    )
                )
            return self.reverse_document_wizard()
        return super().action_invoice_cancel()

    @api.multi
    def action_document_reversal(self, date=None, journal_id=None):
        """ Reverse all moves related to this invoice + set state to cancel """
        # Check document state
        if "cancel" in self.mapped("state"):
            raise ValidationError(_("You are trying to cancel the cancelled document"))
        MoveLine = self.env["account.move.line"]
        move_lines = MoveLine.search([("invoice_id", "in", self.ids)])
        moves = move_lines.mapped("move_id")
        # Set all moves to unreconciled
        move_lines.filtered(lambda x: x.account_id.reconcile).remove_move_reconcile()
        # Important to remove relation with move.line before reverse
        move_lines.write({"invoice_id": False})
        # Create reverse entries
        moves.reverse_moves(date, journal_id)
        # Set state cancelled and unlink with account.move
        self.write(
            {
                "move_id": False,
                "move_name": False,
                "reference": False,
                "state": "cancel",
            }
        )
        return True