feat: Allow state store schema's to be included when schemas are marshalled into JSON output

SarahFrench · SarahFrench · commit d19eb839cc03 · 2025-12-12T11:26:18.000Z
diff --git a/internal/command/jsonprovider/provider.go b/internal/command/jsonprovider/provider.go
@@ -31,6 +31,7 @@ type Provider struct {
 	Functions                map[string]*jsonfunction.FunctionSignature `json:"functions,omitempty"`
 	ResourceIdentitySchemas  map[string]*IdentitySchema                 `json:"resource_identity_schemas,omitempty"`
 	ActionSchemas            map[string]*ActionSchema                   `json:"action_schemas,omitempty"`
+	StateStoreSchemas        map[string]*Schema                         `json:"state_store_schemas,omitempty"`
 }
 
 func newProviders() *Providers {
@@ -69,6 +70,7 @@ func marshalProvider(tps providers.ProviderSchema) *Provider {
 		Functions:                jsonfunction.MarshalProviderFunctions(tps.Functions),
 		ResourceIdentitySchemas:  marshalIdentitySchemas(tps.ResourceTypes),
 		ActionSchemas:            marshalActionSchemas(tps.Actions),
+		StateStoreSchemas:        marshalSchemas(tps.StateStores),
 	}
 
 	// List resource schemas are nested under a "config" block, so we need to

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,7 @@ type Provider struct {`
`31`	`31`	Functions map[string]*jsonfunction.FunctionSignature `json:"functions,omitempty"`
`32`	`32`	ResourceIdentitySchemas map[string]*IdentitySchema `json:"resource_identity_schemas,omitempty"`
`33`	`33`	ActionSchemas map[string]*ActionSchema `json:"action_schemas,omitempty"`
	`34`	+ StateStoreSchemas map[string]*Schema `json:"state_store_schemas,omitempty"`
`34`	`35`	`}`
`35`	`36`
`36`	`37`	`func newProviders() *Providers {`
`@@ -69,6 +70,7 @@ func marshalProvider(tps providers.ProviderSchema) *Provider {`
`69`	`70`	`Functions: jsonfunction.MarshalProviderFunctions(tps.Functions),`
`70`	`71`	`ResourceIdentitySchemas: marshalIdentitySchemas(tps.ResourceTypes),`
`71`	`72`	`ActionSchemas: marshalActionSchemas(tps.Actions),`
	`73`	`+ StateStoreSchemas: marshalSchemas(tps.StateStores),`
`72`	`74`	`}`
`73`	`75`
`74`	`76`	`// List resource schemas are nested under a "config" block, so we need to`