add generate resource config rpc

Author: stephybun

tf5muxserver/diagnostics.go

@@ -68,6 +68,16 @@ func ephemeralResourceMissingError(typeName string) *tfprotov5.Diagnostic {
 	}
 }
 
+func generateResourceConfigMissingError(typeName string) *tfprotov5.Diagnostic {
+	return &tfprotov5.Diagnostic{
+		Severity: tfprotov5.DiagnosticSeverityError,
+		Summary:  "Generate Resource Config Not Implemented",
+		Detail: "The combined provider does not implement the requested generate resource config for the resource type. " +
+			"This is always an issue in the provider implementation and should be reported to the provider developers.\n\n" +
+			"Missing generate resource config for resource type: " + typeName,
+	}
+}
+
 func listResourceDuplicateError(typeName string) *tfprotov5.Diagnostic {
 	return &tfprotov5.Diagnostic{
 		Severity: tfprotov5.DiagnosticSeverityError,

tf5muxserver/mux_server.go

@@ -29,6 +29,9 @@ type muxServer struct {
 	// Routing for ephemeral resource types
 	ephemeralResources map[string]tfprotov5.ProviderServer
 
+	// Routing for generating resource configuration
+	generateResourceConfig map[string]tfprotov5.ProviderServer
+
 	// Routing for list resource types
 	listResources map[string]tfprotov5.ProviderServer
 
@@ -65,6 +68,10 @@ func (s *muxServer) ProviderServer() tfprotov5.ProviderServer {
 	return s
 }
 
+func (s *muxServer) ProviderServers() []tfprotov5.ProviderServer {
+	return s.servers
+}
+
 func (s *muxServer) getActionServer(ctx context.Context, actionType string) (tfprotov5.ProviderServer, []*tfprotov5.Diagnostic, error) {
 	s.serverDiscoveryMutex.RLock()
 	server, ok := s.actions[actionType]
@@ -170,6 +177,41 @@ func (s *muxServer) getEphemeralResourceServer(ctx context.Context, typeName str
 	return server, s.serverDiscoveryDiagnostics, nil
 }
 
+func (s *muxServer) getGenerateResourceConfigServer(ctx context.Context, typeName string) (tfprotov5.ProviderServer, []*tfprotov5.Diagnostic, error) {
+	s.serverDiscoveryMutex.RLock()
+	server, ok := s.resources[typeName]
+	discoveryComplete := s.serverDiscoveryComplete
+	s.serverDiscoveryMutex.RUnlock()
+
+	if discoveryComplete {
+		if ok {
+			return server, s.serverDiscoveryDiagnostics, nil
+		}
+
+		return nil, []*tfprotov5.Diagnostic{
+			generateResourceConfigMissingError(typeName),
+		}, nil
+	}
+
+	err := s.serverDiscovery(ctx)
+
+	if err != nil || diagnosticsHasError(s.serverDiscoveryDiagnostics) {
+		return nil, s.serverDiscoveryDiagnostics, err
+	}
+
+	s.serverDiscoveryMutex.RLock()
+	server, ok = s.resources[typeName]
+	s.serverDiscoveryMutex.RUnlock()
+
+	if !ok {
+		return nil, []*tfprotov5.Diagnostic{
+			generateResourceConfigMissingError(typeName),
+		}, nil
+	}
+
+	return server, s.serverDiscoveryDiagnostics, nil
+}
+
 func (s *muxServer) getListResourceServer(ctx context.Context, typeName string) (tfprotov5.ProviderServer, []*tfprotov5.Diagnostic, error) {
 	s.serverDiscoveryMutex.RLock()
 	server, ok := s.listResources[typeName]

tf5muxserver/mux_server_GenerateResourceConfig.go

@@ -0,0 +1,31 @@
+package tf5muxserver
+
+import (
+	"context"
+	//"fmt"
+
+	//"github.com/hashicorp/terraform-plugin-framework/providerserver"
+	"github.com/hashicorp/terraform-plugin-go/tfprotov5"
+	"github.com/hashicorp/terraform-plugin-mux/internal/logging"
+)
+
+func (s *muxServer) GenerateResourceConfig(ctx context.Context, req *tfprotov5.GenerateResourceConfigRequest) (*tfprotov5.GenerateResourceConfigResponse, error) {
+	rpc := "GenerateResourceConfig"
+	ctx = logging.InitContext(ctx)
+	ctx = logging.RpcContext(ctx, rpc)
+
+	server, diags, err := s.getGenerateResourceConfigServer(ctx, req.TypeName)
+
+	if err != nil {
+		return nil, err
+	}
+
+	// If there is an error diagnostic, return it directly
+	if diagnosticsHasError(diags) {
+		return &tfprotov5.GenerateResourceConfigResponse{
+			Diagnostics: diags,
+		}, nil
+	}
+
+	return server.GenerateResourceConfig(ctx, req)
+}