Allow requesting action schema from plan producer

Author: mutahhir

internal/stacks/stackplan/from_plan.go

@@ -37,6 +37,9 @@ type PlanProducer interface {
 
 	// ResourceSchema returns the schema for a resource type from a provider.
 	ResourceSchema(ctx context.Context, providerTypeAddr addrs.Provider, mode addrs.ResourceMode, resourceType string) (providers.Schema, error)
+
+	// ActionSchema returns the schema for an action type from a provider.
+	ActionSchema(ctx context.Context, providerTypeAddr addrs.Provider, actionType string) (providers.ActionSchema, error)
 }
 
 func FromPlan(ctx context.Context, config *configs.Config, plan *plans.Plan, refreshPlan *plans.Plan, action plans.Action, producer PlanProducer) ([]PlannedChange, tfdiags.Diagnostics) {

internal/stacks/stackruntime/internal/stackeval/component_instance.go

@@ -43,10 +43,12 @@ type ComponentInstance struct {
 	inputVariableValues perEvalPhase[promising.Once[withDiagnostics[cty.Value]]]
 }
 
-var _ Applyable = (*ComponentInstance)(nil)
-var _ Plannable = (*ComponentInstance)(nil)
-var _ ExpressionScope = (*ComponentInstance)(nil)
-var _ ConfigComponentExpressionScope[stackaddrs.AbsComponentInstance] = (*ComponentInstance)(nil)
+var (
+	_ Applyable                                                       = (*ComponentInstance)(nil)
+	_ Plannable                                                       = (*ComponentInstance)(nil)
+	_ ExpressionScope                                                 = (*ComponentInstance)(nil)
+	_ ConfigComponentExpressionScope[stackaddrs.AbsComponentInstance] = (*ComponentInstance)(nil)
+)
 
 func newComponentInstance(call *Component, addr stackaddrs.AbsComponentInstance, repetition instances.RepetitionData, mode plans.Mode, deferred bool) *ComponentInstance {
 	component := &ComponentInstance{
@@ -137,7 +139,6 @@ func (c *ComponentInstance) inputValuesForModulesRuntime(ctx context.Context, ph
 		}
 	}
 	return ret
-
 }
 
 func (c *ComponentInstance) PlanOpts(ctx context.Context, mode plans.Mode, skipRefresh bool) (*terraform.PlanOpts, tfdiags.Diagnostics) {
@@ -795,6 +796,25 @@ func (c *ComponentInstance) ResourceSchema(ctx context.Context, providerTypeAddr
 	return ret, nil
 }
 
+// ActionSchema implements stackplan.PlanProducer.
+func (c *ComponentInstance) ActionSchema(ctx context.Context, providerTypeAddr addrs.Provider, typ string) (providers.ActionSchema, error) {
+	// This should not be able to fail with an error because we should
+	// be retrieving the same schema that was already used to encode
+	// the object we're working with. The error handling here is for
+	// robustness but any error here suggests a bug in Terraform.
+
+	providerType := c.main.ProviderType(providerTypeAddr)
+	providerSchema, err := providerType.Schema(ctx)
+	if err != nil {
+		return providers.ActionSchema{}, err
+	}
+	ret := providerSchema.SchemaForActionType(typ)
+	if ret.ConfigSchema == nil {
+		return providers.ActionSchema{}, fmt.Errorf("schema does not include %q", typ)
+	}
+	return ret, nil
+}
+
 // RequiredComponents implements stackplan.PlanProducer.
 func (c *ComponentInstance) RequiredComponents(ctx context.Context) collections.Set[stackaddrs.AbsComponent] {
 	return c.call.RequiredComponents(ctx)

internal/stacks/stackruntime/internal/stackeval/removed_component_instance.go

@@ -366,3 +366,16 @@ func (r *RemovedComponentInstance) ResourceSchema(ctx context.Context, providerT
 func (r *RemovedComponentInstance) tracingName() string {
 	return r.Addr().String() + " (removed)"
 }
+
+func (r *RemovedComponentInstance) ActionSchema(ctx context.Context, providerTypeAddr addrs.Provider, typ string) (providers.ActionSchema, error) {
+	providerType := r.main.ProviderType(providerTypeAddr)
+	providerSchema, err := providerType.Schema(ctx)
+	if err != nil {
+		return providers.ActionSchema{}, err
+	}
+	ret := providerSchema.SchemaForActionType(typ)
+	if ret.ConfigSchema == nil {
+		return providers.ActionSchema{}, fmt.Errorf("schema does not include %q", typ)
+	}
+	return ret, nil
+}