import importlib
import inspect
from pathlib import Path
import re

from pygments.lexers import LEXERS

for class_name, (module_name, _, _, _, _) in LEXERS.items():
    mod = importlib.import_module(module_name)
    mod = importlib.reload(mod)
    cls = getattr(mod, class_name)

    file = inspect.getsourcefile(cls)
    lines, first_line_index = inspect.getsourcelines(cls)
    first_line_index -= 1 # make it zero-based
    end_line_index = first_line_index + len(lines)

    docstring_finished = False
    for version_added_index, version_added_line in enumerate(lines):
        if m := re.match(r"    .. versionadded:: (.*)", version_added_line):
            version_added = m.group(1)
            cursor = version_added_index
            del lines[version_added_index]
            cursor -= 1
            if not lines[version_added_index-1].strip():
                del lines[version_added_index-1]
                cursor -= 1
            break
    else:
        version_added = ''
        assert lines[1].startswith('    """'), f"{file}:{first_line_index}"
        if lines[1].endswith('"""\n'):
            docstring_finished = True
        cursor = 2

    if not docstring_finished:
        for i, line in enumerate(lines[cursor:], cursor):
            if line == '    """\n':
                cursor = i+1
                break
        else:
            assert False, f"{file}:{first_line_index}"

    unlocked = False
    for insertion_index, line in enumerate(lines[cursor:], cursor):
        if re.match("    (name|url|aliases|filenames|mimetypes) ?=", line):
            unlocked = True
        elif unlocked and not line.startswith("     ") and not re.match(r"    (\[|\])", line):
            break
    else:
        insertion_index += 1

    new_version_added_line = f"    version_added = {version_added!r}\n"
    lines.insert(insertion_index, new_version_added_line)

    all_lines = Path(file).read_text(encoding="utf-8").splitlines(keepends=True)
    all_lines[first_line_index:end_line_index] = lines
    Path(file).write_text(''.join(all_lines), encoding="utf-8")