-
Notifications
You must be signed in to change notification settings - Fork 55
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(monitor): add support for monitoring additional contracts (#2493)
contracts.ToFund didn't quite work well for monitoring contracts we want to withdraw from instead of adding funding to, so I added contracts.ToWithdraw and incorporated it into the monitor. Initially prepared it for the gas pumps. issue: none
- Loading branch information
Showing
5 changed files
with
159 additions
and
75 deletions.
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,124 @@ | ||
| package contracts | ||
|
|
||
| import ( | ||
| "context" | ||
| "math/big" | ||
|
|
||
| "github.com/omni-network/omni/halo/genutil/evm/predeploys" | ||
| "github.com/omni-network/omni/lib/netconf" | ||
|
|
||
| "github.com/ethereum/go-ethereum/common" | ||
| "github.com/ethereum/go-ethereum/params" | ||
|
|
||
| "cosmossdk.io/math" | ||
| ) | ||
|
|
||
| // Contract defines a contract to monitor. | ||
| type Contract struct { | ||
| Name string | ||
| Address common.Address | ||
| OnlyOmniEVM bool | ||
| NotOmniEVM bool | ||
| FundThresholds *FundThresholds | ||
| WithdrawThresholds *WithdrawThresholds | ||
| } | ||
|
|
||
| // ToMonitor returns all contracts for the given network relevant to the monitor. | ||
| func ToMonitor(ctx context.Context, network netconf.ID) ([]Contract, error) { | ||
| addrs, err := GetAddresses(ctx, network) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return []Contract{ | ||
| // Funded contracts | ||
| { | ||
| Name: "gas-station", | ||
| Address: addrs.GasStation, | ||
| OnlyOmniEVM: true, | ||
| NotOmniEVM: false, | ||
| FundThresholds: &FundThresholds{minEther: 200, targetEther: 1000}, // GasStation funds user GasPump requests, and needs a large OMNI balance. | ||
| }, | ||
| // Withdrawal contracts | ||
| { | ||
| Name: "gas-pump", | ||
| Address: addrs.GasPump, | ||
| OnlyOmniEVM: false, | ||
| NotOmniEVM: true, | ||
| WithdrawThresholds: &WithdrawThresholds{maxEther: 10}, | ||
| }, | ||
| // Monitoring contracts | ||
| { | ||
| Name: "staking", | ||
| Address: common.HexToAddress(predeploys.Staking), | ||
| OnlyOmniEVM: true, | ||
| NotOmniEVM: false, | ||
| }, | ||
| { | ||
| Name: "nativeBridge", | ||
| Address: common.HexToAddress(predeploys.OmniBridgeNative), | ||
| OnlyOmniEVM: true, | ||
| NotOmniEVM: false, | ||
| }, | ||
| }, nil | ||
| } | ||
|
|
||
| // ToFund returns all fundable contracts for the given network. | ||
| func ToFund(ctx context.Context, network netconf.ID) ([]Contract, error) { | ||
| contracts, err := ToMonitor(ctx, network) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| var fundContracts []Contract | ||
| for _, contract := range contracts { | ||
| if contract.FundThresholds != nil { | ||
| fundContracts = append(fundContracts, contract) | ||
| } | ||
| } | ||
|
|
||
| return fundContracts, nil | ||
| } | ||
|
|
||
| // FundThresholds defines the thresholds for funding a contract. | ||
| type FundThresholds struct { | ||
| minEther float64 | ||
| targetEther float64 | ||
| } | ||
|
|
||
| // MinBalance returns the minimum balance required for funding a contract. | ||
| func (t FundThresholds) MinBalance() *big.Int { | ||
| gwei := t.minEther * params.GWei | ||
|
|
||
| if gwei < 1 { | ||
| panic("ether float64 must be greater than 1 Gwei") | ||
| } | ||
|
|
||
| return math.NewInt(params.GWei).MulRaw(int64(gwei)).BigInt() | ||
| } | ||
|
|
||
| // TargetBalance returns the target balance to fund a contract to. | ||
| func (t FundThresholds) TargetBalance() *big.Int { | ||
| gwei := t.targetEther * params.GWei | ||
| if gwei < 1 { | ||
| panic("ether float64 must be greater than 1 Gwei") | ||
| } | ||
|
|
||
| return math.NewInt(params.GWei).MulRaw(int64(gwei)).BigInt() | ||
| } | ||
|
|
||
| // WithdrawThresholds defines the thresholds for withdrawing from a contract. | ||
| type WithdrawThresholds struct { | ||
| maxEther float64 | ||
| } | ||
|
|
||
| // MaxBalance returns the max balance a contract can have before a withdrawal. | ||
| func (t WithdrawThresholds) MaxBalance() *big.Int { | ||
| gwei := t.maxEther * params.GWei | ||
|
|
||
| if gwei < 1 { | ||
| panic("ether float64 must be greater than 1 Gwei") | ||
| } | ||
|
|
||
| return math.NewInt(params.GWei).MulRaw(int64(gwei)).BigInt() | ||
| } |
This file was deleted.
Oops, something went wrong.
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