import codecs
import json
import os
import re
from pathlib import Path
from github import Github

CURRENT_FILE = Path(__file__)
ROOT = CURRENT_FILE.parents[1]
GITHUB_TOKEN = os.getenv("GITHUB_TOKEN", None)

# dictionary declared here to be exported to post_gen_project, for faster license access
titles_dict = {}

def main() -> None:
    """
    Script entry point.
    """
    repo = Github(login_or_token=GITHUB_TOKEN).get_repo("github/choosealicense.com")
    license_dir = ROOT / "{{cookiecutter.project_slug}}" / "licenses"
    license_dir.mkdir(exist_ok=True)
    global titles_dict
    for file in repo.get_contents("_licenses", "gh-pages"):
        content = codecs.decode(file.decoded_content)
        # make below line into a dictionary mapping to filename
        titles_dict[content.split("\n", maxsplit=2)[1].replace("title: ", "")] = file.name
        #titles.append(content.split("\n", maxsplit=2)[1].replace("title: ", ""))
        path = license_dir / file.name
        if not path.is_file():
            path.touch()
            (license_dir / file.name).write_text(replace_content_options(content))
    # Put "Not open source" at front so people know it's an option
    front_options = [
        "Not open source",
        "MIT License",
        'BSD 3-Clause "New" or "Revised" License',
        "GNU General Public License v3.0",
        "Apache License 2.0",
    ]
    # update to iterate through dictionary
    titles = [x for x in sorted(titles_dict.keys()) if x not in front_options]
    update_cookiecutter(front_options + titles)


year = (re.compile(r"\[year]"), "{% now 'utc', '%Y' %}")
email = (re.compile(r"\[email]"), "{{ cookiecutter.email }}")
fullname = (re.compile(r"\[fullname]"), "{{ cookiecutter.author_name }}")
project = (re.compile(r"\[project]"), "{{ cookiecutter.project_name }}")
projecturl = (re.compile(r"\[projecturl]"), "{{ cookiecutter.domain_name }}")


def replace_content_options(content) -> str:
    for compiled, replace in (year, email, fullname, project, projecturl):
        content = compiled.sub(replace, content)
    return content


def update_cookiecutter(titles: list):
    with open("cookiecutter.json") as f:
        data = json.load(f)
    data["open_source_license"] = titles
    with open("cookiecutter.json", "wt") as f:
        json.dump(data, f, indent=2)


if __name__ == "__main__":
    main()