all: Implement action RPC support in all mux servers (#314)

* update go module + internal 5to6 / 6to5 * tf5muxserver impl actions * fix all list resource references in Go module * tf6muxserver * grammar * 5to6 and 6to5 impl * update to use main

Author: austinvalle

go.mod

@@ -6,7 +6,7 @@ toolchain go1.23.7
 
 require (
 github.com/google/go-cmp v0.7.0
-github.com/hashicorp/terraform-plugin-go v0.28.1-0.20250616135123-a19df43120ea
+github.com/hashicorp/terraform-plugin-go v0.29.0-alpha.1.0.20250709165734-a8477a15f806
 github.com/hashicorp/terraform-plugin-log v0.9.0
 google.golang.org/grpc v1.73.0
 )

go.sum

@@ -21,8 +21,8 @@ github.com/hashicorp/go-plugin v1.6.3 h1:xgHB+ZUSYeuJi96WtxEjzi23uh7YQpznjGh0U0U
 github.com/hashicorp/go-plugin v1.6.3/go.mod h1:MRobyh+Wc/nYy1V4KAXUiYfzxoYhs7V1mlH1Z7iY2h0=
 github.com/hashicorp/go-uuid v1.0.3 h1:2gKiV6YVmrJ1i2CKKa9obLvRieoRGviZFL26PcT/Co8=
 github.com/hashicorp/go-uuid v1.0.3/go.mod h1:6SBZvOh/SIDV7/2o3Jml5SYk/TvGqwFJ/bN7x4byOro=
-github.com/hashicorp/terraform-plugin-go v0.28.1-0.20250616135123-a19df43120ea h1:U9EAAeQtszGlR7mDS7rY77B/a4/XiMDB8HfAtqLAuAQ=
-github.com/hashicorp/terraform-plugin-go v0.28.1-0.20250616135123-a19df43120ea/go.mod h1:hL//wLEfYo0YVt0TC/VLzia/ADQQto3HEm4/jX2gkdY=
+github.com/hashicorp/terraform-plugin-go v0.29.0-alpha.1.0.20250709165734-a8477a15f806 h1:i3kA1sT/Fk8Ex+VVKdjf9sFOPwS7w3Q73pfbnxKwdjg=
+github.com/hashicorp/terraform-plugin-go v0.29.0-alpha.1.0.20250709165734-a8477a15f806/go.mod h1:hL//wLEfYo0YVt0TC/VLzia/ADQQto3HEm4/jX2gkdY=
 github.com/hashicorp/terraform-plugin-log v0.9.0 h1:i7hOA+vdAItN1/7UrfBqBwvYPQ9TFvymaRGZED3FCV0=
 github.com/hashicorp/terraform-plugin-log v0.9.0/go.mod h1:rKL8egZQ/eXSyDqzLUuwUYLVdlYeamldAHSxjUFADow=
 github.com/hashicorp/terraform-registry-address v0.3.0 h1:HMpK3nqaGFPS9VmgRXrJL/dzHNdheGVKk5k7VlFxzCo=

internal/tf5testserver/tf5testserver.go

@@ -68,6 +68,10 @@ type TestServer struct {
 ValidateListResourceConfigCalled map[string]bool
 
 ListResourceCalled map[string]bool
+
+PlanActionCalled map[string]bool
+
+InvokeActionCalled map[string]bool
 }
 
 func (s *TestServer) ProviderServer() tfprotov5.ProviderServer {
@@ -295,3 +299,21 @@ func (s *TestServer) ListResource(_ context.Context, req *tfprotov5.ListResource
 s.ListResourceCalled[req.TypeName] = true
 return nil, nil
 }
+
+func (s *TestServer) PlanAction(ctx context.Context, req *tfprotov5.PlanActionRequest) (*tfprotov5.PlanActionResponse, error) {
+if s.PlanActionCalled == nil {
+s.PlanActionCalled = make(map[string]bool)
+}
+
+s.PlanActionCalled[req.ActionType] = true
+return nil, nil
+}
+
+func (s *TestServer) InvokeAction(ctx context.Context, req *tfprotov5.InvokeActionRequest) (*tfprotov5.InvokeActionServerStream, error) {
+if s.InvokeActionCalled == nil {
+s.InvokeActionCalled = make(map[string]bool)
+}
+
+s.InvokeActionCalled[req.ActionType] = true
+return nil, nil
+}

internal/tf6testserver/tf6testserver.go

@@ -68,6 +68,10 @@ type TestServer struct {
 ValidateListResourceConfigCalled map[string]bool
 
 ListResourceCalled map[string]bool
+
+PlanActionCalled map[string]bool
+
+InvokeActionCalled map[string]bool
 }
 
 func (s *TestServer) ProviderServer() tfprotov6.ProviderServer {
@@ -295,3 +299,21 @@ func (s *TestServer) ListResource(_ context.Context, req *tfprotov6.ListResource
 s.ListResourceCalled[req.TypeName] = true
 return nil, nil
 }
+
+func (s *TestServer) PlanAction(ctx context.Context, req *tfprotov6.PlanActionRequest) (*tfprotov6.PlanActionResponse, error) {
+if s.PlanActionCalled == nil {
+s.PlanActionCalled = make(map[string]bool)
+}
+
+s.PlanActionCalled[req.ActionType] = true
+return nil, nil
+}
+
+func (s *TestServer) InvokeAction(ctx context.Context, req *tfprotov6.InvokeActionRequest) (*tfprotov6.InvokeActionServerStream, error) {
+if s.InvokeActionCalled == nil {
+s.InvokeActionCalled = make(map[string]bool)
+}
+
+s.InvokeActionCalled[req.ActionType] = true
+return nil, nil
+}

internal/tfprotov5tov6/tfprotov5tov6.go

@@ -4,6 +4,8 @@
 package tfprotov5tov6
 
 import (
+"fmt"
+
 "github.com/hashicorp/terraform-plugin-go/tfprotov5"
 "github.com/hashicorp/terraform-plugin-go/tfprotov6"
 )
@@ -301,6 +303,7 @@ func GetMetadataResponse(in *tfprotov5.GetMetadataResponse) *tfprotov6.GetMetada
 }
 
 resp := &tfprotov6.GetMetadataResponse{
+Actions: make([]tfprotov6.ActionMetadata, 0, len(in.Actions)),
 DataSources: make([]tfprotov6.DataSourceMetadata, 0, len(in.DataSources)),
 Diagnostics: Diagnostics(in.Diagnostics),
 EphemeralResources: make([]tfprotov6.EphemeralResourceMetadata, 0, len(in.Resources)),
@@ -330,6 +333,10 @@ func GetMetadataResponse(in *tfprotov5.GetMetadataResponse) *tfprotov6.GetMetada
 resp.Resources = append(resp.Resources, ResourceMetadata(resource))
 }
 
+for _, action := range in.Actions {
+resp.Actions = append(resp.Actions, ActionMetadata(action))
+}
+
 return resp
 }
 
@@ -376,7 +383,14 @@ func GetProviderSchemaResponse(in *tfprotov5.GetProviderSchemaResponse) *tfproto
 resourceSchemas[k] = Schema(v)
 }
 
+actionSchemas := make(map[string]*tfprotov6.ActionSchema, len(in.ActionSchemas))
+
+for k, v := range in.ActionSchemas {
+actionSchemas[k] = ActionSchema(v)
+}
+
 return &tfprotov6.GetProviderSchemaResponse{
+ActionSchemas: actionSchemas,
 DataSourceSchemas: dataSourceSchemas,
 Diagnostics: Diagnostics(in.Diagnostics),
 EphemeralResourceSchemas: ephemeralResourceSchemas,
@@ -1052,3 +1066,244 @@ func ListResourceResult(in tfprotov5.ListResourceResult) tfprotov6.ListResourceR
 Diagnostics: Diagnostics(in.Diagnostics),
 }
 }
+
+func ActionMetadata(in tfprotov5.ActionMetadata) tfprotov6.ActionMetadata {
+return tfprotov6.ActionMetadata{
+TypeName: in.TypeName,
+}
+}
+
+func ActionSchema(in *tfprotov5.ActionSchema) *tfprotov6.ActionSchema {
+if in == nil {
+return nil
+}
+
+actionSchema := &tfprotov6.ActionSchema{
+Schema: Schema(in.Schema),
+}
+
+switch actionSchemaType := in.Type.(type) {
+case tfprotov5.UnlinkedActionSchemaType:
+actionSchema.Type = tfprotov6.UnlinkedActionSchemaType{}
+case tfprotov5.LifecycleActionSchemaType:
+actionSchema.Type = tfprotov6.LifecycleActionSchemaType{
+Executes: tfprotov6.LifecycleExecutionOrder(actionSchemaType.Executes),
+LinkedResource: LinkedResourceSchema(actionSchemaType.LinkedResource),
+}
+case tfprotov5.LinkedActionSchemaType:
+actionSchema.Type = tfprotov6.LinkedActionSchemaType{
+LinkedResources: LinkedResourceSchemas(actionSchemaType.LinkedResources),
+}
+default:
+// It is not currently possible to create tfprotov5.ActionSchemaType
+// implementations outside the terraform-plugin-go module. If this panic was reached,
+// it implies that a new event type was introduced and needs to be implemented
+// as a new case above.
+panic(fmt.Sprintf("unimplemented tfprotov5.ActionSchemaType type: %T", in.Type))
+}
+
+return actionSchema
+}
+
+func LinkedResourceSchemas(in []*tfprotov5.LinkedResourceSchema) []*tfprotov6.LinkedResourceSchema {
+schemas := make([]*tfprotov6.LinkedResourceSchema, 0, len(in))
+
+for _, schema := range in {
+schemas = append(schemas, LinkedResourceSchema(schema))
+}
+
+return schemas
+}
+
+func LinkedResourceSchema(in *tfprotov5.LinkedResourceSchema) *tfprotov6.LinkedResourceSchema {
+if in == nil {
+return nil
+}
+
+return &tfprotov6.LinkedResourceSchema{
+TypeName: in.TypeName,
+Description: in.Description,
+}
+}
+
+func PlanActionRequest(in *tfprotov5.PlanActionRequest) *tfprotov6.PlanActionRequest {
+if in == nil {
+return nil
+}
+
+return &tfprotov6.PlanActionRequest{
+ActionType: in.ActionType,
+LinkedResources: ProposedLinkedResources(in.LinkedResources),
+Config: DynamicValue(in.Config),
+ClientCapabilities: PlanActionClientCapabilities(in.ClientCapabilities),
+}
+}
+
+func ProposedLinkedResources(in []*tfprotov5.ProposedLinkedResource) []*tfprotov6.ProposedLinkedResource {
+if in == nil {
+return nil
+}
+
+linkedResources := make([]*tfprotov6.ProposedLinkedResource, 0, len(in))
+
+for _, inLinkedResource := range in {
+if inLinkedResource == nil {
+linkedResources = append(linkedResources, nil)
+continue
+}
+
+linkedResources = append(linkedResources, &tfprotov6.ProposedLinkedResource{
+PriorState: DynamicValue(inLinkedResource.PriorState),
+PlannedState: DynamicValue(inLinkedResource.PlannedState),
+Config: DynamicValue(inLinkedResource.Config),
+PriorIdentity: ResourceIdentityData(inLinkedResource.PriorIdentity),
+})
+}
+
+return linkedResources
+}
+
+func PlanActionClientCapabilities(in *tfprotov5.PlanActionClientCapabilities) *tfprotov6.PlanActionClientCapabilities {
+if in == nil {
+return nil
+}
+
+resp := &tfprotov6.PlanActionClientCapabilities{
+DeferralAllowed: in.DeferralAllowed,
+}
+
+return resp
+}
+
+func PlanActionResponse(in *tfprotov5.PlanActionResponse) *tfprotov6.PlanActionResponse {
+if in == nil {
+return nil
+}
+
+return &tfprotov6.PlanActionResponse{
+LinkedResources: PlannedLinkedResources(in.LinkedResources),
+Diagnostics: Diagnostics(in.Diagnostics),
+Deferred: Deferred(in.Deferred),
+}
+}
+
+func PlannedLinkedResources(in []*tfprotov5.PlannedLinkedResource) []*tfprotov6.PlannedLinkedResource {
+if in == nil {
+return nil
+}
+
+linkedResources := make([]*tfprotov6.PlannedLinkedResource, 0, len(in))
+
+for _, inLinkedResource := range in {
+if inLinkedResource == nil {
+linkedResources = append(linkedResources, nil)
+continue
+}
+
+linkedResources = append(linkedResources, &tfprotov6.PlannedLinkedResource{
+PlannedState: DynamicValue(inLinkedResource.PlannedState),
+PlannedIdentity: ResourceIdentityData(inLinkedResource.PlannedIdentity),
+})
+}
+
+return linkedResources
+}
+
+func InvokeActionRequest(in *tfprotov5.InvokeActionRequest) *tfprotov6.InvokeActionRequest {
+if in == nil {
+return nil
+}
+
+return &tfprotov6.InvokeActionRequest{
+ActionType: in.ActionType,
+LinkedResources: InvokeLinkedResources(in.LinkedResources),
+Config: DynamicValue(in.Config),
+}
+}
+
+func InvokeLinkedResources(in []*tfprotov5.InvokeLinkedResource) []*tfprotov6.InvokeLinkedResource {
+if in == nil {
+return nil
+}
+
+linkedResources := make([]*tfprotov6.InvokeLinkedResource, 0, len(in))
+
+for _, inLinkedResource := range in {
+if inLinkedResource == nil {
+linkedResources = append(linkedResources, nil)
+continue
+}
+
+linkedResources = append(linkedResources, &tfprotov6.InvokeLinkedResource{
+PriorState: DynamicValue(inLinkedResource.PriorState),
+PlannedState: DynamicValue(inLinkedResource.PlannedState),
+Config: DynamicValue(inLinkedResource.Config),
+PlannedIdentity: ResourceIdentityData(inLinkedResource.PlannedIdentity),
+})
+}
+
+return linkedResources
+}
+
+func InvokeActionServerStream(in *tfprotov5.InvokeActionServerStream) *tfprotov6.InvokeActionServerStream {
+if in == nil {
+return nil
+}
+
+return &tfprotov6.InvokeActionServerStream{
+Events: func(yield func(tfprotov6.InvokeActionEvent) bool) {
+for res := range in.Events {
+if !yield(InvokeActionEvent(res)) {
+break
+}
+}
+},
+}
+}
+
+func InvokeActionEvent(in tfprotov5.InvokeActionEvent) tfprotov6.InvokeActionEvent {
+switch event := (in.Type).(type) {
+case tfprotov5.ProgressInvokeActionEventType:
+return tfprotov6.InvokeActionEvent{
+Type: tfprotov6.ProgressInvokeActionEventType{
+Message: event.Message,
+},
+}
+case tfprotov5.CompletedInvokeActionEventType:
+return tfprotov6.InvokeActionEvent{
+Type: tfprotov6.CompletedInvokeActionEventType{
+LinkedResources: NewLinkedResources(event.LinkedResources),
+Diagnostics: Diagnostics(event.Diagnostics),
+},
+}
+}
+
+// It is not currently possible to create tfprotov5.InvokeActionEventType
+// implementations outside the terraform-plugin-go module. If this panic was reached,
+// it implies that a new event type was introduced and needs to be implemented
+// as a new case above.
+panic(fmt.Sprintf("unimplemented tfprotov5.InvokeActionEventType type: %T", in.Type))
+}
+
+func NewLinkedResources(in []*tfprotov5.NewLinkedResource) []*tfprotov6.NewLinkedResource {
+if in == nil {
+return nil
+}
+
+linkedResources := make([]*tfprotov6.NewLinkedResource, 0, len(in))
+
+for _, inLinkedResource := range in {
+if inLinkedResource == nil {
+linkedResources = append(linkedResources, nil)
+continue
+}
+
+linkedResources = append(linkedResources, &tfprotov6.NewLinkedResource{
+NewState: DynamicValue(inLinkedResource.NewState),
+NewIdentity: ResourceIdentityData(inLinkedResource.NewIdentity),
+RequiresReplace: inLinkedResource.RequiresReplace,
+})
+}
+
+return linkedResources
+}