package main

import (
	"bufio"
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
)

type OpenAIRequest struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type OpenAIResponse struct {
	Choices []struct {
		Message Message `json:"message"`
	} `json:"choices"`
}

type Config struct {
	Instructions string `json:"instructions"`
	Model        string `json:"model"`
	Ruleset      string `json:"ruleset"`
	configApiKey string `json:"apiKey"`
}

func loadConfig(filename string) (Config, error) {
	var config Config
	file, err := ioutil.ReadFile(filename)
	if err != nil {
		return config, err
	}
	err = json.Unmarshal(file, &config)
	return config, err
}

func main() {
	apiKey := os.Getenv("OPENAI_API_KEY")

	config, err := loadConfig("config.json")
	if err != nil {
		fmt.Println("Error loading config file:", err)
		return
	}

	if apiKey == "" {
		apiKey = config.configApiKey
	}
	if apiKey == "" {
		fmt.Println("Error: Neither config api Variable nor OPENAI_API_KEY environment variable is set.")
		return
	}

	scanner := bufio.NewScanner(os.Stdin)
	messages := []Message{{Role: "system", Content: config.Instructions + " Only include information relevant for " + config.Ruleset + ". Do not include information from other rulesets."}}

	for {
		fmt.Print("Enter your prompt (or type 'exit' to quit): ")
		scanner.Scan()
		prompt := scanner.Text()

		if prompt == "exit" {
			fmt.Println("Goodbye!")
			break
		}

		messages = append(messages, Message{Role: "user", Content: prompt})

		requestBody, err := json.Marshal(OpenAIRequest{
			Model:    config.Model,
			Messages: messages,
		})
		if err != nil {
			fmt.Println("Error creating JSON request:", err)
			continue
		}

		req, err := http.NewRequest("POST", "https://api.openai.com/v1/chat/completions", bytes.NewBuffer(requestBody))
		if err != nil {
			fmt.Println("Error creating request:", err)
			continue
		}
		req.Header.Set("Content-Type", "application/json")
		req.Header.Set("Authorization", "Bearer "+apiKey)

		client := &http.Client{}
		resp, err := client.Do(req)
		if err != nil {
			fmt.Println("Error making request:", err)
			continue
		}
		defer resp.Body.Close()

		body, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			fmt.Println("Error reading response:", err)
			continue
		}

		var openAIResp OpenAIResponse
		if err := json.Unmarshal(body, &openAIResp); err != nil {
			fmt.Println("Error parsing JSON response:", err)
			continue
		}

		if len(openAIResp.Choices) > 0 {
			responseMessage := openAIResp.Choices[0].Message
			fmt.Println("AI Response:", responseMessage.Content)
			messages = append(messages, responseMessage)
		} else {
			fmt.Println("No response from AI.")
		}
	}
}