67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
"screenmark/screenmark/agents/client"
|
|
"screenmark/screenmark/auth"
|
|
"screenmark/screenmark/images"
|
|
"screenmark/screenmark/limits"
|
|
"screenmark/screenmark/models"
|
|
"screenmark/screenmark/notifications"
|
|
"screenmark/screenmark/processor"
|
|
"screenmark/screenmark/stacks"
|
|
|
|
ourmiddleware "screenmark/screenmark/middleware"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
)
|
|
|
|
type TestAiClient struct {
|
|
ImageInfo client.ImageMessageContent
|
|
}
|
|
|
|
func (client TestAiClient) GetImageInfo(imageName string, imageData []byte) (client.ImageMessageContent, error) {
|
|
return client.ImageInfo, nil
|
|
}
|
|
|
|
func setupRouter(db *sql.DB, jwtManager *ourmiddleware.JwtManager) (chi.Router, error) {
|
|
limitsManager := limits.CreateLimitsManager(db)
|
|
|
|
imageModel := models.NewImageModel(db)
|
|
stackModel := models.NewStackModel(db)
|
|
|
|
notifier := notifications.NewNotifier[notifications.Notification](10)
|
|
|
|
imageProcessorLogger := createLogger("Image Processor", os.Stdout)
|
|
imageProcessor, err := processor.NewImageProcessor(imageProcessorLogger, imageModel, stackModel, limitsManager, ¬ifier)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("processor: %w", err)
|
|
}
|
|
|
|
go imageProcessor.Processor.Work()
|
|
|
|
stackHandler := stacks.CreateStackHandler(db, limitsManager, jwtManager)
|
|
authHandler := auth.CreateAuthHandler(db, jwtManager)
|
|
imageHandler := images.CreateImageHandler(db, limitsManager, jwtManager, imageProcessor.Processor)
|
|
|
|
r := chi.NewRouter()
|
|
|
|
r.Use(middleware.Logger)
|
|
r.Use(ourmiddleware.CorsMiddleware)
|
|
|
|
r.Route("/stacks", stackHandler.CreateRoutes)
|
|
r.Route("/auth", authHandler.CreateRoutes)
|
|
r.Route("/images", imageHandler.CreateRoutes)
|
|
|
|
r.Route("/notifications", func(r chi.Router) {
|
|
r.Use(ourmiddleware.GetUserIdFromUrl(jwtManager))
|
|
|
|
r.Get("/", CreateEventsHandler(¬ifier))
|
|
})
|
|
|
|
return r, nil
|
|
}
|