komari/api/client/task.go

package client

import (
	"github.com/gin-gonic/gin"
	"github.com/komari-monitor/komari/database/clients"
	"github.com/komari-monitor/komari/database/models"
	"github.com/komari-monitor/komari/database/tasks"
)

func TaskResult(c *gin.Context) {
	token := c.Query("token")
	clientId, _ := clients.GetClientUUIDByToken(token)
	if clientId == "" {
		c.JSON(400, gin.H{"status": "error", "message": "Invalid or missing token"})
		return
	}
	var req struct {
		TaskId     string    `json:"task_id" binding:"required"`
		Result     string    `json:"result" binding:"required"`
		ExitCode   int       `json:"exit_code"`
		FinishedAt models.LocalTime `json:"finished_at" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(400, gin.H{"status": "error", "message": "Invalid request"})
		return
	}

	if err := tasks.SaveTaskResult(req.TaskId, clientId, req.Result, req.ExitCode, req.FinishedAt); err != nil {
		c.JSON(500, gin.H{"status": "error", "message": "Failed to update task result: " + err.Error()})
		return
	}

	c.JSON(200, gin.H{"status": "success", "message": "Task result updated successfully"})
}