106 lines
3 KiB
Go
106 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
_ "github.com/joho/godotenv/autoload"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
var (
|
|
httpAddr string
|
|
logLevel string
|
|
zapLogLevel zapcore.Level
|
|
postFlagParse = []func(){}
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
_ = err
|
|
|
|
// -- app
|
|
flag.StringVar(&httpAddr, "addr", os.Getenv("HTTP_ADDR"), "Server address")
|
|
|
|
// -- log
|
|
flag.StringVar(&logLevel, "log-level", os.Getenv("LOG_LEVEL"), "Log level")
|
|
qpostFlagParse(func() {
|
|
err := zapLogLevel.UnmarshalText([]byte(logLevel))
|
|
if err != nil {
|
|
zapLogLevel = zapcore.InfoLevel
|
|
}
|
|
})
|
|
|
|
// -- OBS
|
|
flag.StringVar(&defaultObsOpts.Endpoint, "obs-endpoint", os.Getenv("OBS_ENDPOINT"), "OBS host")
|
|
flag.StringVar(&defaultObsOpts.Region, "obs-region", os.Getenv("OBS_REGION"), "OBS region")
|
|
flag.BoolVar(&defaultObsOpts.Secure, "obs-secure", ok1(strconv.ParseBool(os.Getenv("OBS_SECURE"))), "OBS secure transport")
|
|
flag.StringVar(&defaultObsOpts.BucketName, "obs-bucket", os.Getenv("OBS_BUCKET_NAME"), "OBS bucket name")
|
|
|
|
flag.BoolVar(&defaultObsOpts.RedirectSecure, "obs-redirect-secure", ok1(strconv.ParseBool(os.Getenv("OBS_REDIRECT_SECURE"))), "OBS redirect secure transport")
|
|
flag.StringVar(&defaultObsOpts.HostRedirect, "obs-host-redirect", os.Getenv("OBS_HOST_REDIRECT"), "OBS host redirect")
|
|
|
|
// redirect http code
|
|
var obsRedirectCode = int64(defaultObsOpts.RedirectCode)
|
|
if obsRedirectCodeStr := os.Getenv("OBS_REDIRECT_CODE"); obsRedirectCodeStr != "" {
|
|
obsRedirectCode, err = strconv.ParseInt(obsRedirectCodeStr, 10, 64)
|
|
if err != nil {
|
|
obsRedirectCode = int64(defaultObsOpts.RedirectCode)
|
|
}
|
|
}
|
|
flag.IntVar(&defaultObsOpts.RedirectCode, "obs-redirect-code", int(obsRedirectCode), "OBS redirect http code")
|
|
|
|
// url expiry
|
|
var obsUrlExpiry = defaultObsOpts.URLExpiry
|
|
if obsUrlExpiryStr := os.Getenv("OBS_URL_EXPIRY"); obsUrlExpiryStr != "" {
|
|
if obsUrlExpiry, err = time.ParseDuration(obsUrlExpiryStr); err != nil {
|
|
obsUrlExpiry = defaultObsOpts.URLExpiry
|
|
}
|
|
}
|
|
flag.DurationVar(&defaultObsOpts.URLExpiry, "obs-url-expiry", obsUrlExpiry, "OBS url expiry")
|
|
|
|
}
|
|
|
|
func qpostFlagParse(f func()) {
|
|
postFlagParse = append(postFlagParse, f)
|
|
}
|
|
func qpostFlagParseInvoke() {
|
|
for _, f := range postFlagParse {
|
|
f()
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
qpostFlagParseInvoke()
|
|
|
|
zcfg := zap.NewProductionConfig()
|
|
zcfg.Level = zap.NewAtomicLevelAt(zapLogLevel)
|
|
|
|
logger := unwrap1(zcfg.Build())
|
|
defer logger.Sync()
|
|
|
|
sug := logger.Named("main").Sugar()
|
|
sug.Infow("starting",
|
|
"log_level", zapLogLevel,
|
|
"obs_bucket", defaultObsOpts.BucketName,
|
|
"obs_endpoint", defaultObsOpts.Endpoint,
|
|
"obs_redirect_secure", defaultObsOpts.RedirectSecure,
|
|
"obs_host_redirect", defaultObsOpts.HostRedirect,
|
|
"obs_redirect_code", defaultObsOpts.RedirectCode,
|
|
"obs_url_expiry", defaultObsOpts.URLExpiry.String(),
|
|
)
|
|
|
|
client := unwrap1(newObsClient(defaultObsOpts))
|
|
srv.Init(serverOptions{
|
|
Addr: httpAddr,
|
|
Logger: logger.Named("server"),
|
|
OBS: &defaultObsOpts,
|
|
S3: client,
|
|
})
|
|
|
|
srv.Run()
|
|
}
|