from dataclasses import dataclass

import gradio as gr
from graphql_calls import get_tag_commit_date, get_commits


@dataclass
class Contributions:
    additions: int
    deletions: int
    descriptions: list[str]


def get_release_notes(
    token: str,
    repo: str,
    tag: str,
    branch: str,
    contributor_treshold: int,
    ignore_dependabot: bool,
    ignore_direct: bool,
):
    date = get_tag_commit_date(token, repo, tag)
    commits = get_commits(token, repo, date)

    result = ""
    contributors = {}

    for commit in commits:
        if "Hugging Face" not in commit.user.organizations:
            if commit.user.name not in contributors:
                contributors[commit.user.name] = Contributions(
                    additions=commit.additions,
                    deletions=commit.deletions,
                    descriptions=[commit.message],
                )
            else:
                contributors[commit.user.name].additions += commit.additions
                contributors[commit.user.name].deletions += commit.deletions
                contributors[commit.user.name].descriptions += [commit.message]

        if "(#" in commit.message:
            if ignore_dependabot and commit.user.name == 'dependabot[bot]':
                continue

            split = commit.message.split("(#")
            message = split[0]
            number = split[1].strip(")")

            result += f"* {message} by @{commit.user.name} in #{number}\n"
        elif not ignore_direct:
            result += f"* {commit.message} by @{commit.user.name} (direct commit on {branch})\n"

    significant_contributors = {
        k: v for k, v in contributors.items() if (v.additions + v.deletions) > contributor_treshold
    }

    if len(significant_contributors):
        result += (
            "\n## Significant community contributions\n"
            "\nThe following contributors have made significant "
            "changes to the library over the last release:\n\n"
        )

        for significant_contributor, contributions in significant_contributors.items():
            result += f"* @{significant_contributor}\n"
            for description in contributions.descriptions:
                result += f"    * {description}\n"

    return result


demo = gr.Interface(
    fn=get_release_notes,
    inputs=[
        gr.inputs.Textbox(lines=1, placeholder="Your GitHub token"),
        gr.inputs.Textbox(
            lines=1, placeholder="Repository", default="huggingface/transformers"
        ),
        gr.inputs.Textbox(lines=1, placeholder="The tag from which to get commit"),
        gr.inputs.Textbox(
            lines=1,
            placeholder="The linear branch on which the new version tag will be added",
            default="main",
        ),
        gr.inputs.Slider(
            minimum=0,
            maximum=2000,
            default=500,
            label="Threshold for significant contributors",
        ),
        gr.inputs.Checkbox(label="Ignore dependabot commits"),
        gr.inputs.Checkbox(label="Ignore direct commits"),
    ],
    outputs="text",
)


if __name__ == "__main__":
    demo.launch()