feat: Add identifier parsers

pkg/acceptance/helpers/grant_client.go

@@ -0,0 +1,45 @@
+package helpers
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
+	"github.com/stretchr/testify/require"
+)
+
+type GrantClient struct {
+	context *TestClientContext
+	ids     *IdsGenerator
+}
+
+func NewGrantClient(context *TestClientContext, idsGenerator *IdsGenerator) *GrantClient {
+	return &GrantClient{
+		context: context,
+		ids:     idsGenerator,
+	}
+}
+
+func (c *GrantClient) client() sdk.Grants {
+	return c.context.client.Grants
+}
+
+func (c *GrantClient) GrantOnSchemaToAccountRole(t *testing.T, schemaId sdk.DatabaseObjectIdentifier, accountRoleId sdk.AccountObjectIdentifier, privileges ...sdk.SchemaPrivilege) {
+	t.Helper()
+	ctx := context.Background()
+
+	err := c.client().GrantPrivilegesToAccountRole(
+		ctx,
+		&sdk.AccountRoleGrantPrivileges{
+			SchemaPrivileges: privileges,
+		},
+		&sdk.AccountRoleGrantOn{
+			Schema: &sdk.GrantOnSchema{
+				Schema: &schemaId,
+			},
+		},
+		accountRoleId,
+		new(sdk.GrantPrivilegesToAccountRoleOptions),
+	)
+	require.NoError(t, err)
+}

pkg/acceptance/helpers/test_client.go

@@ -24,6 +24,7 @@ type TestClient struct {
 	ExternalVolume            *ExternalVolumeClient
 	FailoverGroup             *FailoverGroupClient
 	FileFormat                *FileFormatClient
+	Grant                     *GrantClient
 	MaskingPolicy             *MaskingPolicyClient
 	MaterializedView          *MaterializedViewClient
 	NetworkPolicy             *NetworkPolicyClient
@@ -77,6 +78,7 @@ func NewTestClient(c *sdk.Client, database string, schema string, warehouse stri
 		ExternalVolume:            NewExternalVolumeClient(context, idsGenerator),
 		FailoverGroup:             NewFailoverGroupClient(context, idsGenerator),
 		FileFormat:                NewFileFormatClient(context, idsGenerator),
+		Grant:                     NewGrantClient(context, idsGenerator),
 		MaskingPolicy:             NewMaskingPolicyClient(context, idsGenerator),
 		MaterializedView:          NewMaterializedViewClient(context, idsGenerator),
 		NetworkPolicy:             NewNetworkPolicyClient(context, idsGenerator),

pkg/datasources/database.go

@@ -83,7 +83,11 @@ func ReadDatabase(d *schema.ResourceData, meta interface{}) error {
 	if err := d.Set("is_current", database.IsCurrent); err != nil {
 		return err
 	}
-	if err := d.Set("origin", database.Origin); err != nil {
+	var origin string
+	if database.Origin != nil {
+		origin = database.Origin.FullyQualifiedName()
+	}
+	if err := d.Set("origin", origin); err != nil {
 		return err
 	}
 	if err := d.Set("retention_time", database.RetentionTime); err != nil {

pkg/helpers/helpers.go

@@ -105,7 +105,7 @@ func DecodeSnowflakeID(id string) sdk.ObjectIdentifier {
 // The following configuration { "some_identifier": "db.name" } will be parsed as an object called "name" that lives
 // inside database called "db", not a database called "db.name". In this case quotes should be used.
 func DecodeSnowflakeParameterID(identifier string) (sdk.ObjectIdentifier, error) {
-	parts, err := ParseIdentifierString(identifier)
+	parts, err := sdk.ParseIdentifierString(identifier)
 	if err != nil {
 		return nil, err
 	}
@@ -126,7 +126,7 @@ func DecodeSnowflakeParameterID(identifier string) (sdk.ObjectIdentifier, error)
 // DecodeSnowflakeAccountIdentifier decodes account identifier (usually passed as one of the parameter in tf configuration) into sdk.AccountIdentifier.
 // Check more in https://docs.snowflake.com/en/sql-reference/sql/create-account#required-parameters.
 func DecodeSnowflakeAccountIdentifier(identifier string) (sdk.AccountIdentifier, error) {
-	parts, err := ParseIdentifierString(identifier)
+	parts, err := sdk.ParseIdentifierString(identifier)
 	if err != nil {
 		return sdk.AccountIdentifier{}, err
 	}
@@ -166,7 +166,7 @@ func ConcatSlices[T any](slices ...[]T) []T {
 // TODO(SNOW-999049): address during identifiers rework
 func ParseRootLocation(location string) (sdk.SchemaObjectIdentifier, string, error) {
 	location = strings.TrimPrefix(location, "@")
-	parts, err := parseIdentifierStringWithOpts(location, func(r *csv.Reader) {
+	parts, err := sdk.ParseIdentifierStringWithOpts(location, func(r *csv.Reader) {
 		r.Comma = '.'
 		r.LazyQuotes = true
 	})