import pathlib
import shutil

import click

from odoo_openupgrade_wizard.cli.cli_options import database_option_required
from odoo_openupgrade_wizard.tools.tools_postgres import (
    check_db_exist,
    execute_pg_dump,
)
from odoo_openupgrade_wizard.tools.tools_system import dump_filestore


@click.command()
@database_option_required
@click.option(
    "--database-path",
    type=click.Path(writable=True, resolve_path=True),
    required=True,
    help="Destination path for the database dump relative to the project "
    "directory.",
)
@click.option(
    "--database-format",
    type=click.Choice(("p", "c", "d", "t")),
    default="c",
    help="Database format (see pg_dump options): plain SQL (p), "
    "custom format compressed (c), directory (d), or tar file (t).",
)
@click.option(
    "--filestore-path",
    type=click.Path(writable=True, resolve_path=True),
    required=True,
    help="Destination path for the filestore backup.",
)
@click.option(
    "--filestore-format",
    type=click.Choice(("d", "t", "tgz")),
    default="tgz",
    help="Filestore format: directory (d), tar file (t), "
    "or tar file compressed with gzip (tgz)",
)
@click.option(
    "--force",
    is_flag=True,
    default=False,
    help="Overwrite files if they already exist.",
)
@click.pass_context
def dumpdb(
    ctx,
    database,
    database_path,
    database_format,
    filestore_path,
    filestore_format,
    force,
):
    """Create a dump of an Odoo database and filestore.

    Creates a backup of the selected Odoo database and
    its associated filestore. Both output locations must
    reside inside the project directory. Existing files
    will be overwritten if --force is specified.
    """

    database_path = pathlib.Path(database_path)
    filestore_path = pathlib.Path(filestore_path)

    # Check that database_path is inside the env_folder_path
    absolute_database_path = database_path.absolute()
    absolute_env_folder_path = ctx.obj["env_folder_path"].resolve().absolute()
    if not str(absolute_database_path).startswith(
        str(absolute_env_folder_path)
    ):
        ctx.fail(
            "database-path should be inside the project path to allow "
            "PostgreSQL to write to it."
        )

    # Fail if dumps already exists and force argument not given
    # Remove file if already exists and force is given
    if not force and database_path.exists():
        ctx.fail(f"{database_path} exists, use --force to overwrite it.")
    elif force and database_path.exists():
        if database_path.is_dir():
            shutil.rmtree(database_path)
        else:
            database_path.unlink()

    if not force and filestore_path.exists():
        ctx.fail(f"{filestore_path} exists, use --force to overwrite it.")
    elif force and filestore_path.exists():
        if filestore_path.is_dir():
            shutil.rmtree(filestore_path)
        else:
            filestore_path.unlink()

    # Normalise database_path
    database_path = absolute_database_path.relative_to(
        absolute_env_folder_path
    )

    check_db_exist(ctx, database, raise_exception=True)

    # dump the database
    output = execute_pg_dump(
        ctx,
        database=database,
        dumpformat=database_format,
        filename=str(database_path),
    )
    if output:
        click.echo(output)

    # dump the filestore
    dump_filestore(
        ctx,
        database=database,
        destpath=filestore_path,
        copyformat=filestore_format,
    )