From 5a8d11b85e858d989cbd43cea00b2f9587d8cbf6 Mon Sep 17 00:00:00 2001
From: whyrusleeping <why@ipfs.io>
Date: Wed, 16 Oct 2019 17:01:41 +0900
Subject: [PATCH] add format flag to chain list command

---
 chain/store/store.go |  5 +++--
 cli/chain.go         | 40 +++++++++++++++++++++++++++-------------
 2 files changed, 30 insertions(+), 15 deletions(-)

diff --git a/chain/store/store.go b/chain/store/store.go
index 5148a4e64..e9b1b8b73 100644
--- a/chain/store/store.go
+++ b/chain/store/store.go
@@ -5,9 +5,10 @@ import (
 	"crypto/sha256"
 	"encoding/json"
 	"fmt"
-	"github.com/filecoin-project/go-lotus/chain/vm"
 	"sync"
 
+	"github.com/filecoin-project/go-lotus/chain/vm"
+
 	"github.com/filecoin-project/go-lotus/build"
 	"github.com/filecoin-project/go-lotus/chain/address"
 	"github.com/filecoin-project/go-lotus/chain/state"
@@ -271,7 +272,7 @@ func (cs *ChainStore) takeHeaviestTipSet(ts *types.TipSet) error {
 			new: ts,
 		}
 	} else {
-		log.Warn("no heaviest tipset found, using %s", ts.Cids())
+		log.Warnf("no heaviest tipset found, using %s", ts.Cids())
 	}
 
 	log.Debugf("New heaviest tipset! %s", ts.Cids())
diff --git a/cli/chain.go b/cli/chain.go
index 7e8896492..d25c9f28f 100644
--- a/cli/chain.go
+++ b/cli/chain.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"strings"
 	"time"
 
 	cid "github.com/ipfs/go-cid"
@@ -279,7 +280,12 @@ var chainListCmd = &cli.Command{
 	Usage: "View a segment of the chain",
 	Flags: []cli.Flag{
 		&cli.Uint64Flag{Name: "height"},
-		&cli.UintFlag{Name: "count", Value: 30},
+		&cli.IntFlag{Name: "count", Value: 30},
+		&cli.StringFlag{
+			Name:  "format",
+			Usage: "specify the format to print out tipsets",
+			Value: "<height>: (<time>) <blocks>",
+		},
 	},
 	Action: func(cctx *cli.Context) error {
 		api, closer, err := GetFullNodeAPI(cctx)
@@ -300,15 +306,15 @@ var chainListCmd = &cli.Command{
 			return err
 		}
 
-		count := cctx.Uint("count")
+		count := cctx.Int("count")
 		if count < 1 {
 			return nil
 		}
 
-		tss := make([]*types.TipSet, count)
-		tss[0] = head
+		tss := make([]*types.TipSet, 0, count)
+		tss = append(tss, head)
 
-		for i := 1; i < len(tss); i++ {
+		for i := 1; i < count; i++ {
 			if head.Height() == 0 {
 				break
 			}
@@ -318,18 +324,26 @@ var chainListCmd = &cli.Command{
 				return err
 			}
 
-			tss[i] = head
+			tss = append(tss, head)
 		}
 
 		for i := len(tss) - 1; i >= 0; i-- {
-			mints := tss[i].MinTimestamp()
-			t := time.Unix(int64(mints), 0)
-			fmt.Printf("%d: (%s) [ ", tss[i].Height(), t.Format(time.Stamp))
-			for _, b := range tss[i].Blocks() {
-				fmt.Printf("%s: %s,", b.Cid(), b.Miner)
-			}
-			fmt.Println("]")
+			printTipSet(cctx.String("format"), tss[i])
 		}
 		return nil
 	},
 }
+
+func printTipSet(format string, ts *types.TipSet) {
+	format = strings.ReplaceAll(format, "<height>", fmt.Sprint(ts.Height()))
+	format = strings.ReplaceAll(format, "<time>", time.Unix(int64(ts.MinTimestamp()), 0).Format(time.Stamp))
+	blks := "[ "
+	for _, b := range ts.Blocks() {
+		blks += fmt.Sprintf("%s: %s,", b.Cid(), b.Miner)
+	}
+	blks += " ]"
+	format = strings.ReplaceAll(format, "<blocks>", blks)
+	format = strings.ReplaceAll(format, "<weight>", fmt.Sprint(ts.Blocks()[0].ParentWeight))
+
+	fmt.Println(format)
+}