add go require & replace

Signed-off-by: Danny Brito <[email protected]>

Author: DannyBrito

docs/spec.schema.json

@@ -900,6 +900,10 @@
 					],
 					"description": "Auth is the git authorization to use for gomods. The keys are the hosts, and the values are the auth to use for that host."
 				},
+				"edits": {
+					"$ref": "#/$defs/GomodEdits",
+					"description": "Edits contains go.mod manipulation directives (replace, require) that are applied\nbefore downloading module dependencies."
+				},
 				"paths": {
 					"items": {
 						"type": [
@@ -1048,6 +1052,36 @@
 			],
 			"description": "GoBuildCache is a cache for Go build artifacts."
 		},
+		"GomodEdits": {
+			"properties": {
+				"replace": {
+					"items": {
+						"$ref": "#/$defs/GomodReplace"
+					},
+					"type": [
+						"array"
+					],
+					"description": "Replace applies go.mod replace directives before downloading module dependencies.\nEach entry can be either a string \"old:new\" or a struct with Old and New fields."
+				},
+				"require": {
+					"items": {
+						"$ref": "#/$defs/GomodRequire"
+					},
+					"type": [
+						"array"
+					],
+					"description": "Require applies go.mod require directives before downloading module dependencies.\nEach entry can be either a string \"module:version\" or a struct with Module and Version fields."
+				}
+			},
+			"additionalProperties": {
+				"not": {}
+			},
+			"type": [
+				"object",
+				"null"
+			],
+			"description": "GomodEdits groups the go.mod manipulation directives that can be applied before downloading module dependencies."
+		},
 		"GomodGitAuth": {
 			"properties": {
 				"header": {
@@ -1102,6 +1136,62 @@
 				"null"
 			]
 		},
+		"GomodReplace": {
+			"required": [
+				"old",
+				"new"
+			],
+			"properties": {
+				"new": {
+					"type": [
+						"string"
+					],
+					"description": "New is the replacement module path or local directory"
+				},
+				"old": {
+					"type": [
+						"string"
+					],
+					"description": "Old is the module path to replace (can include @version)"
+				}
+			},
+			"additionalProperties": {
+				"not": {}
+			},
+			"type": [
+				"object",
+				"null"
+			],
+			"description": "GomodReplace represents a go.mod replace directive."
+		},
+		"GomodRequire": {
+			"required": [
+				"module",
+				"version"
+			],
+			"properties": {
+				"module": {
+					"type": [
+						"string"
+					],
+					"description": "Module is the module path that appears in go.mod"
+				},
+				"version": {
+					"type": [
+						"string"
+					],
+					"description": "Version is the target module and version (e.g., \"module@version\")"
+				}
+			},
+			"additionalProperties": {
+				"not": {}
+			},
+			"type": [
+				"object",
+				"null"
+			],
+			"description": "GomodRequire represents a go.mod require directive."
+		},
 		"ImageConfig": {
 			"properties": {
 				"base": {

frontend/debug/handle_sources.go

@@ -3,11 +3,11 @@ package debug
 import (
 	"context"
 
-	"github.com/project-dalec/dalec"
-	"github.com/project-dalec/dalec/frontend"
 	"github.com/moby/buildkit/client/llb"
 	gwclient "github.com/moby/buildkit/frontend/gateway/client"
 	ocispecs "github.com/opencontainers/image-spec/specs-go/v1"
+	"github.com/project-dalec/dalec"
+	"github.com/project-dalec/dalec/frontend"
 )
 
 // Sources is a handler that outputs all the sources.
@@ -64,6 +64,13 @@ func PatchedSources(ctx context.Context, client gwclient.Client) (*gwclient.Resu
 		}
 
 		pc := dalec.Platform(platform)
+
+		// Preprocess the spec to generate patches for gomod edits and other generators
+		// This must happen before getting sources so that generated patch contexts are available
+		if err := spec.Preprocess(client, sOpt, worker, pc); err != nil {
+			return nil, nil, err
+		}
+
 		sources, err := dalec.Sources(spec, sOpt, pc)
 		if err != nil {
 			return nil, nil, err

frontend/gateway.go

@@ -6,7 +6,6 @@ import (
 	"sync"
 	"sync/atomic"
 
-	"github.com/project-dalec/dalec"
 	"github.com/moby/buildkit/client/llb"
 	"github.com/moby/buildkit/frontend/dockerui"
 	gwclient "github.com/moby/buildkit/frontend/gateway/client"
@@ -15,6 +14,7 @@ import (
 	"github.com/opencontainers/go-digest"
 	ocispecs "github.com/opencontainers/image-spec/specs-go/v1"
 	"github.com/pkg/errors"
+	"github.com/project-dalec/dalec"
 )
 
 const (
@@ -128,6 +128,7 @@ func SourceOptFromUIClient(ctx context.Context, c gwclient.Client, dc *dockerui.
 			return st, err
 		},
 		GitCredHelperOpt: withCredHelper(c),
+		GeneratedStates:  make(map[string]llb.State), // Initialize map for generated states
 	}
 }