mr/issue: Add subscribe/unsubscribe subcommands

For projects where one is only tangentially involved, it sometimes
makes sense to subscribe to an individual issue or merge request
rather than the project as a whole.

Support that with corresponding subcommands.

cmd/issue_subscribe.go

@@ -0,0 +1,44 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/rsteube/carapace"
+	"github.com/spf13/cobra"
+	"github.com/zaquestion/lab/internal/action"
+	lab "github.com/zaquestion/lab/internal/gitlab"
+)
+
+var issueSubscribeCmd = &cobra.Command{
+	Use:              "subscribe [remote] <id>",
+	Aliases:          []string{},
+	Short:            "Subscribe to issue",
+	Long:             ``,
+	PersistentPreRun: LabPersistentPreRun,
+	Run: func(cmd *cobra.Command, args []string) {
+		rn, id, err := parseArgsRemoteAndID(args)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		p, err := lab.FindProject(rn)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = lab.IssueSubscribe(p.ID, int(id))
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Subscribed to issue #%d\n", id)
+	},
+}
+
+func init() {
+	issueCmd.AddCommand(issueSubscribeCmd)
+	carapace.Gen(issueSubscribeCmd).PositionalCompletion(
+		action.Remotes(),
+		action.Issues(issueList),
+	)
+}

cmd/issue_unsubscribe.go

@@ -0,0 +1,44 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/rsteube/carapace"
+	"github.com/spf13/cobra"
+	"github.com/zaquestion/lab/internal/action"
+	lab "github.com/zaquestion/lab/internal/gitlab"
+)
+
+var issueUnsubscribeCmd = &cobra.Command{
+	Use:              "unsubscribe [remote] <id>",
+	Aliases:          []string{},
+	Short:            "Unubscribe from issue",
+	Long:             ``,
+	PersistentPreRun: LabPersistentPreRun,
+	Run: func(cmd *cobra.Command, args []string) {
+		rn, id, err := parseArgsRemoteAndID(args)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		p, err := lab.FindProject(rn)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = lab.IssueUnsubscribe(p.ID, int(id))
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Unsubscribed from issue #%d\n", id)
+	},
+}
+
+func init() {
+	issueCmd.AddCommand(issueUnsubscribeCmd)
+	carapace.Gen(issueUnsubscribeCmd).PositionalCompletion(
+		action.Remotes(),
+		action.Issues(issueList),
+	)
+}

cmd/mr_subscribe.go

@@ -0,0 +1,44 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/rsteube/carapace"
+	"github.com/spf13/cobra"
+	"github.com/zaquestion/lab/internal/action"
+	lab "github.com/zaquestion/lab/internal/gitlab"
+)
+
+var mrSubscribeCmd = &cobra.Command{
+	Use:              "subscribe [remote] <id>",
+	Aliases:          []string{},
+	Short:            "Subscribe to merge request",
+	Long:             ``,
+	PersistentPreRun: LabPersistentPreRun,
+	Run: func(cmd *cobra.Command, args []string) {
+		rn, id, err := parseArgsWithGitBranchMR(args)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		p, err := lab.FindProject(rn)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = lab.MRSubscribe(p.ID, int(id))
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Subscribed to merge request !%d\n", id)
+	},
+}
+
+func init() {
+	mrCmd.AddCommand(mrSubscribeCmd)
+	carapace.Gen(mrSubscribeCmd).PositionalCompletion(
+		action.Remotes(),
+		action.MergeRequests(mrList),
+	)
+}

cmd/mr_unsubscribe.go

@@ -0,0 +1,44 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/rsteube/carapace"
+	"github.com/spf13/cobra"
+	"github.com/zaquestion/lab/internal/action"
+	lab "github.com/zaquestion/lab/internal/gitlab"
+)
+
+var mrUnsubscribeCmd = &cobra.Command{
+	Use:              "unsubscribe [remote] <id>",
+	Aliases:          []string{},
+	Short:            "Unubscribe from merge request",
+	Long:             ``,
+	PersistentPreRun: LabPersistentPreRun,
+	Run: func(cmd *cobra.Command, args []string) {
+		rn, id, err := parseArgsWithGitBranchMR(args)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		p, err := lab.FindProject(rn)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = lab.MRUnsubscribe(p.ID, int(id))
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Unsubscribed from merge request !%d\n", id)
+	},
+}
+
+func init() {
+	mrCmd.AddCommand(mrUnsubscribeCmd)
+	carapace.Gen(mrUnsubscribeCmd).PositionalCompletion(
+		action.Remotes(),
+		action.MergeRequests(mrList),
+	)
+}

internal/gitlab/gitlab.go

@@ -535,6 +535,30 @@ func MRUnapprove(pid interface{}, id int) error {
 	return nil
 }
 
+// MRSubscribe subscribes to an mr on a GitLab project
+func MRSubscribe(pid interface{}, id int) error {
+	_, resp, err := lab.MergeRequests.SubscribeToMergeRequest(pid, id, nil)
+	if resp != nil && resp.StatusCode == http.StatusNotModified {
+		return errors.New("Already subscribed")
+	}
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// MRUnsubscribe unsubscribes from a previously mr on a GitLab project
+func MRUnsubscribe(pid interface{}, id int) error {
+	_, resp, err := lab.MergeRequests.UnsubscribeFromMergeRequest(pid, id, nil)
+	if resp != nil && resp.StatusCode == http.StatusNotModified {
+		return errors.New("Not subscribed")
+	}
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 // MRThumbUp places a thumb up/down on a merge request
 func MRThumbUp(pid interface{}, id int) error {
 	_, _, err := lab.AwardEmoji.CreateMergeRequestAwardEmoji(pid, id, &gitlab.CreateAwardEmojiOptions{
@@ -744,6 +768,30 @@ func IssueListDiscussions(project string, issueNum int) ([]*gitlab.Discussion, e
 	return discussions, nil
 }
 
+// IssueSubscribe subscribes to an issue on a GitLab project
+func IssueSubscribe(pid interface{}, id int) error {
+	_, resp, err := lab.Issues.SubscribeToIssue(pid, id, nil)
+	if resp != nil && resp.StatusCode == http.StatusNotModified {
+		return errors.New("Already subscribed")
+	}
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// IssueUnsubscribe unsubscribes from an issue on a GitLab project
+func IssueUnsubscribe(pid interface{}, id int) error {
+	_, resp, err := lab.Issues.UnsubscribeFromIssue(pid, id, nil)
+	if resp != nil && resp.StatusCode == http.StatusNotModified {
+		return errors.New("Not subscribed")
+	}
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 // GetCommit returns top Commit by ref (hash, branch or tag).
 func GetCommit(pid interface{}, ref string) (*gitlab.Commit, error) {
 	c, _, err := lab.Commits.GetCommit(pid, ref)