geo/server/internal/tenant/transport/desktop_task_handler.go

package transport

import (
	"errors"
	"io"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"

	"github.com/geo-platform/tenant-api/internal/bootstrap"
	"github.com/geo-platform/tenant-api/internal/shared/auth"
	"github.com/geo-platform/tenant-api/internal/shared/response"
	"github.com/geo-platform/tenant-api/internal/tenant/app"
	"github.com/geo-platform/tenant-api/internal/tenant/repository"
)

type DesktopTaskHandler struct {
	svc *app.DesktopTaskService
}

func NewDesktopTaskHandler(a *bootstrap.App) *DesktopTaskHandler {
	return &DesktopTaskHandler{
		svc: app.NewDesktopTaskService(
			repository.NewDesktopTaskRepository(a.DB),
			a.RabbitMQ,
			a.Logger,
		),
	}
}

func (h *DesktopTaskHandler) Lease(c *gin.Context) {
	var req app.LeaseDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	var routeTaskID *uuid.UUID
	if rawID := c.Param("id"); rawID != "" {
		parsed, err := uuid.Parse(rawID)
		if err != nil {
			response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
			return
		}
		routeTaskID = &parsed
	}

	data, err := h.svc.Lease(c.Request.Context(), MustDesktopClient(c.Request.Context()), req, routeTaskID, c.Query("from_parked") == "true")
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) Extend(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.ExtendDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.Extend(c.Request.Context(), MustDesktopClient(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) Park(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.ParkDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.Park(c.Request.Context(), MustDesktopClient(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) Result(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.CompleteDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.Complete(c.Request.Context(), MustDesktopClient(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) Cancel(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.CancelDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.Cancel(c.Request.Context(), MustDesktopClient(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) TenantCancel(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.CancelDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil && !errors.Is(err, io.EOF) {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.TenantCancel(c.Request.Context(), auth.MustActor(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}

func (h *DesktopTaskHandler) Reconcile(c *gin.Context) {
	desktopID, err := uuid.Parse(c.Param("id"))
	if err != nil {
		response.Error(c, response.ErrBadRequest(40084, "invalid_desktop_task_id", "task id must be a uuid"))
		return
	}

	var req app.ReconcileDesktopTaskRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
		return
	}

	data, err := h.svc.Reconcile(c.Request.Context(), auth.MustActor(c.Request.Context()), desktopID, req)
	if err != nil {
		response.Error(c, err)
		return
	}
	response.Success(c, data)
}