From 9fa29e33d5a77d26cb0b5602eb81d83b6c00982b Mon Sep 17 00:00:00 2001
From: Heng Lu <henglu@microsoft.com>
Date: Fri, 7 May 2021 09:37:59 +0800
Subject: [PATCH] add resource azurerm_cosmosdb_sql_user_defined_function

---
 .../internal/services/cosmos/client/client.go |   5 +
 .../cosmos/cosmosdb_sql_function_resource.go  | 156 ++++++++++++++
 .../cosmosdb_sql_function_resource_test.go    | 190 ++++++++++++++++++
 .../services/cosmos/parse/sql_function.go     |  87 ++++++++
 .../cosmos/parse/sql_function_test.go         | 160 +++++++++++++++
 .../internal/services/cosmos/registration.go  |   1 +
 .../internal/services/cosmos/resourceids.go   |   1 +
 .../cosmos/validate/sql_function_id.go        |  23 +++
 .../cosmos/validate/sql_function_id_test.go   | 112 +++++++++++
 ...db_sql_user_defined_function.html.markdown |  74 +++++++
 10 files changed, 809 insertions(+)
 create mode 100644 azurerm/internal/services/cosmos/cosmosdb_sql_function_resource.go
 create mode 100644 azurerm/internal/services/cosmos/cosmosdb_sql_function_resource_test.go
 create mode 100644 azurerm/internal/services/cosmos/parse/sql_function.go
 create mode 100644 azurerm/internal/services/cosmos/parse/sql_function_test.go
 create mode 100644 azurerm/internal/services/cosmos/validate/sql_function_id.go
 create mode 100644 azurerm/internal/services/cosmos/validate/sql_function_id_test.go
 create mode 100644 website/docs/r/cosmosdb_sql_user_defined_function.html.markdown

diff --git a/azurerm/internal/services/cosmos/client/client.go b/azurerm/internal/services/cosmos/client/client.go
index 72dec52fcf60..d56d2278eed3 100644
--- a/azurerm/internal/services/cosmos/client/client.go
+++ b/azurerm/internal/services/cosmos/client/client.go
@@ -12,6 +12,7 @@ type Client struct {
 	MongoDbClient           *documentdb.MongoDBResourcesClient
 	NotebookWorkspaceClient *documentdb.NotebookWorkspacesClient
 	SqlClient               *documentdb.SQLResourcesClient
+	SqlResourceClient       *documentdb.SQLResourcesClient
 	TableClient             *documentdb.TableResourcesClient
 }
 
@@ -34,6 +35,9 @@ func NewClient(o *common.ClientOptions) *Client {
 	sqlClient := documentdb.NewSQLResourcesClientWithBaseURI(o.ResourceManagerEndpoint, o.SubscriptionId)
 	o.ConfigureClient(&sqlClient.Client, o.ResourceManagerAuthorizer)
 
+	sqlResourceClient := documentdb.NewSQLResourcesClientWithBaseURI(o.ResourceManagerEndpoint, o.SubscriptionId)
+	o.ConfigureClient(&sqlResourceClient.Client, o.ResourceManagerAuthorizer)
+
 	tableClient := documentdb.NewTableResourcesClientWithBaseURI(o.ResourceManagerEndpoint, o.SubscriptionId)
 	o.ConfigureClient(&tableClient.Client, o.ResourceManagerAuthorizer)
 
@@ -44,6 +48,7 @@ func NewClient(o *common.ClientOptions) *Client {
 		MongoDbClient:           &mongoDbClient,
 		NotebookWorkspaceClient: &notebookWorkspaceClient,
 		SqlClient:               &sqlClient,
+		SqlResourceClient:       &sqlResourceClient,
 		TableClient:             &tableClient,
 	}
 }
diff --git a/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource.go b/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource.go
new file mode 100644
index 000000000000..a1b6d30fed3a
--- /dev/null
+++ b/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource.go
@@ -0,0 +1,156 @@
+package cosmos
+
+import (
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/Azure/azure-sdk-for-go/services/cosmos-db/mgmt/2021-01-15/documentdb"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/validation"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/tf"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/clients"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos/parse"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos/validate"
+	azSchema "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/tf/schema"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/timeouts"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/utils"
+)
+
+func resourceCosmosDbSQLFunction() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceCosmosDbSQLFunctionCreateUpdate,
+		Read:   resourceCosmosDbSQLFunctionRead,
+		Update: resourceCosmosDbSQLFunctionCreateUpdate,
+		Delete: resourceCosmosDbSQLFunctionDelete,
+
+		Timeouts: &schema.ResourceTimeout{
+			Create: schema.DefaultTimeout(30 * time.Minute),
+			Read:   schema.DefaultTimeout(5 * time.Minute),
+			Update: schema.DefaultTimeout(30 * time.Minute),
+			Delete: schema.DefaultTimeout(30 * time.Minute),
+		},
+
+		Importer: azSchema.ValidateResourceIDPriorToImport(func(id string) error {
+			_, err := parse.SqlFunctionID(id)
+			return err
+		}),
+
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+
+			"container_id": {
+				Type:         schema.TypeString,
+				Required:     true,
+				ForceNew:     true,
+				ValidateFunc: validate.SqlContainerID,
+			},
+
+			"body": {
+				Type:         schema.TypeString,
+				Required:     true,
+				ValidateFunc: validation.StringIsNotEmpty,
+			},
+		},
+	}
+}
+func resourceCosmosDbSQLFunctionCreateUpdate(d *schema.ResourceData, meta interface{}) error {
+	subscriptionId := meta.(*clients.Client).Account.SubscriptionId
+	client := meta.(*clients.Client).Cosmos.SqlResourceClient
+	ctx, cancel := timeouts.ForCreateUpdate(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	name := d.Get("name").(string)
+	containerId, _ := parse.SqlContainerID(d.Get("container_id").(string))
+	body := d.Get("body").(string)
+
+	id := parse.NewSqlFunctionID(subscriptionId, containerId.ResourceGroup, containerId.DatabaseAccountName, containerId.SqlDatabaseName, containerId.ContainerName, name)
+
+	if d.IsNewResource() {
+		existing, err := client.GetSQLUserDefinedFunction(ctx, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName)
+		if err != nil {
+			if !utils.ResponseWasNotFound(existing.Response) {
+				return fmt.Errorf("checking for existing CosmosDb SqlFunction %q: %+v", id, err)
+			}
+		}
+		if !utils.ResponseWasNotFound(existing.Response) {
+			return tf.ImportAsExistsError("azurerm_cosmosdb_sql_function", id.ID())
+		}
+	}
+
+	createUpdateSqlUserDefinedFunctionParameters := documentdb.SQLUserDefinedFunctionCreateUpdateParameters{
+		SQLUserDefinedFunctionCreateUpdateProperties: &documentdb.SQLUserDefinedFunctionCreateUpdateProperties{
+			Resource: &documentdb.SQLUserDefinedFunctionResource{
+				ID:   &name,
+				Body: &body,
+			},
+			Options: &documentdb.CreateUpdateOptions{},
+		},
+	}
+	future, err := client.CreateUpdateSQLUserDefinedFunction(ctx, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName, createUpdateSqlUserDefinedFunctionParameters)
+	if err != nil {
+		return fmt.Errorf("creating/updating CosmosDb SqlFunction %q: %+v", id, err)
+	}
+
+	if err := future.WaitForCompletionRef(ctx, client.Client); err != nil {
+		return fmt.Errorf("waiting for creation/update of the CosmosDb SqlFunction %q: %+v", id, err)
+	}
+
+	d.SetId(id.ID())
+	return resourceCosmosDbSQLFunctionRead(d, meta)
+}
+
+func resourceCosmosDbSQLFunctionRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*clients.Client).Cosmos.SqlResourceClient
+	ctx, cancel := timeouts.ForRead(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	id, err := parse.SqlFunctionID(d.Id())
+	if err != nil {
+		return err
+	}
+
+	resp, err := client.GetSQLUserDefinedFunction(ctx, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName)
+	if err != nil {
+		if utils.ResponseWasNotFound(resp.Response) {
+			log.Printf("[INFO] CosmosDb SqlFunction %q does not exist - removing from state", d.Id())
+			d.SetId("")
+			return nil
+		}
+		return fmt.Errorf("retrieving CosmosDb SqlFunction %q: %+v", id, err)
+	}
+	containerId := parse.NewSqlContainerID(id.SubscriptionId, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName)
+	d.Set("name", id.UserDefinedFunctionName)
+	d.Set("container_id", containerId.ID())
+	if props := resp.SQLUserDefinedFunctionGetProperties; props != nil {
+		if props.Resource != nil {
+			d.Set("body", props.Resource.Body)
+		}
+	}
+	return nil
+}
+
+func resourceCosmosDbSQLFunctionDelete(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*clients.Client).Cosmos.SqlResourceClient
+	ctx, cancel := timeouts.ForDelete(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	id, err := parse.SqlFunctionID(d.Id())
+	if err != nil {
+		return err
+	}
+
+	future, err := client.DeleteSQLUserDefinedFunction(ctx, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName)
+	if err != nil {
+		return fmt.Errorf("deleting CosmosDb SqlFunction %q: %+v", id, err)
+	}
+
+	if err := future.WaitForCompletionRef(ctx, client.Client); err != nil {
+		return fmt.Errorf("waiting for deletion of the CosmosDb SqlFunction %q: %+v", id, err)
+	}
+	return nil
+}
diff --git a/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource_test.go b/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource_test.go
new file mode 100644
index 000000000000..7bb9c6334ae0
--- /dev/null
+++ b/azurerm/internal/services/cosmos/cosmosdb_sql_function_resource_test.go
@@ -0,0 +1,190 @@
+package cosmos_test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/Azure/azure-sdk-for-go/services/cosmos-db/mgmt/2021-01-15/documentdb"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/terraform"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/acceptance"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/acceptance/check"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/clients"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos/parse"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/utils"
+)
+
+type CosmosDbSQLFunctionResource struct{}
+
+func TestAccCosmosDbSQLFunction_basic(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_cosmosdb_sql_function", "test")
+	r := CosmosDbSQLFunctionResource{}
+	data.ResourceTest(t, r, []resource.TestStep{
+		{
+			Config: r.basic(data),
+			Check: resource.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
+func TestAccCosmosDbSQLFunction_requiresImport(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_cosmosdb_sql_function", "test")
+	r := CosmosDbSQLFunctionResource{}
+	data.ResourceTest(t, r, []resource.TestStep{
+		{
+			Config: r.basic(data),
+			Check: resource.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.RequiresImportErrorStep(r.requiresImport),
+	})
+}
+
+func TestAccCosmosDbSQLFunction_update(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_cosmosdb_sql_function", "test")
+	r := CosmosDbSQLFunctionResource{}
+	data.ResourceTest(t, r, []resource.TestStep{
+		{
+			Config: r.basic(data),
+			Check: resource.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+		{
+			Config: r.update(data),
+			Check: resource.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+		{
+			Config: r.basic(data),
+			Check: resource.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
+func (r CosmosDbSQLFunctionResource) Exists(ctx context.Context, client *clients.Client, state *terraform.InstanceState) (*bool, error) {
+	id, err := parse.SqlFunctionID(state.ID)
+	if err != nil {
+		return nil, err
+	}
+	resp, err := client.Cosmos.SqlResourceClient.GetSQLUserDefinedFunction(ctx, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName)
+	if err != nil {
+		if utils.ResponseWasNotFound(resp.Response) {
+			return utils.Bool(false), nil
+		}
+		return nil, fmt.Errorf("retrieving CosmosDb SQLFunction %q: %+v", id, err)
+	}
+	return utils.Bool(true), nil
+}
+
+func (r CosmosDbSQLFunctionResource) template(data acceptance.TestData) string {
+	return fmt.Sprintf(`
+provider "azurerm" {
+  features {}
+}
+
+resource "azurerm_resource_group" "test" {
+  name     = "acctestRG-cosmos-%[2]d"
+  location = "%[1]s"
+}
+
+resource "azurerm_cosmosdb_account" "test" {
+  name                = "acctest-cosmos-%[2]d"
+  location            = azurerm_resource_group.test.location
+  resource_group_name = azurerm_resource_group.test.name
+  offer_type          = "Standard"
+  kind                = "%[3]s"
+
+  consistency_policy {
+    consistency_level = "%[4]s"
+  }
+
+  geo_location {
+    location          = azurerm_resource_group.test.location
+    failover_priority = 0
+  }
+}
+
+resource "azurerm_cosmosdb_sql_database" "test" {
+  name                = "acctest-sql-database-%[2]d"
+  resource_group_name = azurerm_cosmosdb_account.test.resource_group_name
+  account_name        = azurerm_cosmosdb_account.test.name
+}
+
+resource "azurerm_cosmosdb_sql_container" "test" {
+  name                = "acctest-sql-container-%[2]d"
+  resource_group_name = azurerm_cosmosdb_account.test.resource_group_name
+  account_name        = azurerm_cosmosdb_account.test.name
+  database_name       = azurerm_cosmosdb_sql_database.test.name
+  partition_key_path  = "/definition/id"
+}
+	`, data.Locations.Primary, data.RandomInteger, string(documentdb.GlobalDocumentDB), string(documentdb.Session))
+}
+
+func (r CosmosDbSQLFunctionResource) basic(data acceptance.TestData) string {
+	template := r.template(data)
+	return fmt.Sprintf(`
+%s
+
+resource "azurerm_cosmosdb_sql_function" "test" {
+  name         = "acctest-dssdf-%d"
+  container_id = azurerm_cosmosdb_sql_container.test.id
+  body         = <<BODY
+  	function test() {
+		var context = getContext();
+		var response = context.getResponse();
+		response.setBody('Hello, World');
+	}
+BODY
+}
+`, template, data.RandomInteger)
+}
+
+func (r CosmosDbSQLFunctionResource) requiresImport(data acceptance.TestData) string {
+	config := r.basic(data)
+	return fmt.Sprintf(`
+%s
+
+resource "azurerm_cosmosdb_sql_function" "import" {
+  name         = azurerm_cosmosdb_sql_function.test.name
+  container_id = azurerm_cosmosdb_sql_function.test.container_id
+  body         = <<BODY
+  	function test() {
+		var context = getContext();
+		var response = context.getResponse();
+		response.setBody('Hello, World');
+	}
+BODY
+}
+`, config)
+}
+
+func (r CosmosDbSQLFunctionResource) update(data acceptance.TestData) string {
+	template := r.template(data)
+	return fmt.Sprintf(`
+%s
+
+resource "azurerm_cosmosdb_sql_function" "test" {
+  name         = "acctest-dssdf-%d"
+  container_id = azurerm_cosmosdb_sql_container.test.id
+  body         = <<BODY
+	function test() {
+		var context = getContext();
+		var response = context.getResponse();
+		response.setBody('Welcome To Terraform');
+	}
+BODY
+}
+`, template, data.RandomInteger)
+}
diff --git a/azurerm/internal/services/cosmos/parse/sql_function.go b/azurerm/internal/services/cosmos/parse/sql_function.go
new file mode 100644
index 000000000000..a952d92e9643
--- /dev/null
+++ b/azurerm/internal/services/cosmos/parse/sql_function.go
@@ -0,0 +1,87 @@
+package parse
+
+// NOTE: this file is generated via 'go:generate' - manual changes will be overwritten
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/azure"
+)
+
+type SqlFunctionId struct {
+	SubscriptionId          string
+	ResourceGroup           string
+	DatabaseAccountName     string
+	SqlDatabaseName         string
+	ContainerName           string
+	UserDefinedFunctionName string
+}
+
+func NewSqlFunctionID(subscriptionId, resourceGroup, databaseAccountName, sqlDatabaseName, containerName, userDefinedFunctionName string) SqlFunctionId {
+	return SqlFunctionId{
+		SubscriptionId:          subscriptionId,
+		ResourceGroup:           resourceGroup,
+		DatabaseAccountName:     databaseAccountName,
+		SqlDatabaseName:         sqlDatabaseName,
+		ContainerName:           containerName,
+		UserDefinedFunctionName: userDefinedFunctionName,
+	}
+}
+
+func (id SqlFunctionId) String() string {
+	segments := []string{
+		fmt.Sprintf("User Defined Function Name %q", id.UserDefinedFunctionName),
+		fmt.Sprintf("Container Name %q", id.ContainerName),
+		fmt.Sprintf("Sql Database Name %q", id.SqlDatabaseName),
+		fmt.Sprintf("Database Account Name %q", id.DatabaseAccountName),
+		fmt.Sprintf("Resource Group %q", id.ResourceGroup),
+	}
+	segmentsStr := strings.Join(segments, " / ")
+	return fmt.Sprintf("%s: (%s)", "Sql Function", segmentsStr)
+}
+
+func (id SqlFunctionId) ID() string {
+	fmtString := "/subscriptions/%s/resourceGroups/%s/providers/Microsoft.DocumentDB/databaseAccounts/%s/sqlDatabases/%s/containers/%s/userDefinedFunctions/%s"
+	return fmt.Sprintf(fmtString, id.SubscriptionId, id.ResourceGroup, id.DatabaseAccountName, id.SqlDatabaseName, id.ContainerName, id.UserDefinedFunctionName)
+}
+
+// SqlFunctionID parses a SqlFunction ID into an SqlFunctionId struct
+func SqlFunctionID(input string) (*SqlFunctionId, error) {
+	id, err := azure.ParseAzureResourceID(input)
+	if err != nil {
+		return nil, err
+	}
+
+	resourceId := SqlFunctionId{
+		SubscriptionId: id.SubscriptionID,
+		ResourceGroup:  id.ResourceGroup,
+	}
+
+	if resourceId.SubscriptionId == "" {
+		return nil, fmt.Errorf("ID was missing the 'subscriptions' element")
+	}
+
+	if resourceId.ResourceGroup == "" {
+		return nil, fmt.Errorf("ID was missing the 'resourceGroups' element")
+	}
+
+	if resourceId.DatabaseAccountName, err = id.PopSegment("databaseAccounts"); err != nil {
+		return nil, err
+	}
+	if resourceId.SqlDatabaseName, err = id.PopSegment("sqlDatabases"); err != nil {
+		return nil, err
+	}
+	if resourceId.ContainerName, err = id.PopSegment("containers"); err != nil {
+		return nil, err
+	}
+	if resourceId.UserDefinedFunctionName, err = id.PopSegment("userDefinedFunctions"); err != nil {
+		return nil, err
+	}
+
+	if err := id.ValidateNoEmptySegments(input); err != nil {
+		return nil, err
+	}
+
+	return &resourceId, nil
+}
diff --git a/azurerm/internal/services/cosmos/parse/sql_function_test.go b/azurerm/internal/services/cosmos/parse/sql_function_test.go
new file mode 100644
index 000000000000..ec30f4970079
--- /dev/null
+++ b/azurerm/internal/services/cosmos/parse/sql_function_test.go
@@ -0,0 +1,160 @@
+package parse
+
+// NOTE: this file is generated via 'go:generate' - manual changes will be overwritten
+
+import (
+	"testing"
+
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/resourceid"
+)
+
+var _ resourceid.Formatter = SqlFunctionId{}
+
+func TestSqlFunctionIDFormatter(t *testing.T) {
+	actual := NewSqlFunctionID("12345678-1234-9876-4563-123456789012", "resourceGroup1", "account1", "database1", "container1", "userDefinedFunction1").ID()
+	expected := "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/userDefinedFunction1"
+	if actual != expected {
+		t.Fatalf("Expected %q but got %q", expected, actual)
+	}
+}
+
+func TestSqlFunctionID(t *testing.T) {
+	testData := []struct {
+		Input    string
+		Error    bool
+		Expected *SqlFunctionId
+	}{
+
+		{
+			// empty
+			Input: "",
+			Error: true,
+		},
+
+		{
+			// missing SubscriptionId
+			Input: "/",
+			Error: true,
+		},
+
+		{
+			// missing value for SubscriptionId
+			Input: "/subscriptions/",
+			Error: true,
+		},
+
+		{
+			// missing ResourceGroup
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/",
+			Error: true,
+		},
+
+		{
+			// missing value for ResourceGroup
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/",
+			Error: true,
+		},
+
+		{
+			// missing DatabaseAccountName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/",
+			Error: true,
+		},
+
+		{
+			// missing value for DatabaseAccountName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/",
+			Error: true,
+		},
+
+		{
+			// missing SqlDatabaseName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/",
+			Error: true,
+		},
+
+		{
+			// missing value for SqlDatabaseName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/",
+			Error: true,
+		},
+
+		{
+			// missing ContainerName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/",
+			Error: true,
+		},
+
+		{
+			// missing value for ContainerName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/",
+			Error: true,
+		},
+
+		{
+			// missing UserDefinedFunctionName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/",
+			Error: true,
+		},
+
+		{
+			// missing value for UserDefinedFunctionName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/",
+			Error: true,
+		},
+
+		{
+			// valid
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/userDefinedFunction1",
+			Expected: &SqlFunctionId{
+				SubscriptionId:          "12345678-1234-9876-4563-123456789012",
+				ResourceGroup:           "resourceGroup1",
+				DatabaseAccountName:     "account1",
+				SqlDatabaseName:         "database1",
+				ContainerName:           "container1",
+				UserDefinedFunctionName: "userDefinedFunction1",
+			},
+		},
+
+		{
+			// upper-cased
+			Input: "/SUBSCRIPTIONS/12345678-1234-9876-4563-123456789012/RESOURCEGROUPS/RESOURCEGROUP1/PROVIDERS/MICROSOFT.DOCUMENTDB/DATABASEACCOUNTS/ACCOUNT1/SQLDATABASES/DATABASE1/CONTAINERS/CONTAINER1/USERDEFINEDFUNCTIONS/USERDEFINEDFUNCTION1",
+			Error: true,
+		},
+	}
+
+	for _, v := range testData {
+		t.Logf("[DEBUG] Testing %q", v.Input)
+
+		actual, err := SqlFunctionID(v.Input)
+		if err != nil {
+			if v.Error {
+				continue
+			}
+
+			t.Fatalf("Expect a value but got an error: %s", err)
+		}
+		if v.Error {
+			t.Fatal("Expect an error but didn't get one")
+		}
+
+		if actual.SubscriptionId != v.Expected.SubscriptionId {
+			t.Fatalf("Expected %q but got %q for SubscriptionId", v.Expected.SubscriptionId, actual.SubscriptionId)
+		}
+		if actual.ResourceGroup != v.Expected.ResourceGroup {
+			t.Fatalf("Expected %q but got %q for ResourceGroup", v.Expected.ResourceGroup, actual.ResourceGroup)
+		}
+		if actual.DatabaseAccountName != v.Expected.DatabaseAccountName {
+			t.Fatalf("Expected %q but got %q for DatabaseAccountName", v.Expected.DatabaseAccountName, actual.DatabaseAccountName)
+		}
+		if actual.SqlDatabaseName != v.Expected.SqlDatabaseName {
+			t.Fatalf("Expected %q but got %q for SqlDatabaseName", v.Expected.SqlDatabaseName, actual.SqlDatabaseName)
+		}
+		if actual.ContainerName != v.Expected.ContainerName {
+			t.Fatalf("Expected %q but got %q for ContainerName", v.Expected.ContainerName, actual.ContainerName)
+		}
+		if actual.UserDefinedFunctionName != v.Expected.UserDefinedFunctionName {
+			t.Fatalf("Expected %q but got %q for UserDefinedFunctionName", v.Expected.UserDefinedFunctionName, actual.UserDefinedFunctionName)
+		}
+	}
+}
diff --git a/azurerm/internal/services/cosmos/registration.go b/azurerm/internal/services/cosmos/registration.go
index a29bb1805b5f..fd334e565eaf 100644
--- a/azurerm/internal/services/cosmos/registration.go
+++ b/azurerm/internal/services/cosmos/registration.go
@@ -38,6 +38,7 @@ func (r Registration) SupportedResources() map[string]*schema.Resource {
 		"azurerm_cosmosdb_notebook_workspace":   resourceCosmosDbNotebookWorkspace(),
 		"azurerm_cosmosdb_sql_container":        resourceCosmosDbSQLContainer(),
 		"azurerm_cosmosdb_sql_database":         resourceCosmosDbSQLDatabase(),
+		"azurerm_cosmosdb_sql_function":         resourceCosmosDbSQLFunction(),
 		"azurerm_cosmosdb_sql_stored_procedure": resourceCosmosDbSQLStoredProcedure(),
 		"azurerm_cosmosdb_table":                resourceCosmosDbTable(),
 	}
diff --git a/azurerm/internal/services/cosmos/resourceids.go b/azurerm/internal/services/cosmos/resourceids.go
index af2a39ae35b5..6af4e0b0d872 100644
--- a/azurerm/internal/services/cosmos/resourceids.go
+++ b/azurerm/internal/services/cosmos/resourceids.go
@@ -10,5 +10,6 @@ package cosmos
 //go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=NotebookWorkspace -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/notebookWorkspaces/notebookWorkspace1
 //go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=SqlContainer -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resGroup1/providers/Microsoft.DocumentDB/databaseAccounts/acc1/sqlDatabases/db1/containers/container1
 //go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=SqlDatabase -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resGroup1/providers/Microsoft.DocumentDB/databaseAccounts/acc1/sqlDatabases/db1
+//go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=SqlFunction -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/userDefinedFunction1
 //go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=SqlStoredProcedure -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resGroup1/providers/Microsoft.DocumentDB/databaseAccounts/acc1/sqlDatabases/db1/containers/container1/storedProcedures/sproc1
 //go:generate go run ../../tools/generator-resource-id/main.go -path=./ -name=Table -id=/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resGroup1/providers/Microsoft.DocumentDB/databaseAccounts/acc1/tables/table1
diff --git a/azurerm/internal/services/cosmos/validate/sql_function_id.go b/azurerm/internal/services/cosmos/validate/sql_function_id.go
new file mode 100644
index 000000000000..282fc52948ed
--- /dev/null
+++ b/azurerm/internal/services/cosmos/validate/sql_function_id.go
@@ -0,0 +1,23 @@
+package validate
+
+// NOTE: this file is generated via 'go:generate' - manual changes will be overwritten
+
+import (
+	"fmt"
+
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos/parse"
+)
+
+func SqlFunctionID(input interface{}, key string) (warnings []string, errors []error) {
+	v, ok := input.(string)
+	if !ok {
+		errors = append(errors, fmt.Errorf("expected %q to be a string", key))
+		return
+	}
+
+	if _, err := parse.SqlFunctionID(v); err != nil {
+		errors = append(errors, err)
+	}
+
+	return
+}
diff --git a/azurerm/internal/services/cosmos/validate/sql_function_id_test.go b/azurerm/internal/services/cosmos/validate/sql_function_id_test.go
new file mode 100644
index 000000000000..a4ccc555c8cc
--- /dev/null
+++ b/azurerm/internal/services/cosmos/validate/sql_function_id_test.go
@@ -0,0 +1,112 @@
+package validate
+
+// NOTE: this file is generated via 'go:generate' - manual changes will be overwritten
+
+import "testing"
+
+func TestSqlFunctionID(t *testing.T) {
+	cases := []struct {
+		Input string
+		Valid bool
+	}{
+
+		{
+			// empty
+			Input: "",
+			Valid: false,
+		},
+
+		{
+			// missing SubscriptionId
+			Input: "/",
+			Valid: false,
+		},
+
+		{
+			// missing value for SubscriptionId
+			Input: "/subscriptions/",
+			Valid: false,
+		},
+
+		{
+			// missing ResourceGroup
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/",
+			Valid: false,
+		},
+
+		{
+			// missing value for ResourceGroup
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/",
+			Valid: false,
+		},
+
+		{
+			// missing DatabaseAccountName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/",
+			Valid: false,
+		},
+
+		{
+			// missing value for DatabaseAccountName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/",
+			Valid: false,
+		},
+
+		{
+			// missing SqlDatabaseName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/",
+			Valid: false,
+		},
+
+		{
+			// missing value for SqlDatabaseName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/",
+			Valid: false,
+		},
+
+		{
+			// missing ContainerName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/",
+			Valid: false,
+		},
+
+		{
+			// missing value for ContainerName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/",
+			Valid: false,
+		},
+
+		{
+			// missing UserDefinedFunctionName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/",
+			Valid: false,
+		},
+
+		{
+			// missing value for UserDefinedFunctionName
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/",
+			Valid: false,
+		},
+
+		{
+			// valid
+			Input: "/subscriptions/12345678-1234-9876-4563-123456789012/resourceGroups/resourceGroup1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/userDefinedFunction1",
+			Valid: true,
+		},
+
+		{
+			// upper-cased
+			Input: "/SUBSCRIPTIONS/12345678-1234-9876-4563-123456789012/RESOURCEGROUPS/RESOURCEGROUP1/PROVIDERS/MICROSOFT.DOCUMENTDB/DATABASEACCOUNTS/ACCOUNT1/SQLDATABASES/DATABASE1/CONTAINERS/CONTAINER1/USERDEFINEDFUNCTIONS/USERDEFINEDFUNCTION1",
+			Valid: false,
+		},
+	}
+	for _, tc := range cases {
+		t.Logf("[DEBUG] Testing Value %s", tc.Input)
+		_, errors := SqlFunctionID(tc.Input, "test")
+		valid := len(errors) == 0
+
+		if tc.Valid != valid {
+			t.Fatalf("Expected %t but got %t", tc.Valid, valid)
+		}
+	}
+}
diff --git a/website/docs/r/cosmosdb_sql_user_defined_function.html.markdown b/website/docs/r/cosmosdb_sql_user_defined_function.html.markdown
new file mode 100644
index 000000000000..76739d1fd256
--- /dev/null
+++ b/website/docs/r/cosmosdb_sql_user_defined_function.html.markdown
@@ -0,0 +1,74 @@
+---
+subcategory: "CosmosDB (DocumentDB)"
+layout: "azurerm"
+page_title: "Azure Resource Manager: azurerm_cosmosdb_sql_user_defined_function"
+description: |-
+  Manages an SQL User Defined Function.
+---
+
+# azurerm_cosmosdb_sql_user_defined_function
+
+Manages an SQL User Defined Function.
+
+## Example Usage
+
+```hcl
+data "azurerm_cosmosdb_account" "example" {
+  name                = "tfex-cosmosdb-account"
+  resource_group_name = "tfex-cosmosdb-account-rg"
+}
+
+resource "azurerm_cosmosdb_sql_database" "example" {
+  name                = "tfex-cosmos-db"
+  resource_group_name = data.azurerm_cosmosdb_account.example.resource_group_name
+  account_name        = data.azurerm_cosmosdb_account.example.name
+  throughput          = 400
+}
+
+resource "azurerm_cosmosdb_sql_container" "example" {
+  name                = "example-container"
+  resource_group_name = azurerm_cosmosdb_account.example.resource_group_name
+  account_name        = azurerm_cosmosdb_account.example.name
+  database_name       = azurerm_cosmosdb_sql_database.example.name
+  partition_key_path  = "/id"
+}
+
+resource "azurerm_cosmosdb_sql_user_defined_function" "example" {
+  name         = "test-function"
+  container_id = azurerm_cosmosdb_sql_container.example.id
+  body         = "function trigger(){}"
+}
+```
+
+## Arguments Reference
+
+The following arguments are supported:
+
+* `name` - (Required) The name which should be used for this SQL User Defined Function. Changing this forces a new SQL User Defined Function to be created.
+
+* `container_id` - (Required) The id of the Cosmos DB SQL Container to create the SQL User Defined Function within. Changing this forces a new SQL User Defined Function to be created.
+
+* `body` - (Required) Body of the User Defined Function.
+
+## Attributes Reference
+
+In addition to the Arguments listed above - the following Attributes are exported: 
+
+* `id` - The ID of the SQL User Defined Function.
+
+## Timeouts
+
+The `timeouts` block allows you to specify [timeouts](https://www.terraform.io/docs/configuration/resources.html#timeouts) for certain actions:
+
+* `create` - (Defaults to 30 minutes) Used when creating the SQL User Defined Function.
+* `read` - (Defaults to 5 minutes) Used when retrieving the SQL User Defined Function.
+* `update` - (Defaults to 30 minutes) Used when updating the SQL User Defined Function.
+* `delete` - (Defaults to 30 minutes) Used when deleting the SQL User Defined Function.
+
+## Import
+
+SQL User Defined Functions can be imported using the `resource id`, e.g.
+
+```shell
+terraform import azurerm_cosmosdb_sql_user_defined_function.example /subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/group1/providers/Microsoft.DocumentDB/databaseAccounts/account1/sqlDatabases/database1/containers/container1/userDefinedFunctions/userDefinedFunction1
+```