WIP Add draft state, iterate on UI design

Author: stefanhaller

pkg/commands/git_commands/github.go

@@ -69,6 +69,7 @@ type PullRequestNode struct {
 	Url                 string                `json:"url"`
 	HeadRepositoryOwner GithubRepositoryOwner `json:"headRepositoryOwner"`
 	State               string                `json:"state"`
+	IsDraft             bool                  `json:"isDraft"`
 }
 
 type GithubRepositoryOwner struct {
@@ -90,6 +91,7 @@ func fetchPullRequestsQuery(branches []string, owner string, repo string) string
           state
           number
           url
+		  isDraft
           headRepositoryOwner {
             login
           }
@@ -212,7 +214,7 @@ func (self *GitHubCommands) FetchRecentPRsAux(repoOwner string, repoName string,
 				HeadRefName: node.HeadRefName,
 				Number:      node.Number,
 				Title:       node.Title,
-				State:       node.State,
+				State:       lo.Ternary(node.IsDraft && node.State != "CLOSED", "DRAFT", node.State),
 				Url:         node.Url,
 				HeadRepositoryOwner: models.GithubRepositoryOwner{
 					Login: node.HeadRepositoryOwner.Login,

pkg/commands/models/github.go

@@ -5,7 +5,7 @@ type GithubPullRequest struct {
 	HeadRefName         string                `json:"headRefName"`
 	Number              int                   `json:"number"`
 	Title               string                `json:"title"`
-	State               string                `json:"state"` // "MERGED", "OPEN", "CLOSED"
+	State               string                `json:"state"` // "MERGED", "OPEN", "CLOSED", "DRAFT"
 	Url                 string                `json:"url"`
 	HeadRepositoryOwner GithubRepositoryOwner `json:"headRepositoryOwner"`
 }

pkg/gui/controllers/branches_controller.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"fmt"
 	"strconv"
+	"strings"
 
 	"github.com/jesseduffield/gocui"
 	"github.com/jesseduffield/lazygit/pkg/commands/git_commands"
@@ -206,10 +207,12 @@ func (self *BranchesController) GetOnRenderToMain() func() {
 				)
 
 				if pr, ok := prs[branch.Name]; ok {
-					ptyTask.Prefix = fmt.Sprintf("%s %s (%s)\n\n",
+					ptyTask.Prefix = fmt.Sprintf("%s %s  %s\n",
 						coloredPrNumber(pr),
-						style.FgYellow.Sprint(style.PrintHyperlink(pr.Title, pr.Url)),
-						pr.State)
+						style.PrintHyperlink(pr.Title, pr.Url),
+						coloredStateText(pr.State))
+					ptyTask.Prefix += strings.Repeat("─", self.c.Contexts().Normal.GetView().InnerWidth()) + "\n"
+
 				}
 			}
 
@@ -224,6 +227,26 @@ func (self *BranchesController) GetOnRenderToMain() func() {
 	}
 }
 
+func stateText(state string) string {
+	// TODO: add icons only if nerd fonts are used
+	switch state {
+	case "OPEN":
+		return " Open"
+	case "CLOSED":
+		return " Closed"
+	case "MERGED":
+		return " Merged"
+	case "DRAFT":
+		return " Draft"
+	default:
+		return ""
+	}
+}
+
+func coloredStateText(state string) string {
+	return prColor(state).Sprint(stateText(state))
+}
+
 func coloredPrNumber(pr *models.GithubPullRequest) string {
 	return prColor(pr.State).Sprint("#" + strconv.Itoa(pr.Number))
 }
@@ -236,6 +259,8 @@ func prColor(state string) style.TextStyle {
 		return style.FgRed
 	case "MERGED":
 		return style.FgMagenta
+	case "DRAFT":
+		return style.FgBlackLighter
 	default:
 		return style.FgDefault
 	}

pkg/gui/presentation/branches.go

@@ -269,6 +269,8 @@ func prColor(state string) style.TextStyle {
 		return style.FgRed
 	case "MERGED":
 		return style.FgMagenta
+	case "DRAFT":
+		return style.FgBlackLighter
 	default:
 		return style.FgDefault
 	}