# Copyright 2019 Apps2GROW - Henrik Norlin
# Copyright 2019 Tecnativa - Pedro M. Baeza
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from openupgradelib import openupgrade

_model_renames = [
    ("account.asset.category", "account.asset.profile"),
    ("account.asset.depreciation.line", "account.asset.line"),
    ("account.asset.asset", "account.asset"),
]

_table_renames = [
    (old.replace(".", "_"), new.replace(".", "_")) for (old, new) in _model_renames
]

_column_copies = {
    "account_asset": [("method_number", None, None), ("method_time", None, None)],
    "account_asset_profile": [
        ("method_number", None, None),
        ("method_time", None, None),
    ],
}

_column_renames = {
    "account_asset": [("method_period", None)],
    "account_asset_profile": [("method_period", None)],
}

_field_renames = [
    ("account.asset", "account_asset", "category_id", "profile_id"),
    ("account.asset", "account_asset", "currency_id", "company_currency_id"),
    ("account.asset", "account_asset", "date", "date_start"),
    ("account.asset", "account_asset", "value", "purchase_value"),
    ("account.asset.line", "account_asset_line", "depreciation_date", "line_date"),
    (
        "account.asset.profile",
        "account_asset_profile",
        "account_depreciation_expense_id",
        "account_expense_depreciation_id",
    ),
    (
        "account.invoice.line",
        "account_invoice_line",
        "asset_category_id",
        "asset_profile_id",
    ),
]


def handle_account_asset_disposal_migration(env):
    """Take care of potentially installed `account_asset_disposal` module.

    In this phase we rename stuff for adapting to the new data structure.
    """
    cr = env.cr
    if not openupgrade.column_exists(cr, "account_asset", "disposal_move_id"):
        return
    openupgrade.copy_columns(cr, {"account_asset": [("state", None, None)]})
    openupgrade.rename_columns(cr, {"account_asset": [("disposal_move_id", None)]})
    openupgrade.map_values(
        cr,
        openupgrade.get_legacy_name("state"),
        "state",
        [("disposed", "removed")],
        table="account_asset",
    )
    openupgrade.rename_fields(
        env,
        [
            ("account.asset", "account_asset", "disposal_date", "date_remove"),
            (
                "account.asset.profile",
                "account_asset_profile",
                "account_loss_id",
                "account_residual_value_id",
            ),
        ],
    )


@openupgrade.migrate()
def migrate(env, version):
    cr = env.cr
    if openupgrade.table_exists(cr, "account_asset_asset"):
        # `account_asset` module was installed in v11
        if openupgrade.table_exists(cr, "account_asset"):
            # `account_asset_management` module also installed in v11
            # TODO: Merge in existing tables assets if both module installed
            pass
        else:
            openupgrade.rename_models(cr, _model_renames)
            openupgrade.rename_tables(cr, _table_renames)
            openupgrade.copy_columns(cr, _column_copies)
            openupgrade.rename_columns(cr, _column_renames)
            openupgrade.rename_fields(env, _field_renames)
            handle_account_asset_disposal_migration(env)
            if openupgrade.column_exists(cr, "account_asset", "analytic_account_id"):
                # account_asset_analytic module in OCA/account_analytic
                openupgrade.rename_fields(
                    env,
                    [
                        (
                            "account.asset",
                            "account_asset",
                            "analytic_account_id",
                            "account_analytic_id",
                        )
                    ],
                )