97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"shop/internal/auth"
|
|
"shop/internal/config"
|
|
"shop/internal/database"
|
|
"shop/internal/handlers"
|
|
"shop/internal/middleware"
|
|
"shop/internal/repository"
|
|
)
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
cfg := config.Load()
|
|
|
|
pool, err := database.Connect(ctx)
|
|
if err != nil {
|
|
log.Fatalf("database: %v", err)
|
|
}
|
|
defer pool.Close()
|
|
|
|
if err := database.Migrate(ctx, pool); err != nil {
|
|
log.Fatalf("migrate: %v", err)
|
|
}
|
|
|
|
productRepo := repository.NewProductRepository(pool)
|
|
adminRepo := repository.NewAdminRepository(pool)
|
|
|
|
if cfg.AdminEmail != "" && cfg.AdminPassword != "" {
|
|
if err := adminRepo.Upsert(ctx, cfg.AdminEmail, cfg.AdminPassword); err != nil {
|
|
log.Fatalf("admin seed: %v", err)
|
|
}
|
|
log.Printf("admin ready: %s", cfg.AdminEmail)
|
|
} else {
|
|
log.Println("warning: set ADMIN_EMAIL and ADMIN_PASSWORD in .env to create admin")
|
|
}
|
|
|
|
home, err := handlers.NewHomeHandler(productRepo)
|
|
if err != nil {
|
|
log.Fatalf("handler: %v", err)
|
|
}
|
|
|
|
session := auth.NewSession(cfg.SessionSecret)
|
|
admin := handlers.NewAdminHandler(adminRepo, productRepo, session, home.Templates())
|
|
|
|
mux := http.NewServeMux()
|
|
mux.Handle("GET /static/", http.StripPrefix("/static/", home.Static()))
|
|
mux.HandleFunc("GET /health", home.Health)
|
|
mux.HandleFunc("GET /{$}", home.Home)
|
|
|
|
mux.HandleFunc("GET /admin/login", admin.LoginPage)
|
|
mux.HandleFunc("POST /admin/login", admin.Login)
|
|
mux.HandleFunc("POST /admin/logout", admin.Logout)
|
|
mux.HandleFunc("GET /admin/{$}", admin.Dashboard)
|
|
|
|
addr := ":" + getEnv("APP_PORT", "8080")
|
|
srv := &http.Server{
|
|
Addr: addr,
|
|
Handler: middleware.TrustProxy(mux),
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 30 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
log.Printf("server listening on %s", addr)
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("listen: %v", err)
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
|
log.Printf("shutdown: %v", err)
|
|
}
|
|
}
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|