Introduce target-aware resolved reference storage

internal/framework/pipeline/profile.go

@@ -60,23 +60,32 @@ type ReferenceUnbind struct {
 	SlotName string `json:"slot_name"`
 }
 
-type ResolvedArtifactLane struct {
-	ID           string
-	Extract      ModuleBinding
-	Merge        ModuleBinding
-	Normalize    ModuleBinding
-	Validators   []ModuleBinding
-	References   []ReferenceBinding     `json:"references,omitempty"`
+type ResolvedReferenceTarget struct {
+	Stage        ModuleStage            `json:"stage"`
+	LaneID       string                 `json:"lane_id,omitempty"`
+	Module       string                 `json:"module"`
+	Bindings     []ReferenceBinding     `json:"bindings,omitempty"`
 	ReferenceSet contracts.ReferenceSet `json:"-"`
 }
 
+type ResolvedArtifactLane struct {
+	ID                  string
+	Extract             ModuleBinding
+	Merge               ModuleBinding
+	Normalize           ModuleBinding
+	Validators          []ModuleBinding
+	ExtractReferences   ResolvedReferenceTarget `json:"extract_references"`
+	NormalizeReferences ResolvedReferenceTarget `json:"normalize_references"`
+}
+
 type ResolvedPipeline struct {
-	ID            string
-	Digest        string
-	Input         ModuleBinding
-	Chunk         ModuleBinding
-	ArtifactLanes []ResolvedArtifactLane
-	Output        ModuleBinding
+	ID              string
+	Digest          string
+	Input           ModuleBinding
+	Chunk           ModuleBinding
+	ChunkReferences ResolvedReferenceTarget `json:"chunk_references"`
+	ArtifactLanes   []ResolvedArtifactLane
+	Output          ModuleBinding
 }
 
 type ModuleCatalog struct {
@@ -140,10 +149,11 @@ func ResolvePipeline(profile PipelineProfile, options ResolveOptions, catalog Mo
 	}
 
 	resolved := ResolvedPipeline{
-		ID:     pipelineID,
-		Input:  input,
-		Chunk:  chunk,
-		Output: resolveBinding(profile.Output, DefaultOutputModule),
+		ID:              pipelineID,
+		Input:           input,
+		Chunk:           chunk,
+		ChunkReferences: referenceTarget(StageChunk, "", chunk.Module, nil),
+		Output:          resolveBinding(profile.Output, DefaultOutputModule),
 	}
 	outputCapabilities := capabilities.clone()
 
@@ -206,7 +216,7 @@ func resolveArtifactLane(
 	if err != nil {
 		return ResolvedArtifactLane{}, nil, err
 	}
-	lane.References = references
+	lane.ExtractReferences = referenceTarget(StageExtract, laneID, lane.Extract.Module, references)
 	capabilities.add(extractSpec.Provides...)
 
 	mergeSpec, err := mergerSpec(catalog, lane.Merge.Module)
@@ -225,6 +235,7 @@ func resolveArtifactLane(
 	if missing, ok := capabilities.missing(normalizeSpec.Requires); ok {
 		return ResolvedArtifactLane{}, nil, capabilityError(pipelineID, laneID, StageNormalize, lane.Normalize.Module, missing)
 	}
+	lane.NormalizeReferences = referenceTarget(StageNormalize, laneID, lane.Normalize.Module, nil)
 	capabilities.add(normalizeSpec.Provides...)
 
 	for _, validator := range lane.Validators {
@@ -241,6 +252,15 @@ func resolveArtifactLane(
 	return lane, capabilities, nil
 }
 
+func referenceTarget(stage ModuleStage, laneID string, module string, bindings []ReferenceBinding) ResolvedReferenceTarget {
+	return ResolvedReferenceTarget{
+		Stage:    stage,
+		LaneID:   strings.TrimSpace(laneID),
+		Module:   strings.TrimSpace(module),
+		Bindings: append([]ReferenceBinding(nil), bindings...),
+	}
+}
+
 func validatePipelineReferenceDefaults(
 	pipelineID string,
 	pipelineReferences map[string]string,
@@ -532,17 +552,19 @@ func selectedArtifactLanes(pipelineID string, artifacts map[string]ArtifactLaneP
 
 func resolvedPipelineDigest(resolved ResolvedPipeline) (string, error) {
 	withoutDigest := struct {
-		ID            string
-		Input         ModuleBinding
-		Chunk         ModuleBinding
-		ArtifactLanes []ResolvedArtifactLane
-		Output        ModuleBinding
+		ID              string
+		Input           ModuleBinding
+		Chunk           ModuleBinding
+		ChunkReferences ResolvedReferenceTarget
+		ArtifactLanes   []ResolvedArtifactLane
+		Output          ModuleBinding
 	}{
-		ID:            resolved.ID,
-		Input:         resolved.Input,
-		Chunk:         resolved.Chunk,
-		ArtifactLanes: resolved.ArtifactLanes,
-		Output:        resolved.Output,
+		ID:              resolved.ID,
+		Input:           resolved.Input,
+		Chunk:           resolved.Chunk,
+		ChunkReferences: resolved.ChunkReferences,
+		ArtifactLanes:   resolved.ArtifactLanes,
+		Output:          resolved.Output,
 	}
 	encoded, err := json.Marshal(withoutDigest)
 	if err != nil {