( c *gin.Context, manager TaskService, req contract.CreateTaskBlockRequest, )
| 316 | } |
| 317 | |
| 318 | func (h *BaseHandlers) taskBlockClaimToken( |
| 319 | c *gin.Context, |
| 320 | manager TaskService, |
| 321 | req contract.CreateTaskBlockRequest, |
| 322 | ) (string, error) { |
| 323 | claimToken := strings.TrimSpace(c.GetHeader(taskClaimTokenHeader)) |
| 324 | if claimToken != "" || strings.TrimSpace(req.RunID) == "" { |
| 325 | return claimToken, nil |
| 326 | } |
| 327 | credentials := agentCallerCredentialsFromRequest(c) |
| 328 | if !hasAgentCallerIdentityCredentials(credentials) { |
| 329 | return "", nil |
| 330 | } |
| 331 | caller, err := h.resolveAgentCallerForWorkspace( |
| 332 | c.Request.Context(), |
| 333 | credentials, |
| 334 | "tasks."+taskActionBlock, |
| 335 | "", |
| 336 | ) |
| 337 | if err != nil { |
| 338 | return "", err |
| 339 | } |
| 340 | handle, err := h.lookupAgentTaskLease(c.Request.Context(), manager, caller, req.RunID) |
| 341 | if err != nil { |
| 342 | return "", err |
| 343 | } |
| 344 | return handle.ClaimToken, nil |
| 345 | } |
| 346 | |
| 347 | // ListTaskBlocks returns task block rows for one task. |
| 348 | func (h *BaseHandlers) ListTaskBlocks(c *gin.Context) { |
no test coverage detected