adapters

var ConfigureLaneLegAsDest = cldf_ops.NewSequence(
	"aptos/sequences/ccip/tooling-api/configure-lane-leg-as-dest",
	semver.MustParse("1.6.0"),
	"Configures lane leg as dest on CCIP 1.6.0",
	func(b cldf_ops.Bundle, chains cldf_chain.BlockChains, input lanes.UpdateLanesInput) (sequences.OnChainOutput, error) {
		chainSelector := input.Dest.Selector
		chain := chains.AptosChains()[chainSelector]
		deps := buildAptosDeps(chain, chainSelector, input.Dest.OffRamp)

		var result sequences.OnChainOutput
		isEnabled := !input.IsDisabled

		offRampReport, err := cldf_ops.ExecuteOperation(b, operation.UpdateOffRampSourcesOp, deps, operation.UpdateOffRampSourcesInput{
			Updates: map[uint64]v1_6.OffRampSourceUpdate{
				input.Source.Selector: {
					IsEnabled:                 isEnabled,
					TestRouter:                input.TestRouter,
					IsRMNVerificationDisabled: !input.Source.RMNVerificationEnabled,
				},
			},
		})
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("update offramp sources: %w", err)
		}
		appendBatchOp(&result, chainSelector, offRampReport.Output)

		return result, nil
	},
)

var ConfigureLaneLegAsSource = cldf_ops.NewSequence(
	"aptos/sequences/ccip/tooling-api/configure-lane-leg-as-source",
	semver.MustParse("1.6.0"),
	"Configures lane leg as source on CCIP 1.6.0",
	func(b cldf_ops.Bundle, chains cldf_chain.BlockChains, input lanes.UpdateLanesInput) (sequences.OnChainOutput, error) {
		chainSelector := input.Source.Selector
		chain := chains.AptosChains()[chainSelector]
		deps := buildAptosDeps(chain, chainSelector, input.Source.OnRamp)

		var result sequences.OnChainOutput
		isEnabled := !input.IsDisabled

		fqReport, err := cldf_ops.ExecuteOperation(b, operation.UpdateFeeQuoterDestsOp, deps, operation.UpdateFeeQuoterDestsInput{
			Updates: map[uint64]aptos_fee_quoter.DestChainConfig{
				input.Dest.Selector: v1_6.ToAptosFeeQuoterDestChainConfig(input.Dest.FeeQuoterDestChainConfig),
			},
		})
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("update fee quoter dests: %w", err)
		}
		appendBatchOp(&result, chainSelector, fqReport.Output)

		onRampReport, err := cldf_ops.ExecuteOperation(b, operation.UpdateOnRampDestsOp, deps, operation.UpdateOnRampDestsInput{
			Updates: map[uint64]v1_6.OnRampDestinationUpdate{
				input.Dest.Selector: {
					IsEnabled:        isEnabled,
					TestRouter:       input.TestRouter,
					AllowListEnabled: input.Dest.AllowListEnabled,
				},
			},
		})
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("update onramp dests: %w", err)
		}
		appendBatchOp(&result, chainSelector, onRampReport.Output)

		if err := appendApplyAllowedOfframpUpdates(b, &result, chainSelector, deps); err != nil {
			return sequences.OnChainOutput{}, err
		}

		priceReport, err := cldf_ops.ExecuteOperation(b, operation.UpdateFeeQuoterPricesOp, deps, operation.UpdateFeeQuoterPricesInput{
			TokenPrices: input.Source.TokenPrices,
			GasPrices: map[uint64]*big.Int{
				input.Dest.Selector: input.Dest.GasPrice,
			},
		})
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("update fee quoter prices: %w", err)
		}
		appendBatchOp(&result, chainSelector, priceReport.Output)

		onRampVersion := input.ExtraConfigs.OnRampVersion
		if onRampVersion == nil {
			onRampVersion = []byte{1, 6, 0}
		}
		routerReport, err := cldf_ops.ExecuteOperation(b, operation.UpdateRouterOp, deps, operation.UpdateRouterDestInput{
			Updates: []aptos_router.OnRampSet{{
				DestChainSelector: input.Dest.Selector,
				OnRampVersion:     onRampVersion,
			}},
		})
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("update router: %w", err)
		}
		appendBatchOp(&result, chainSelector, []mcmstypes.Transaction{routerReport.Output})

		return result, nil
	},
)

var SetOCR3Config = cldf_ops.NewSequence(
	"aptos/sequences/ccip/tooling-api/set-ocr3-config",
	semver.MustParse("1.6.0"),
	"Set OCR3 Config on Aptos chains",
	func(b cldf_ops.Bundle, chains cldf_chain.BlockChains, input deployops.SetOCR3ConfigInput) (sequences.OnChainOutput, error) {
		chainSelector := input.ChainSelector
		chain := chains.AptosChains()[chainSelector]

		ccipBytes, err := getCCIPAccountBytes(input.Datastore, chainSelector)
		if err != nil {
			return sequences.OnChainOutput{}, fmt.Errorf("get ccip address: %w", err)
		}
		deps := buildAptosDeps(chain, chainSelector, ccipBytes)

		var result sequences.OnChainOutput
		for pluginType, cfg := range input.Configs {
			report, err := cldf_ops.ExecuteOperation(b, operation.SetOcr3ConfigOp, deps, operation.SetOcr3ConfigInput{
				OcrPluginType: uint8(pluginType),
				OCRConfigArgs: intoOCRArgs(cfg),
			})
			if err != nil {
				return sequences.OnChainOutput{}, fmt.Errorf("set ocr3 config for plugin %d: %w", pluginType, err)
			}
			appendBatchOp(&result, chainSelector, []mcmstypes.Transaction{report.Output})
		}

		return result, nil
	},
)