package contracts import ( "context" "errors" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/ethclient" "github.com/status-im/status-go/contracts/directory" "github.com/status-im/status-go/contracts/ethscan" "github.com/status-im/status-go/contracts/hop" hopBridge "github.com/status-im/status-go/contracts/hop/bridge" hopSwap "github.com/status-im/status-go/contracts/hop/swap" hopWrapper "github.com/status-im/status-go/contracts/hop/wrapper" "github.com/status-im/status-go/contracts/registrar" "github.com/status-im/status-go/contracts/resolver" "github.com/status-im/status-go/contracts/snt" "github.com/status-im/status-go/contracts/stickers" "github.com/status-im/status-go/rpc" ) type ContractMaker struct { RPCClient *rpc.Client } func (c *ContractMaker) NewRegistryWithAddress(chainID uint64, address common.Address) (*resolver.ENSRegistryWithFallback, error) { backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return resolver.NewENSRegistryWithFallback( address, backend, ) } func (c *ContractMaker) NewRegistry(chainID uint64) (*resolver.ENSRegistryWithFallback, error) { contractAddr, err := resolver.ContractAddress(chainID) if err != nil { return nil, err } return c.NewRegistryWithAddress(chainID, contractAddr) } func (c *ContractMaker) NewPublicResolver(chainID uint64, resolverAddress *common.Address) (*resolver.PublicResolver, error) { backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return resolver.NewPublicResolver(*resolverAddress, backend) } func (c *ContractMaker) NewUsernameRegistrar(chainID uint64, contractAddr common.Address) (*registrar.UsernameRegistrar, error) { backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return registrar.NewUsernameRegistrar( contractAddr, backend, ) } func (c *ContractMaker) NewSNT(chainID uint64) (*snt.SNT, error) { contractAddr, err := snt.ContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return snt.NewSNT(contractAddr, backend) } func (c *ContractMaker) NewStickerType(chainID uint64) (*stickers.StickerType, error) { contractAddr, err := stickers.StickerTypeContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return stickers.NewStickerType( contractAddr, backend, ) } func (c *ContractMaker) NewStickerMarket(chainID uint64) (*stickers.StickerMarket, error) { contractAddr, err := stickers.StickerMarketContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return stickers.NewStickerMarket( contractAddr, backend, ) } func (c *ContractMaker) NewStickerPack(chainID uint64) (*stickers.StickerPack, error) { contractAddr, err := stickers.StickerPackContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return stickers.NewStickerPack( contractAddr, backend, ) } func (c *ContractMaker) NewDirectory(chainID uint64) (*directory.Directory, error) { contractAddr, err := directory.ContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return directory.NewDirectory( contractAddr, backend, ) } func (c *ContractMaker) NewDirectoryWithBackend(chainID uint64, backend *ethclient.Client) (*directory.Directory, error) { contractAddr, err := directory.ContractAddress(chainID) if err != nil { return nil, err } return directory.NewDirectory( contractAddr, backend, ) } func (c *ContractMaker) NewEthScan(chainID uint64) (*ethscan.BalanceScanner, error) { contractAddr, err := ethscan.ContractAddress(chainID) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } bytecode, err := backend.CodeAt(context.Background(), contractAddr, nil) if err != nil { return nil, err } if len(bytecode) == 0 { return nil, errors.New("is not a contract") } return ethscan.NewBalanceScanner( contractAddr, backend, ) } func (c *ContractMaker) NewHopL2SaddlSwap(chainID uint64, symbol string) (*hopSwap.HopSwap, error) { contractAddr, err := hop.L2SaddleSwapContractAddress(chainID, symbol) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return hopSwap.NewHopSwap( contractAddr, backend, ) } func (c *ContractMaker) NewHopL1Bridge(chainID uint64, symbol string) (*hopBridge.HopBridge, error) { contractAddr, err := hop.L1BridgeContractAddress(chainID, symbol) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return hopBridge.NewHopBridge( contractAddr, backend, ) } func (c *ContractMaker) NewHopL2AmmWrapper(chainID uint64, symbol string) (*hopWrapper.HopWrapper, error) { contractAddr, err := hop.L2AmmWrapperContractAddress(chainID, symbol) if err != nil { return nil, err } backend, err := c.RPCClient.EthClient(chainID) if err != nil { return nil, err } return hopWrapper.NewHopWrapper( contractAddr, backend, ) }