refactor: changed file structure

pkg/config/config.go

@@ -0,0 +1,118 @@
+package config
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"projectreshoot/pkg/logging"
+	"projectreshoot/pkg/tmdb"
+
+	"github.com/joho/godotenv"
+	"github.com/pkg/errors"
+	"github.com/rs/zerolog"
+)
+
+type Config struct {
+	Host               string        // Host to listen on
+	Port               string        // Port to listen on
+	TrustedHost        string        // Domain/Hostname to accept as trusted
+	SSL                bool          // Flag for SSL Mode
+	GZIP               bool          // Flag for GZIP compression on requests
+	ReadHeaderTimeout  time.Duration // Timeout for reading request headers in seconds
+	WriteTimeout       time.Duration // Timeout for writing requests in seconds
+	IdleTimeout        time.Duration // Timeout for idle connections in seconds
+	DBName             string        // Filename of the db - hardcoded and doubles as DB version
+	DBLockTimeout      time.Duration // Timeout for acquiring database lock
+	SecretKey          string        // Secret key for signing tokens
+	AccessTokenExpiry  int64         // Access token expiry in minutes
+	RefreshTokenExpiry int64         // Refresh token expiry in minutes
+	TokenFreshTime     int64         // Time for tokens to stay fresh in minutes
+	LogLevel           zerolog.Level // Log level for global logging. Defaults to info
+	LogOutput          string        // "file", "console", or "both". Defaults to console
+	LogDir             string        // Path to create log files
+	TMDBToken          string        // Read access token for TMDB API
+	TMDBConfig         *tmdb.Config  // Config data for interfacing with TMDB
+}
+
+// Load the application configuration and get a pointer to the Config object
+func GetConfig(args map[string]string) (*Config, error) {
+	godotenv.Load(".env")
+	var (
+		host      string
+		port      string
+		logLevel  zerolog.Level
+		logOutput string
+		valid     bool
+	)
+
+	if args["host"] != "" {
+		host = args["host"]
+	} else {
+		host = GetEnvDefault("HOST", "127.0.0.1")
+	}
+	if args["port"] != "" {
+		port = args["port"]
+	} else {
+		port = GetEnvDefault("PORT", "3010")
+	}
+	if args["loglevel"] != "" {
+		logLevel = logging.GetLogLevel(args["loglevel"])
+	} else {
+		logLevel = logging.GetLogLevel(GetEnvDefault("LOG_LEVEL", "info"))
+	}
+	if args["logoutput"] != "" {
+		opts := map[string]string{
+			"both":    "both",
+			"file":    "file",
+			"console": "console",
+		}
+		logOutput, valid = opts[args["logoutput"]]
+		if !valid {
+			logOutput = "console"
+			fmt.Println(
+				"Log output type was not parsed correctly. Defaulting to console only",
+			)
+		}
+	} else {
+		logOutput = GetEnvDefault("LOG_OUTPUT", "console")
+	}
+	if logOutput != "both" && logOutput != "console" && logOutput != "file" {
+		logOutput = "console"
+	}
+	tmdbcfg, err := tmdb.GetConfig(os.Getenv("TMDB_API_TOKEN"))
+	if err != nil {
+		return nil, errors.Wrap(err, "tmdb.GetConfig")
+	}
+
+	config := &Config{
+		Host:               host,
+		Port:               port,
+		TrustedHost:        GetEnvDefault("TRUSTED_HOST", "127.0.0.1"),
+		SSL:                GetEnvBool("SSL_MODE", false),
+		GZIP:               GetEnvBool("GZIP", false),
+		ReadHeaderTimeout:  GetEnvDur("READ_HEADER_TIMEOUT", 2),
+		WriteTimeout:       GetEnvDur("WRITE_TIMEOUT", 10),
+		IdleTimeout:        GetEnvDur("IDLE_TIMEOUT", 120),
+		DBName:             "00001",
+		DBLockTimeout:      GetEnvDur("DB_LOCK_TIMEOUT", 60),
+		SecretKey:          os.Getenv("SECRET_KEY"),
+		AccessTokenExpiry:  GetEnvInt64("ACCESS_TOKEN_EXPIRY", 5),
+		RefreshTokenExpiry: GetEnvInt64("REFRESH_TOKEN_EXPIRY", 1440), // defaults to 1 day
+		TokenFreshTime:     GetEnvInt64("TOKEN_FRESH_TIME", 5),
+		LogLevel:           logLevel,
+		LogOutput:          logOutput,
+		LogDir:             GetEnvDefault("LOG_DIR", ""),
+		TMDBToken:          os.Getenv("TMDB_API_TOKEN"),
+		TMDBConfig:         tmdbcfg,
+	}
+
+	if config.SecretKey == "" && args["dbver"] != "true" {
+		return nil, errors.New("Envar not set: SECRET_KEY")
+	}
+	if config.TMDBToken == "" && args["dbver"] != "true" {
+		return nil, errors.New("Envar not set: TMDB_API_TOKEN")
+	}
+
+	return config, nil
+}

pkg/config/environment.go

@@ -0,0 +1,94 @@
+package config
+
+import (
+	"os"
+	"strconv"
+	"strings"
+	"time"
+)
+
+// Get an environment variable, specifying a default value if its not set
+func GetEnvDefault(key string, defaultValue string) string {
+	val, exists := os.LookupEnv(key)
+	if !exists {
+		return defaultValue
+	}
+	return val
+}
+
+// Get an environment variable as a time.Duration, specifying a default value if its
+// not set or can't be parsed properly
+func GetEnvDur(key string, defaultValue time.Duration) time.Duration {
+	val, exists := os.LookupEnv(key)
+	if !exists {
+		return time.Duration(defaultValue)
+	}
+
+	intVal, err := strconv.Atoi(val)
+	if err != nil {
+		return time.Duration(defaultValue)
+	}
+	return time.Duration(intVal)
+
+}
+
+// Get an environment variable as an int, specifying a default value if its
+// not set or can't be parsed properly into an int
+func GetEnvInt(key string, defaultValue int) int {
+	val, exists := os.LookupEnv(key)
+	if !exists {
+		return defaultValue
+	}
+
+	intVal, err := strconv.Atoi(val)
+	if err != nil {
+		return defaultValue
+	}
+	return intVal
+
+}
+
+// Get an environment variable as an int64, specifying a default value if its
+// not set or can't be parsed properly into an int64
+func GetEnvInt64(key string, defaultValue int64) int64 {
+	val, exists := os.LookupEnv(key)
+	if !exists {
+		return defaultValue
+	}
+
+	intVal, err := strconv.ParseInt(val, 10, 64)
+	if err != nil {
+		return defaultValue
+	}
+	return intVal
+
+}
+
+// Get an environment variable as a boolean, specifying a default value if its
+// not set or can't be parsed properly into a bool
+func GetEnvBool(key string, defaultValue bool) bool {
+	val, exists := os.LookupEnv(key)
+	if !exists {
+		return defaultValue
+	}
+	truthy := map[string]bool{
+		"true": true, "t": true, "yes": true, "y": true, "on": true, "1": true,
+		"enable": true, "enabled": true, "active": true, "affirmative": true,
+	}
+
+	falsy := map[string]bool{
+		"false": false, "f": false, "no": false, "n": false, "off": false, "0": false,
+		"disable": false, "disabled": false, "inactive": false, "negative": false,
+	}
+
+	normalized := strings.TrimSpace(strings.ToLower(val))
+
+	if val, ok := truthy[normalized]; ok {
+		return val
+	}
+	if val, ok := falsy[normalized]; ok {
+		return val
+	}
+
+	return defaultValue
+}