fix: re-add tasks (#442)

internal/task/task.go

@@ -15,6 +15,7 @@ import (
 	entChannel "github.com/zibbp/ganymede/ent/channel"
 	entVod "github.com/zibbp/ganymede/ent/vod"
 	"github.com/zibbp/ganymede/internal/archive"
+	"github.com/zibbp/ganymede/internal/auth"
 	"github.com/zibbp/ganymede/internal/database"
 	"github.com/zibbp/ganymede/internal/live"
 	"github.com/zibbp/ganymede/internal/twitch"
@@ -31,6 +32,46 @@ func NewService(store *database.Database, liveService *live.Service, archiveServ
 	return &Service{Store: store, LiveService: liveService, ArchiveService: archiveService}
 }
 
+func (s *Service) StartTask(c echo.Context, task string) error {
+	log.Info().Msgf("Manually starting task %s", task)
+
+	switch task {
+	case "check_live":
+		err := s.LiveService.Check()
+		if err != nil {
+			return fmt.Errorf("error checking live: %v", err)
+		}
+
+	case "check_vod":
+		go s.LiveService.CheckVodWatchedChannels()
+
+	case "get_jwks":
+		err := auth.FetchJWKS()
+		if err != nil {
+			return fmt.Errorf("error fetching jwks: %v", err)
+		}
+
+	case "twitch_auth":
+		err := twitch.Authenticate()
+		if err != nil {
+			return fmt.Errorf("error authenticating twitch: %v", err)
+		}
+
+	case "storage_migration":
+		go func() {
+			err := s.StorageMigration()
+			if err != nil {
+				log.Error().Err(err).Msg("Error migrating storage")
+			}
+		}()
+
+	case "prune_videos":
+		go PruneVideos()
+	}
+
+	return nil
+}
+
 func (s *Service) StorageMigration() error {
 	// Get all videos in db
 	videos, err := s.Store.Client.Vod.Query().WithChannel().All(context.Background())

internal/transport/http/handler.go

@@ -264,7 +264,8 @@ func groupV1Routes(e *echo.Group, h *Handler) {
 	execGroup.POST("/ffprobe", h.GetFfprobeData, auth.GuardMiddleware, auth.GetUserMiddleware, auth.UserRoleMiddleware(utils.ArchiverRole))
 
 	// Task
-	// taskGroup := e.Group("/task")
+	taskGroup := e.Group("/task")
+	taskGroup.POST("/start", h.StartTask, auth.GuardMiddleware, auth.GetUserMiddleware, auth.UserRoleMiddleware(utils.AdminRole))
 
 	// Notification
 	notificationGroup := e.Group("/notification")

internal/transport/http/task.go

@@ -1,4 +1,41 @@
 package http
 
+import (
+	"net/http"
+
+	"github.com/labstack/echo/v4"
+)
+
 type TaskService interface {
+	StartTask(c echo.Context, task string) error
+}
+
+type StartTaskRequest struct {
+	Task string `json:"task" validate:"required,oneof=check_live check_vod get_jwks twitch_auth storage_migration prune_videos"`
+}
+
+// StartTask godoc
+//
+//	@Summary		Start a task
+//	@Description	Start a task
+//	@Tags			task
+//	@Accept			json
+//	@Produce		json
+//	@Param			body	body	StartTaskRequest	true	"StartTaskRequest"
+//	@Success		200
+//	@Failure		500	{object}	utils.ErrorResponse
+//	@Router			/task/start [post]
+//	@Security		ApiKeyCookieAuth
+func (h *Handler) StartTask(c echo.Context) error {
+	str := new(StartTaskRequest)
+	if err := c.Bind(str); err != nil {
+		return echo.NewHTTPError(http.StatusBadRequest, err.Error())
+	}
+	if err := c.Validate(str); err != nil {
+		return echo.NewHTTPError(http.StatusBadRequest, err.Error())
+	}
+	if err := h.Service.TaskService.StartTask(c, str.Task); err != nil {
+		return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
+	}
+	return c.NoContent(http.StatusOK)
 }