Add state flag

main.go

@@ -25,8 +25,10 @@ var (
 )
 
 func main() {
+	var state shared.PullRequestState
 	var dryRun bool
 	var debug bool
+	flag.Var(&state, "state", "Filter by state: {open|closed|merged}")
 	flag.BoolVar(&dryRun, "dry-run", false, "Show branches to delete")
 	flag.BoolVar(&debug, "debug", false, "Enable debug logs")
 	flag.Usage = func() {
@@ -46,7 +48,7 @@ func main() {
 	args := flag.Args()
 
 	if len(args) == 0 {
-		runMain(dryRun, debug)
+		runMain(state, dryRun, debug)
 	} else {
 		subcmd, args := args[0], args[1:]
 		switch subcmd {
@@ -76,7 +78,7 @@ func main() {
 	}
 }
 
-func runMain(dryRun bool, debug bool) {
+func runMain(state shared.PullRequestState, dryRun bool, debug bool) {
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
 	defer stop()
 

main_test.go

@@ -9,12 +9,13 @@ import (
 
 	"github.com/fatih/color"
 	"github.com/stretchr/testify/assert"
+	"github.com/seachicken/gh-poi/shared"
 )
 
 func Test_DeletingBranchesWhenDryRunOptionIsFalse(t *testing.T) {
 	onlyCI(t)
 
-	results := captureOutput(func() { runMain(false, false) })
+	results := captureOutput(func() { runMain(shared.Merged, false, false) })
 
 	expected := fmt.Sprintf("%s %s", green("✔"), "Deleting branches...")
 	assert.Contains(t, results, expected)
@@ -23,7 +24,7 @@ func Test_DeletingBranchesWhenDryRunOptionIsFalse(t *testing.T) {
 func Test_DoNotDeleteBranchesWhenDryRunOptionIsTrue(t *testing.T) {
 	onlyCI(t)
 
-	results := captureOutput(func() { runMain(true, false) })
+	results := captureOutput(func() { runMain(shared.Merged, true, false) })
 
 	expected := fmt.Sprintf("%s %s", hiBlack("-"), "Deleting branches...")
 	assert.Contains(t, results, expected)
@@ -33,12 +34,12 @@ func Test_ProtectAndUnprotect(t *testing.T) {
 	onlyCI(t)
 
 	runProtect([]string{"main"}, false)
-	protectResults := captureOutput(func() { runMain(true, false) })
+	protectResults := captureOutput(func() { runMain(shared.Merged, true, false) })
 	expected := fmt.Sprintf("main %s", hiBlack("[protected]"))
 	assert.Contains(t, protectResults, expected)
 
 	runUnprotect([]string{"main"}, false)
-	unprotectResults := captureOutput(func() { runMain(true, false) })
+	unprotectResults := captureOutput(func() { runMain(shared.Merged, true, false) })
 	assert.NotContains(t, unprotectResults, expected)
 }
 

shared/State.go

@@ -0,0 +1,22 @@
+package shared
+
+import (
+// "slices"
+)
+
+type State string
+
+//const (
+//	Open   State = "open"
+//	Closed State = "closed"
+//	Merged State = "merged"
+//	All    State = "all"
+//)
+//
+//func (s State) Set(value string) error {
+//	if slices.Contains([]State{"open", "closed", "merged"}, value) {
+//		return value
+//	} else {
+//		return errors.New("invalid state")
+//	}
+//}

shared/pull_request.go

@@ -1,5 +1,11 @@
 package shared
 
+import (
+	"slices"
+
+	"github.com/pkg/errors"
+)
+
 type (
 	PullRequestState int
 
@@ -19,3 +25,32 @@ const (
 	Merged
 	Open
 )
+
+func (s PullRequestState) String() string {
+	switch s {
+	case Closed:
+		return "closed"
+	case Merged:
+		return "merged"
+	case Open:
+		return "open"
+	default:
+		return ""
+	}
+}
+
+func (s PullRequestState) Set(value string) error {
+	if slices.Contains([]string{"closed", "merged", "open"}, value) {
+		switch value {
+		case "closed":
+			s = Closed
+		case "merged":
+			s = Merged
+		case "open":
+			s = Open
+		}
+		return nil
+	} else {
+		return errors.New("invalid state")
+	}
+}