Add Assignees, Reviewers and ReviewersTeams (#214)

nobe4 · Nov 12, 2024 · 67f0927 · 67f0927
1 parent 0e84be6
commit 67f0927
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/internal/gh/enrichments.go b/internal/gh/enrichments.go
@@ -10,9 +10,12 @@ import (
 )
 
 type ThreadExtra struct {
-	User    notifications.User `json:"user"`
-	State   string             `json:"state"`
-	HtmlUrl string             `json:"html_url"`
+	User           notifications.User   `json:"user"`
+	State          string               `json:"state"`
+	HtmlUrl        string               `json:"html_url"`
+	Assignees      []notifications.User `json:"assignees"`
+	Reviewers      []notifications.User `json:"requested_reviewers"`
+	ReviewersTeams []notifications.Team `json:"requested_teams"`
 }
 
 func (c *Client) Enrich(n *notifications.Notification) error {
@@ -27,6 +30,9 @@ func (c *Client) Enrich(n *notifications.Notification) error {
 	n.Author = threadExtra.User
 	n.Subject.State = threadExtra.State
 	n.Subject.HtmlUrl = threadExtra.HtmlUrl
+	n.Assignees = threadExtra.Assignees
+	n.Reviewers = threadExtra.Reviewers
+	n.ReviewersTeams = threadExtra.ReviewersTeams
 
 	LastCommentor, err := c.getLastCommentor(n)
 	if err != nil {

diff --git a/internal/notifications/notifications.go b/internal/notifications/notifications.go
@@ -27,8 +27,11 @@ type Notification struct {
 	Subject    Subject    `json:"subject"`
 
 	// Enriched API fields
-	Author          User `json:"author"`
-	LatestCommentor User `json:"latest_commentor"`
+	Author          User   `json:"author"`
+	LatestCommentor User   `json:"latest_commentor"`
+	Assignees       []User `json:"assignees"`
+	Reviewers       []User `json:"requested_reviewers"`
+	ReviewersTeams  []Team `json:"requested_teams"`
 
 	// gh-not specific fields
 	// Those fields are not part of the GitHub API and will persist between
@@ -82,6 +85,11 @@ type User struct {
 	Type  string `json:"type"`
 }
 
+type Team struct {
+	Name string `json:"name"`
+	Id   uint   `json:"id"`
+}
+
 func (n Notifications) Equal(others Notifications) bool {
 	if len(n) != len(others) {
 		return false