Merge pull request #50 from vitessio/close-milestones

Add close milestone

Author: frouioui

go/interactive/release/close_milestone.go

@@ -22,6 +22,7 @@ import (
 	tea "github.com/charmbracelet/bubbletea"
 	"vitess.io/vitess-releaser/go/interactive/ui"
 	"vitess.io/vitess-releaser/go/releaser"
+	"vitess.io/vitess-releaser/go/releaser/release"
 	"vitess.io/vitess-releaser/go/releaser/steps"
 )
 
@@ -44,9 +45,19 @@ func CloseMilestoneItem(ctx context.Context) *ui.MenuItem {
 type closeMilestoneUrl string
 
 func closeMilestoneUpdate(mi *ui.MenuItem, msg tea.Msg) (*ui.MenuItem, tea.Cmd) {
+	_, ok := msg.(closeMilestoneUrl)
+	if !ok {
+		return mi, nil
+	}
+
+	mi.Info = mi.State.Issue.CloseMilestone.URL
+	mi.IsDone = mi.State.Issue.CloseMilestone.Done
 	return mi, nil
 }
 
 func closeMilestoneAct(mi *ui.MenuItem) (*ui.MenuItem, tea.Cmd) {
-	return mi, nil
+	pl, back := release.CloseMilestone(mi.State)
+	return mi, tea.Batch(func() tea.Msg {
+		return closeMilestoneUrl(back())
+	}, ui.PushDialog(ui.NewProgressDialog("Close Milestone", pl)))
 }

go/releaser/github/milestone.go

@@ -20,21 +20,23 @@ import (
 	"encoding/json"
 	"fmt"
 	"log"
+	"strconv"
 	"strings"
 
 	"github.com/cli/go-gh"
 )
 
 type Milestone struct {
-	URL string `json:"url"`
+	URL    string `json:"url"`
+	Number int    `json:"number"`
 }
 
 func GetMilestonesByName(repo, name string) []Milestone {
 	stdOut, _, err := gh.Exec(
 		"milestone", "list",
 		"--query", name,
 		"--repo", repo,
-		"--json", "url",
+		"--json", "url,number",
 		"--state", "all",
 	)
 	if err != nil {
@@ -62,4 +64,24 @@ func CreateNewMilestone(repo, name string) string {
 	out := strings.ReplaceAll(stdOut.String(), "\n", "")
 	idx := strings.LastIndex(out, fmt.Sprintf("https://github.com/%s/milestone/", repo))
 	return out[idx:]
-}
+}
+
+func CloseMilestone(repo, name string) string {
+	ms := GetMilestonesByName(repo, name)
+	if len(ms) != 1 {
+		log.Fatalf("expected to find one milestone found %d", len(ms))
+	}
+
+	stdOut, _, err := gh.Exec(
+		"milestone", "edit",
+		strconv.Itoa(ms[0].Number),
+		"--repo", repo,
+		"--state", "closed",
+	)
+	if err != nil {
+		log.Fatal(err.Error())
+	}
+	out := strings.ReplaceAll(stdOut.String(), "\n", "")
+	idx := strings.LastIndex(out, fmt.Sprintf("https://github.com/%s/milestone/", repo))
+	return out[idx:]
+}

go/releaser/github/pr.go

@@ -179,3 +179,38 @@ func GetMergedPRsAndAuthorsByMilestone(repo, milestone string) (prs []PR, author
 	sort.Strings(authors)
 	return prs, authors
 }
+
+func GetOpenedPRsByMilestone(repo, milestone string) []PR {
+	byteRes, _, err := gh.Exec(
+		"pr", "list",
+		"-s", "open",
+		"-S", fmt.Sprintf("milestone:%s", milestone),
+		"--json", "number,title,labels,author",
+		"--limit", "5000",
+		"--repo", repo,
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	var prs []PR
+	err = json.Unmarshal(byteRes.Bytes(), &prs)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return prs
+}
+
+func AssignMilestoneToPRs(repo, milestone string, prs []PR) {
+	for _, pr := range prs {
+		_, _, err := gh.Exec(
+			"pr", "edit",
+			strconv.Itoa(pr.Number),
+			"--milestone", milestone,
+			"--repo", repo,
+		)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+}

go/releaser/release/milestone.go

@@ -0,0 +1,59 @@
+/*
+Copyright 2023 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package release
+
+import (
+	"fmt"
+
+	"vitess.io/vitess-releaser/go/releaser"
+	"vitess.io/vitess-releaser/go/releaser/github"
+	"vitess.io/vitess-releaser/go/releaser/logging"
+)
+
+func CloseMilestone(state *releaser.State) (*logging.ProgressLogging, func() string) {
+	pl := &logging.ProgressLogging{
+		TotalSteps: 5,
+	}
+
+	return pl, func() string {
+		milestone := fmt.Sprintf("v%s", state.Release)
+		nextRelease := releaser.FindVersionAfterNextRelease(state)
+		nextMilestone := fmt.Sprintf("v%s", nextRelease)
+
+		pl.NewStepf("Get opened Pull Requests for Milestone %s", milestone)
+		prs := github.GetOpenedPRsByMilestone(state.VitessRepo, milestone)
+
+		if len(prs) > 0 {
+			pl.NewStepf("Move %d Pull Requests to the %s Milestone", len(prs), nextMilestone)
+			github.AssignMilestoneToPRs(state.VitessRepo, nextMilestone, prs)
+		} else {
+			pl.TotalSteps--
+		}
+
+		pl.NewStepf("Close Milestone %s", milestone)
+		url := github.CloseMilestone(state.VitessRepo, milestone)
+
+		pl.NewStepf("Update Issue %s on GitHub", state.IssueLink)
+		state.Issue.CloseMilestone.Done = true
+		state.Issue.CloseMilestone.URL = url
+		_, fn := state.UploadIssue()
+		issueLink := fn()
+
+		pl.NewStepf("Issue updated, see: %s", issueLink)
+		return url
+	}
+}