-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add sqllite db connection and userop data type
- Loading branch information
1 parent
caa1e86
commit 6a77b0e
Showing
6 changed files
with
343 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,25 @@ | ||
| package data | ||
|
|
||
| import ( | ||
| "gorm.io/driver/sqlite" | ||
| "gorm.io/gorm" | ||
| ) | ||
|
|
||
| func OpenConnection(isInMemory bool) (*gorm.DB, error) { | ||
| var db *gorm.DB | ||
| var err error | ||
|
|
||
| if isInMemory { | ||
| db, err = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) | ||
| } else { | ||
| db, err = gorm.Open(sqlite.Open("betsy.db"), &gorm.Config{}) | ||
| } | ||
|
|
||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| db.AutoMigrate(&UserOpV7Hexify{}) | ||
|
|
||
| return db, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,279 @@ | ||
| package data | ||
|
|
||
| import ( | ||
| "errors" | ||
|
|
||
| "github.com/ethereum/go-ethereum/common" | ||
| "github.com/ethereum/go-ethereum/common/hexutil" | ||
| "github.com/transeptorlabs/betsy/contracts/entrypoint" | ||
| "github.com/transeptorlabs/betsy/internal/utils" | ||
| "gorm.io/gorm" | ||
| ) | ||
|
|
||
| // UserOpV7Hexify is a struct used to store user operations in a database. Contains all EIP-4337 with all hex fields. | ||
| type UserOpV7Hexify struct { | ||
| gorm.Model // Embedding gorm.Model adds fields ID(primaryKey), CreatedAt, UpdatedAt, DeletedAt | ||
| UserOpHash string `gorm:"index;unique;not null"` | ||
| Sender string `json:"sender" mapstructure:"sender" validate:"required" gorm:"not null"` | ||
| Nonce string `json:"nonce" mapstructure:"nonce" validate:"required" gorm:"not null"` | ||
|
|
||
| // (optional) | ||
| Factory string `json:"factory" mapstructure:"factory" validate:"required"` | ||
| FactoryData string `json:"factoryData" mapstructure:"factoryData"` | ||
|
|
||
| CallData string `json:"callData" mapstructure:"callData" validate:"required" gorm:"not null"` | ||
| CallGasLimit string `json:"callGasLimit" mapstructure:"callGasLimit" validate:"required" gorm:"not null"` | ||
| VerificationGasLimit string `json:"verificationGasLimit" mapstructure:"verificationGasLimit" validate:"required" gorm:"not null"` | ||
| PreVerificationGas string `json:"preVerificationGas" mapstructure:"preVerificationGas" validate:"required" gorm:"not null" ` | ||
| MaxFeePerGas string `json:"maxFeePerGas" mapstructure:"maxFeePerGas" validate:"required" gorm:"not null"` | ||
| MaxPriorityFeePerGas string `json:"maxPriorityFeePerGas" mapstructure:"maxPriorityFeePerGas" validate:"required" gorm:"not null"` | ||
|
|
||
| // (optional) | ||
| Paymaster string `json:"paymaster" mapstructure:"paymaster" validate:"required"` | ||
| PaymasterVerificationGasLimit string `json:"paymasterVerificationGasLimit" mapstructure:"paymasterVerificationGasLimit"` | ||
| PaymasterPostOpGasLimit string `json:"paymasterPostOpGasLimit" mapstructure:"paymasterPostOpGasLimit"` | ||
| PaymasterData string `json:"paymasterAndData" mapstructure:"paymasterAndData"` | ||
|
|
||
| Signature string `gorm:"not null" json:"signature" mapstructure:"signature"` | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) BeforeCreate(tx *gorm.DB) (err error) { | ||
| ok, invalidAddress := op.IsValidAddresses() | ||
| if !ok { | ||
| return errors.New("can't save invalid eth address:" + invalidAddress) | ||
| } | ||
|
|
||
| err = op.SetUserOpHash() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) IsValidAddresses() (bool, string) { | ||
| isSenderAddress := common.IsHexAddress(op.Sender) | ||
| if !isSenderAddress { | ||
| return isSenderAddress, op.Sender | ||
| } | ||
|
|
||
| return true, "" | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) SetUserOpHash() error { | ||
| op.UserOpHash = "" | ||
| return nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) GetInitCode() ([]byte, error) { | ||
| if op.Factory == "0x" { | ||
| return []byte{}, nil | ||
| } | ||
|
|
||
| if op.FactoryData == "0x" { | ||
| return nil, errors.New("Got Factory but missing FactoryData") | ||
| } | ||
|
|
||
| factoryDecoded, err := hexutil.Decode(op.Factory) | ||
| if err != nil { | ||
| return nil, errors.New("factory (bytes) conversion failed") | ||
| } | ||
|
|
||
| factoryDataDecoded, err := hexutil.Decode(op.FactoryData) | ||
| if err != nil { | ||
| return nil, errors.New("factoryData (bytes) conversion failed") | ||
| } | ||
|
|
||
| concatenatedBytes := append(factoryDecoded, factoryDataDecoded...) | ||
|
|
||
| return concatenatedBytes, nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) GetAccountGasLimits() ([32]byte, error) { | ||
| verificationGasLimitDecoded, err := hexutil.Decode(op.VerificationGasLimit) | ||
| if err != nil { | ||
| return [32]byte{}, errors.New("verificationGasLimit (bytes) conversion failed") | ||
| } | ||
|
|
||
| callGasLimitDecoded, err := hexutil.Decode(op.CallGasLimit) | ||
| if err != nil { | ||
| return [32]byte{}, errors.New("callGasLimit (bytes) conversion failed") | ||
| } | ||
|
|
||
| // Truncate if longer than 16 bytes or Pad with leading zeros if shorter than 16 bytes | ||
| verificationGasLimitDecodedPadded, err := utils.PadToBytes16(verificationGasLimitDecoded) | ||
| if err != nil { | ||
| return [32]byte{}, err | ||
| } | ||
|
|
||
| callGasLimitDecodedPadded, err := utils.PadToBytes16(callGasLimitDecoded) | ||
| if err != nil { | ||
| return [32]byte{}, err | ||
| } | ||
|
|
||
| // Concatenate the byte slices | ||
| concatenatedBytes := append(verificationGasLimitDecodedPadded, callGasLimitDecodedPadded...) | ||
|
|
||
| if len(concatenatedBytes) != 32 { | ||
| return [32]byte{}, errors.New("concatenatedBytes(verificationGasLimitDecodedPadded, callGasLimitDecodedPadded) is not equal to 32 bytes") | ||
| } | ||
|
|
||
| // Convert concatenatedBytes to [32]byte | ||
| var result [32]byte | ||
| copy(result[:], concatenatedBytes) | ||
|
|
||
| return result, nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) GasFees() ([32]byte, error) { | ||
| maxPriorityFeePerGasDecoded, err := hexutil.Decode(op.MaxPriorityFeePerGas) | ||
| if err != nil { | ||
| return [32]byte{}, errors.New("maxPriorityFeePerGas (bytes) conversion failed") | ||
| } | ||
|
|
||
| maxFeePerGasDecoded, err := hexutil.Decode(op.MaxFeePerGas) | ||
| if err != nil { | ||
| return [32]byte{}, errors.New("maxFeePerGas (bytes) conversion failed") | ||
| } | ||
|
|
||
| // Truncate if longer than 16 bytes or Pad with leading zeros if shorter than 16 bytes | ||
| maxPriorityFeePerGasDecodedPadded, err := utils.PadToBytes16(maxPriorityFeePerGasDecoded) | ||
| if err != nil { | ||
| return [32]byte{}, err | ||
| } | ||
|
|
||
| maxFeePerGasDecodedPadded, err := utils.PadToBytes16(maxFeePerGasDecoded) | ||
| if err != nil { | ||
| return [32]byte{}, err | ||
| } | ||
|
|
||
| // Concatenate the byte slices | ||
| concatenatedBytes := append(maxPriorityFeePerGasDecodedPadded, maxFeePerGasDecodedPadded...) | ||
|
|
||
| if len(concatenatedBytes) != 32 { | ||
| return [32]byte{}, errors.New("concatenatedBytes(maxPriorityFeePerGasDecodedPadded, maxFeePerGasDecodedPadded) is not equal to 32 bytes") | ||
| } | ||
|
|
||
| // Convert concatenatedBytes to [32]byte | ||
| var result [32]byte | ||
| copy(result[:], concatenatedBytes) | ||
|
|
||
| return result, nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) GetPaymasterAndData() ([]byte, error) { | ||
| if op.Paymaster == "0x" { | ||
| return []byte{}, nil | ||
| } | ||
|
|
||
| if op.PreVerificationGas == "0x" { | ||
| return nil, errors.New("Got Paymaster but missing PreVerificationGas") | ||
| } | ||
|
|
||
| if op.PaymasterPostOpGasLimit == "0x" { | ||
| return nil, errors.New("Got Paymaster but missing PaymasterPostOpGasLimit") | ||
| } | ||
|
|
||
| // Decode all paymaster values | ||
| paymasterDecoded, err := hexutil.Decode(op.Paymaster) | ||
| if err != nil { | ||
| return nil, errors.New("paymasterDecoded (bytes) conversion failed") | ||
| } | ||
|
|
||
| preVerificationGasDecoded, err := hexutil.Decode(op.PreVerificationGas) | ||
| if err != nil { | ||
| return []byte{}, errors.New("preVerificationGas (bytes) conversion failed") | ||
| } | ||
|
|
||
| paymasterPostOpGasLimitDecoded, err := hexutil.Decode(op.PaymasterPostOpGasLimit) | ||
| if err != nil { | ||
| return []byte{}, errors.New("paymasterPostOpGasLimit (bytes) conversion failed") | ||
| } | ||
|
|
||
| // Truncate if longer than 16 bytes or Pad with leading zeros if shorter than 16 bytes | ||
| preVerificationGasDecodedPadded, err := utils.PadToBytes16(preVerificationGasDecoded) | ||
| if err != nil { | ||
| return []byte{}, err | ||
| } | ||
|
|
||
| paymasterPostOpGasLimitDecodedPadded, err := utils.PadToBytes16(paymasterPostOpGasLimitDecoded) | ||
| if err != nil { | ||
| return []byte{}, err | ||
| } | ||
| concatenatedGasBytes := append(preVerificationGasDecodedPadded, paymasterPostOpGasLimitDecodedPadded...) | ||
|
|
||
| // Concatenate the byte slices | ||
| concatenatedPaymasterBytes := append( | ||
| paymasterDecoded, | ||
| concatenatedGasBytes..., | ||
| ) | ||
|
|
||
| var paymasterDataDecoded []byte | ||
| if op.PaymasterData == "0x" { | ||
| paymasterDataDecoded = []byte{} | ||
| } else { | ||
| paymasterDataDecoded, err = hexutil.Decode(op.PaymasterData) | ||
| if err != nil { | ||
| return nil, errors.New("paymasterData (bytes) conversion failed") | ||
| } | ||
| } | ||
|
|
||
| return append( | ||
| concatenatedPaymasterBytes, | ||
| paymasterDataDecoded..., | ||
| ), nil | ||
| } | ||
|
|
||
| func (op *UserOpV7Hexify) PackUserOp() (*entrypoint.PackedUserOperation, error) { | ||
| nonceDecoded, err := hexutil.DecodeBig(op.Nonce) | ||
| if err != nil { | ||
| return nil, errors.New("nonce (bigInt) conversion failed") | ||
| } | ||
|
|
||
| initCodeDecoded, err := op.GetInitCode() | ||
| if err != nil { | ||
| return nil, errors.New("nonce (bigInt) conversion failed") | ||
| } | ||
|
|
||
| callDataDecoded, err := hexutil.Decode(op.CallData) | ||
| if err != nil { | ||
| return nil, errors.New("calldata (bytes) conversion failed") | ||
| } | ||
|
|
||
| preVerificationGaseDecoded, err := hexutil.DecodeBig(op.PreVerificationGas) | ||
| if err != nil { | ||
| return nil, errors.New("preVerificationGas (bigInt) conversion failed") | ||
| } | ||
|
|
||
| accountGasLimitsDecoded, err := op.GetAccountGasLimits() | ||
| if err != nil { | ||
| return nil, errors.New("accountGasLimit (bytes 32) conversion failed") | ||
| } | ||
|
|
||
| gasFeesDecoded, err := op.GasFees() | ||
| if err != nil { | ||
| return nil, errors.New("gasFees (bytes 32) conversion failed") | ||
| } | ||
|
|
||
| paymasterAndDataDecoded, err := op.GetPaymasterAndData() | ||
| if err != nil { | ||
| return nil, errors.New("paymasterAndData (bytes) conversion failed") | ||
| } | ||
|
|
||
| signatureDecoded, err := hexutil.Decode(op.Signature) | ||
| if err != nil { | ||
| return nil, errors.New("signature (bytes) conversion failed") | ||
| } | ||
|
|
||
| return &entrypoint.PackedUserOperation{ | ||
| Sender: common.HexToAddress(op.Sender), | ||
| Nonce: nonceDecoded, | ||
| InitCode: initCodeDecoded, | ||
| CallData: callDataDecoded, | ||
| AccountGasLimits: accountGasLimitsDecoded, | ||
| PreVerificationGas: preVerificationGaseDecoded, | ||
| GasFees: gasFeesDecoded, | ||
| PaymasterAndData: paymasterAndDataDecoded, | ||
| Signature: signatureDecoded, | ||
| }, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| package mempool |
Oops, something went wrong.