from dataclasses import dataclass
import os
import gradio as gr
from graphql_calls import get_tag_commit_date, get_commits
from github import fetch_github_info


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


def get_release_notes(
    tag: str,
    branch: str,
    contributor_treshhold: int,
    ignore_dependabot: bool,
    ignore_direct: bool,
):
    try:
        token = os.getenv("GITHUB_TOKEN")
        repo = "huggingface/diffusers"
        date = get_tag_commit_date(token, repo, tag)
        commits = get_commits(token, repo, branch, date)
    except ValueError as e:
        return str(e)

    result = ""
    contributors = {}

    for commit in commits[::-1]:
        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_treshhold and k != "<NOT FOUND>"
    }

    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


article = """
Mostly copied from [lysandre/github-release](https://huggingface.co/spaces/lysandre/github-release) but suited for
Diffusers.
"""
github_info = fetch_github_info("https://github.com/huggingface/diffusers")

demo = gr.Interface(
    title="Automatic release notes for 🤗 Diffusers",
    article=article,
    description="**See instructions below the form.**",
    fn=get_release_notes,
    inputs=[
        gr.components.Textbox(
            lines=1,
            value=github_info["second_last_tag"] if github_info else None,
            placeholder="The tag from which to get commit",
        ),
        gr.components.Textbox(
            lines=1,
            placeholder="The linear branch on which the new version tag will be added",
            value=github_info["latest_release_branch"] if github_info else None,
        ),
        gr.components.Slider(
            minimum=0,
            maximum=2000,
            value=500,
            label="Threshold for significant contributors",
        ),
        gr.components.Checkbox(label="Ignore dependabot commits"),
        gr.components.Checkbox(label="Ignore direct commits"),
    ],
    outputs="text",
    allow_flagging="never"
)


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