//nolint:gomnd
package pit

import (
	"fmt"

	"github.com/shiningrush/fastflow/pkg/entity"
	"github.com/shiningrush/fastflow/pkg/entity/run"
	"github.com/tidbcloud/dataflow-service/app/container"
	"github.com/tidbcloud/dataflow-service/app/workflow"
	"github.com/tidbcloud/dataflow-service/app/workflow/action"
	backup_base "github.com/tidbcloud/dataflow-service/app/workflow/action/backup/base"
	pit_common "github.com/tidbcloud/dataflow-service/app/workflow/action/backup/pit/common"
	pit_disable "github.com/tidbcloud/dataflow-service/app/workflow/action/backup/pit/disable"
	crrf_auto "github.com/tidbcloud/dataflow-service/app/workflow/action/crrf/auto"
)

func GetDisableDag() (*entity.Dag, []run.Action) {
	base := backup_base.NewBaseAction(
		container.Global().TransProvider,
		container.Global().BackupRepo,
		container.Global().BackupScheduleRepo,
		container.Global().LogBackupRepo,
		container.Global().BackupRegionalFacade,
		container.Global().ClusterJobModel,
		container.Global().MgmtSvcGrpcClient,
		container.Global().PitBackupSwitchRepo,
		container.Global().FeatureGate,
	)

	actions := []run.Action{
		pit_disable.NewStopLogBackupAction(base),
		pit_disable.NewUpdateSwitchStatusToDisabledAction(base, container.Global().PitBackupSwitchRepo),
		action.MustNewActionWrapperV2(crrf_auto.NewAutoCRRFDisableFromPitAction(
			container.Global().PitBackupSwitchRepo,
			container.Global().CRRFModel,
		), action.SetRunAfterRetryFun(crrf_auto.RetryFunc)),
		pit_common.NewUpdateBackupSettingStatusToNormalAction(base,
			container.Global().BackupSettingRepo,
			container.Global().BackupSettingModel),
	}
	dag := &entity.Dag{
		BaseInfo: entity.BaseInfo{
			ID: workflow.DagDisablePitBackup,
		},
		Name: workflow.DagDisablePitBackup,
		Vars: entity.DagVars{
			workflow.KeyPitBackupSwitchID:  {},
			workflow.KeyDisableLogBackupID: {},
		},
		Status: entity.DagStatusNormal,
		Tasks: []entity.Task{
			{
				// Disable CRRF first while the log backup is still running, so
				// crrf_disable_daemon can locate the log backup CR by name.
				// Stopping the log backup first leaves the fallback path with
				// nothing to find and causes crrf_disable_daemon to fail.
				ID:         workflow.ActionAutoCRRFDisableFromPit,
				ActionName: workflow.ActionAutoCRRFDisableFromPit,
				Params: map[string]interface{}{
					workflow.KeyPitBackupSwitchID: fmt.Sprintf("{{%v}}", workflow.KeyPitBackupSwitchID),
				},
				TimeoutSecs: 40 * 60,
			},
			{
				ID:          workflow.ActionStopLogBackup,
				ActionName:  workflow.ActionStopLogBackup,
				DependOn:    []string{workflow.ActionAutoCRRFDisableFromPit},
				TimeoutSecs: 20 * 60,
			},
			{
				ID:          workflow.ActionUpdatePitBackupSwitchStatusToDisabled,
				ActionName:  workflow.ActionUpdatePitBackupSwitchStatusToDisabled,
				DependOn:    []string{workflow.ActionStopLogBackup},
				TimeoutSecs: 5 * 60,
			},
			{
				ID:          workflow.ActionUpdatePitBackupSettingStatusToNormal,
				ActionName:  workflow.ActionUpdatePitBackupSettingStatusToNormal,
				DependOn:    []string{workflow.ActionUpdatePitBackupSwitchStatusToDisabled},
				TimeoutSecs: 5 * 60,
			},
		},
	}
	return dag, actions
}