package service

import (
	"ai-search/utils"

	"github.com/ilyakaznacheev/cleanenv"
	"github.com/joho/godotenv"
)

type AppSettings struct {
	ListenAddr             string       `env:"LISTEN_ADDR" env-default:":8080"`
	DBPath                 string       `env:"DB_PATH" env-default:"/litefs/db"`
	OpenAIEndpint          string       `env:"OPENAI_ENDPOINT" env-required:"true"`
	OpenAIAPIKey           string       `env:"OPENAI_API_KEY" env-required:"true"`
	OpenAIWriterPrompt     string       `env:"OPENAI_WRITER_PROMPT" env-default:"您是一名人工智能写作助手，可以根据先前文本的上下文继续现有文本。给予后面的字符比开始的字符更多的权重/优先级。而且绝对一定要尽量多使用中文！将您的回答限制在 200 个字符以内，但请确保构建完整的句子。"`
	HttpProxy              string       `env:"HTTP_PROXY"`
	IsDebug                bool         `env:"DEBUG" env-default:"false"`
	ChatMaxTokens          int          `env:"CHAT_MAX_TOKENS" env-default:"400"`
	OpenAIChatEndpoint     string       `env:"OPENAI_CHAT_ENDPOINT" env-required:"true"`
	OpenAIChatAPIKey       string       `env:"OPENAI_CHAT_API_KEY" env-required:"true"`
	OpenAIChatNetworkDelay int          `env:"OPENAI_CHAT_NETWORK_DELAY" env-default:"5"`
	OpenAIChatQueueLen     int          `env:"OPENAI_CHAT_QUEUE_LEN" env-default:"10"`
	CacheSize              int          `env:"CACHE_SIZE" env-default:"100"` // in MB
	RAGSearchCount         int          `env:"RAG_SEARCH_COUNT" env-default:"8"`
	RAGSearchCacheTime     int          `env:"RAG_SEARCH_CACHE_TIME" env-default:"1200"` // sec
	RPCEndpoints           RPCEndpoints `env-prefix:"RPC_"`
	Prompts                Prompts      `env-prefix:"PROMPT_"`
	RAGParams              RAGParams    `env-prefix:"RAG_"`
}

type RPCEndpoints struct {
	SearchURL string `env:"SEARCH_URL" env-required:"true"`
}

type Prompts struct {
	RAGPath           string `env:"RAG_PATH" env-default:""`
	MoreQuestionsPath string `env:"MORE_QUESTIONS_PATH" env-default:""`
}

type RAGParams struct {
	MaxTokens                int     `env:"MAX_TOKENS" env-default:"2048"`
	MoreQuestionsMaxTokens   int     `env:"MORE_QUESTIONS_MAX_TOKENS" env-default:"1024"`
	Temperature              float32 `env:"TEMPERATURE" env-default:"0.9"`
	MoreQuestionsTemperature float32 `env:"MORE_QUESTIONS_TEMPERATURE" env-default:"0.7"`
}

var (
	appSetting *AppSettings
)

func init() {
	godotenv.Load()
	conf := &AppSettings{}
	if err := cleanenv.ReadEnv(conf); err != nil {
		panic(err)
	}
	appSetting = conf
	InitCache()
	if len(appSetting.Prompts.MoreQuestionsPath) >= 0 {
		moreQuestionsPrompt = utils.GetFileContent(appSetting.Prompts.MoreQuestionsPath)
	}
	if len(appSetting.Prompts.RAGPath) >= 0 {
		ragPrompt = utils.GetFileContent(appSetting.Prompts.RAGPath)
	}
}

func GetSettings() *AppSettings {
	return appSetting
}