83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"os"
|
|
"screenmark/screenmark/agents/client"
|
|
"screenmark/screenmark/auth"
|
|
"screenmark/screenmark/imageprocessor"
|
|
"screenmark/screenmark/images"
|
|
"screenmark/screenmark/limits"
|
|
"screenmark/screenmark/models"
|
|
"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) chi.Router {
|
|
imageModel := models.NewImageModel(db)
|
|
stackModel := models.NewListModel(db)
|
|
|
|
limitsManager := limits.CreateLimitsManager(db)
|
|
|
|
notifier := NewNotifier[imageprocessor.Notification](10)
|
|
|
|
// TODO: should extract these into a notification manager
|
|
// And actually make them the same code.
|
|
// The events are basically the same.
|
|
|
|
go ListenNewImageEvents(db)
|
|
|
|
logger := createLogger("Image Processor", os.Stdout)
|
|
processingChan := getProcessingImageStatusChannel(db)
|
|
|
|
imageProcessor := imageprocessor.NewImageProcessor(logger, imageModel, processingChan, func(userID string, n imageprocessor.Notification) {
|
|
if err := notifier.SendAndCreate(userID, n); err != nil {
|
|
logger.Error("sending notification", "err", err)
|
|
}
|
|
})
|
|
|
|
go imageProcessor.Work()
|
|
|
|
go ListenNewStackEvents(db)
|
|
go ListenProcessingStackStatus(db, stackModel, ¬ifier)
|
|
|
|
stackHandler := stacks.CreateStackHandler(db, limitsManager)
|
|
authHandler := auth.CreateAuthHandler(db)
|
|
imageHandler := images.CreateImageHandler(db, limitsManager, imageProcessor)
|
|
|
|
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)
|
|
|
|
r.Get("/", CreateEventsHandler(¬ifier))
|
|
})
|
|
|
|
logWriter := DatabaseWriter{
|
|
dbPool: db,
|
|
}
|
|
|
|
r.Route("/logs", createLogHandler(&logWriter))
|
|
|
|
return r
|
|
}
|