mcmsops

var AddModulesMCMSOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "package", "add_modules"),
	semver.MustParse("0.1.0"),
	"Add modules to the MCMS registry",
	addModulesHandler,
)

var AllOperationsMCMS = []cld_ops.Operation[any, any, any]{
	*MCMSAcceptOwnershipOp.AsUntyped(),
	*MCMSTransferOwnershipOp.AsUntyped(),
	*MCMSExecuteTransferOwnershipOp.AsUntyped(),
	*SetConfigMCMSOp.AsUntyped(),
	*AddModulesMCMSOp.AsUntyped(),
}

var DeployMCMSOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "package", "deploy"),
	semver.MustParse("0.1.0"),
	"Deploys the MCMS contract",
	handler,
)

var DeployMCMSSequence = cld_ops.NewSequence(
	"sui-deploy-mcms-seq",
	semver.MustParse("0.1.0"),
	"Deploys the MCMS package, sets the initial configuration, init the ownership transfer to self and generates the proposal to accept the ownership",
	func(env cld_ops.Bundle, deps sui_ops.OpTxDeps, input DeployMCMSSeqInput) (DeployMCMSSeqOutput, error) {

		deployReport, err := cld_ops.ExecuteOperation(env, DeployMCMSOp, deps, cld_ops.EmptyInput{})
		if err != nil {
			return DeployMCMSSeqOutput{}, fmt.Errorf("failed to deploy MCMS: %w", err)
		}

		roleConfigs := []struct {
			config *types.Config
			role   suisdk.TimelockRole
			name   string
		}{
			{input.Bypasser, suisdk.TimelockRoleBypasser, "Bypasser"},
			{input.Canceller, suisdk.TimelockRoleCanceller, "Canceller"},
			{input.Proposer, suisdk.TimelockRoleProposer, "Proposer"},
		}

		for _, roleConfig := range roleConfigs {
			if roleConfig.config != nil {
				setConfigInput := MCMSSetConfigInput{
					ChainSelector: input.ChainSelector,
					McmsPackageID: deployReport.Output.PackageId,
					OwnerCap:      deployReport.Output.Objects.McmsAccountOwnerCapObjectId,
					McmsObjectID:  deployReport.Output.Objects.McmsMultisigStateObjectId,
					Role:          roleConfig.role,
					Config:        *roleConfig.config,
				}

				_, err = cld_ops.ExecuteOperation(env, SetConfigMCMSOp, deps, setConfigInput)
				if err != nil {
					return DeployMCMSSeqOutput{}, fmt.Errorf("failed to set config for role %s: %w", roleConfig.name, err)
				}

				env.Logger.Infow("Set MCMS config", "role", roleConfig.name, "chainSelector", input.ChainSelector)
			}
		}

		transferOwnershipInput := MCMSTransferOwnershipInput{
			McmsPackageID:   deployReport.Output.PackageId,
			OwnerCap:        deployReport.Output.Objects.McmsAccountOwnerCapObjectId,
			AccountObjectID: deployReport.Output.Objects.McmsAccountStateObjectId,
		}
		_, err = cld_ops.ExecuteOperation(env, MCMSTransferOwnershipOp, deps, transferOwnershipInput)
		if err != nil {
			return DeployMCMSSeqOutput{}, fmt.Errorf("failed to transfer ownership to MCMS: %w", err)
		}

		proposalInput := ProposalGenerateInput{
			Defs: []cld_ops.Definition{
				MCMSAcceptOwnershipOp.Def(),
			},
			Inputs: []any{
				MCMSAcceptOwnershipInput{
					McmsPackageID:   deployReport.Output.PackageId,
					AccountObjectID: deployReport.Output.Objects.McmsAccountStateObjectId,
				},
			},

			MmcsPackageID:      deployReport.Output.PackageId,
			McmsStateObjID:     deployReport.Output.Objects.McmsMultisigStateObjectId,
			TimelockObjID:      deployReport.Output.Objects.TimelockObjectId,
			AccountObjID:       deployReport.Output.Objects.McmsAccountStateObjectId,
			RegistryObjID:      deployReport.Output.Objects.McmsRegistryObjectId,
			DeployerStateObjID: deployReport.Output.Objects.McmsDeployerStateObjectId,
			ChainSelector:      uint64(input.ChainSelector),

			TimelockConfig: utils.TimelockConfig{
				MCMSAction:   types.TimelockActionSchedule,
				MinDelay:     0,
				OverrideRoot: false,
			},
		}

		acceptOwnershipProposalReport, err := cld_ops.ExecuteSequence(env, MCMSDynamicProposalGenerateSeq, deps, proposalInput)
		if err != nil {
			return DeployMCMSSeqOutput{}, fmt.Errorf("failed to generate accept ownership proposal: %w", err)
		}

		output := DeployMCMSSeqOutput{
			AcceptOwnershipProposal: acceptOwnershipProposalReport.Output,
			PackageId:               deployReport.Output.PackageId,
			Objects:                 deployReport.Output.Objects,
		}

		return output, nil
	},
)

var MCMSAcceptOwnershipOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "mcms_account", "accept_ownership"),
	semver.MustParse("0.1.0"),
	"Accept ownership of the MCMS contract as MCMS",
	acceptOwnershipHandler,
)

var MCMSDynamicProposalGenerateSeq = cld_ops.NewSequence(
	sui_ops.NewSuiOperationName("mcms", "proposal", "generate"),
	semver.MustParse("0.1.0"),
	"Generates an MCMS timelock proposal that batches multiple operations based on the provided definitions and inputs",
	generateProposalHandler,
)

var MCMSExecuteTransferOwnershipOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "mcms_account", "execute_transfer_ownership"),
	semver.MustParse("0.1.0"),
	"Execute transfer ownership of the MCMS contract to itself",
	executeTransferOwnershipHandler,
)

var MCMSTransferOwnershipOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "mcms_account", "transfer_ownership"),
	semver.MustParse("0.1.0"),
	"Init the transfer ownership of the MCMS contract to itself",
	transferOwnershipHandler,
)

var SetConfigMCMSOp = cld_ops.NewOperation(
	sui_ops.NewSuiOperationName("mcms", "mcms", "set_config"),
	semver.MustParse("0.1.0"),
	"Set config in the MCMS contract",
	setConfigMcmsHandler,
)