from os.path import exists
from shutil import copytree, rmtree

from flask_admin import __file__ as flaskAdminPath


def copyAdminTemplates():
    src = flaskAdminPath.removesuffix("__init__.py") + "templates/bootstrap4/admin"
    if not exists(src):
        print("Templates could not be found in", src)
        print("You can also copy them manually.")
        return False

    dist = "advlabdb/templates/admin"
    if exists(dist):
        while True:
            ans = input(
                f"The directory {dist} already exists. Enter 'o' to override it and update the templates or enter 's' to stop the process: "
            ).lower()
            if ans == "s":
                print("Process stopped!")
                return False
            elif ans == "o":
                break
        rmtree(dist)
        print("Old templates deleted!")

    copytree(src, dist)
    print("Copied", src, "->", dist)

    print(
        f"""
        _________
        | WARNING
        | -------
        |
        | You might have to edit the file {dist}/base.html by adding nav in the following way:
        | This line:\t<ul class="navbar-nav mr-auto">
        | Becomes:\t<ul class="nav navbar-nav mr-auto">
        |
        | This will prevent the navigation bar from expanding such that some elements can not be seen.
        | Refer to this pull request: https://github.com/flask-admin/flask-admin/pull/2233
        |
        | If the above pull request is merged and flask-admin is on a new release after the merge,
        | then this step is not needed.
        _________
        """
    )

    return True


if __name__ == "__main__":
    if not copyAdminTemplates():
        print("Did not copy!")