main.go

@@ -99,26 +99,14 @@ func FormatComment(body string, beenWrapped bool) string {
 	return fmt.Sprintf("%s%s%s", config.Language.Opening, body, config.Language.Closing)
 }
 
-func PatchFiles(wrapper *wrap.Wrapper, commentsByFile map[string]([]*gitea.PullReviewComment)) {
-	for filepath, comments := range commentsByFile {
-		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() {
-			for (nextComment < commentsLength) && comments[nextComment].LineNum == i {
+func WriteCommentsInFile(
+	wrapper *wrap.Wrapper,
+	outputFile *os.File,
+	lineNum uint64,
+	comments []*gitea.PullReviewComment,
+	nextComment, commentsLength int,
+) int {
+	for (nextComment < commentsLength) && comments[nextComment].LineNum == lineNum {
 		body := comments[nextComment].Body
 
 		beenWrapped := len(body)+len(config.Language.Opening)+len(config.Language.Closing) > 100
@@ -141,10 +129,36 @@ func PatchFiles(wrapper *wrap.Wrapper, commentsByFile map[string]([]*gitea.PullR
 		fmt.Printf("L%04d:\n%s\n", comments[nextComment].LineNum, comment)
 		nextComment++
 	}
+	return nextComment
+}
 
+func ProcessFile(wrapper *wrap.Wrapper, 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(wrapper, outputFile, i, comments, nextComment, commentsLength)
 		fmt.Fprintln(outputFile, scanner.Text())
 		i++
 	}
+}
+
+func PatchFiles(wrapper *wrap.Wrapper, commentsByFile map[string]([]*gitea.PullReviewComment)) {
+	for filepath, comments := range commentsByFile {
+		ProcessFile(wrapper, filepath, comments)
 	}
 }