Mikkel Georgsen
59aa89b199
- router.go: NewRouter accepts intakeHandler http.Handler, registers POST /api/intake - config.go: adds NetBoxDefaultDeviceTypeID/RoleID/SiteID fields with defaults and env bindings - main.go: creates netbox.Client, ai.Orchestrator, inventory.CatalogUpdater, handlers.IntakeHandler - main.go: replaces NoOpHandler with NewNetBoxOpHandler(nbClient) for WAQ worker - main.go: uses typed interface variable for WAQ to avoid nil-interface-wrapping bug
100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
hwlab "git.georgsen.dk/hwlab"
|
|
"git.georgsen.dk/hwlab/internal/ai"
|
|
"git.georgsen.dk/hwlab/internal/api"
|
|
"git.georgsen.dk/hwlab/internal/api/handlers"
|
|
"git.georgsen.dk/hwlab/internal/config"
|
|
"git.georgsen.dk/hwlab/internal/inventory"
|
|
"git.georgsen.dk/hwlab/internal/netbox"
|
|
"git.georgsen.dk/hwlab/internal/queue"
|
|
)
|
|
|
|
func main() {
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
log.Fatalf("config: %v", err)
|
|
}
|
|
|
|
staticFS, err := fs.Sub(hwlab.StaticFiles, "web/dist")
|
|
if err != nil {
|
|
log.Fatalf("embed: %v", err)
|
|
}
|
|
|
|
// Context for graceful shutdown
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
// NetBox client (required — fatal if misconfigured)
|
|
nbClient, err := netbox.NewClient(cfg.NetBoxURL, cfg.NetBoxToken)
|
|
if err != nil {
|
|
log.Fatalf("netbox client: %v", err)
|
|
}
|
|
|
|
// AI tier clients and orchestrator
|
|
tier1 := ai.NewTierClient(cfg.AI.Tier1)
|
|
tier2 := ai.NewTierClient(cfg.AI.Tier2)
|
|
orch := ai.NewOrchestrator(tier1, tier2, cfg.AI.ConfidenceThreshold)
|
|
|
|
// Catalog updater
|
|
catalogUpdater := inventory.NewCatalogUpdater(nbClient)
|
|
|
|
// Start write-ahead queue worker (non-fatal if DragonFlyDB unavailable).
|
|
// waqForHandler is typed as the interface so that a nil *WAQ is not wrapped in a non-nil interface.
|
|
var waqForHandler handlers.IntakeWAQ
|
|
waqInstance, waqErr := queue.NewWAQ(cfg.DragonflyURL)
|
|
if waqErr != nil {
|
|
log.Printf("WARNING: WAQ unavailable (%v) — NetBox operations will not be queued during downtime", waqErr)
|
|
} else {
|
|
nbHandler := queue.NewNetBoxOpHandler(nbClient)
|
|
retryInterval := time.Duration(cfg.WAQRetryIntervalSeconds) * time.Second
|
|
go waqInstance.RunWorker(ctx, nbHandler, cfg.WAQMaxAttempts, retryInterval)
|
|
defer waqInstance.Close()
|
|
waqForHandler = waqInstance
|
|
log.Printf("WAQ worker started")
|
|
}
|
|
|
|
// Intake handler — waqForHandler may be nil; handler handles nil gracefully
|
|
intakeHandler := handlers.NewIntakeHandler(
|
|
orch,
|
|
nbClient,
|
|
catalogUpdater,
|
|
waqForHandler,
|
|
cfg.NetBoxDefaultDeviceTypeID,
|
|
cfg.NetBoxDefaultRoleID,
|
|
cfg.NetBoxDefaultSiteID,
|
|
cfg.AI.QuickAddEnabled,
|
|
cfg.AI.QuickAddThreshold,
|
|
)
|
|
|
|
router := api.NewRouter(staticFS, intakeHandler)
|
|
addr := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
|
|
log.Printf("HWLab starting on %s", addr)
|
|
|
|
srv := &http.Server{Addr: addr, Handler: router}
|
|
go func() {
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("server: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Wait for shutdown signal
|
|
<-ctx.Done()
|
|
log.Println("Shutting down...")
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
|
log.Printf("server shutdown: %v", err)
|
|
}
|
|
log.Println("Shutdown complete")
|
|
}
|