mv pkg/omni pkg/contract_watcher

2019-03-21 18:33:56 -05:00
parent 073378de6d
commit 24879a85aa
60 changed files with 117 additions and 117 deletions
@@ -0,0 +1,35 @@
+// VulcanizeDB
+// Copyright © 2019 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package getter_test
+
+import (
+	"io/ioutil"
+	"log"
+	"testing"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+func TestRepository(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Getter Suite Test")
+}
+
+var _ = BeforeSuite(func() {
+	log.SetOutput(ioutil.Discard)
+})
@@ -0,0 +1,55 @@
+// VulcanizeDB
+// Copyright © 2019 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package getter_test
+
+import (
+	"github.com/ethereum/go-ethereum/ethclient"
+	"github.com/ethereum/go-ethereum/rpc"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/contract_watcher/shared/constants"
+	"github.com/vulcanize/vulcanizedb/pkg/contract_watcher/shared/getter"
+	"github.com/vulcanize/vulcanizedb/pkg/geth"
+	"github.com/vulcanize/vulcanizedb/pkg/geth/client"
+	rpc2 "github.com/vulcanize/vulcanizedb/pkg/geth/converters/rpc"
+	"github.com/vulcanize/vulcanizedb/pkg/geth/node"
+)
+
+var _ = Describe("Interface Getter", func() {
+	Describe("GetAbi", func() {
+		It("Constructs and returns a custom abi based on results from supportsInterface calls", func() {
+			expectedABI := `[` + constants.AddrChangeInterface + `,` + constants.NameChangeInterface + `,` + constants.ContentChangeInterface + `,` + constants.AbiChangeInterface + `,` + constants.PubkeyChangeInterface + `]`
+
+			blockNumber := int64(6885696)
+			infuraIPC := "https://mainnet.infura.io/v3/b09888c1113640cc9ab42750ce750c05"
+			rawRpcClient, err := rpc.Dial(infuraIPC)
+			Expect(err).NotTo(HaveOccurred())
+			rpcClient := client.NewRpcClient(rawRpcClient, infuraIPC)
+			ethClient := ethclient.NewClient(rawRpcClient)
+			blockChainClient := client.NewEthClient(ethClient)
+			node := node.MakeNode(rpcClient)
+			transactionConverter := rpc2.NewRpcTransactionConverter(ethClient)
+			blockChain := geth.NewBlockChain(blockChainClient, rpcClient, node, transactionConverter)
+			interfaceGetter := getter.NewInterfaceGetter(blockChain)
+			abi := interfaceGetter.GetABI(constants.PublicResolverAddress, blockNumber)
+			Expect(abi).To(Equal(expectedABI))
+			_, err = geth.ParseAbi(abi)
+			Expect(err).ToNot(HaveOccurred())
+		})
+	})
+})
@@ -0,0 +1,105 @@
+// VulcanizeDB
+// Copyright © 2019 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package getter
+
+import (
+	"github.com/vulcanize/vulcanizedb/pkg/contract_watcher/shared/constants"
+	"github.com/vulcanize/vulcanizedb/pkg/contract_watcher/shared/fetcher"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+)
+
+type InterfaceGetter interface {
+	GetABI(resolverAddr string, blockNumber int64) string
+	GetBlockChain() core.BlockChain
+}
+
+type interfaceGetter struct {
+	fetcher.Fetcher
+}
+
+func NewInterfaceGetter(blockChain core.BlockChain) *interfaceGetter {
+	return &interfaceGetter{
+		Fetcher: fetcher.Fetcher{
+			BlockChain: blockChain,
+		},
+	}
+}
+
+// Used to construct a custom ABI based on the results from calling supportsInterface
+func (g *interfaceGetter) GetABI(resolverAddr string, blockNumber int64) string {
+	a := constants.SupportsInterfaceABI
+	args := make([]interface{}, 1)
+	args[0] = constants.MetaSig.Bytes()
+	supports, err := g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err != nil || !supports {
+		return ""
+	}
+	abiStr := `[`
+	args[0] = constants.AddrChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.AddrChangeInterface + ","
+	}
+	args[0] = constants.NameChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.NameChangeInterface + ","
+	}
+	args[0] = constants.ContentChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.ContentChangeInterface + ","
+	}
+	args[0] = constants.AbiChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.AbiChangeInterface + ","
+	}
+	args[0] = constants.PubkeyChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.PubkeyChangeInterface + ","
+	}
+	args[0] = constants.ContentHashChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.ContenthashChangeInterface + ","
+	}
+	args[0] = constants.MultihashChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.MultihashChangeInterface + ","
+	}
+	args[0] = constants.TextChangeSig.Bytes()
+	supports, err = g.getSupportsInterface(a, resolverAddr, blockNumber, args)
+	if err == nil && supports {
+		abiStr += constants.TextChangeInterface + ","
+	}
+	abiStr = abiStr[:len(abiStr)-1] + `]`
+
+	return abiStr
+}
+
+// Use this method to check whether or not a contract supports a given method/event interface
+func (g *interfaceGetter) getSupportsInterface(contractAbi, contractAddress string, blockNumber int64, methodArgs []interface{}) (bool, error) {
+	return g.Fetcher.FetchBool("supportsInterface", contractAbi, contractAddress, blockNumber, methodArgs)
+}
+
+// Method to retrieve the Getter's blockchain
+func (g *interfaceGetter) GetBlockChain() core.BlockChain {
+	return g.Fetcher.BlockChain
+}