package env

import (
	"os"
	"strconv"
	"strings"

	tempest "github.com/Amatsagu/Tempest"
)

func GetEnvVariables() Env {
	ids := strings.Split(os.Getenv("DISCORD_SERVER_IDS"), ",")

	serverIDs := []tempest.Snowflake{}
	for _, id := range ids {
		serverIDs = append(serverIDs, tempest.StringToSnowflake(id))
	}

	halResPercent, err := strconv.Atoi(os.Getenv("HAL_RESPONSE_PERCENT"))
	if err != nil {
		panic(err)
	}

	return Env{
		AppID:          tempest.StringToSnowflake(os.Getenv("HAL_APP_ID")),
		PublicKey:      os.Getenv("HAL_PUBLIC_KEY"),
		Token:          os.Getenv("HAL_TOKEN"),
		ServerIDs:      serverIDs,
		OpenaiHalToken: os.Getenv("OPENAI_HAL"),

		GeminiToken:        os.Getenv("GEMINI_API_KEY"),
		HalResponsePercent: halResPercent,
		LlmEndpoint:        os.Getenv("LLM_ENDPOINT"),
		LlmToken:           os.Getenv("LLM_TOKEN"),
		LlmModel:           os.Getenv("LLM_MODEL"),
	}
}

type Env struct {
	AppID              tempest.Snowflake
	PublicKey          string
	Token              string
	ServerIDs          []tempest.Snowflake
	OpenaiHalToken     string
	GeminiToken        string
	HalResponsePercent int
	LlmEndpoint        string
	LlmToken           string
	LlmModel           string
}