forward relay metadata for remote capability execution

(cherry picked from commit 16e4fc9)

Author: nadahalli

core/capabilities/confidentialrelay/handler.go

@@ -354,12 +354,20 @@ func (h *Handler) handleCapabilityExecute(ctx context.Context, gatewayID string,
 		return h.errorResponse(ctx, gatewayID, req, jsonrpc.ErrInvalidParams, fmt.Errorf("failed to unmarshal capability request: %w", err))
 	}
 
+	referenceID := params.ReferenceID
+	if referenceID == "" {
+		referenceID = req.ID
+	}
+
 	capReq := capabilities.CapabilityRequest{
 		Payload:      sdkReq.Payload,
 		Method:       sdkReq.Method,
 		CapabilityId: params.CapabilityID,
 		Metadata: capabilities.RequestMetadata{
-			WorkflowID: params.WorkflowID,
+			WorkflowID:          params.WorkflowID,
+			WorkflowOwner:       params.Owner,
+			WorkflowExecutionID: params.ExecutionID,
+			ReferenceID:         referenceID,
 		},
 	}
 

core/capabilities/confidentialrelay/handler_test.go

@@ -182,6 +182,9 @@ func TestHandler_HandleGatewayMessage(t *testing.T) {
 			req: func(t *testing.T) *jsonrpc.Request[json.RawMessage] {
 				return makeRequest(t, confidentialrelaytypes.MethodCapabilityExec, confidentialrelaytypes.CapabilityRequestParams{
 					WorkflowID:   "wf-1",
+					Owner:        "0xowner",
+					ExecutionID:  "32c631d295ef5e32deb99a10ee6804bc4af13855687559d7ff6552ac6dbb2ce1",
+					ReferenceID:  "17",
 					CapabilityID: "my-cap@1.0.0",
 					Payload:      makeCapabilityPayload(t, map[string]any{"key": "val"}),
 					Attestation:  testAttestationB64,
@@ -199,6 +202,14 @@ func TestHandler_HandleGatewayMessage(t *testing.T) {
 				assert.Equal(t, "result-proto-bytes", string(capResp.GetPayload().GetValue()))
 				assert.Empty(t, result.Error)
 			},
+			checkExecutable: func(t *testing.T, reg *mockCapRegistry) {
+				exec := reg.executables["my-cap@1.0.0"]
+				require.NotNil(t, exec.lastRequest, "Execute should have been called")
+				assert.Equal(t, "wf-1", exec.lastRequest.Metadata.WorkflowID)
+				assert.Equal(t, "0xowner", exec.lastRequest.Metadata.WorkflowOwner)
+				assert.Equal(t, "32c631d295ef5e32deb99a10ee6804bc4af13855687559d7ff6552ac6dbb2ce1", exec.lastRequest.Metadata.WorkflowExecutionID)
+				assert.Equal(t, "17", exec.lastRequest.Metadata.ReferenceID)
+			},
 		},
 		{
 			name: "capability execute sets Inputs from Payload for backward compat",