package rpc

import (
	"github.com/go-chi/chi/v5"
	"github.com/marcopeocchi/yt-dlp-web-ui/server/config"
	"github.com/marcopeocchi/yt-dlp-web-ui/server/internal"
	middlewares "github.com/marcopeocchi/yt-dlp-web-ui/server/middleware"
)

// Dependency injection container.
func Container(db *internal.MemoryDB, mq *internal.MessageQueue) *Service {
	return &Service{
		db: db,
		mq: mq,
	}
}

// RPC service must be registered before applying this router!
func ApplyRouter() func(chi.Router) {
	return func(r chi.Router) {
		r.Route("/ws", func(r chi.Router) {
			if config.Instance().RequireAuth {
				r.Use(middlewares.WebSocketAuthentication)
			}
			r.Get("/", WebSocket)
		})

		r.Route("/http", func(r chi.Router) {
			if config.Instance().RequireAuth {
				r.Use(middlewares.Authenticated)
			}
			r.Post("/", Post)
		})
	}
}