Agent par défaut

backend/internal/http/router.go

@@ -1,6 +1,7 @@
 package http
 
 import (
+	"context"
 	"embed"
 	"fmt"
 	"io/fs"
@@ -30,7 +31,7 @@ var indexhtml []byte
 var robotstxt []byte
 
 type Router struct {
-	echo *echo.Echo
+	server *http.Server
 }
 
 type Dependencies struct {
@@ -44,6 +45,8 @@ type Dependencies struct {
 
 func NewRouter(deps Dependencies) *Router {
 
+	cfg := config.GetConfig()
+
 	e := echo.New()
 	e.Use(middleware.RequestLogger())
 	e.Use(middleware.CORS("http://localhost:5173"))
@@ -76,14 +79,23 @@ func NewRouter(deps Dependencies) *Router {
 	_ = handlers.NewUserGroup("/user", api, deps.UserService, deps.TokenManager.TokenMiddleware)
 	_ = handlers.NewMiscGroup("/misc", api, deps.TokenManager.TokenMiddleware)
 	_ = handlers.NewModelGroup("/provider", api, deps.ProviderService, deps.TokenManager.TokenMiddleware)
-	_ = handlers.NewAgentGroup("/agent", api, deps.AgentService, deps.TokenManager.TokenMiddleware)
+	_ = handlers.NewAgentGroup("/agent", api, deps.AgentService, deps.UserService, deps.TokenManager.TokenMiddleware)
+	_ = handlers.NewTeamGroup("/team", api, deps.UserService, deps.TokenManager.TokenMiddleware)
+
+	server := http.Server{
+		Addr:    fmt.Sprintf(":%d", cfg.HttpPort),
+		Handler: e,
+	}
 
 	return &Router{
-		echo: e,
+		&server,
 	}
 }
 
 func (r *Router) Start() error {
-	cfg := config.GetConfig()
-	return r.echo.Start(fmt.Sprintf(":%d", cfg.HttpPort))
+	return r.server.ListenAndServe()
+}
+
+func (r *Router) Stop(ctx context.Context) error {
+	return r.server.Shutdown(ctx)
 }