package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)

func GetMergeRequests(projectId string, mergeRequestId string) map[string]interface{} {
	mergeRequestURL :=
		"https://git.wosai-inc.com/api/v4/projects/:id/merge_requests/:merge_request_iid/changes"
	// 替换:id 和 :merge_request_iid 为对应的值
	mergeRequestURL = replaceParameters(mergeRequestURL,
		map[string]string{
			":id":                projectId,      // 替换为 GitLab 项目的 ID
			":merge_request_iid": mergeRequestId, // 替换为 Merge Request 的 IID
		})

	// 创建 HTTP 请求
	client := http.Client{}
	request, err := http.NewRequest("GET", mergeRequestURL, nil)
	if err != nil {
		fmt.Println("Error creating request", err)
		return nil
	}

	// 设置 GitLab 认证令牌
	request.Header.Add("Private-Token", "你的token")

	// 发送 HTTP 请求
	response, err := client.Do(request)
	if err != nil {
		fmt.Println("Error sending request", err)
		return nil
	}

	// 解析响应数据
	body, err := ioutil.ReadAll(response.Body)
	if err != nil {
		fmt.Println("Error reading response", err)
		return nil
	}

	// 将响应数据解析为 JSON
	var mergeRequest map[string]interface{}
	err = json.Unmarshal(body, &mergeRequest)
	if err != nil {
		fmt.Println("Error decoding JSON", err)
		return nil
	}

	// 输出 Merge Request 标题
	fmt.Println("Merge Request title:", mergeRequest["title"])
	return mergeRequest
}

// replaceParameters 用于替换 URL 中的参数
func replaceParameters(url string, parameters map[string]string) string {
	for key, value := range parameters {
		url = strings.ReplaceAll(url, key, value)
	}
	return url
}

{
  Role:    openai.ChatMessageRoleSystem,
  Content: "你是是一位资深编程专家，负责代码变更的审查工作。需要给出审查建议。在建议的开始需明确对此代码变更给出「拒绝」或「接受」的决定，并且以格式「变更评分：实际的分数」给变更打分，分数区间为0~100分。然后，以精炼的语言、严厉的语气指出存在的问题。如果你觉得必要的情况下，可直接给出修改后的内容。建议中的语句可以使用emoji结尾。你的反馈内容必须使用严谨的markdown格式。",
}

{
	Role:    openai.ChatMessageRoleSystem,
	Content: "你是一个天才小女孩，精通编程工作，性格很傲娇又高傲，负责对前辈的代码变更进行审查，用后辈的态度、活泼轻快的方式的指出存在的问题。使用markdown格式。可以包含emoji。",
}

package main

func main() {
	//读取Gitlab MR，并获取对应changes内容
  //参数为projectid 、mergeRequestId
	runProjectCodeReview("projectid", "mergeRequestId") 
}

package main

import (
	"context"
	"fmt"
	"github.com/charmbracelet/glamour"
	"github.com/sashabaranov/go-openai"
	"os"
	"strings"
)

var client *openai.Client

func init() {
	fmt.Println("====== open AI client init======")
	// 设置代理地址
	proxyURL := "http://127.0.0.1:7890"
	os.Setenv("HTTP_PROXY", proxyURL)
	os.Setenv("HTTPS_PROXY", proxyURL)

	apiKey := os.Getenv("OPENAI_API_KEY")
	if apiKey == "" {
		fmt.Println("Error: OPENAI_API_KEY environment variable not set")
		os.Exit(1)
	}

	client = openai.NewClient(apiKey)
}

func GetCodeReview(input string) {
	resp, err := client.CreateChatCompletion(
		context.Background(),
		openai.ChatCompletionRequest{
			Model: openai.GPT3Dot5Turbo0301,
			Messages: []openai.ChatCompletionMessage{
				{
					Role:    openai.ChatMessageRoleSystem,
					Content: "你是是一位资深编程专家，负责代码变更的审查工作。需要给出审查建议。在建议的开始需明确对此代码变更给出「拒绝」或「接受」的决定，并且以格式「变更评分：实际的分数」给变更打分，分数区间为0~100分。然后，以精炼的语言、严厉的语气指出存在的问题。如果你觉得必要的情况下，可直接给出修改后的内容。建议中的语句可以使用emoji结尾。你的反馈内容必须使用严谨的markdown格式。",
				},
				{
					Role:    openai.ChatMessageRoleSystem,
					Content: "你是一个天才小女孩，精通编程工作，性格很傲娇又高傲，负责对前辈的代码变更进行审查，用后辈的态度、活泼轻快的方式的指出存在的问题。使用markdown格式。可以包含emoji。",
				},
				{
					Role:    openai.ChatMessageRoleUser,
					Content: input,
				},
			},
		},
	)
	if err != nil {
		fmt.Printf("Error: %v\n", err)
		return
	}
	fmt.Println("Bot AI review意见: ", formatOutput(resp.Choices[0].Message.Content))
}

func formatOutput(text string) string {
	// Render the text using the glamour library for better formatting
	styled, err := glamour.Render(text, "dark")
	if err != nil {
		fmt.Println("Error:", err)
		os.Exit(1)
	}
	return styled
}

func runProjectCodeReview(projectId string, mergeRequestId string) {
	mergeRequest := GetMergeRequests(projectId, mergeRequestId)
	changes := mergeRequest["changes"]
	ignoreFileName := []string{".yml", "test", ".mod", ".sum", "config", "json", "lock", "css", "html", "Settings"}
	for _, change := range changes.([]interface{}) {
		changeMap := change.(map[string]interface{})
		newPath := changeMap["new_path"]
		var ignore = false
		for _, s := range ignoreFileName {
			if strings.LastIndex(newPath.(string), s) != -1 {
				ignore = true
				break
			}
		}
		if ignore {
			continue
		}
		fmt.Printf("对 %s review中...\n", newPath)
		input := "请review这部分代码变更:" + changeMap["diff"].(string)
		GetCodeReview(input)
	}
}