frag-review/main.go

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
	"sort"
	"strconv"

	"code.gitea.io/sdk/gitea"
)

type ByLineNum []*gitea.PullReviewComment

func (a ByLineNum) Len() int           { return len(a) }
func (a ByLineNum) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
func (a ByLineNum) Less(i, j int) bool { return a[i].LineNum < a[j].LineNum }

func ExitOnError(err error) {
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
}

func RetrieveComments(client *gitea.Client, prID, reviewID int64) []*gitea.PullReviewComment {
	comments, _, err := client.ListPullReviewComments(
		config.Gitea.Owner, config.Gitea.Repository, prID, reviewID,
	)
	ExitOnError(err)
	return comments
}

func SplitByFile(
	splitComments map[string]([]*gitea.PullReviewComment),
	comments []*gitea.PullReviewComment,
) map[string]([]*gitea.PullReviewComment) {
	for _, comment := range comments {
		arr, found := splitComments[comment.Path]

		if !found {
			arr = make([]*gitea.PullReviewComment, 0)
		}

		splitComments[comment.Path] = append(arr, comment)
	}

	return splitComments
}

func BackUpSource(filepath string) {
	inputFile, err := os.Open(filepath)
	ExitOnError(err)
	defer inputFile.Close()

	BackUpSourceFile, err := os.Create(filepath + ".bck")
	ExitOnError(err)
	defer BackUpSourceFile.Close()

	_, err = io.Copy(BackUpSourceFile, inputFile)
	ExitOnError(err)
}

func WriteCommentsInFile(
	outputFile *os.File,
	lineNum uint64,
	comments []*gitea.PullReviewComment,
	nextComment, commentsLength int,
) int {
	for (nextComment < commentsLength) && comments[nextComment].LineNum == lineNum {
		comment := comments[nextComment].Body
		fmt.Fprint(outputFile, comment)
		fmt.Printf("L%04d:\n%s\n", comments[nextComment].LineNum, comment)
		nextComment++
	}
	return nextComment
}

func ProcessFile(filepath string, comments []*gitea.PullReviewComment) {
	fmt.Printf("FILE: %s\n", filepath)

	BackUpSource(filepath)

	inputFile, err := os.Open(filepath + ".bck")
	ExitOnError(err)
	defer inputFile.Close()

	outputFile, err := os.Create(filepath)
	ExitOnError(err)
	defer outputFile.Close()

	var i uint64 = 1
	nextComment, commentsLength := 0, len(comments)

	scanner := bufio.NewScanner(inputFile)
	for scanner.Scan() {
		nextComment = WriteCommentsInFile(outputFile, i, comments, nextComment, commentsLength)
		fmt.Fprintln(outputFile, scanner.Text())
		i++
	}
}

func PatchFiles(commentsByFile map[string]([]*gitea.PullReviewComment)) {
	for filepath, comments := range commentsByFile {
		ProcessFile(filepath, comments)
	}
}

func GetReviewsPerPR(client *gitea.Client, prID int64) []int64 {
	reviews, _, err := client.ListPullReviews(
		config.Gitea.Owner, config.Gitea.Repository, prID, gitea.ListPullReviewsOptions{},
	)
	ExitOnError(err)

	result := make([]int64, 0)
	for _, review := range reviews {
		result = append(result, review.ID)
	}

	return result
}

var config Config = Config{}

func main() {
	token := os.Getenv("GITEA_TOKEN")
	LoadConfig(&config, os.Getenv("FRAG_REVIEW_CONFIG"))

	client, err := gitea.NewClient(config.Gitea.InstanceURL, gitea.SetToken(token))
	ExitOnError(err)

	if len(os.Args) < 2 {
		fmt.Fprintf(os.Stderr, "Usage: %s pr-id\n", os.Args[0])
		os.Exit(1)
	}

	prID, err := strconv.ParseInt(os.Args[1], 10, 64)
	ExitOnError(err)

	reviews := GetReviewsPerPR(client, prID)

	splitComments := make(map[string]([]*gitea.PullReviewComment))
	for _, reviewID := range reviews {
		comments := RetrieveComments(client, prID, reviewID)
		sort.Sort(ByLineNum(comments))
		splitComments = SplitByFile(splitComments, comments)
	}

	ProcessComments(&splitComments)
	PatchFiles(splitComments)
}