Merge pull request #26 from 8thlight/erc20-take-2

Erc20 take 2

cmd/erc20.go

@@ -0,0 +1,70 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+	"github.com/vulcanize/vulcanizedb/examples/erc20_watcher/every_block"
+	"github.com/vulcanize/vulcanizedb/libraries/shared"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+	"github.com/vulcanize/vulcanizedb/pkg/geth"
+	"log"
+	"time"
+)
+
+// erc20Cmd represents the erc20 command
+var erc20Cmd = &cobra.Command{
+	Use:   "erc20",
+	Short: "Fetches and persists token supply",
+	Long: `Fetches the totalSupply for the configured token from each block and persists it in Vulcanize DB.
+vulcanizedb erc20 --config environments/public
+
+Expects an ethereum node to be running and requires a .toml config file:
+
+  [database]
+  name = "vulcanize_public"
+  hostname = "localhost"
+  port = 5432
+
+  [client]
+  ipcPath = "/Users/user/Library/Ethereum/geth.ipc"
+`,
+	Run: func(cmd *cobra.Command, args []string) {
+		watchERC20s()
+	},
+}
+
+func watchERC20s() {
+	ticker := time.NewTicker(5 * time.Second)
+	defer ticker.Stop()
+	blockchain := geth.NewBlockchain(ipc)
+	db, err := postgres.NewDB(databaseConfig, blockchain.Node())
+	if err != nil {
+		log.Fatal("Failed to initialize database.")
+	}
+	watcher := shared.Watcher{
+		DB:         *db,
+		Blockchain: blockchain,
+	}
+
+	watcher.AddTransformers(every_block.TransformerInitializers())
+	for range ticker.C {
+		watcher.Execute()
+	}
+}
+
+func init() {
+	rootCmd.AddCommand(erc20Cmd)
+}