From 3062575fa76e5c2622e0c8567ce50ddf0e06ac75 Mon Sep 17 00:00:00 2001
From: gizmo385 <gizmo385@users.noreply.github.com>
Date: Fri, 21 Jun 2024 06:30:35 +0000
Subject: [PATCH] Add very basic WIP issues table

---
 lazy_github/ui/widgets/issues.py | 35 +++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/lazy_github/ui/widgets/issues.py b/lazy_github/ui/widgets/issues.py
index 944a43f..fb231e7 100644
--- a/lazy_github/ui/widgets/issues.py
+++ b/lazy_github/ui/widgets/issues.py
@@ -1,3 +1,6 @@
+from typing import Dict
+
+from github.Issue import Issue
 from textual.app import ComposeResult
 
 from lazy_github.lib.messages import RepoSelected
@@ -5,10 +8,40 @@
 
 
 class IssuesContainer(LazyGithubContainer):
+    issues: Dict[int, Issue] = {}
+    status_column_index = -1
+    number_column_index = -1
+    title_column_index = -1
+
     def compose(self) -> ComposeResult:
         self.border_title = "[3] Issues"
         yield LazyGithubDataTable(id="issues_table")
 
+    @property
+    def table(self) -> LazyGithubDataTable:
+        return self.query_one("#issues_table", LazyGithubDataTable)
+
+    def on_mount(self) -> None:
+        self.table.cursor_type = "row"
+        self.table.add_column("Status", key="status")
+        self.table.add_column("Number", key="number")
+        self.table.add_column("Author", key="author")
+        self.table.add_column("Title", key="title")
+
+        self.status_column_index = self.table.get_column_index("status")
+        self.number_column_index = self.table.get_column_index("number")
+        self.title_column_index = self.table.get_column_index("title")
+
     async def on_repo_selected(self, message: RepoSelected) -> None:
-        # TODO: Load the issues for the selected repo
         message.stop()
+        self.table.clear()
+        self.issues = {}
+        issues = message.repo.get_issues(state="all", sort="updated", direction="desc")
+        rows = []
+        for issue in issues:
+            # TODO: Currently, this also includes PRs because the Github API classifies all PRs as issues. The PyGithub
+            # library makes this even more problematic because there seemingly isn't a way to distinguish between an
+            # issue and a PR without making an additional API request. This is quite slow >:(
+            self.issues[issue.number] = issue
+            rows.append((issue.state, issue.number, issue.user.login, issue.title))
+        self.table.add_rows(rows)