From 6b38834ade0a41c04542320def414ce21b743d44 Mon Sep 17 00:00:00 2001
From: edwardfeng-db <jinghao.feng@databricks.com>
Date: Wed, 21 Aug 2024 15:34:13 +0200
Subject: [PATCH] update

---
 .../tfschema/customizable_schema.go           | 126 ++++++++++++++++++
 1 file changed, 126 insertions(+)
 create mode 100644 pluginframework/tfschema/customizable_schema.go

diff --git a/pluginframework/tfschema/customizable_schema.go b/pluginframework/tfschema/customizable_schema.go
new file mode 100644
index 0000000000..a7b1e50f08
--- /dev/null
+++ b/pluginframework/tfschema/customizable_schema.go
@@ -0,0 +1,126 @@
+package tfschema
+
+import (
+	"fmt"
+)
+
+type CustomizableSchemaPluginFramework struct {
+	attr AttributeBuilder
+}
+
+func ConstructCustomizableSchema(attributes map[string]AttributeBuilder) *CustomizableSchemaPluginFramework {
+	attr := AttributeBuilder(SingleNestedAttribute{Attributes: attributes})
+	return &CustomizableSchemaPluginFramework{attr: attr}
+}
+
+// Converts CustomizableSchema into a map from string to Attribute.
+func (s *CustomizableSchemaPluginFramework) ToAttributeMap() map[string]AttributeBuilder {
+	return attributeToMap(&s.attr)
+}
+
+func attributeToMap(attr *AttributeBuilder) map[string]AttributeBuilder {
+	var m map[string]AttributeBuilder
+	switch attr := (*attr).(type) {
+	case SingleNestedAttribute:
+		m = attr.Attributes
+	case ListNestedAttribute:
+		m = attr.NestedObject.Attributes
+	case MapNestedAttribute:
+		m = attr.NestedObject.Attributes
+	default:
+		panic(fmt.Errorf("cannot convert to map, attribute is not nested"))
+	}
+
+	return m
+}
+
+func (s *CustomizableSchemaPluginFramework) AddValidator(v any, path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.AddValidators(v)
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+
+	return s
+}
+
+func (s *CustomizableSchemaPluginFramework) SetOptional(path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetOptional()
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+
+	return s
+}
+
+func (s *CustomizableSchemaPluginFramework) SetRequired(path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetRequired()
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+
+	return s
+}
+
+func (s *CustomizableSchemaPluginFramework) SetSensitive(path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetSensitive()
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+	return s
+}
+
+func (s *CustomizableSchemaPluginFramework) SetDeprecated(msg string, path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetDeprecated(msg)
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+
+	return s
+}
+
+func (s *CustomizableSchemaPluginFramework) SetComputed(path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetComputed()
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+	return s
+}
+
+// SetReadOnly sets the schema to be read-only (i.e. computed, non-optional).
+// This should be used for fields that are not user-configurable but are returned
+// by the platform.
+func (s *CustomizableSchemaPluginFramework) SetReadOnly(path ...string) *CustomizableSchemaPluginFramework {
+	cb := func(attr AttributeBuilder) AttributeBuilder {
+		return attr.SetReadOnly()
+	}
+
+	navigateSchemaWithCallback(&s.attr, cb, path...)
+
+	return s
+}
+
+// Helper function for navigating through schema attributes, panics if path does not exist or invalid.
+func navigateSchemaWithCallback(s *AttributeBuilder, cb func(AttributeBuilder) AttributeBuilder, path ...string) (AttributeBuilder, error) {
+	current_scm := s
+	for i, p := range path {
+		m := attributeToMap(current_scm)
+
+		v, ok := m[p]
+		if !ok {
+			return nil, fmt.Errorf("missing key %s", p)
+		}
+
+		if i == len(path)-1 {
+			m[p] = cb(v)
+			return m[p], nil
+		}
+		current_scm = &v
+	}
+	return nil, fmt.Errorf("path %v is incomplete", path)
+}