# Part of Odoo. See LICENSE file for full copyright and licensing details.
import datetime
from collections import defaultdict
from zoneinfo import ZoneInfo

from odoo import models


class HrVersion(models.Model):
    _inherit = 'hr.version'

    def _get_version_work_entries_values(self, date_start, date_stop):
        # Add the work entries difference for french payroll
        # Work entries by default are not generated on days the employee does not work
        # So we have to fill the gaps with work entries for those periods
        result = super()._get_version_work_entries_values(date_start, date_stop)
        fr_versions = self.filtered(lambda c: c.company_id.country_id.code == 'FR' and c.resource_calendar_id != c.company_id.resource_calendar_id)
        if not fr_versions:
            return result
        start_dt = date_start.replace(tzinfo=datetime.UTC) if not date_start.tzinfo else date_start
        end_dt = date_stop.replace(tzinfo=datetime.UTC) if not date_stop.tzinfo else date_stop
        # l10n_fr_date_to_changed is False when no adjustment had to be done
        all_leaves = self.env['hr.leave'].search([
            ('employee_id', 'in', fr_versions.employee_id.ids),
            ('state', '=', 'validate'),
            ('date_from', '<=', end_dt.astimezone(datetime.UTC).replace(tzinfo=None)),
            ('date_to', '>=', start_dt.astimezone(datetime.UTC).replace(tzinfo=None)),
            ('l10n_fr_date_to_changed', '=', True),
        ])
        leaves_per_employee = defaultdict(lambda: self.env['hr.leave'])
        for leave in all_leaves:
            leaves_per_employee[leave.employee_id] |= leave
        for version in fr_versions:
            employee = version.employee_id
            company = version.company_id
            company_calendar = company.resource_calendar_id
            resource = employee.resource_id
            tz = ZoneInfo(version._get_tz())

            for leave in leaves_per_employee[employee]:
                leave_start_dt = max(start_dt, leave.date_from.astimezone(tz))
                leave_end_dt_fr = min(end_dt, leave.date_to.astimezone(tz))

                # Compute the attendances for the company calendar and the employee calendar
                # and then compute and keep the difference between those two
                resources_per_tz = version._get_resources_per_tz()
                company_attendances = company_calendar._attendance_intervals_batch(
                    leave_start_dt, leave_end_dt_fr, resources_per_tz=resources_per_tz,
                )[resource.id]
                # Dates on which work entries should already be generated
                employee_dates = set()
                for vals in result:
                    employee_dates.add(vals['date_start'].date())
                    employee_dates.add(vals['date_stop'].date())
                leave_work_entry_type = leave.work_entry_type_id
                result += [{
                    'date_start': interval[0].astimezone(datetime.UTC).replace(tzinfo=None),
                    'date_stop': interval[1].astimezone(datetime.UTC).replace(tzinfo=None),
                    'work_entry_type_id': leave_work_entry_type,
                    'employee_id': employee,
                    'company_id': version.company_id,
                    'version_id': version,
                    'leave_ids': leave,
                } for interval in company_attendances if interval[0].date() not in employee_dates]

        return result