#!/usr/bin/env python3

import re
import subprocess
from pathlib import Path


GIT_DEFAULT_BRANCH = "main"
GIT_TRACKED_FILES = list(
    map(
        Path,
        filter(
            None,
            map(
                str.strip,
                subprocess.check_output(
                    [
                        "git",
                        "ls-tree",
                        "-r",
                        "--name-only",
                        GIT_DEFAULT_BRANCH,
                    ],
                    text=True,
                ).splitlines(),
            ),
        ),
    )
)


COPYRIGHT_PATTERN = re.compile(
    r'''
    """\n
    Copyright
    .*?         # non-greedy match
    License\.\n
    """
    ''',
    flags=re.VERBOSE | re.IGNORECASE | re.DOTALL,
)


def replace_quoted_string_to_comments(match):
    string = match.group().replace('"""', "").strip()
    string = "\n".join(f"# {line}".strip() for line in string.splitlines())
    string += "\n# " + "=" * 78
    return string


for file in GIT_TRACKED_FILES:
    try:
        content = file.read_text(encoding="utf-8")
    except UnicodeError:
        continue  # non-text file

    updated_content = COPYRIGHT_PATTERN.sub(replace_quoted_string_to_comments, content)
    file.write_text(updated_content, encoding="utf-8")