extract option defaults logic

Signed-off-by: Cian Johnston <[email protected]>

Author: johnstcn

envbuilder.go

@@ -26,7 +26,6 @@ import (
 
 	"github.com/coder/envbuilder/constants"
 	"github.com/coder/envbuilder/git"
-	"github.com/coder/envbuilder/internal/chmodfs"
 	"github.com/coder/envbuilder/options"
 
 	"github.com/GoogleContainerTools/kaniko/pkg/config"
@@ -42,7 +41,6 @@ import (
 	_ "github.com/distribution/distribution/v3/registry/storage/driver/filesystem"
 	"github.com/docker/cli/cli/config/configfile"
 	"github.com/fatih/color"
-	"github.com/go-git/go-billy/v5/osfs"
 	"github.com/go-git/go-git/v5/plumbing/transport"
 	v1 "github.com/google/go-containerregistry/pkg/v1"
 	"github.com/google/go-containerregistry/pkg/v1/remote"
@@ -61,23 +59,8 @@ type DockerConfig configfile.ConfigFile
 // Filesystem is the filesystem to use for all operations.
 // Defaults to the host filesystem.
 func Run(ctx context.Context, opts options.Options) error {
-	// Temporarily removed these from the default settings to prevent conflicts
-	// between current and legacy environment variables that add default values.
-	// Once the legacy environment variables are phased out, this can be
-	// reinstated to the previous default values.
-	if len(opts.IgnorePaths) == 0 {
-		opts.IgnorePaths = []string{
-			"/var/run",
-			// KinD adds these paths to pods, so ignore them by default.
-			"/product_uuid", "/product_name",
-		}
-	}
-	if opts.InitScript == "" {
-		opts.InitScript = "sleep infinity"
-	}
-	if opts.InitCommand == "" {
-		opts.InitCommand = "/bin/sh"
-	}
+	opts.SetDefaults()
+
 	if opts.CacheRepo == "" && opts.PushImage {
 		return fmt.Errorf("--cache-repo must be set when using --push-image")
 	}
@@ -90,12 +73,6 @@ func Run(ctx context.Context, opts options.Options) error {
 			return fmt.Errorf("parse init args: %w", err)
 		}
 	}
-	if opts.Filesystem == nil {
-		opts.Filesystem = chmodfs.New(osfs.New("/"))
-	}
-	if opts.WorkspaceFolder == "" {
-		opts.WorkspaceFolder = options.DefaultWorkspaceFolder(opts.GitURL)
-	}
 
 	stageNumber := 0
 	startStage := func(format string, args ...any) func(format string, args ...any) {

options/defaults.go

@@ -0,0 +1,58 @@
+package options
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/go-git/go-billy/v5/osfs"
+
+	giturls "github.com/chainguard-dev/git-urls"
+	"github.com/coder/envbuilder/constants"
+	"github.com/coder/envbuilder/internal/chmodfs"
+)
+
+// DefaultWorkspaceFolder returns the default workspace folder
+// for a given repository URL.
+func DefaultWorkspaceFolder(repoURL string) string {
+	if repoURL == "" {
+		return constants.EmptyWorkspaceDir
+	}
+	parsed, err := giturls.Parse(repoURL)
+	if err != nil {
+		return constants.EmptyWorkspaceDir
+	}
+	name := strings.Split(parsed.Path, "/")
+	hasOwnerAndRepo := len(name) >= 2
+	if !hasOwnerAndRepo {
+		return constants.EmptyWorkspaceDir
+	}
+	repo := strings.TrimSuffix(name[len(name)-1], ".git")
+	return fmt.Sprintf("/workspaces/%s", repo)
+}
+
+func (o *Options) SetDefaults() {
+	// Temporarily removed these from the default settings to prevent conflicts
+	// between current and legacy environment variables that add default values.
+	// Once the legacy environment variables are phased out, this can be
+	// reinstated to the previous default values.
+	if len(o.IgnorePaths) == 0 {
+		o.IgnorePaths = []string{
+			"/var/run",
+			// KinD adds these paths to pods, so ignore them by default.
+			"/product_uuid", "/product_name",
+		}
+	}
+	if o.InitScript == "" {
+		o.InitScript = "sleep infinity"
+	}
+	if o.InitCommand == "" {
+		o.InitCommand = "/bin/sh"
+	}
+
+	if o.Filesystem == nil {
+		o.Filesystem = chmodfs.New(osfs.New("/"))
+	}
+	if o.WorkspaceFolder == "" {
+		o.WorkspaceFolder = DefaultWorkspaceFolder(o.GitURL)
+	}
+}

options/defaults_test.go

@@ -0,0 +1,93 @@
+package options_test
+
+import (
+	"testing"
+
+	"github.com/coder/envbuilder/internal/chmodfs"
+	"github.com/go-git/go-billy/v5/osfs"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/coder/envbuilder/constants"
+	"github.com/coder/envbuilder/options"
+	"github.com/stretchr/testify/require"
+)
+
+func TestDefaultWorkspaceFolder(t *testing.T) {
+	t.Parallel()
+
+	successTests := []struct {
+		name     string
+		gitURL   string
+		expected string
+	}{
+		{
+			name:     "HTTP",
+			gitURL:   "https://github.com/coder/envbuilder.git",
+			expected: "/workspaces/envbuilder",
+		},
+		{
+			name:     "SSH",
+			gitURL:   "[email protected]:coder/envbuilder.git",
+			expected: "/workspaces/envbuilder",
+		},
+		{
+			name:     "username and password",
+			gitURL:   "https://username:[email protected]/coder/envbuilder.git",
+			expected: "/workspaces/envbuilder",
+		},
+		{
+			name:     "fragment",
+			gitURL:   "https://github.com/coder/envbuilder.git#feature-branch",
+			expected: "/workspaces/envbuilder",
+		},
+		{
+			name:     "empty",
+			gitURL:   "",
+			expected: constants.EmptyWorkspaceDir,
+		},
+	}
+	for _, tt := range successTests {
+		t.Run(tt.name, func(t *testing.T) {
+			dir := options.DefaultWorkspaceFolder(tt.gitURL)
+			require.Equal(t, tt.expected, dir)
+		})
+	}
+
+	invalidTests := []struct {
+		name       string
+		invalidURL string
+	}{
+		{
+			name:       "simple text",
+			invalidURL: "not a valid URL",
+		},
+		{
+			name:       "website URL",
+			invalidURL: "www.google.com",
+		},
+	}
+	for _, tt := range invalidTests {
+		t.Run(tt.name, func(t *testing.T) {
+			dir := options.DefaultWorkspaceFolder(tt.invalidURL)
+			require.Equal(t, constants.EmptyWorkspaceDir, dir)
+		})
+	}
+}
+
+func TestOptions_SetDefaults(t *testing.T) {
+	t.Parallel()
+
+	expected := options.Options{
+		InitScript:      "sleep infinity",
+		InitCommand:     "/bin/sh",
+		IgnorePaths:     []string{"/var/run", "/product_uuid", "/product_name"},
+		Filesystem:      chmodfs.New(osfs.New("/")),
+		GitURL:          "",
+		WorkspaceFolder: constants.EmptyWorkspaceDir,
+	}
+
+	var actual options.Options
+	actual.SetDefaults()
+	assert.Equal(t, expected, actual)
+}