mirror of
https://github.com/Forceu/Gokapi.git
synced 2026-02-04 07:28:32 -06:00
101 lines
3.0 KiB
Go
101 lines
3.0 KiB
Go
package environment
|
|
|
|
import (
|
|
"fmt"
|
|
envParser "github.com/caarlos0/env/v6"
|
|
"github.com/forceu/gokapi/internal/configuration/database"
|
|
"github.com/forceu/gokapi/internal/environment/flagparser"
|
|
"github.com/forceu/gokapi/internal/helper"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
// DefaultPort for the webserver
|
|
const DefaultPort = 53842
|
|
|
|
// Environment is a struct containing available env variables
|
|
type Environment struct {
|
|
ConfigDir string `env:"CONFIG_DIR" envDefault:"config"`
|
|
ConfigFile string `env:"CONFIG_FILE" envDefault:"config.json"`
|
|
DataDir string `env:"DATA_DIR" envDefault:"data"`
|
|
WebserverPort int `env:"PORT" envDefault:"53842"`
|
|
LengthId int `env:"LENGTH_ID" envDefault:"15"`
|
|
MaxMemory int `env:"MAX_MEMORY_UPLOAD" envDefault:"40"`
|
|
MaxFileSize int `env:"MAX_FILESIZE" envDefault:"102400"` // 102400==100GB
|
|
AwsBucket string `env:"AWS_BUCKET"`
|
|
AwsRegion string `env:"AWS_REGION"`
|
|
AwsKeyId string `env:"AWS_KEY"`
|
|
AwsKeySecret string `env:"AWS_KEY_SECRET"`
|
|
AwsEndpoint string `env:"AWS_ENDPOINT"`
|
|
ConfigPath string
|
|
FileDbPath string
|
|
FileDb string `env:"FILE_DB" envDefault:"filestorage.db"`
|
|
}
|
|
|
|
// New parses the env variables
|
|
func New() Environment {
|
|
result := Environment{WebserverPort: DefaultPort}
|
|
err := envParser.Parse(&result, envParser.Options{
|
|
Prefix: "GOKAPI_",
|
|
})
|
|
if err != nil {
|
|
fmt.Println("Error parsing env variables:", err)
|
|
osExit(1)
|
|
return Environment{}
|
|
}
|
|
helper.Check(err)
|
|
|
|
flags := flagparser.ParseFlags()
|
|
if flags.IsPortSet {
|
|
result.WebserverPort = flags.Port
|
|
}
|
|
if flags.IsConfigDirSet {
|
|
result.ConfigDir = flags.ConfigDir
|
|
}
|
|
if flags.IsDataDirSet {
|
|
result.DataDir = flags.DataDir
|
|
}
|
|
|
|
result.ConfigPath = result.ConfigDir + "/" + result.ConfigFile
|
|
if flags.IsConfigPathSet {
|
|
result.ConfigPath = flags.ConfigPath
|
|
}
|
|
result.FileDbPath = result.DataDir + "/" + result.FileDb
|
|
if IsDockerInstance() && os.Getenv("TMPDIR") == "" {
|
|
os.Setenv("TMPDIR", result.DataDir)
|
|
}
|
|
if result.LengthId < 5 {
|
|
result.LengthId = 5
|
|
}
|
|
if result.LengthId > database.GetLengthAvailable() {
|
|
result.LengthId = database.GetLengthAvailable()
|
|
fmt.Println("Reduced ID length to " + strconv.Itoa(database.GetLengthAvailable()) + " due to database constraints")
|
|
}
|
|
if result.MaxMemory < 5 {
|
|
result.MaxMemory = 5
|
|
}
|
|
if result.MaxFileSize < 1 {
|
|
result.MaxFileSize = 5
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
// IsAwsProvided returns true if all required env variables have been set for using AWS S3 / Backblaze
|
|
func (e *Environment) IsAwsProvided() bool {
|
|
return e.AwsBucket != "" &&
|
|
e.AwsRegion != "" &&
|
|
e.AwsKeyId != "" &&
|
|
e.AwsKeySecret != ""
|
|
}
|
|
|
|
// GetConfigPaths returns the config paths to config files and the directory containing the files. The following results are returned:
|
|
// Path to config file, Path to directory containing config file, Name of config file, Path to AWS config file
|
|
func GetConfigPaths() (string, string, string, string) {
|
|
env := New()
|
|
awsConfigPAth := env.ConfigDir + "/cloudconfig.yml"
|
|
return env.ConfigPath, env.ConfigDir, env.ConfigFile, awsConfigPAth
|
|
}
|
|
|
|
var osExit = os.Exit
|