WIP pulling some notifications

nate smith · nate smith · commit f6d0cb92d572 · 2022-02-24T13:51:09.000-06:00
diff --git a/pkg/cmd/status/status.go b/pkg/cmd/status/status.go
@@ -1,20 +1,27 @@
 package status
 
 import (
+	"fmt"
+	"net/http"
+	"net/url"
+
 	"github.com/cli/cli/v2/api"
+	"github.com/cli/cli/v2/internal/ghinstance"
 	"github.com/cli/cli/v2/internal/ghrepo"
 	"github.com/cli/cli/v2/pkg/cmdutil"
 	"github.com/spf13/cobra"
 )
 
 type StatusOptions struct {
 	BaseRepo        func() (ghrepo.Interface, error)
+	HttpClient      func() (*http.Client, error)
 	HasRepoOverride bool
 	Org             string
 }
 
 func NewCmdStatus(f *cmdutil.Factory, runF func(*StatusOptions) error) *cobra.Command {
 	opts := &StatusOptions{}
+	opts.HttpClient = f.HttpClient
 	cmd := &cobra.Command{
 		Use:   "status",
 		Short: "Print information about relevant issues, pull requests, and notifications across repositories",
@@ -36,19 +43,102 @@ func NewCmdStatus(f *cmdutil.Factory, runF func(*StatusOptions) error) *cobra.Co
 	return cmd
 }
 
-type gqlResults struct {
-	ReviewRequests []api.PullRequest
-	AssignedPRs    []api.PullRequest
-	AssignedIssues []api.PullRequest
+/*
+• "repo activity"
+	• using notification API
+	• new issues
+	• new prs
+	• comments
+• mentions
+	• using notifications API
+• review requests
+	• using search API
+• pr assignments
+	• using search API
+• issue assignments
+	• using search API
+*/
+
+type Notification struct {
+	Reason  string
+	Subject struct {
+		Title            string
+		LatestCommentURL string `json:"latest_comment_url"`
+		URL              string
+		Type             string
+	}
+	Repository struct {
+		FullName string `json:"full_name"`
+	}
+}
+
+func getNotifications(client *http.Client) ([]Notification, error) {
+	apiClient := api.NewClientFromHTTP(client)
+	query := url.Values{}
+	query.Add("per_page", "100")
+	// TODO might want to get multiple pages since I'm sorting the results into buckets
+	p := fmt.Sprintf("notifications?%s", query.Encode())
+	var resp []Notification
+	err := apiClient.REST(ghinstance.Default(), "GET", p, nil, &resp)
+	if err != nil {
+		return nil, fmt.Errorf("could not get notifications: %w", err)
+	}
+
+	return resp, nil
 }
 
 func statusRun(opts *StatusOptions) error {
+	client, err := opts.HttpClient()
+	if err != nil {
+		return fmt.Errorf("could not create client: %w", err)
+	}
+	ns, err := getNotifications(client)
+	if err != nil {
+		return err
+	}
+
+	mentions := []Notification{}
+	newIssues := []Notification{}
+	newPRs := []Notification{}
+	comments := []Notification{}
+
+	for _, n := range ns {
+		if n.Reason == "mention" {
+			mentions = append(mentions, n)
+			continue
+		}
+		if n.Subject.LatestCommentURL == n.Subject.URL {
+			if n.Subject.Type == "PullRequest" {
+				newPRs = append(newPRs, n)
+			} else if n.Subject.Type == "Issue" {
+				newIssues = append(newIssues, n)
+			} else {
+				// TODO i donno
+				fmt.Printf("DBG %#v\n", n)
+			}
+		} else {
+			comments = append(comments, n)
+		}
+	}
+
+	// this is picking up stuff like team mentions. i should handle those explicitly.
+
+	fmt.Println("MENTIONS")
+	fmt.Printf("%#v\n", mentions)
+	fmt.Println("NEW PRs")
+	fmt.Printf("%#v\n", newPRs)
+	fmt.Println("NEW ISSUES")
+	fmt.Printf("%#v\n", newIssues)
+	fmt.Println("COMMENTS")
+	fmt.Printf("%#v\n", comments)
+
 	// should i attempt to shoehorn all of this into a single giant graphql
 	// query? i guess everything that is in graphql should be trated that way.
 
 	// TODO review requests -- GQL search
 	// TODO pr assignments -- GQL search
 	// TODO issue assignments -- GQL search
+	// TODO discussions -- GQL search
 	// TODO mentions -- GQL, apparently. can this include discussions? continue to study mislav's extension
 
 	// TODO figure out if this could work:
