Add support for creating GitHub draft releases

Author: chand1012

cmd/release.go

@@ -0,0 +1,120 @@
+/*
+Copyright © 2023 Chandler <chandler@chand1012.dev>
+*/
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/chand1012/ottodocs/pkg/ai"
+	"github.com/chand1012/ottodocs/pkg/config"
+	"github.com/chand1012/ottodocs/pkg/constants"
+	"github.com/chand1012/ottodocs/pkg/gh"
+	"github.com/chand1012/ottodocs/pkg/git"
+	"github.com/chand1012/ottodocs/pkg/utils"
+	l "github.com/charmbracelet/log"
+	"github.com/spf13/cobra"
+)
+
+// releaseCmd represents the release command
+var releaseCmd = &cobra.Command{
+	Use:   "release",
+	Short: "Generate GitHub release notes from git commit logs",
+	Long: `This command generates GitHub release notes from git commit logs.
+It will create a new release given a tag and post it to GitHub as a draft.`,
+	Aliases: []string{"r"},
+	PreRun: func(cmd *cobra.Command, args []string) {
+		if verbose {
+			log.SetLevel(l.DebugLevel)
+		}
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		c, err := config.Load()
+		if err != nil {
+			log.Errorf("Error loading config: %s", err)
+			os.Exit(1)
+		}
+
+		if previousTag == "" {
+			previousTag, err = utils.Input("Previous tag: ")
+			if err != nil {
+				log.Errorf("Error getting previous tag: %s", err)
+				os.Exit(1)
+			}
+		}
+
+		if currentTag == "" {
+			currentTag, err = utils.Input("Current tag: ")
+			if err != nil {
+				log.Errorf("Error getting current tag: %s", err)
+				os.Exit(1)
+			}
+		}
+
+		fmt.Print("Release notes: ")
+
+		// get the log between the tags
+		gitLog, err := git.LogBetween(previousTag, currentTag)
+		if err != nil {
+			log.Errorf("Error getting log between tags: %s", err)
+			os.Exit(1)
+		}
+
+		prompt := constants.RELEASE_PROMPT + gitLog
+
+		stream, err := ai.SimpleStreamRequest(prompt, c)
+		if err != nil {
+			log.Errorf("Error getting response: %s", err)
+			os.Exit(1)
+		}
+
+		releaseNotes, err := utils.PrintChatCompletionStream(stream)
+		if err != nil {
+			log.Errorf("Error printing completion stream: %s", err)
+			os.Exit(1)
+		}
+
+		if !force {
+			confirm, err := utils.Input("Create release? (y/n): ")
+			if err != nil {
+				log.Errorf("Error getting confirmation: %s", err)
+				os.Exit(1)
+			}
+			confirm = strings.ToLower(confirm)
+			if confirm != "y" {
+				os.Exit(0)
+			}
+		}
+
+		origin, err := git.GetRemote("origin")
+		if err != nil {
+			log.Errorf("Error getting remote: %s", err)
+			os.Exit(1)
+		}
+
+		owner, repo, err := git.ExtractOriginInfo(origin)
+		if err != nil {
+			log.Errorf("Error extracting origin info: %s", err)
+			os.Exit(1)
+		}
+
+		err = gh.CreateDraftRelease(owner, repo, currentTag, releaseNotes, currentTag, c)
+		if err != nil {
+			log.Errorf("Error creating release: %s", err)
+			os.Exit(1)
+		}
+
+		fmt.Println("Release created successfully!")
+	},
+}
+
+func init() {
+	RootCmd.AddCommand(releaseCmd)
+
+	releaseCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "Verbose output")
+	releaseCmd.Flags().BoolVarP(&force, "force", "f", false, "Force overwrite of existing file")
+	releaseCmd.Flags().StringVarP(&previousTag, "prev-tag", "p", "", "Previous tag")
+	releaseCmd.Flags().StringVarP(&currentTag, "tag", "t", "", "Current tag")
+}

cmd/vars.go

@@ -44,6 +44,9 @@ var startLine int
 var endLine int
 var appendFile bool
 
+var previousTag string
+var currentTag string
+
 var log = l.NewWithOptions(os.Stderr, l.Options{
 	Level:           l.InfoLevel,
 	ReportTimestamp: false,

pkg/constants/prompts.go

@@ -41,3 +41,12 @@ var PR_BODY_PROMPT string = "You are a helpful assistant who writes pull request
 var COMPRESS_DIFF_PROMPT string = "You are a helpful assistant who describes git diff changes. You will be given a Git diff and you should use it to create a description of the changes. The description should be no longer than 75 characters long and should describe the changes in the diff. Do not include the file names in the description."
 
 var EDIT_CODE_PROMPT string = `You are a helpful assistant who edits code. You will be given a file what you are trying to accomplish in the edit and you should edit it to the best of your abilities. Only edit the code you are told. All code to edit will be preceded by "EDIT:". If "EDIT:" is omitted, assume you must edit the entire file. The goal of the edit will be preceded by "GOAL:". The entire file will be preceded by "FILE:". If there is not content after "FILE:", assume you are writing new code. Make sure to use the language specified in the task. The output code should be unformatted. Use no markdown.\n`
+
+var RELEASE_PROMPT string = `You are a helpful assistant who creates GitHub release notes from git commit logs. You will be given a series of git commit messages and your task is to summarize these messages into release notes. The release notes should:
+- Be concise and informative about the changes made in the release.
+- Be written in plain English.
+- Group related changes together under appropriate headings.
+- Exclude unnecessary details, such as the specific file names that were changed.
+
+Commit log:
+`

pkg/gh/issues.go

@@ -111,3 +111,44 @@ type Comment struct {
 	// The username of the user who made the comment. This is optional.
 	Username string `json:"username,omitempty"`
 }
+
+func CreateDraftRelease(owner, repo, title, body, tag string, conf *config.Config) error {
+	if conf.GHToken == "" {
+		return fmt.Errorf("no GitHub token found")
+	}
+
+	releaseURL := fmt.Sprintf("https://api.github.com/repos/%s/%s/releases", owner, repo)
+	data := map[string]interface{}{
+		"title":      title,
+		"body":       body,
+		"tag_name":   tag,
+		"draft":      true,
+		"prerelease": false,
+	}
+
+	payload, err := json.Marshal(data)
+	if err != nil {
+		return err
+	}
+
+	releaseReq, err := http.NewRequest("POST", releaseURL, strings.NewReader(string(payload)))
+	if err != nil {
+		return err
+	}
+
+	releaseReq.Header.Set("Authorization", fmt.Sprintf("Bearer %s", conf.GHToken))
+	releaseReq.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{}
+	releaseResp, err := client.Do(releaseReq)
+	if err != nil {
+		return err
+	}
+	defer releaseResp.Body.Close()
+
+	if releaseResp.StatusCode != http.StatusCreated {
+		return fmt.Errorf("failed to create release draft: %s", releaseResp.Status)
+	}
+
+	return nil
+}