#!/usr/bin/env python3

import argparse
import re
from pathlib import Path


PATTERN = r'R"JSON\((.*?)\)JSON"'


def use_uppercase(cpp_content: str) -> str:
    return cpp_content.replace('R"json(', 'R"JSON(').replace(')json"', ')JSON"')


def fix_json_style(cpp_content: str) -> str:
    def replace_json(match):
        raw_json = match.group(1)

        raw_json = (
            raw_json.replace(" :", ":")
            .replace(" ,", ",")
            .replace(" null", "null")
            .replace(':"', ': "')
            .replace(',"', ', "')
            .replace('":{', '": {')
            .replace('":[', '": [')
            .replace('":true', '": true')
            .replace('":false', '": false')
            .replace('":null', '": null')
        )
        for digit in range(10):
            raw_json = raw_json.replace(f'":{digit}', f'": {digit}')
        return f'R"JSON({raw_json})JSON"'

    return re.sub(PATTERN, replace_json, cpp_content, flags=re.DOTALL)


def fix_colon_spacing(cpp_content: str) -> str:
    def replace_json(match):
        raw_json = match.group(1)
        raw_json = re.sub(r'":\n\s*(\[|\{)', r'": \1', raw_json)
        return f'R"JSON({raw_json})JSON"'
    return re.sub(PATTERN, replace_json, cpp_content, flags=re.DOTALL)


def fix_indentation(cpp_content: str) -> str:
    if "JSON(" not in cpp_content:
        return cpp_content

    lines = cpp_content.splitlines()

    ends_with_newline = cpp_content.endswith('\n')

    def find_indentation(line: str) -> int:
        return len(line) - len(line.lstrip())

    for (line_num, (line, next_line)) in enumerate(zip(lines[:-1], lines[1:])):
        if "JSON(" in line and ")JSON" not in line:
            indent = find_indentation(line)
            next_indent = find_indentation(next_line)

            by_how_much = next_indent - (indent + 4)
            if by_how_much != 0:
                print(
                    f"Indentation error at line: {line_num + 2}: expected {indent + 4} spaces, found {next_indent} spaces"
                )

                for i in range(line_num + 1, len(lines)):
                    if ")JSON" in lines[i]:
                        lines[i] = " " * indent + lines[i].lstrip()
                        break
                    lines[i] = lines[i][by_how_much:] if by_how_much > 0 else " " * (-by_how_much) + lines[i]

    result = "\n".join(lines)

    if ends_with_newline:
        result += "\n"
    return result


def process_file(file_path: Path, dry_run: bool) -> bool:
    content = file_path.read_text(encoding="utf-8")

    new_content = content
    new_content = use_uppercase(new_content)
    new_content = fix_json_style(new_content)
    new_content = fix_colon_spacing(new_content)
    new_content = fix_indentation(new_content)

    if new_content != content:
        print(f"Processing file: {file_path}")
        if dry_run:
            print("Dry run: changes won't be written to the file.")
        else:
            print("Writing changes to file.")
            file_path.write_text(new_content, encoding="utf-8")
    return new_content == content


def main():
    parser = argparse.ArgumentParser(
        description="Fix JSON style in C++ files",
    )
    parser.add_argument(
        "--dry-run",
        default=False,
        action="store_true",
        help="Don't modify files, just print what would be changed",
    )
    parser.add_argument(
        "files",
        nargs="*",
        help="Specific files to process",
    )

    args = parser.parse_args()

    success = True
    for file in args.files:
        success = success and process_file(Path(file), dry_run=args.dry_run)
    if not success:
        print("Errors occurred while processing files.")
        exit(1)


if __name__ == "__main__":
    main()