# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import api, models, _, fields
from odoo.tools import float_compare
from odoo.tools.misc import DEFAULT_SERVER_DATE_FORMAT


TRIAL_BALANCE_END_COLUMN_GROUP_KEY = '_trial_balance_end_column_group'


class TrialBalanceCustomHandler(models.AbstractModel):
    _name = 'account.trial.balance.report.handler'
    _inherit = 'account.report.custom.handler'
    _description = 'Trial Balance Custom Handler'

    def _dynamic_lines_generator(self, report, options, all_column_groups_expression_totals, warnings=None):
        def _update_column(line, column_key, new_value):
            line['columns'][column_key]['no_format'] = new_value
            line['columns'][column_key]['is_zero'] = self.env.company.currency_id.is_zero(new_value)

        def _update_balance_columns(line, debit_column_key, credit_column_key, balance_column_key=None):
            debit_value = line['columns'][debit_column_key]['no_format'] if debit_column_key is not None else False
            credit_value = line['columns'][credit_column_key]['no_format'] if credit_column_key is not None else False

            if debit_value and credit_value:
                new_debit_value = 0.0
                new_credit_value = 0.0

                if self.env.company.currency_id.compare_amounts(debit_value, credit_value) == 1:
                    new_debit_value = debit_value - credit_value
                else:
                    new_credit_value = (debit_value - credit_value) * -1

                _update_column(line, debit_column_key, new_debit_value)
                _update_column(line, credit_column_key, new_credit_value)

            if balance_column_key is not None:
                _update_column(line, balance_column_key, debit_value - credit_value)

        lines = [line[1] for line in self.env['account.general.ledger.report.handler']._dynamic_lines_generator(report, options, all_column_groups_expression_totals, warnings=warnings)]

        # We need to find the index of debit and credit columns for initial and end balance in case of extra custom columns
        init_balance_debit_index = next((index for index, column in enumerate(options['columns']) if column.get('expression_label') == 'debit'), None)
        init_balance_credit_index = next((index for index, column in enumerate(options['columns']) if column.get('expression_label') == 'credit'), None)

        end_balance_debit_index = next((index for index, column in enumerate(options['columns']) if column.get('expression_label') == 'debit' and column.get('column_group_key') == TRIAL_BALANCE_END_COLUMN_GROUP_KEY), None)
        end_balance_credit_index = next((index for index, column in enumerate(options['columns']) if column.get('expression_label') == 'credit' and column.get('column_group_key') == TRIAL_BALANCE_END_COLUMN_GROUP_KEY), None)
        end_balance_balance_index = next((index for index, column in enumerate(options['columns']) if column.get('expression_label') == 'balance' and column.get('column_group_key') == TRIAL_BALANCE_END_COLUMN_GROUP_KEY), None)

        currency = self.env.company.currency_id
        for line in lines[:-1]:
            # Initial balance
            _update_balance_columns(line, init_balance_debit_index, init_balance_credit_index)

            # End balance: sum all the previous columns for both debit and credit
            if end_balance_debit_index is not None:
                end_balance_debit_sum = sum(
                    currency.round(column['no_format'])
                    for index, column in enumerate(line['columns'])
                    if column.get('expression_label') == 'debit' and index != end_balance_debit_index and column['no_format'] is not None
                )
                _update_column(line, end_balance_debit_index, end_balance_debit_sum)

            if end_balance_credit_index is not None:
                end_balance_credit_sum = sum(
                    currency.round(column['no_format'])
                    for index, column in enumerate(line['columns'])
                    if column.get('expression_label') == 'credit' and index != end_balance_credit_index and column['no_format'] is not None
                )
                _update_column(line, end_balance_credit_index, end_balance_credit_sum)

            _update_balance_columns(line, end_balance_debit_index, end_balance_credit_index, end_balance_balance_index)

            line.pop('expand_function', None)
            line.pop('groupby', None)
            line.update({
                'unfoldable': False,
                'unfolded': False,
            })

            res_model = report._get_model_info_from_id(line['id'])[0]
            if res_model == 'account.account':
                line['caret_options'] = 'trial_balance'

        # Total line
        if lines:
            total_line = lines[-1]

            for index in (init_balance_debit_index, init_balance_credit_index, end_balance_debit_index, end_balance_credit_index):
                if index is not None:
                    total_line['columns'][index]['no_format'] = sum(currency.round(line['columns'][index]['no_format']) for line in lines[:-1] if report._get_model_info_from_id(line['id'])[0] == 'account.account')

        return [(0, line) for line in lines]

    def _caret_options_initializer(self):
        return {
            'trial_balance': [
                {'name': _("General Ledger"), 'action': 'caret_option_open_general_ledger'},
                {'name': _("Journal Items"), 'action': 'open_journal_items'},
            ],
        }

    def _get_column_group_creation_data(self, report, options, previous_options=None):
        """
        Return tuple of tuples containing a reference to the column_group creation function and on which side ('left' | 'right') of the report the column_group goes
        """
        return (
            (self._create_column_group_initial_balance, 'left'),
            (self._create_column_group_end_balance, 'right'),
        )

    @api.model
    def _create_and_append_column_group(self, report, options, header_name, forced_options, side_to_append, group_vals, exclude_initial_balance=False, append_col_groups=True):
        header_element = [{'name': header_name, 'forced_options': forced_options}]
        column_headers = [header_element, *options['column_headers'][1:]]
        column_group_vals = report._generate_columns_group_vals_recursively(column_headers, group_vals)

        if exclude_initial_balance:
            # This column group must not include initial balance; we use a special option key for that in general ledger
            for column_group in column_group_vals:
                column_group['forced_options']['general_ledger_strict_range'] = True

        columns, column_groups = report._build_columns_from_column_group_vals(forced_options, column_group_vals)

        side_to_append['column_headers'] += header_element
        if append_col_groups:
            side_to_append['column_groups'] |= column_groups
        side_to_append['columns'] += columns

    def _custom_options_initializer(self, report, options, previous_options):
        """ Modifies the provided options to add a column group for initial balance and end balance, as well as the appropriate columns.
        """
        default_group_vals = {'horizontal_groupby_element': {}, 'forced_options': {}}
        left_side = {'column_headers': [], 'column_groups': {}, 'columns': []}
        right_side = {'column_headers': [], 'column_groups': {}, 'columns': []}

        # Columns between initial and end balance must not include initial balance; we use a special option key for that in general ledger
        for column_group in options['column_groups'].values():
            column_group['forced_options']['general_ledger_strict_range'] = True

        if options.get('comparison') and not options['comparison'].get('periods'):
            options['comparison']['period_order'] = 'ascending'

        # Create column groups
        for function, side in self._get_column_group_creation_data(report, options, previous_options):
            function(report, options, previous_options, default_group_vals, left_side if side == 'left' else right_side)

        # Update options
        options['column_headers'][0] = left_side['column_headers'] + options['column_headers'][0] + right_side['column_headers']
        options['column_groups'].update(left_side['column_groups'])
        options['column_groups'].update(right_side['column_groups'])
        options['columns'] = left_side['columns'] + options['columns'] + right_side['columns']
        options['ignore_totals_below_sections'] = True  # So that GL does not compute them

        # All the periods displayed between initial and end balance need to use the same rates, so we manually change the period key.
        # account.report will then compute the currency table periods accordingly
        middle_periods_period_key = '_trial_balance_middle_periods'
        for col_group in options['column_groups'].values():
            col_group_date = col_group['forced_options'].get('date')
            if col_group_date:
                col_group_date['currency_table_period_key'] = middle_periods_period_key

        report._init_options_order_column(options, previous_options)

    def _custom_line_postprocessor(self, report, options, lines):
        # If the hierarchy is enabled, ensure to add the o_account_coa_column_contrast class to the hierarchy lines
        if options.get('hierarchy'):
            for line in lines:
                model, dummy = report._get_model_info_from_id(line['id'])
                if model == 'account.group':
                    line_classes = line.get('class', '')
                    line['class'] = line_classes + ' o_account_coa_column_contrast_hierarchy'

        return lines

    def _create_column_group_initial_balance(self, report, options, previous_options, default_group_vals, side_to_append):
        initial_balance_options = self.env['account.general.ledger.report.handler']._get_options_initial_balance(options)
        initial_forced_options = {
            'date': initial_balance_options['date'],
            'include_current_year_in_unaff_earnings': initial_balance_options['include_current_year_in_unaff_earnings'],
            'no_impact_on_currency_table': True,
        }

        self._create_and_append_column_group(
            report,
            options,
            _("Initial Balance"),
            initial_forced_options,
            side_to_append,
            default_group_vals,
        )

    def _create_column_group_end_balance(self, report, options, previous_options, default_group_vals, side_to_append):
        end_date_to = options['date']['date_to']
        end_date_from = options['comparison']['periods'][-1]['date_from'] if options.get('comparison', {}).get('periods') else options['date']['date_from']
        end_forced_options = {
            'date': report._get_dates_period(
                fields.Date.from_string(end_date_from),
                fields.Date.from_string(end_date_to),
                'range',
            ),
        }

        self._create_and_append_column_group(
            report,
            options,
            _("End Balance"),
            end_forced_options,
            side_to_append,
            default_group_vals,
            append_col_groups=False,
        )

        # We don't add end_column_groups on purpose: they shouldn't be computed, since we'll just sum the values of other groups in that one.
        # So, we don't want to run any SQL for it. We also force a dedicated column_group_key on the end columns, to better identify them.
        for column_data in side_to_append['columns'][-len(report.column_ids):]:
            column_data['column_group_key'] = TRIAL_BALANCE_END_COLUMN_GROUP_KEY