# Copyright 2012-2018 Camptocamp SA
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from flectra import models, fields, api


class AccountAccountLine(models.Model):
    _inherit = 'account.move.line'
    # By convention added columns stats with gl_.
    gl_foreign_balance = fields.Float(string='Aggregated Amount currency')
    gl_balance = fields.Float(string='Aggregated Amount')
    gl_revaluated_balance = fields.Float(string='Revaluated Amount')
    gl_currency_rate = fields.Float(string='Currency rate')


class AccountAccount(models.Model):
    _inherit = 'account.account'

    currency_revaluation = fields.Boolean(
        string="Allow Currency revaluation",
        default=False,
    )

    _sql_mapping = {
        'balance': "COALESCE(SUM(debit),0) - COALESCE(SUM(credit), 0) as "
                   "balance",
        'debit': "COALESCE(SUM(debit), 0) as debit",
        'credit': "COALESCE(SUM(credit), 0) as credit",
        'foreign_balance': "COALESCE(SUM(amount_currency), 0) as foreign_"
                           "balance",
    }

    def init(self):
        # all receivable, payable and Bank and Cash accounts should
        # have currency_revaluation True by default
        res = super().init()
        accounts = self.env["account.account"].search([
            ('user_type_id.id', 'in', self._get_revaluation_account_types()),
            ('currency_revaluation', '=', False),
        ])
        accounts.write({
            'currency_revaluation': True,
        })
        return res

    def _get_revaluation_account_types(self):
        return [
            self.env.ref("account.data_account_type_receivable").id,
            self.env.ref("account.data_account_type_payable").id,
            self.env.ref("account.data_account_type_liquidity").id,
        ]

    @api.multi
    @api.onchange('user_type_id')
    def _onchange_user_type_id(self):
        revaluation_accounts = self._get_revaluation_account_types()
        for rec in self:
            if rec.user_type_id.id in revaluation_accounts:
                rec.currency_revaluation = True

    @api.multi
    def _revaluation_query(self, revaluation_date):

        tables, where_clause, where_clause_params = \
            self.env['account.move.line']._query_get()

        query = ("with amount as ( SELECT aml.account_id, aml.partner_id, "
                 "aml.currency_id, aml.debit, aml.credit, aml.amount_currency "
                 "FROM account_move_line aml LEFT JOIN "
                 "account_partial_reconcile aprc ON (aml.balance < 0 "
                 "AND aml.id = aprc.credit_move_id) LEFT JOIN "
                 "account_move_line amlcf ON (aml.balance < 0 "
                 "AND aprc.debit_move_id = amlcf.id "
                 "AND amlcf.date < %s ) LEFT JOIN "
                 "account_partial_reconcile aprd ON (aml.balance > 0 "
                 "AND aml.id = aprd.debit_move_id) LEFT JOIN "
                 "account_move_line amldf ON (aml.balance > 0 "
                 "AND aprd.credit_move_id = amldf.id "
                 "AND amldf.date < %s ) "
                 "WHERE aml.account_id IN %s "
                 "AND aml.date <= %s "
                 "AND aml.currency_id IS NOT NULL "
                 "GROUP BY aml.id "
                 "HAVING aml.full_reconcile_id IS NULL "
                 "OR (MAX(amldf.id) IS NULL AND MAX(amlcf.id) IS NULL)"
                 ") SELECT account_id as id, partner_id, currency_id, " +
                 ', '.join(self._sql_mapping.values()) +
                 " FROM amount " +
                 (("WHERE " + where_clause) if where_clause else " ") +
                 " GROUP BY account_id, currency_id, partner_id")

        params = []
        params.append(revaluation_date)
        params.append(revaluation_date)
        params.append(tuple(self.ids))
        params.append(revaluation_date)
        params += where_clause_params
        return query, params

    @api.multi
    def compute_revaluations(self, revaluation_date):
        accounts = {}
        # compute for each account the balance/debit/credit from the move lines
        query, params = self._revaluation_query(revaluation_date)
        self.env.cr.execute(query, params)

        lines = self.env.cr.dictfetchall()

        for line in lines:
            # generate a tree
            # - account_id
            # -- currency_id
            # --- partner_id
            # ----- balances
            account_id, currency_id, partner_id = \
                line['id'], line['currency_id'], line['partner_id']

            accounts.setdefault(account_id, {})
            accounts[account_id].setdefault(currency_id, {})
            accounts[account_id][currency_id].\
                setdefault(partner_id, {})
            accounts[account_id][currency_id][partner_id] = line

        return accounts