Fix/wallet balance cmd

cmd/address.go

@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/howeyc/gopass"
 
@@ -261,10 +262,35 @@ var balanceCmd = &cmds.Command{
 			return err
 		}
 
-		return printOneString(re, (types.FIL)(balance).String())
+		isDone, err := isSyncDone(req.Context, env)
+		if err != nil {
+			return err
+		}
+
+		var balanceStr string
+		if balance.Equals(big.NewInt(0)) && !isDone {
+			balanceStr = fmt.Sprintf("%s (warning: may display 0 if chain sync in progress)", types.FIL(balance))
+		} else {
+			balanceStr = (types.FIL)(balance).String()
+		}
+
+		return printOneString(re, balanceStr)
 	},
 }
 
+func isSyncDone(ctx context.Context, env cmds.Environment) (bool, error) {
+	head, err := getEnv(env).ChainAPI.ChainHead(ctx)
+	if err != nil {
+		return false, err
+	}
+	params, err := getEnv(env).ChainAPI.StateGetNetworkParams(ctx)
+	if err != nil {
+		return false, err
+	}
+
+	return time.Now().Unix()-int64(head.MinTimestamp()) < int64(params.BlockDelaySecs), nil
+}
+
 // WalletSerializeResult is the type wallet export and import return and expect.
 type WalletSerializeResult struct {
 	KeyInfo []*key.KeyInfo

cmd/address_integration_test.go

@@ -59,7 +59,7 @@ func TestWalletBalance(t *testing.T) {
 
 	t.Log("[success] not found, zero")
 	balance := cmdClient.RunSuccess(ctx, "wallet", "balance", addr.String()).ReadStdout()
-	assert.Equal(t, "0 FIL\n", balance)
+	assert.Equal(t, "0 FIL (warning: may display 0 if chain sync in progress)\n", balance)
 
 	t.Log("[success] balance 1394000000000000000000000000")
 	balance = cmdClient.RunSuccess(ctx, "wallet", "balance", builtin.RewardActorAddr.String()).ReadStdout()
@@ -69,7 +69,7 @@ func TestWalletBalance(t *testing.T) {
 	var addrNew cmd.AddressResult
 	cmdClient.RunSuccessFirstLine(ctx, "wallet", "new")
 	balance = cmdClient.RunSuccess(ctx, "wallet", "balance", addrNew.Address.String()).ReadStdout()
-	assert.Equal(t, "0 FIL\n", balance)
+	assert.Equal(t, "0 FIL (warning: may display 0 if chain sync in progress)\n", balance)
 }
 
 func TestWalletLoadFromFile(t *testing.T) {

pkg/events/filter/index.go

@@ -168,6 +168,10 @@ func (ei *EventIndex) migrateToVersion2(ctx context.Context, chainStore *chain.S
 	currTS := chainStore.GetHead()
 
 	for int64(currTS.Height()) >= minHeight.Int64 {
+		if currTS.Height() == 0 {
+			break
+		}
+
 		if currTS.Height()%1000 == 0 {
 			log.Infof("Migrating height %d (remaining %d)", currTS.Height(), int64(currTS.Height())-minHeight.Int64)
 		}