add repo mode

Author: mafredri

internal/provider/cached_image_data_source.go

@@ -13,6 +13,7 @@ import (
 	kconfig "github.com/GoogleContainerTools/kaniko/pkg/config"
 	"github.com/coder/envbuilder"
 	"github.com/coder/envbuilder/constants"
+	ebgit "github.com/coder/envbuilder/git"
 	eblog "github.com/coder/envbuilder/log"
 	eboptions "github.com/coder/envbuilder/options"
 	"github.com/go-git/go-billy/v5/osfs"
@@ -315,6 +316,22 @@ func (d *CachedImageDataSource) Read(ctx context.Context, req datasource.ReadReq
 		SkipRebuild:         false,
 	}
 
+	cloneOpts, err := ebgit.CloneOptionsFromOptions(opts)
+	if err != nil {
+		resp.Diagnostics.AddError("Failed to create git clone options", err.Error())
+		return
+	}
+
+	w := ebgit.ProgressWriter(func(line string) { tflog.Info(ctx, line) })
+	defer w.Close()
+	cloneOpts.Progress = w
+
+	err = ebgit.ShallowCloneRepo(ctx, cloneOpts)
+	if err != nil {
+		resp.Diagnostics.AddError("Failed to clone git repository", err.Error())
+		return
+	}
+
 	image, err := envbuilder.RunCacheProbe(ctx, opts)
 	data.Exists = types.BoolValue(err == nil)
 	if err != nil {