# Part of Odoo. See LICENSE file for full copyright and licensing details.

from datetime import datetime

from dateutil.relativedelta import relativedelta

from odoo import api, fields, models
from odoo.exceptions import UserError


class ContributionRegisterReport(models.AbstractModel):
    _name = "report.payroll.report_contributionregister"
    _description = "Payroll Contribution Register Report"

    def _get_payslip_lines(self, register_ids, date_from, date_to):
        result = {}
        self.env.cr.execute(
            """
            SELECT pl.id from hr_payslip_line as pl
            LEFT JOIN hr_payslip AS hp on (pl.slip_id = hp.id)
            WHERE (hp.date_from >= %s) AND (hp.date_to <= %s)
            AND pl.register_id in %s
            AND hp.state = 'done'
            ORDER BY pl.slip_id, pl.sequence""",
            (date_from, date_to, tuple(register_ids)),
        )
        line_ids = [x[0] for x in self.env.cr.fetchall()]
        for line in self.env["hr.payslip.line"].browse(line_ids):
            result.setdefault(line.register_id.id, self.env["hr.payslip.line"])
            result[line.register_id.id] += line
        return result

    @api.model
    def _get_report_values(self, docids, data=None):
        if not data.get("form"):
            raise UserError(
                self.env._("Form content is missing, this report cannot be printed.")
            )

        register_ids = self.env.context.get("active_ids", [])
        contrib_registers = self.env["hr.contribution.register"].browse(register_ids)
        date_from = data["form"].get("date_from", fields.Date.today())
        date_to = data["form"].get(
            "date_to",
            str(datetime.now() + relativedelta(months=+1, day=1, days=-1))[:10],
        )
        lines_data = self._get_payslip_lines(register_ids, date_from, date_to)
        lines_total = {}
        for register in contrib_registers:
            lines = lines_data.get(register.id)
            lines_total[register.id] = lines and sum(lines.mapped("total")) or 0.0
        return {
            "doc_ids": register_ids,
            "doc_model": "hr.contribution.register",
            "docs": contrib_registers,
            "data": data,
            "lines_data": lines_data,
            "lines_total": lines_total,
        }