from rich.table import Table
from rich.prompt import Prompt
from rich import print
from api import API
import httpx


class GHub:
    user = None
    repos = None

    def load_user(self, user_name: str) -> bool:
        self.user_name = user_name
        self.api = API()
        try:
            self.user = self.api.get(f"/users/{user_name}")
            self.repos = self.api.get_with_pagination(
                f"https://api.github.com/users/{user_name}/repos"
            )
            return True
        except httpx.HTTPStatusError as e:
            print(e)
            return False

    def repos_trimmed(self) -> list:
        for repo in self.repos:
            yield [
                str(f)
                for f in [
                    repo["name"],
                    repo["description"],
                    repo["stargazers_count"],
                    repo["watchers_count"],
                    repo["forks_count"],
                    repo["open_issues_count"],
                ]
            ]

    def repos_table(self) -> Table:
        table = Table(show_header=True, header_style="bold magenta")
        table.add_column("Name")
        table.add_column("Description")
        table.add_column("Stargazers")
        table.add_column("Watchers")
        table.add_column("Forks")
        table.add_column("Open Issues")
        for repo in self.repos_trimmed():
            table.add_row(*repo)
        return table