package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"

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

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)
	}
}

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)

	comments := GetComments(client, prID)
	ProcessComments(&comments)
	PatchFiles(comments)
}