110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/jacek/pamietnik/backend/internal/api"
|
|
"github.com/jacek/pamietnik/backend/internal/auth"
|
|
"github.com/jacek/pamietnik/backend/internal/db"
|
|
)
|
|
|
|
func main() {
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
slog.SetDefault(logger)
|
|
|
|
dsn := getenv("DATABASE_URL", "postgres://pamietnik:pamietnik@localhost:5432/pamietnik?sslmode=disable")
|
|
addr := getenv("LISTEN_ADDR", ":8080")
|
|
uploadDir := getenv("UPLOAD_DIR", "./uploads")
|
|
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
pool, err := db.NewPool(ctx, dsn)
|
|
if err != nil {
|
|
slog.Error("connect db", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
defer pool.Close()
|
|
slog.Info("database connected")
|
|
|
|
if err := db.InitSchema(ctx, pool); err != nil {
|
|
slog.Error("init schema", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
slog.Info("schema ready")
|
|
|
|
if adminUser := os.Getenv("ADMIN_USER"); adminUser != "" {
|
|
adminPass := os.Getenv("ADMIN_PASSWORD")
|
|
if adminPass == "" {
|
|
slog.Error("ADMIN_USER set but ADMIN_PASSWORD is empty")
|
|
os.Exit(1)
|
|
}
|
|
hash, err := auth.HashPassword(adminPass)
|
|
if err != nil {
|
|
slog.Error("hash admin password", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
created, err := db.SeedAdminUser(ctx, pool, adminUser, hash)
|
|
if err != nil {
|
|
slog.Error("seed admin user", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
if created {
|
|
slog.Info("admin user created", "username", adminUser)
|
|
}
|
|
}
|
|
|
|
if err := os.MkdirAll(uploadDir, 0o755); err != nil {
|
|
slog.Error("create upload dir", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
authStore := auth.NewStore(pool)
|
|
tpStore := db.NewTrackpointStore(pool)
|
|
stopStore := db.NewStopStore(pool)
|
|
suggStore := db.NewSuggestionStore(pool)
|
|
journalStore := db.NewJournalStore(pool)
|
|
userStore := db.NewUserStore(pool)
|
|
|
|
router := api.NewRouter(authStore, tpStore, stopStore, suggStore, journalStore, userStore, uploadDir)
|
|
|
|
srv := &http.Server{
|
|
Addr: addr,
|
|
Handler: router,
|
|
ReadTimeout: 15 * time.Second,
|
|
WriteTimeout: 30 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
slog.Info("server starting", "addr", addr)
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
slog.Error("server error", "err", err)
|
|
cancel()
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
slog.Info("shutting down")
|
|
|
|
shutCtx, shutCancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer shutCancel()
|
|
if err := srv.Shutdown(shutCtx); err != nil {
|
|
slog.Error("shutdown error", "err", err)
|
|
}
|
|
slog.Info("server stopped")
|
|
}
|
|
|
|
func getenv(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|