#!/usr/bin/env python3


import json


from commands.base import BaseCommand
from gitlab import get_mrs_for_branch, get_comments


class Comments(BaseCommand):
    width = 5

    @staticmethod
    def sort_comments(comments):
        result = {}

        for comment in comments:
            author = comment["author"]["username"], comment["author"]["name"]
            if author not in result:
                result[author] = list()

            result[author].append(comment)

        return result

    @staticmethod
    def print_comments(comments):
        comments = Comments.sort_comments(comments)

        for author in comments:
            login, name = author
            print(f"***** {name} ({login}) *****")

            for comment in comments[author]:
                if comment["system"]:
                    continue

                if comment["type"] == "DiffNote":
                    body = comment["body"].replace(
                        "\n", "\n" + " " * (Comments.width + 2)
                    )
                    print(
                        f"""{f'L{comment["position"]["new_line"]}':>{Comments.width}}: {body}"""
                    )
                else:
                    print(f"""[{comment["created_at"]}]\n{comment["body"]}""")
            print()
        print()

    def exec(self, login: str, submission: str):
        mrs = get_mrs_for_branch(self.branch)
        if not mrs:
            return

        iid = mrs[0]["iid"]
        comments = get_comments(iid)

        print(f"# {self.branch}")
        self.print_comments(comments)
        print(f"# {self.branch}")