hhs
106 lines
2.1 KiB
Go
106 lines
2.1 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/gin-contrib/cors"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
type ServerConfig struct {
|
|
Port string
|
|
}
|
|
|
|
type Neo4jConfig struct {
|
|
URI string
|
|
Username string
|
|
Password string
|
|
}
|
|
|
|
type DataConfig struct {
|
|
FilePath string
|
|
}
|
|
|
|
type CORSConfig struct {
|
|
AllowOrigins []string
|
|
AllowMethods []string
|
|
AllowHeaders []string
|
|
ExposeHeaders []string
|
|
AllowCredentials bool
|
|
MaxAge time.Duration
|
|
}
|
|
|
|
type LLMConfig struct {
|
|
Provider string
|
|
APIKey string
|
|
Model string
|
|
BaseURL string
|
|
}
|
|
|
|
type Config struct {
|
|
Server ServerConfig
|
|
Neo4j Neo4jConfig
|
|
Data DataConfig
|
|
CORS CORSConfig
|
|
LLM LLMConfig
|
|
}
|
|
|
|
func Load() (*Config, error) {
|
|
var err error
|
|
err = godotenv.Load("../configs/.env")
|
|
err = godotenv.Load("./configs/.env")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error loading .env file: %w", err)
|
|
}
|
|
|
|
cfg := &Config{
|
|
Server: ServerConfig{
|
|
Port: getEnv("PORT", "3001"),
|
|
},
|
|
Neo4j: Neo4jConfig{
|
|
URI: os.Getenv("NEO4J_URI"),
|
|
Username: os.Getenv("NEO4J_USERNAME"),
|
|
Password: os.Getenv("NEO4J_PASSWORD"),
|
|
},
|
|
Data: DataConfig{
|
|
FilePath: getEnv("DATA_FILE", "data.json"),
|
|
},
|
|
CORS: CORSConfig{
|
|
AllowOrigins: []string{"http://localhost:5173", "http://localhost:3000"},
|
|
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
|
AllowHeaders: []string{"Origin", "Content-Type", "Authorization", "Accept"},
|
|
ExposeHeaders: []string{"Content-Length", "Content-Type"},
|
|
AllowCredentials: true,
|
|
MaxAge: 12 * time.Hour,
|
|
},
|
|
LLM: LLMConfig{
|
|
Provider: os.Getenv("PROVIDER"),
|
|
APIKey: os.Getenv("API_KEY"),
|
|
Model: os.Getenv("MODEL"),
|
|
BaseURL: os.Getenv("BASE_URL"),
|
|
},
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|
|
|
|
func (c *CORSConfig) ToGinConfig() cors.Config {
|
|
return cors.Config{
|
|
AllowOrigins: c.AllowOrigins,
|
|
AllowMethods: c.AllowMethods,
|
|
AllowHeaders: c.AllowHeaders,
|
|
ExposeHeaders: c.ExposeHeaders,
|
|
AllowCredentials: c.AllowCredentials,
|
|
MaxAge: c.MaxAge,
|
|
}
|
|
}
|
|
|
|
func getEnv(key, defaultValue string) string {
|
|
if value := os.Getenv(key); value != "" {
|
|
return value
|
|
}
|
|
return defaultValue
|
|
}
|