package config

import (
	"fmt"

	"github.com/BurntSushi/toml"
)

type Config struct {
	APIURL        string `toml:"APIURL"`
	ShowSys       bool   `toml:"ShowSys"`
	LogFile       string `toml:"LogFile"`
	UserRole      string `toml:"UserRole"`
	ToolRole      string `toml:"ToolRole"`
	AssistantRole string `toml:"AssistantRole"`
	AssistantIcon string `toml:"AssistantIcon"`
	UserIcon      string `toml:"UserIcon"`
	ToolIcon      string `toml:"ToolIcon"`
	ChunkLimit    uint32 `toml:"ChunkLimit"`
}

func LoadConfigOrDefault(fn string) *Config {
	if fn == "" {
		fn = "config.toml"
	}
	config := &Config{}
	_, err := toml.DecodeFile(fn, &config)
	if err != nil {
		fmt.Println("failed to read config from file, loading default")
		config.APIURL = "http://localhost:8080/v1/chat/completions"
		config.ShowSys = true
		config.LogFile = "log.txt"
		config.UserRole = "user"
		config.ToolRole = "tool"
		config.AssistantRole = "assistant"
		config.ChunkLimit = 8192
	}
	return config
}