testutils

type Any2SuiRampMessage struct {
	Header        RampMessageHeader      `json:"header"`
	Sender        []byte                 `json:"sender"`
	Data          []byte                 `json:"data"`
	Receiver      []byte                 `json:"receiver"`  // address in Move becomes []byte for 32-byte Sui address
	GasLimit      *big.Int               `json:"gas_limit"` // u256 in Move becomes *big.Int in Go
	TokenReceiver []byte                 `json:"token_receiver"`
	TokenAmounts  []Any2SuiTokenTransfer `json:"token_amounts"`
}

type Any2SuiTokenTransfer struct {
	SourcePoolAddress []byte   `json:"source_pool_address"`
	DestTokenAddress  []byte   `json:"dest_token_address"` // address in Move becomes []byte for 32-byte Sui address
	DestGasAmount     uint32   `json:"dest_gas_amount"`
	ExtraData         []byte   `json:"extra_data"`
	Amount            *big.Int `json:"amount"` // u256 in Move becomes *big.Int in Go
}

type CommitReport struct {
	PriceUpdates         PriceUpdates
	BlessedMerkleRoots   []MerkleRoot
	UnblessedMerkleRoots []MerkleRoot
	RMNSignatures        [][]byte
}

type ContractObject struct {
	ObjectID    string
	PackageName string
	StructName  string
}

type Contracts struct {
	Path     string
	Name     string
	ModuleID string
	Objects  []ContractObject
}

type ECDSAKeyPair struct {
	PrivateKey ecdsa.PrivateKey
	PublicKey  ecdsa.PublicKey
	Address    []byte
}

type ExecutionReport struct {
	SourceChainSelector uint64             `json:"source_chain_selector"`
	Message             Any2SuiRampMessage `json:"message"`
	OffchainTokenData   [][]byte           `json:"offchain_token_data"`
	Proofs              [][]byte           `json:"proofs"`
}

type FakeSuiPTBClient struct {
	// Status controls the simulated response for GetTransactionStatus
	Status client.TransactionResult
	// CoinsData controls the simulated response for GetCoinsByAddress
	CoinsData  []models.CoinData
	MockClient sui.ISuiAPI
}

type GasPriceUpdate struct {
	DestChainSelector uint64   // u64 in Move
	UsdPerUnitGas     *big.Int // u256 in Move becomes *big.Int in Go
}

type MerkleRoot struct {
	SourceChainSelector uint64 // u64 in Move
	OnRampAddress       []byte // vector<u8> in Move
	MinSeqNr            uint64 // u64 in Move
	MaxSeqNr            uint64 // u64 in Move
	MerkleRoot          []byte // vector<u8> in Move (32 bytes)
}

type MerkleTree [][32]byte

type NodeEnvType int

type ObjectChange struct {
	Type            string   `json:"type"` // "published", "created", etc.
	Sender          string   `json:"sender,omitempty"`
	Owner           Owner    `json:"owner,omitempty"`
	ObjectType      string   `json:"objectType,omitempty"`
	ObjectID        string   `json:"objectId,omitempty"`
	Version         string   `json:"version,omitempty"`
	PreviousVersion string   `json:"previousVersion,omitempty"`
	Digest          string   `json:"digest,omitempty"`
	PackageID       string   `json:"packageId,omitempty"` // Only in type == "published"
	Modules         []string `json:"modules,omitempty"`   // Only in type == "published"
}

type Op struct {
	Role         uint8
	ChainID      *big.Int
	MultiSig     []byte
	Nonce        uint64
	To           []byte
	ModuleName   string
	FunctionName string
	Data         []byte
}

type Owner struct {
	AddressOwner *string      `json:"AddressOwner,omitempty"`
	Shared       *SharedOwner `json:"Shared,omitempty"`
	Immutable    *string      `json:"Immutable,omitempty"`
}

type PriceUpdates struct {
	TokenPriceUpdates []TokenPriceUpdate
	GasPriceUpdates   []GasPriceUpdate
}

type RampMessageHeader struct {
	MessageId           []byte `json:"message_id"`
	SourceChainSelector uint64 `json:"source_chain_selector"`
	DestChainSelector   uint64 `json:"dest_chain_selector"`
	SequenceNumber      uint64 `json:"sequence_number"`
	Nonce               uint64 `json:"nonce"`
}

type RootMetadata struct {
	Role                 uint8
	ChainID              *big.Int
	MultiSig             []byte
	PreOpCount           uint64
	PostOpCount          uint64
	OverridePreviousRoot bool
}

type SharedOwner struct {
	InitialSharedVersion int `json:"initial_shared_version"`
}

type StatefulFakeSuiPTBClient struct {
	CoinsData                    []models.CoinData
	GasBudgetThreshold           uint64 // Minimum gas budget required for success
	CallCount                    int    // Track number of calls to GetTransactionStatus
	ForcedTransactionStatusError string // Force the transaction status to be this value
	CurrentGasBudget             uint64 // Track the current gas budget being tested
	// MockClient allows tests to inject custom ISuiAPI behavior when GetClient() is called.
	// If nil, GetClient() will return nil, which is fine for most tests that don't use it.
	// Tests that need specific ISuiAPI behavior should set this field explicitly.
	MockClient sui.ISuiAPI
}

type TestKeystore struct {
	Keys map[string]ed25519.PrivateKey
	// contains filtered or unexported fields
}

type TestState struct {
	AccountAddress  string
	PublicKeyBytes  []byte
	SuiGateway      *client.PTBClient
	KeystoreGateway loop.Keystore
	TxManager       *txm.SuiTxm
	TxStore         *txm.InMemoryStore
	Contracts       []Contracts
	Cmd             exec.Cmd
}

type TimelockOperation struct {
	Target       []byte
	ModuleName   string
	FunctionName string
	Data         []byte
}

type TokenPoolType string

type TokenPriceUpdate struct {
	SourceToken []byte   // address in Move becomes []byte for 32-byte Sui address
	UsdPerToken *big.Int // u256 in Move becomes *big.Int in Go
}

type TokenToolDetails struct {
	TokenPoolPackageId string
	TokenPoolType      TokenPoolType
}

type TxnMetaWithObjectChanges struct {
	ObjectChanges []ObjectChange `json:"objectChanges"`
}