[Feature] Add databricks_registered_model data source (#4033)

## Changes
<!-- Summary of your changes that are easy to understand -->

## Tests
<!-- 
How is this tested? Please see the checklist below and also describe any
other relevant tests
-->

- [x] `make test` run locally
- [x] relevant change in `docs/` folder
- [x] covered with integration tests in `internal/acceptance`
- [x] relevant acceptance tests are passing
- [x] using Go SDK

---------

Co-authored-by: Miles Yucht <[email protected]>

Author: alexott

docs/data-sources/registered_model.md

@@ -0,0 +1,53 @@
+---
+subcategory: "Unity Catalog"
+---
+# databricks_registered_model Data Source
+
+-> This resource can only be used with a workspace-level provider!
+
+This resource allows you to get information about [Model in Unity Catalog](https://docs.databricks.com/en/mlflow/models-in-uc.html) in Databricks.
+
+## Example Usage
+
+```hcl
+data "databricks_registered_model" "this" {
+  full_name = "main.default.my_model"
+}
+```
+
+## Argument Reference
+
+The following arguments are supported:
+
+* `full_name` - (Required, String) The fully-qualified name of the registered model (`catalog_name.schema_name.name`).
+* `include_aliases` - (Optional, Boolean) flag to specify if list of aliases should be included into output.
+* `include_browse` - (Optional, Boolean) flag to specify if include registered models in the response for which the principal can only access selective metadata for.
+
+## Attribute Reference
+
+The following attributes are exported:
+
+* `model_info` - block with information about the model in Unity Catalog:
+  * `aliases` - the list of aliases associated with this model. Each item is object consisting of following attributes:
+    * `alias_name` - string with the name of alias
+    * `version_num` - associated model version
+  * `catalog_name` - The name of the catalog where the schema and the registered model reside.
+  * `comment` - The comment attached to the registered model.
+  * `created_at` - the Unix timestamp at the model's creation
+  * `created_by` - the identifier of the user who created the model
+  * `full_name` - The fully-qualified name of the registered model (`catalog_name.schema_name.name`).
+  * `metastore_id` - the unique identifier of the metastore
+  * `name` - The name of the registered model.
+  * `owner` - Name of the registered model owner.
+  * `schema_name` - The name of the schema where the registered model resides.
+  * `storage_location` - The storage location under which model version data files are stored.
+  * `updated_at` - the timestamp of the last time changes were made to the model
+  * `updated_by` - the identifier of the user who updated the model last time
+
+## Related Resources
+
+The following resources are often used in the same context:
+
+* [databricks_registered_model](../resources/schema.md) resource to manage models within Unity Catalog.
+* [databricks_model_serving](../resources/model_serving.md) to serve this model on a Databricks serving endpoint.
+* [databricks_mlflow_experiment](../resources/mlflow_experiment.md) to manage [MLflow experiments](https://docs.databricks.com/data/data-sources/mlflow-experiment.html) in Databricks.

internal/acceptance/registered_model_test.go

@@ -43,6 +43,13 @@ func TestUcAccRegisteredModel(t *testing.T) {
 				owner = "account users"
 				comment = "new comment"
 			}
+			data "databricks_registered_model" "model" {
+				full_name = databricks_registered_model.model.id
+				include_model_versions = true
+			}
+			output "model" {
+				value = data.databricks_registered_model.model
+			}
 		`,
 		},
 	)

internal/providers/pluginfw/pluginfw.go

@@ -19,6 +19,7 @@ import (
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/cluster"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/library"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/qualitymonitor"
+	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/registered_model"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/volume"
 
 	"github.com/hashicorp/terraform-plugin-framework/datasource"
@@ -52,6 +53,7 @@ func (p *DatabricksProviderPluginFramework) DataSources(ctx context.Context) []f
 	return []func() datasource.DataSource{
 		cluster.DataSourceCluster,
 		volume.DataSourceVolumes,
+		registered_model.DataSourceRegisteredModel,
 	}
 }
 

internal/providers/pluginfw/resources/registered_model/data_registered_model.go

@@ -0,0 +1,90 @@
+package registered_model
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/databricks/databricks-sdk-go/apierr"
+	"github.com/databricks/databricks-sdk-go/service/catalog"
+	"github.com/databricks/terraform-provider-databricks/common"
+	pluginfwcommon "github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/common"
+	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/converters"
+	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/tfschema"
+	"github.com/databricks/terraform-provider-databricks/internal/service/catalog_tf"
+	"github.com/hashicorp/terraform-plugin-framework/datasource"
+	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+)
+
+func DataSourceRegisteredModel() datasource.DataSource {
+	return &RegisteredModelDataSource{}
+}
+
+var _ datasource.DataSourceWithConfigure = &RegisteredModelDataSource{}
+
+type RegisteredModelDataSource struct {
+	Client *common.DatabricksClient
+}
+
+type RegisteredModelData struct {
+	FullName       types.String                     `tfsdk:"full_name"`
+	IncludeAliases types.Bool                       `tfsdk:"include_aliases" tf:"optional"`
+	IncludeBrowse  types.Bool                       `tfsdk:"include_browse" tf:"optional"`
+	ModelInfo      []catalog_tf.RegisteredModelInfo `tfsdk:"model_info" tf:"optional,computed"`
+}
+
+func (d *RegisteredModelDataSource) Metadata(ctx context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
+	resp.TypeName = "databricks_registered_model"
+}
+
+func (d *RegisteredModelDataSource) Schema(ctx context.Context, req datasource.SchemaRequest, resp *datasource.SchemaResponse) {
+	attrs, blocks := tfschema.DataSourceStructToSchemaMap(RegisteredModelData{}, nil)
+	resp.Schema = schema.Schema{
+		Attributes: attrs,
+		Blocks:     blocks,
+	}
+}
+
+func (d *RegisteredModelDataSource) Configure(_ context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
+	if d.Client == nil {
+		d.Client = pluginfwcommon.ConfigureDataSource(req, resp)
+	}
+}
+
+func (d *RegisteredModelDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
+	w, diags := d.Client.GetWorkspaceClient()
+	resp.Diagnostics.Append(diags...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	var registeredModel RegisteredModelData
+	diags = req.Config.Get(ctx, &registeredModel)
+	resp.Diagnostics.Append(diags...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+	modelFullName := registeredModel.FullName.ValueString()
+	modelInfoSdk, err := w.RegisteredModels.Get(ctx, catalog.GetRegisteredModelRequest{
+		FullName:       modelFullName,
+		IncludeAliases: registeredModel.IncludeAliases.ValueBool(),
+		IncludeBrowse:  registeredModel.IncludeBrowse.ValueBool(),
+	})
+	if err != nil {
+		if apierr.IsMissing(err) {
+			resp.State.RemoveResource(ctx)
+		}
+		resp.Diagnostics.AddError(fmt.Sprintf("failed to get registered model %s", modelFullName), err.Error())
+		return
+	}
+	var modelInfo catalog_tf.RegisteredModelInfo
+	resp.Diagnostics.Append(converters.GoSdkToTfSdkStruct(ctx, modelInfoSdk, &modelInfo)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+	if modelInfo.Aliases == nil {
+		modelInfo.Aliases = []catalog_tf.RegisteredModelAlias{}
+	}
+	registeredModel.ModelInfo = append(registeredModel.ModelInfo, modelInfo)
+	resp.Diagnostics.Append(resp.State.Set(ctx, registeredModel)...)
+}