70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"runtime"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/SowinskiBraeden/SulliCartShare/models"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
// Config holds the project config values
|
|
type Config struct {
|
|
URL string
|
|
DatabaseName string
|
|
BaseURL string
|
|
Port string
|
|
}
|
|
|
|
// New sets up all config related services
|
|
func New() *Config {
|
|
|
|
if runtime.GOOS == "windows" {
|
|
_ = godotenv.Load()
|
|
}
|
|
|
|
//setup zap logger and replace default Logger
|
|
logger, err := setLogger(os.Getenv("ENV"))
|
|
if err != nil {
|
|
// if we get an error, we will just set the default to debug and move on
|
|
zap.S().With(err).Warn("issue setting logger")
|
|
}
|
|
defer logger.Sync()
|
|
_ = zap.ReplaceGlobals(logger)
|
|
|
|
return &Config{
|
|
URL: os.Getenv("DB_URI"),
|
|
DatabaseName: os.Getenv("DB_NAME"),
|
|
BaseURL: os.Getenv("BASE_URL"),
|
|
Port: os.Getenv("PORT"),
|
|
}
|
|
}
|
|
|
|
// ErrorStatus is a useful function that will log, write http headers and body for a
|
|
// given message, status code and error
|
|
func ErrorStatus(message string, httpStatusCode int, w http.ResponseWriter, err error) {
|
|
zap.S().With(err).Error(message)
|
|
w.WriteHeader(httpStatusCode)
|
|
b, _ := json.Marshal(models.ErrorMessageResponse{Response: models.MessageError{Message: message, Error: err.Error()}})
|
|
w.Write(b)
|
|
}
|
|
|
|
// setLogger is a helper function to set the Logger based on the environment
|
|
func setLogger(env string) (*zap.Logger, error) {
|
|
switch env {
|
|
case "production":
|
|
return zap.NewProduction()
|
|
case "development":
|
|
return zap.NewDevelopment()
|
|
case "local":
|
|
return zap.NewExample(), nil
|
|
default:
|
|
return zap.NewExample(), fmt.Errorf("cannon find ENV car so defaulting to debug logging")
|
|
}
|
|
}
|