import argparse
import os
import re
import shutil

from pip._internal.operations import freeze as pip

import cms
import example


def main():
    parser = argparse.ArgumentParser(description="SimpleCMS")
    parser.add_argument("project_name", nargs="?", default=".")
    project_name = parser.parse_args().project_name
    if project_name == ".":
        project_dir = os.getcwd()
        project_name = os.path.basename(project_dir)
    else:
        project_dir = os.path.join(os.getcwd(), project_name)
    if re.match(r"^\w+$", project_name):
        if (
            input(f"Do you want to create a new project in `{project_dir}` ?\033[1D")
            in "Yy"
        ):
            create_project(project_name, project_dir)
    else:
        print(f"Invalid project name: {project_name}")


def create_project(project_name, project_dir):
    os.makedirs(project_dir, exist_ok=True)
    with open(os.path.join(project_dir, "requirements.txt"), "w") as f:
        for line in pip.freeze():
            if "django_simplecms" in line:
                line = f"django-simplecms=={cms.__version__}"
            print(line, file=f)

    shutil.copytree(
        os.path.dirname(example.__file__),
        os.path.join(project_dir, project_name),
        dirs_exist_ok=True,
    )
    with open(
        os.open(
            os.path.join(project_dir, "manage.py"), os.O_CREAT | os.O_WRONLY, 0o755
        ),
        "w",
    ) as f:
        print(
            "#!/usr/bin/env python",
            "import os, sys",
            f"os.environ.setdefault('DJANGO_SETTINGS_MODULE', '{project_name}.settings')",
            "from django.core.management import execute_from_command_line",
            "execute_from_command_line(sys.argv)",
            sep="\n",
            file=f,
        )
    with open(os.path.join(project_dir, project_name, "wsgi.py"), "w") as f:
        print(
            "import os",
            "from django.core.wsgi import get_wsgi_application",
            f"os.environ.setdefault('DJANGO_SETTINGS_MODULE', '{project_name}.settings')",
            "application = get_wsgi_application()",
            sep="\n",
            file=f,
        )
    with open(os.path.join(project_dir, ".gitignore"), "w") as f:
        print("*.pyc\n__pycache__/", file=f)

    print(
        f'Successfully created project "{project_name}"',
        "",
        "Things to do next:",
        "- create a database",
        "- ./manage.py makemigrations",
        "- ./manage.py migrate",
        "- ./manage.py createsuperuser",
        "- ./manage.py runserver",
        sep="\n",
    )


if __name__ == "__main__":
    main()