extract option defaults logic

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

Author: johnstcn

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)
+}