mirror of https://github.com/JustKato/FreePad.git
90 lines
1.7 KiB
Go
90 lines
1.7 KiB
Go
package helper
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
func GetDomainBase() string {
|
|
domainBase, domainExists := os.LookupEnv("DOMAIN_BASE")
|
|
|
|
if !domainExists {
|
|
os.Setenv("DOMAIN_BASE", "http://localhost:8080")
|
|
domainBase = "http://localhost:8080"
|
|
}
|
|
|
|
return domainBase
|
|
}
|
|
|
|
func GetApiBanLimit() int {
|
|
banLimit, exists := os.LookupEnv("API_BAN_LIMIT")
|
|
|
|
if !exists {
|
|
os.Setenv("API_BAN_LIMIT", "300")
|
|
banLimit = "300"
|
|
}
|
|
|
|
// Try and convert the string into an integer
|
|
rez, err := strconv.Atoi(banLimit)
|
|
// Check if the conversion has failed
|
|
if err != nil {
|
|
// Simply return the default
|
|
return 300
|
|
}
|
|
|
|
return rez
|
|
}
|
|
|
|
func GetMaximumPadSize() int {
|
|
// Lookup if the maximum pad size variable exists.
|
|
maxPadSize, exists := os.LookupEnv("MAXIMUM_PAD_SIZE")
|
|
|
|
// Check if this environment variable has bee nset
|
|
if !exists {
|
|
// Set the variable ourselves to the default string value
|
|
maxPadSize = "524288"
|
|
}
|
|
|
|
// Try and convert the string into an integer
|
|
rez, err := strconv.Atoi(maxPadSize)
|
|
// Check if the conversion has failed
|
|
if err != nil {
|
|
// Simply return the default
|
|
return 524288
|
|
}
|
|
|
|
// Return the resulting value
|
|
return rez
|
|
}
|
|
|
|
func GetCacheMapLimit() int {
|
|
cacheMapLimit, domainExists := os.LookupEnv("CACHE_MAP_LIMIT")
|
|
|
|
if !domainExists {
|
|
os.Setenv("CACHE_MAP_LIMIT", "25")
|
|
cacheMapLimit = "25"
|
|
}
|
|
|
|
rez, err := strconv.Atoi(cacheMapLimit)
|
|
if err != nil {
|
|
return 25
|
|
}
|
|
|
|
return rez
|
|
}
|
|
|
|
// Get the admin token used to authenticate as an admin
|
|
func GetAdminToken() string {
|
|
// Get the admin login from the environment
|
|
adminToken, exists := os.LookupEnv("ADMIN_TOKEN")
|
|
|
|
// Check if the admin token was defined
|
|
if !exists {
|
|
// The admin token was not defined, disable admin logins
|
|
return ""
|
|
}
|
|
|
|
// Return the admin token
|
|
return adminToken
|
|
}
|