package scale

import (
	"context"

	appworkflow "github.com/tidbcloud/serverless-service/internal/workflow"
	"github.com/tidbcloud/serverless-service/internal/workflow/action"
	workflowchangefeed "github.com/tidbcloud/serverless-service/internal/workflow/action/changefeed"
	"go.uber.org/zap"
)

type ActionFinalStep struct {
	hook appworkflow.CDCNewArchHook
}

func NewFinalAction(hook appworkflow.CDCNewArchHook) *action.Wrapper {
	return action.MustNewActionWrapper(appworkflow.ActionScaleNewArchFinal, &ActionFinalStep{hook: hook})
}

func (a *ActionFinalStep) Params() any {
	return &workflowchangefeed.CommonParams{}
}

func (a *ActionFinalStep) Run(ctx context.Context, logger *zap.Logger, params any) error {
	input := params.(*workflowchangefeed.CommonParams)
	return a.hook.FinalizeScaleChangefeed(ctx, input.ClusterID, input.ChangefeedID)
}