nitter/nitters/gitea/pr_nitter.go

package gitea

import (
	"bytes"
	"embed"
	"strings"
	"text/template"

	"code.gitea.io/sdk/gitea"
	"github.com/golangci/golangci-lint/pkg/report"
	"github.com/golangci/golangci-lint/pkg/result"
	"golang.org/x/exp/slog"

	"code.icb4dc0.de/prskr/nitter/nitters"
)

const summaryFooter = "\n\n(Created by nitter)"

var (
	_ nitters.Nitter = (*prNitter)(nil)

	//go:embed templates/*
	templatesFS embed.FS
	templates   *template.Template
)

func init() {
	if tmpl, err := template.New("issue_templates").ParseFS(templatesFS, "templates/*.tmpl.md"); err != nil {
		panic(err)
	} else {
		templates = tmpl
	}
}

func NewPRNitter(cli Client, cfg *GiteaPRConfig) *prNitter {
	return &prNitter{
		Client: cli,
		cfg:    cfg,
	}
}

type prNitter struct {
	Client
	cfg *GiteaPRConfig
}

func (p prNitter) Report(report *report.Data, issues []result.Issue) error {
	review, err := p.preparePullReview(report, issues)
	if err != nil {
		return err
	}

	if err := p.cleanOutdatedReviews(); err != nil {
		return err
	}

	if _, _, err := p.CreatePullReview(p.cfg.Namespace, p.cfg.Repo, p.cfg.PRIndex, *review); err != nil {
		slog.Error("Failed to submit new PR review", err)
		return err
	}

	return nil
}

func (p prNitter) preparePullReview(report *report.Data, issues []result.Issue) (*gitea.CreatePullReviewOptions, error) {
	templateBuf := bytes.Buffer{}
	summaryData := map[string]any{
		"Report": report,
		"Issues": issues,
	}

	if err := templates.ExecuteTemplate(&templateBuf, "issue_summary.tmpl.md", summaryData); err != nil {
		return nil, err
	}

	pullReviewOptions := &gitea.CreatePullReviewOptions{
		State:    p.cfg.ReviewState,
		Body:     templateBuf.String(),
		Comments: make([]gitea.CreatePullReviewComment, 0, len(issues)),
	}

	if len(issues) == 0 {
		pullReviewOptions.State = gitea.ReviewStateApproved
	} else {
		templateBuf.Reset()
		for i := range issues {
			templateData := map[string]any{
				"Issue": issues[i],
			}
			if err := templates.ExecuteTemplate(&templateBuf, "issue_comment.tmpl.md", templateData); err != nil {
				return nil, err
			}

			if _, err := templateBuf.WriteString(summaryFooter); err != nil {
				slog.Error("Failed to append summary footer", err)
				return nil, err
			}

			pullReviewOptions.Comments = append(pullReviewOptions.Comments, gitea.CreatePullReviewComment{
				Path:       issues[i].Pos.Filename,
				Body:       templateBuf.String(),
				NewLineNum: int64(issues[i].Pos.Line),
			})

			templateBuf.Reset()
		}
	}

	return pullReviewOptions, nil
}

func (p prNitter) cleanOutdatedReviews() error {
	me, _, err := p.GetMyUserInfo()
	if err != nil {
		return err
	}

	slog.Debug("Running as", slog.String("name", me.UserName))

	reviews, _, err := p.ListPullReviews(p.cfg.Namespace, p.cfg.Repo, p.cfg.PRIndex, gitea.ListPullReviewsOptions{})
	if err != nil {
		return err
	}

	for i := range reviews {
		review := reviews[i]
		if review.Reviewer.ID == me.ID && strings.HasSuffix(review.Body, summaryFooter) {
			if _, err = p.DeletePullReview(p.cfg.Namespace, p.cfg.Repo, p.cfg.PRIndex, review.ID); err != nil {
				slog.Error("Failed to delete existing review", err, slog.Int64("reviewId", review.ID))
				return err
			}
		}
	}

	return nil
}