core

type AccountStorage interface {
	// SaveAccount saves the given account
	SaveAccount(account ValidatorAccount) error

	// DeleteAccount deletes account by uuid
	DeleteAccount(accountID uuid.UUID) error

	// OpenAccount returns nil,nil if no account was found
	OpenAccount(accountID uuid.UUID) (ValidatorAccount, error)

	// SetEncryptor sets the given encryptor to the wallet.
	SetEncryptor(encryptor encryptor.Encryptor, password []byte)
}

type AttestationSlashStatus struct {
	Attestation *phase0.AttestationData
	Status      VoteDetectionType
}

type HDKey struct {
	// contains filtered or unexported fields
}

type MasterDerivableKey struct {
	// contains filtered or unexported fields
}

type Network string

type ProposalDetectionType string

type ProposalSlashStatus struct {
	Slot   phase0.Slot
	Status ProposalDetectionType
	Error  error
}

type SlashingProtector interface {
	IsSlashableAttestation(pubKey []byte, attestation *phase0.AttestationData) (*AttestationSlashStatus, error)
	IsSlashableProposal(pubKey []byte, slot phase0.Slot) (*ProposalSlashStatus, error)
	UpdateHighestAttestation(pubKey []byte, attestation *phase0.AttestationData) error
	UpdateHighestProposal(pubKey []byte, slot phase0.Slot) error
	FetchHighestAttestation(pubKey []byte) (*phase0.AttestationData, bool, error)
	FetchHighestProposal(pubKey []byte) (phase0.Slot, bool, error)
}

type SlashingStore interface {
	SaveHighestAttestation(pubKey []byte, attestation *phase0.AttestationData) error
	RetrieveHighestAttestation(pubKey []byte) (*phase0.AttestationData, bool, error)
	SaveHighestProposal(pubKey []byte, slot phase0.Slot) error
	RetrieveHighestProposal(pubKey []byte) (phase0.Slot, bool, error)
}

type Storage interface {
	WalletStorage
	AccountStorage

	// Name returns storage name.
	Name() string

	// Network returns the network storage is related to.
	Network() Network
}

type ValidatorAccount interface {
	// ID provides the ID for the account.
	ID() uuid.UUID

	// Name provides the name for the account.
	Name() string

	// BasePath provides the basePath of the account.
	BasePath() string

	// ValidatorPublicKey provides the public key for the validation key.
	ValidatorPublicKey() []byte

	// WithdrawalPublicKey provides the public key for the withdrawal key.
	WithdrawalPublicKey() []byte

	// ValidationKeySign signs data with the validation key.
	ValidationKeySign(data []byte) ([]byte, error)

	// GetDepositData returns deposit data
	GetDepositData() (map[string]interface{}, error)

	// SetContext sets the given context
	SetContext(ctx *WalletContext)
}

type ValidatorInfo struct {
	Index                 phase0.ValidatorIndex
	Pubkey                phase0.BLSPubKey
	WithdrawalCredentials []byte
	ToExecutionAddress    bellatrix.ExecutionAddress
}

type VoteDetectionType string

type Wallet interface {
	// ID provides the ID for the wallet.
	ID() uuid.UUID

	// Type provides the type of the wallet.
	Type() WalletType

	// CreateValidatorAccount creates a new validation (validator) key pair in the wallet.
	// Keep in mind ND wallets will probably not allow this function, use AddValidatorAccount.
	CreateValidatorAccount(seed []byte, indexPointer *int) (ValidatorAccount, error)

	// CreateValidatorAccountFromPrivateKey creates validator account from Private Key
	CreateValidatorAccountFromPrivateKey(privateKey []byte, indexPointer *int) (ValidatorAccount, error)

	// AddValidatorAccount used to specifically add an account.
	// Keep in mind HD wallets will probably not allow this function, use CreateValidatorAccount.
	AddValidatorAccount(account ValidatorAccount) error

	// Accounts provides all accounts in the wallet.
	Accounts() []ValidatorAccount

	// AccountByID provides a single account from the wallet given its ID.
	// This will error if the account is not found.
	// should return account = nil if not found (not an error!)
	AccountByID(id uuid.UUID) (ValidatorAccount, error)

	// AccountByPublicKey provides a single account from the wallet given its public key.
	// This will error if the account is not found.
	// should return account = nil if not found (not an error!)
	AccountByPublicKey(pubKey string) (ValidatorAccount, error)

	// DeleteAccountByPublicKey delete an account from the wallet given its public key.
	// This will error if the account is not found.
	// should return nil if not error otherwise the error
	DeleteAccountByPublicKey(pubKey string) error

	// SetContext sets the given context
	SetContext(ctx *WalletContext)
}

type WalletContext struct {
	Storage        Storage
	WithdrawalMode bool
}

type WalletStorage interface {
	// SaveWallet stores the given wallet.
	SaveWallet(wallet Wallet) error

	// OpenWallet returns nil,err if no wallet was found
	OpenWallet() (Wallet, error)

	// ListAccounts returns an empty array for no accounts
	ListAccounts() ([]ValidatorAccount, error)
}

type WalletType = string