package main

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"strings"

	"github.com/gofiber/fiber/v2"
)

var GroqErrorCodes map[string]string

func init() {
	GroqErrorCodes = make(map[string]string)
	GroqErrorCodes["400"] = "Provider error: Invalid Request - Please contact the support."
	GroqErrorCodes["404"] = "Provider error: The requested resource could not be found. Most likely the model is not available anymore."
	GroqErrorCodes["422"] = "Provider error: The request was well-formed but could not be followed due to semantic errors. Verify the data provided for correctness and completeness."
	GroqErrorCodes["429"] = "Provider error: Too many requests were sent in a given timeframe. Implement request throttling and respect rate limits."
	GroqErrorCodes["500"] = "Provider error: A generic error occurred on Gorq's server. Try the request again later or contact support if the issue persists."
	GroqErrorCodes["502"] = "Provider error: The server received an invalid response from an upstream server. This may be a temporary issue; retrying the request might resolve it."
	GroqErrorCodes["503"] = "Provider error: The server is not ready to handle the request, often due to maintenance or overload. Wait before retrying the request."
}

func RequestGroq(c *fiber.Ctx, llm LLM, messages []Message, testApiKey string) string {
	model := llm.Model.ModelID
	temperature := float64(llm.Temperature)
	context := llm.Context
	maxTokens := int(llm.MaxToken)

	var apiKey string
	if testApiKey == "" {
		err := edgeGlobalClient.WithGlobals(map[string]interface{}{"ext::auth::client_token": c.Cookies("jade-edgedb-auth-token")}).QuerySingle(edgeCtx, `
		with
			filtered_keys := (
				select Key {
					key
				} filter .company.name = <str>$0 AND .<keys[is Setting].<setting[is User] = global currentUser
			)
		select filtered_keys.key limit 1
	`, &apiKey, "groq")
		if err != nil {
			logErrorCode.Println("04-00-0000")
			return "JADE internal error: 04-00-0000. Please contact the support."
		}
	} else {
		apiKey = testApiKey
	}

	url := "https://api.groq.com/openai/v1/chat/completions"

	requestBody := OpenaiChatCompletionRequest{
		Model:       model,
		Messages:    Message2RequestMessage(messages, context),
		MaxTokens:   maxTokens,
		Temperature: temperature,
	}

	jsonBody, err := json.Marshal(requestBody)
	if err != nil {
		logErrorCode.Println("04-01-0001")
		return "JADE internal error: 04-01-0001. Please contact the support."
	}

	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
	if err != nil {
		logErrorCode.Println("04-02-0002")
		return "JADE internal error: 04-02-0002. Please contact the support."
	}

	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Authorization", "Bearer "+apiKey)

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		logErrorCode.Println("04-02-0003")
		return "JADE internal error: 04-02-0003. Please contact the support."
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		logErrorCode.Println("04-01-0004")
		return "JADE internal error: 04-01-0004. Please contact the support."
	}

	for key, value := range GroqErrorCodes {
		if strings.Contains(resp.Status, key) {
			return value
		}
	}

	var chatCompletionResponse OpenaiChatCompletionResponse
	err = json.Unmarshal(body, &chatCompletionResponse)
	if err != nil {
		logErrorCode.Println("04-01-0005")
		return "JADE internal error: 04-01-0005. Please contact the support."
	}

	if testApiKey != "" {
		return chatCompletionResponse.Choices[0].Message.Content
	}

	var usedModelInfo ModelInfo
	usedModelInfo, found := getModelInfoByID(llm.Model.ID)
	if !found {
		logErrorCode.Println("04-00-0006")
		return "JADE internal error: 04-00-0006. Please contact the support."
	}

	var inputCost float32 = float32(chatCompletionResponse.Usage.PromptTokens) * usedModelInfo.InputPrice
	var outputCost float32 = float32(chatCompletionResponse.Usage.CompletionTokens) * usedModelInfo.OutputPrice
	addUsage(c, inputCost, outputCost, chatCompletionResponse.Usage.PromptTokens, chatCompletionResponse.Usage.CompletionTokens, model)

	if len(chatCompletionResponse.Choices) == 0 {
		logErrorCode.Println("04-03-0007 -", resp.Status, "-", string(body))
		return "JADE internal error: 04-03-0007. Please contact the support."
	}

	return chatCompletionResponse.Choices[0].Message.Content
}