From ed64b87ce29bbcf60c99eb456f4de472c0b6adbc Mon Sep 17 00:00:00 2001
From: whyrusleeping <why@ipfs.io>
Date: Thu, 3 Oct 2019 13:45:14 -0600
Subject: [PATCH] clean up create block

---
 chain/gen/mining.go | 39 +++++++++++----------------------------
 1 file changed, 11 insertions(+), 28 deletions(-)

diff --git a/chain/gen/mining.go b/chain/gen/mining.go
index dcff62df8..d86238159 100644
--- a/chain/gen/mining.go
+++ b/chain/gen/mining.go
@@ -11,8 +11,8 @@ import (
 	cbg "github.com/whyrusleeping/cbor-gen"
 	"golang.org/x/xerrors"
 
-	"github.com/filecoin-project/go-lotus/chain/actors"
 	"github.com/filecoin-project/go-lotus/chain/address"
+	"github.com/filecoin-project/go-lotus/chain/state"
 	"github.com/filecoin-project/go-lotus/chain/stmgr"
 	"github.com/filecoin-project/go-lotus/chain/types"
 	"github.com/filecoin-project/go-lotus/chain/vm"
@@ -27,30 +27,10 @@ func MinerCreateBlock(ctx context.Context, sm *stmgr.StateManager, w *wallet.Wal
 
 	height := parents.Height() + uint64(len(tickets))
 
-	r := vm.NewChainRand(sm.ChainStore(), parents.Cids(), parents.Height(), tickets)
-	vmi, err := vm.NewVM(st, height, r, miner, sm.ChainStore())
-	if err != nil {
-		return nil, err
-	}
-
-	owner, err := stmgr.GetMinerOwner(ctx, sm, st, miner)
-	if err != nil {
-		return nil, xerrors.Errorf("failed to get miner owner: %w", err)
-	}
-
 	worker, err := stmgr.GetMinerWorker(ctx, sm, st, miner)
 	if err != nil {
 		return nil, xerrors.Errorf("failed to get miner worker: %w", err)
 	}
-	networkBalance, err := vmi.ActorBalance(actors.NetworkAddress)
-	if err != nil {
-		return nil, xerrors.Errorf("failed to get network balance: %w", err)
-	}
-
-	// apply miner reward
-	if err := vmi.TransferFunds(actors.NetworkAddress, owner, vm.MiningReward(networkBalance)); err != nil {
-		return nil, err
-	}
 
 	next := &types.BlockHeader{
 		Miner:                 miner,
@@ -113,19 +93,22 @@ func MinerCreateBlock(ctx context.Context, sm *stmgr.StateManager, w *wallet.Wal
 	pweight := sm.ChainStore().Weight(parents)
 	next.ParentWeight = types.NewInt(pweight)
 
-	// TODO: set timestamp
+	cst := hamt.CSTFromBstore(sm.ChainStore().Blockstore())
+	tree, err := state.LoadStateTree(cst, st)
+	if err != nil {
+		return nil, xerrors.Errorf("failed to load state tree: %w", err)
+	}
+
+	waddr, err := vm.ResolveToKeyAddr(tree, cst, worker)
+	if err != nil {
+		return nil, xerrors.Errorf("failed to resolve miner address to key address: %w", err)
+	}
 
 	nosigbytes, err := next.SigningBytes()
 	if err != nil {
 		return nil, xerrors.Errorf("failed to get signing bytes for block: %w", err)
 	}
 
-	cst := hamt.CSTFromBstore(sm.ChainStore().Blockstore())
-	waddr, err := vm.ResolveToKeyAddr(vmi.StateTree(), cst, worker)
-	if err != nil {
-		return nil, xerrors.Errorf("failed to resolve miner address to key address: %w", err)
-	}
-
 	sig, err := w.Sign(ctx, waddr, nosigbytes)
 	if err != nil {
 		return nil, xerrors.Errorf("failed to sign new block: %w", err)