#!/usr/bin/python3 -ttu
import argparse
import re
import sys

from typing import List, Set


BASE_CLASSES = ("RESTObject", "RESTObjectList", "RESTManager")
MIXIN_CLASSES = (
    "AccessRequestMixin",
    "BadgeRenderMixin",
    "CreateMixin",
    "CRUDMixin",
    "DeleteMixin",
    "DownloadMixin",
    "GetMixin",
    "GetWithoutIdMixin",
    "ListMixin",
    "NoUpdateMixin",
    "ObjectDeleteMixin",
    "ParticipantsMixin",
    "RefreshMixin",
    "RetrieveMixin",
    "SaveMixin",
    "SetMixin",
    "SubscribableMixin",
    "TimeTrackingMixin",
    "TodoMixin",
    "UpdateMixin",
    "UserAgentDetailMixin",
)


def main() -> int:
    args = parse_args()

    fixup_file(filename=args.filename)
    return 0


def fixup_file(*, filename: str):
    content = []
    with open(filename, "r") as in_file:
        for line in in_file.readlines():
            line = line.rstrip()
            content.append(line)

    base_classes = scan_classes(classes=BASE_CLASSES, content=content)
    mixin_classes = scan_classes(classes=MIXIN_CLASSES, content=content)

    output = []
    for line in content:

        if line.startswith("from gitlab.base import REST"):
            print("This file seems to already have been processed. Exiting")
            return
        if line.startswith("from gitlab.mixins import ("):
            print("This file seems to already have been processed. Exiting")
            return

        if line.startswith("from gitlab.base import"):
            if not base_classes:
                continue
            output.append(
                "from gitlab.base import {}".format(",".join(sorted(base_classes)))
            )
            continue

        if line.startswith("from gitlab.mixins import"):
            if not mixin_classes:
                continue
            output.append(
                "from gitlab.mixins import {}".format(",".join(sorted(mixin_classes)))
            )
            continue

        output.append(line)

    with open(filename, "w") as out_file:
        for line in output:
            print(line, file=out_file)


def scan_classes(*, classes: List[str], content: List[str]) -> Set[str]:
    result = set()
    for line in content:
        for class_name in classes:
            match = re.search(r"\b" + class_name + r"\b", line)
            if match:
                result.add(class_name)
    return result


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser()

    parser.add_argument("-f", "--filename", required=True)

    args = parser.parse_args()
    return args


if "__main__" == __name__:
    sys.exit(main())