package main

import (
	"context"

	"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 BrTableCache() steps.Step {
	var (
		srcTidb            = &resource.ResourceHandle{}
		dstTidb            = &resource.ResourceHandle{}
		brHost             = &resource.ResourceHandle{}
		mcHost             = &resource.ResourceHandle{}
		backupOpts         = &host.BackupDbOpts{}
		mcRmOpts           = &host.McRmOpts{}
		uniq               = uuid.New().String()
		globalVariableOpts = &steps.TiDBVariableList
		adminCheck         string
		isCached           = false // table cache should not be restored
	)

	return steps.
		NewBuilder().
		LoadEnv().
		LoadResourceFromEnv(utils.ResourcePrefix+"src_tidb", srcTidb).
		LoadResourceFromEnv(utils.ResourcePrefix+"dst_tidb", dstTidb).
		LoadResourceFromEnv(utils.ResourcePrefix+"br", brHost).
		LoadResourceFromEnv(utils.ResourcePrefix+"mc", mcHost).
		LoadResourceFromEnv(utils.ParamPrefix+"br_opts", backupOpts).
		LoadResourceFromEnv(utils.ParamPrefix+"mc_rm_opts", mcRmOpts).
		LoadResourceFromEnv(utils.ParamPrefix+"global_variable", &globalVariableOpts).
		LoadParamFromEnv(utils.ParamPrefix+"admin_check", &adminCheck).
		NewUniqPath(&backupOpts.Storage, uniq).
		NewUniqPath(&mcRmOpts.Path, uniq).
		// this br case should be after lighting case
		// alter table to enable CACHE as lightning imported table is not cached
		ExecuteSQL(srcTidb, "", "ALTER TABLE sbtest.sbtest1 CACHE;").
		BrBackupDb(brHost, srcTidb, backupOpts).
		ConfigGlobalVariable(globalVariableOpts, srcTidb, dstTidb).
		DropDatabasesIfExists(dstTidb, &backupOpts.Database).
		BrRestoreDb(brHost, dstTidb, backupOpts).
		CheckDatabasesCrc32(&backupOpts.Crc32Checksum, &backupOpts.Database, dstTidb).
		CheckTableCache(&isCached, "sbtest", "sbtest1", dstTidb).
		AdminCheck(dstTidb, &adminCheck).
		McRemove(mcRmOpts, mcHost).
		Build()
}

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