142 lines
2.9 KiB
Go
142 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"RenewCMS/api"
|
|
validator "RenewCMS/api/pkg/validator"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
var version = "dev"
|
|
|
|
const (
|
|
readHeaderTimeout = 10 * time.Second
|
|
readTimeout = 30 * time.Second
|
|
writeTimeout = 60 * time.Second
|
|
idleTimeout = 120 * time.Second
|
|
shutdownTimeout = 10 * time.Second
|
|
)
|
|
|
|
var possibleEnvFileLocations = []string{".env", "../.env"}
|
|
|
|
var requiredEnvVars = []string{
|
|
"PUBLIC_URL",
|
|
"JWT_SECRET",
|
|
"SMTP_EMAIL",
|
|
"SMTP_PASSWORD",
|
|
"SMTP_HOST",
|
|
"SMTP_PORT",
|
|
}
|
|
|
|
var envDefaults = map[string]string{
|
|
"ENVIRONMENT": "production",
|
|
"PORT": "8080",
|
|
"DB_FILE": "./data/renewcms.db",
|
|
"UPLOAD_DIR": "./data/uploads",
|
|
}
|
|
|
|
func main() {
|
|
if err := run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func run() error {
|
|
initEnvVariables()
|
|
|
|
container, closeDb := api.InitContainer()
|
|
defer closeDatabase(closeDb)
|
|
validator.InitValidator()
|
|
|
|
server := newServer(api.InitRoutes(container))
|
|
return serveUntilStopped(server)
|
|
}
|
|
|
|
func initEnvVariables() {
|
|
loadEnvFile()
|
|
applyEnvDefaults()
|
|
ensureRequiredEnvVars()
|
|
}
|
|
|
|
func loadEnvFile() {
|
|
for _, envLocation := range possibleEnvFileLocations {
|
|
if godotenv.Load(envLocation) == nil {
|
|
return
|
|
}
|
|
}
|
|
log.Println("WARNING: Could not load any .env file")
|
|
}
|
|
|
|
func applyEnvDefaults() {
|
|
for envVar, value := range envDefaults {
|
|
if _, isSet := os.LookupEnv(envVar); !isSet {
|
|
_ = os.Setenv(envVar, value)
|
|
}
|
|
}
|
|
}
|
|
|
|
func ensureRequiredEnvVars() {
|
|
for _, envVar := range requiredEnvVars {
|
|
if _, isSet := os.LookupEnv(envVar); !isSet {
|
|
panic(fmt.Sprintf("Environment variable %s is not set", envVar))
|
|
}
|
|
}
|
|
}
|
|
|
|
func closeDatabase(closeDb func() error) {
|
|
if err := closeDb(); err != nil {
|
|
log.Println("error closing database:", err)
|
|
}
|
|
}
|
|
|
|
func newServer(handler http.Handler) *http.Server {
|
|
return &http.Server{
|
|
Addr: ":" + os.Getenv("PORT"),
|
|
Handler: handler,
|
|
ReadHeaderTimeout: readHeaderTimeout,
|
|
ReadTimeout: readTimeout,
|
|
WriteTimeout: writeTimeout,
|
|
IdleTimeout: idleTimeout,
|
|
}
|
|
}
|
|
|
|
func serveUntilStopped(server *http.Server) error {
|
|
stopSignal, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
serverErr := make(chan error, 1)
|
|
go listen(server, serverErr)
|
|
|
|
select {
|
|
case err := <-serverErr:
|
|
return err
|
|
case <-stopSignal.Done():
|
|
return shutdown(server)
|
|
}
|
|
}
|
|
|
|
func listen(server *http.Server, serverErr chan<- error) {
|
|
log.Printf("RenewCMS %s (%s) listening on %s, public URL %s", version, os.Getenv("ENVIRONMENT"), server.Addr, os.Getenv("PUBLIC_URL"))
|
|
err := server.ListenAndServe()
|
|
if errors.Is(err, http.ErrServerClosed) {
|
|
err = nil
|
|
}
|
|
serverErr <- err
|
|
}
|
|
|
|
func shutdown(server *http.Server) error {
|
|
log.Println("shutting down...")
|
|
ctx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
|
|
defer cancel()
|
|
return server.Shutdown(ctx)
|
|
}
|