feat: added workspace base environments

Author: MyroslavLevchyk

README.md

@@ -376,6 +376,8 @@ No modules.
 | [databricks_database_instance.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/database_instance) | resource |
 | [databricks_disable_legacy_dbfs_setting.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/disable_legacy_dbfs_setting) | resource |
 | [databricks_entitlements.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/entitlements) | resource |
+| [databricks_environments_default_workspace_base_environment.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/environments_default_workspace_base_environment) | resource |
+| [databricks_environments_workspace_base_environment.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/environments_workspace_base_environment) | resource |
 | [databricks_group.this](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/group) | resource |
 | [databricks_ip_access_list.allowed_list](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/ip_access_list) | resource |
 | [databricks_mount.adls](https://registry.terraform.io/providers/databricks/databricks/latest/docs/resources/mount) | resource |
@@ -417,6 +419,7 @@ No modules.
 | <a name="input_sql_endpoint"></a> [sql\_endpoint](#input\_sql\_endpoint) | Set of objects with parameters to configure SQL Endpoint and assign permissions to it for certain custom groups | <pre>set(object({<br/>    name                      = string<br/>    cluster_size              = optional(string, "2X-Small")<br/>    min_num_clusters          = optional(number, 0)<br/>    max_num_clusters          = optional(number, 1)<br/>    auto_stop_mins            = optional(string, "30")<br/>    enable_photon             = optional(bool, false)<br/>    enable_serverless_compute = optional(bool, false)<br/>    spot_instance_policy      = optional(string, "COST_OPTIMIZED")<br/>    warehouse_type            = optional(string, "PRO")<br/>    permissions = optional(set(object({<br/>      group_name       = string<br/>      permission_level = string<br/>    })), [])<br/>  }))</pre> | `[]` | no |
 | <a name="input_suffix"></a> [suffix](#input\_suffix) | Optional suffix that would be added to the end of resources names. | `string` | `""` | no |
 | <a name="input_workspace_admin_token_enabled"></a> [workspace\_admin\_token\_enabled](#input\_workspace\_admin\_token\_enabled) | Boolean flag to specify whether to create Workspace Admin Token | `bool` | n/a | yes |
+| <a name="input_workspace_base_environments"></a> [workspace\_base\_environments](#input\_workspace\_base\_environments) | Workspace base environments configuration.<br/>Base environments require environment definition files (YAML) to exist and be accessible before Terraform creates the base environment resources.<br/>Prerequisites:<br/>- CPU environment files can be stored either in a workspace directory or in a Databricks volume.<br/>- GPU environment files can only be stored in a workspace directory.<br/>Examples:<br/>- Workspace path: /Workspace/path/to/environment.yaml<br/>- Volume path: /Volumes/<catalog>/<schema>/<volume>/environment.yaml<br/>set\_as\_default specifies whether the environment should be used as the workspace default environment. | <pre>map(object({<br/>    display_name          = string<br/>    filepath              = string<br/>    base_environment_type = optional(string, "CPU")<br/>    set_as_default        = optional(bool, false)<br/>  }))</pre> | `{}` | no |
 
 ## Outputs
 

variables.tf

@@ -293,3 +293,26 @@ When set to true:
 - Note: This setting only takes effect when disabling access. Re-enabling must be done manually via the Databricks UI.
 DESCRIPTION
 }
+
+# Workspace environments
+variable "workspace_base_environments" {
+  description = <<DESCRIPTION
+Workspace base environments configuration.
+Base environments require environment definition files (YAML) to exist and be accessible before Terraform creates the base environment resources.
+Prerequisites:
+- CPU environment files can be stored either in a workspace directory or in a Databricks volume.
+- GPU environment files can only be stored in a workspace directory.
+Examples:
+- Workspace path: /Workspace/path/to/environment.yaml
+- Volume path: /Volumes/<catalog>/<schema>/<volume>/environment.yaml
+set_as_default specifies whether the environment should be used as the workspace default environment.
+DESCRIPTION
+  type = map(object({
+    display_name          = string
+    filepath              = string
+    base_environment_type = optional(string, "CPU")
+    set_as_default        = optional(bool, false)
+  }))
+
+  default = {}
+}

workspace_environments.tf

@@ -0,0 +1,26 @@
+resource "databricks_environments_workspace_base_environment" "this" {
+  for_each = var.workspace_base_environments
+
+  display_name          = each.value.display_name
+  filepath              = each.value.filepath
+  base_environment_type = each.value.base_environment_type
+}
+
+resource "databricks_environments_default_workspace_base_environment" "this" {
+  count = (
+    contains(keys(var.workspace_base_environments), "cpu") ||
+    contains(keys(var.workspace_base_environments), "gpu")
+  ) ? 1 : 0
+
+  cpu_workspace_base_environment = (
+    try(var.workspace_base_environments.cpu.set_as_default, false)
+    ? databricks_environments_workspace_base_environment.this["cpu"].name
+    : null
+  )
+
+  gpu_workspace_base_environment = (
+    try(var.workspace_base_environments.gpu.set_as_default, false)
+    ? databricks_environments_workspace_base_environment.this["gpu"].name
+    : null
+  )
+}