package main

import (
	"context"
	"fmt"

	"github.com/google/uuid"
	"github.com/pingcap/test-infra/caselib/pkg/host"
	"github.com/pingcap/test-infra/caselib/pkg/steps"
	"github.com/pingcap/test-infra/caselib/pkg/utils"
	"github.com/pingcap/test-infra/common/model/resource"
	_ "github.com/pingcap/test-infra/sdk/resource/impl/k8s"
	"go.uber.org/zap"
)

func BrIncrementalDDLComp() steps.Step {
	var (
		srcTidb                  = &resource.ResourceHandle{}
		dstTidb                  = &resource.ResourceHandle{}
		lightningRes             = &resource.ResourceHandle{}
		brHost                   = &resource.ResourceHandle{}
		mcHost                   = &resource.ResourceHandle{}
		lightningOpts            = &steps.LightningImportOpts{}
		lightningOptsIncremental = &steps.LightningImportOpts{}
		backupOpts               = &host.BackupFullOpts{}
		mcRmOpts                 = &host.McRmOpts{}
		uniq                     = uuid.New().String()
		uniqFull                 = fmt.Sprintf("%s/Full", uniq)
		uniqIncr                 = fmt.Sprintf("%s/Incremental", uniq)
		dropDatabases            string
		ddlStatsment             string
	)

	return steps.
		NewBuilder().
		LoadEnv().
		LoadResourceFromEnv(utils.ResourcePrefix+"src_tidb", srcTidb).
		LoadResourceFromEnv(utils.ResourcePrefix+"dst_tidb", dstTidb).
		LoadResourceFromEnv(utils.ResourcePrefix+"lightning", lightningRes).
		LoadParamFromEnv(utils.ParamPrefix+"drop_databases", &dropDatabases).
		LoadParamFromEnv(utils.ParamPrefix+"ddl_statements", &ddlStatsment).
		LoadResourceFromEnv(utils.ResourcePrefix+"br", brHost).
		LoadResourceFromEnv(utils.ResourcePrefix+"mc", mcHost).
		LoadResourceFromEnv(utils.ParamPrefix+"br_opts", backupOpts).
		LoadResourceFromEnv(utils.ParamPrefix+"mc_rm_opts", mcRmOpts).
		NewUniqPath(&backupOpts.Storage, uniqFull).
		NewUniqPath(&backupOpts.StorageIncremental, uniqIncr).
		NewUniqPath(&mcRmOpts.Path, uniq).
		LoadResourceFromEnv(utils.ParamPrefix+"import_basic_opts", lightningOpts).
		LoadResourceFromEnv(utils.ParamPrefix+"import_basic_opts_incremental", lightningOptsIncremental).
		DropDatabasesIfExists(srcTidb, &dropDatabases).
		LightningImport(lightningOpts, lightningRes, srcTidb).
		BrBackupFull(brHost, srcTidb, backupOpts).
		ExecuteSqlStatements(srcTidb, &ddlStatsment).
		LightningImport(lightningOptsIncremental, lightningRes, srcTidb).
		BrBackupFullIncremental(brHost, srcTidb, backupOpts).
		DropUserDatabases(dstTidb).
		BrRestoreFull(brHost, dstTidb, backupOpts).
		BrRestoreFullIncremental(brHost, dstTidb, backupOpts).
		CheckDatabasesCrc32(&backupOpts.Crc32Checksum, &dropDatabases, dstTidb).
		McRemove(mcRmOpts, mcHost).
		Build()
}

func main() {
	_, logger := utils.GetCtxAndLogger()
	ctx := steps.New(context.Background())
	if err := BrIncrementalDDLComp().Execute(ctx); err != nil {
		logger.Fatal("BrIncrementalDDLComp failed", zap.Error(err))
	}
	logger.Info("BrIncrementalDDLComp passed")
}