118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"io/fs"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/wotra/wotra/internal/config"
|
|
"github.com/wotra/wotra/internal/handler"
|
|
"github.com/wotra/wotra/internal/service"
|
|
"github.com/wotra/wotra/internal/store"
|
|
)
|
|
|
|
func main() {
|
|
logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
|
|
slog.SetDefault(logger)
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
slog.Error("config error", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
tz, err := time.LoadLocation(cfg.Timezone)
|
|
if err != nil {
|
|
slog.Error("invalid timezone", "tz", cfg.Timezone, "err", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
db, err := store.Open(cfg.DBPath)
|
|
if err != nil {
|
|
slog.Error("database error", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
defer db.Close()
|
|
|
|
entryStore := store.NewEntryStore(db)
|
|
closedDayStore := store.NewClosedDayStore(db)
|
|
closedWeekStore := store.NewClosedWeekStore(db)
|
|
settingsStore := store.NewSettingsStore(db)
|
|
syncStore := store.NewSyncStore(db)
|
|
|
|
entrySvc := service.NewEntryService(entryStore, closedDayStore, settingsStore, tz)
|
|
daySvc := service.NewDayService(entryStore, closedDayStore, settingsStore, tz)
|
|
settingsSvc := service.NewSettingsService(settingsStore)
|
|
weekSvc := service.NewWeekService(closedDayStore, closedWeekStore, settingsStore, db, tz)
|
|
|
|
// Background goroutine: auto-stop entries that cross midnight
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
go runMidnightGuard(ctx, entrySvc)
|
|
|
|
// Static SPA files (embedded or from disk for dev)
|
|
var staticFS fs.FS
|
|
if webFS != nil {
|
|
staticFS = webFS
|
|
}
|
|
|
|
router := handler.NewRouter(cfg.AuthToken, entrySvc, daySvc, settingsSvc, weekSvc, syncStore, staticFS)
|
|
|
|
srv := &http.Server{
|
|
Addr: ":" + cfg.Port,
|
|
Handler: router,
|
|
ReadTimeout: 15 * time.Second,
|
|
WriteTimeout: 15 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
slog.Info("starting server", "port", cfg.Port, "db", cfg.DBPath, "tz", cfg.Timezone)
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
slog.Error("server error", "err", err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
<-quit
|
|
slog.Info("shutting down")
|
|
cancel()
|
|
|
|
shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer shutdownCancel()
|
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
|
slog.Error("shutdown error", "err", err)
|
|
}
|
|
}
|
|
|
|
// runMidnightGuard checks every minute whether any running entries span into a new day
|
|
// and auto-stops them.
|
|
func runMidnightGuard(ctx context.Context, svc *service.EntryService) {
|
|
ticker := time.NewTicker(time.Minute)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
ids, err := svc.AutoStopStalledEntries(ctx)
|
|
if err != nil {
|
|
slog.Error("midnight guard error", "err", err)
|
|
continue
|
|
}
|
|
if len(ids) > 0 {
|
|
slog.Info("auto-stopped entries crossing midnight", "ids", ids)
|
|
}
|
|
}
|
|
}
|
|
}
|