WIP: [Feature] Add databricks_git_folder resource to replace databricks_repo

internal/providers/sdkv2/sdkv2.go

@@ -149,6 +149,7 @@ func DatabricksProvider() *schema.Provider {
 			"databricks_external_location":               catalog.ResourceExternalLocation().ToResource(),
 			"databricks_file":                            storage.ResourceFile().ToResource(),
 			"databricks_git_credential":                  repos.ResourceGitCredential().ToResource(),
+			"databricks_git_folder":                      repos.ResourceGitFolder().ToResource(),
 			"databricks_global_init_script":              workspace.ResourceGlobalInitScript().ToResource(),
 			"databricks_grant":                           catalog.ResourceGrant().ToResource(),
 			"databricks_grants":                          catalog.ResourceGrants().ToResource(),

repos/resource_git_folder.go

@@ -0,0 +1,157 @@
+package repos
+
+import (
+	"context"
+	"fmt"
+	"net/url"
+	"regexp"
+	"strconv"
+	"strings"
+
+	"github.com/databricks/databricks-sdk-go/service/workspace"
+	"github.com/databricks/terraform-provider-databricks/common"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+type gitFolderStruct workspace.RepoInfo
+
+var aliasMap = map[string]string{
+	"provider": "git_provider",
+}
+
+func (s gitFolderStruct) Aliases() map[string]map[string]string {
+	return map[string]map[string]string{
+		"repos.gitFolderStruct": {
+			"provider": "git_provider",
+		},
+	}
+}
+
+type gitFolderCreateStruct workspace.CreateRepoRequest
+
+func (s gitFolderCreateStruct) Aliases() map[string]map[string]string {
+	return map[string]map[string]string{
+		"repos.gitFolderStruct": {
+			"provider": "git_provider",
+		},
+	}
+}
+
+func (s gitFolderCreateStruct) CustomizeSchema(m *common.CustomizableSchema) *common.CustomizableSchema {
+	return m
+}
+
+type gitFolderUpdateStruct workspace.UpdateRepoRequest
+
+func (s gitFolderUpdateStruct) Aliases() map[string]map[string]string {
+	return map[string]map[string]string{
+		"repos.gitFolderStruct": {
+			"provider": "git_provider",
+		},
+	}
+}
+
+func (s gitFolderUpdateStruct) CustomizeSchema(m *common.CustomizableSchema) *common.CustomizableSchema {
+	return m
+}
+
+var (
+	gitProvidersMap = map[string]string{
+		"github.com":    "gitHub",
+		"dev.azure.com": "azureDevOpsServices",
+		"gitlab.com":    "gitLab",
+		"bitbucket.org": "bitbucketCloud",
+	}
+	awsCodeCommitRegex = regexp.MustCompile(`^git-codecommit\.[^.]+\.amazonaws\.com$`)
+)
+
+func ResourceGitFolder() common.Resource {
+	s := common.StructToSchema(gitFolderStruct{}, func(s map[string]*schema.Schema) map[string]*schema.Schema {
+		s["provider"].DiffSuppressFunc = common.EqualFoldDiffSuppress
+		delete(s, "id")
+		return s
+	})
+	return common.Resource{
+		Create: func(ctx context.Context, d *schema.ResourceData, c *common.DatabricksClient) error {
+			var create gitFolderCreateStruct
+			if create.Provider == "" { // trying to infer Git Provider from the URL
+				create.Provider = GetGitProviderFromUrl(create.Url)
+			}
+			if create.Provider == "" {
+				return fmt.Errorf("git_provider isn't specified and we can't detect provider from URL")
+			}
+			common.DataToStructPointer(d, s, &create)
+			ws, err := c.WorkspaceClient()
+			if err != nil {
+				return err
+			}
+			repo, err := ws.Repos.Create(ctx, workspace.CreateRepoRequest{
+				Path:           create.Path,
+				Provider:       create.Provider,
+				SparseCheckout: create.SparseCheckout,
+				Url:            create.Url,
+			})
+			if err != nil {
+				return err
+			}
+			d.Set("id", repo.Id)
+			common.StructToData(repo, s, d)
+			return nil
+		},
+		Read: func(ctx context.Context, d *schema.ResourceData, c *common.DatabricksClient) error {
+			ws, err := c.WorkspaceClient()
+			if err != nil {
+				return err
+			}
+			id, err := strconv.ParseInt(d.Id(), 10, 64)
+			if err != nil {
+				return err
+			}
+			repo, err := ws.Repos.GetByRepoId(ctx, id)
+			if err != nil {
+				return err
+			}
+			return common.StructToData(repo, s, d)
+		},
+		Update: func(ctx context.Context, d *schema.ResourceData, c *common.DatabricksClient) error {
+			var update gitFolderUpdateStruct
+			common.DataToStructPointer(d, s, &update)
+			ws, err := c.WorkspaceClient()
+			if err != nil {
+				return err
+			}
+			repo := ws.Repos.Update(ctx, workspace.UpdateRepoRequest{
+				Branch:         update.Branch,
+				RepoId:         update.RepoId,
+				SparseCheckout: update.SparseCheckout,
+				Tag:            update.Tag,
+			})
+			return common.StructToData(repo, s, d)
+		},
+		Delete: func(ctx context.Context, d *schema.ResourceData, c *common.DatabricksClient) error {
+			ws, err := c.WorkspaceClient()
+			if err != nil {
+				return err
+			}
+			id, err := strconv.ParseInt(d.Id(), 10, 64)
+			if err != nil {
+				return err
+			}
+			return ws.Repos.DeleteByRepoId(ctx, id)
+		},
+		Schema: s,
+	}
+}
+
+func GetGitProviderFromUrl(uri string) string {
+	provider := ""
+	u, err := url.Parse(uri)
+	if err == nil {
+		lhost := strings.ToLower(u.Host)
+		provider = gitProvidersMap[lhost]
+		if provider == "" && awsCodeCommitRegex.FindStringSubmatch(lhost) != nil {
+			provider = "awsCodeCommit"
+		}
+	}
+	return provider
+}