pkg/email: add patch FormatPatch function

Fixes #6673

1 files changed, 46 insertions, 0 deletions

diff --git a/pkg/email/patch.go b/pkg/email/patch.go
index 81ea9b26c..b33bb09bf 100644
--- a/pkg/email/patch.go
+++ b/pkg/email/patch.go
@@ -6,8 +6,13 @@ package email
 import (
 	"bufio"
 	"bytes"
+	"fmt"
+	"net/mail"
 	"regexp"
 	"strings"
+	"text/template"
+
+	"github.com/google/syzkaller/pkg/aflow/ai"
 )
 
 func ParsePatch(message []byte) (diff string) {
@@ -50,6 +55,47 @@ func ParsePatch(message []byte) (diff string) {
 	return
 }
 
+func FormatPatchDescription(description string, authors []string, recipients []ai.Recipient) string {
+	buf := new(bytes.Buffer)
+	var to, cc []mail.Address
+	for _, recipient := range recipients {
+		addr := mail.Address{Name: recipient.Name, Address: recipient.Email}
+		if recipient.To {
+			to = append(to, addr)
+		} else {
+			cc = append(cc, addr)
+		}
+	}
+	err := patchTemplate.Execute(buf, map[string]any{
+		"description": strings.TrimSpace(description),
+		"authors":     authors,
+		"to":          to,
+		"cc":          cc,
+	})
+	if err != nil {
+		panic(err)
+	}
+	return buf.String()
+}
+
+func FormatPatch(description, diff, baseCommit string, authors []string, recipients []ai.Recipient) string {
+	return FormatPatchDescription(description, authors, recipients) +
+		fmt.Sprintf("%v\nbase-commit: %v\n", diff, baseCommit)
+}
+
+var patchTemplate = template.Must(template.New("").Parse(`{{.description}}
+{{range $addr := .authors}}
+Signed-off-by: {{$addr}}{{end}}
+{{- range $addr := .to}}
+To: {{$addr.String}}{{end}}
+{{- range $addr := .cc}}
+Cc: {{$addr.String}}{{end}}
+---
+This patch was generated by Google Gemini LLM model.
+It was pre-reviewed and Signed-off-by a human, but please review carefully.
+
+`))
+
 var diffRegexps = []*regexp.Regexp{
 	regexp.MustCompile(`^(---|\+\+\+) [^\s]`),
 	regexp.MustCompile(`^diff --git`),