import re
from asyncio import sleep

from github import Github, InputGitAuthor, AuthenticatedUser
import datetime

from download_manager import DownloadManager
from make_bar_graph import build_graph


class LinesOfCode:

    def __init__(self, user: AuthenticatedUser, ghtoken, repositoryData, ignored_repos):
        self.g = Github(ghtoken)
        self.user = user
        self.repositoryData = repositoryData
        self.ignored_repos = ignored_repos

    async def calculateLoc(self):
        result = self.repositoryData
        yearly_data = {}
        total = len(result['data']['user']['repositories']['edges'])
        for ind, repo in enumerate(result['data']['user']['repositories']['edges']):
            if repo['node']['name'] not in self.ignored_repos:
                print(f"{ind}/{total}", "Retrieving repo:", repo['node']["owner"]["login"], repo['node']['name'])
                await self.getCommitStat(repo['node'], yearly_data)
                await sleep(0.7)
        return yearly_data

    async def plotLoc(self, yearly_data):
        await build_graph(yearly_data)
        self.pushChart()

    def getQuarter(self, timeStamp):
        month = datetime.datetime.fromisoformat(timeStamp).month
        if month >= 1 and month <= 3:
            return 1
        elif month >= 4 and month <= 6:
            return 2
        elif month >= 7 and month <= 9:
            return 3
        elif month >= 10 and month <= 12:
            return 4

    async def getCommitStat(self, repoDetails, yearly_data):
        commit_data = await DownloadManager.get_remote_graphql("repository_commit_list", owner=repoDetails["owner"]["login"], name=repoDetails['name'], id=self.user.node_id)

        if commit_data["data"]["repository"] is None:
            print("\tSkipping:", repoDetails['name'])
            return

        for commit in [commit["node"] for branch in commit_data["data"]["repository"]["refs"]["edges"] for commit in branch["node"]["target"]["history"]["edges"]]:
            date = re.search(r'\d+-\d+-\d+', commit["committedDate"]).group(0)
            curr_year = datetime.datetime.fromisoformat(date).year
            quarter = self.getQuarter(date)

            if repoDetails['primaryLanguage'] is not None:
                if curr_year not in yearly_data:
                    yearly_data[curr_year] = {}
                if quarter not in yearly_data[curr_year]:
                    yearly_data[curr_year][quarter] = {}
                if repoDetails['primaryLanguage']['name'] not in yearly_data[curr_year][quarter]:
                    yearly_data[curr_year][quarter][repoDetails['primaryLanguage']['name']] = 0
                yearly_data[curr_year][quarter][repoDetails['primaryLanguage']['name']] += (commit["additions"] - commit["deletions"])


    def pushChart(self):
        repo = self.g.get_repo(f"{self.user.login}/{self.user.login}")
        committer = InputGitAuthor('readme-bot', '41898282+github-actions[bot]@users.noreply.github.com')
        with open('bar_graph.png', 'rb') as input_file:
            data = input_file.read()
        try:
            contents = repo.get_contents("charts/bar_graph.png")
            repo.update_file(contents.path, "Charts Updated", data, contents.sha, committer=committer)
        except Exception as e:
            repo.create_file("charts/bar_graph.png", "Charts Added", data, committer=committer)