core: support extracting governance state from state trie

1 files changed, 31 insertions, 0 deletions

diff --git a/core/state/gov.go b/core/state/gov.go
new file mode 100644
index 000000000..8b94cfab2
--- /dev/null
+++ b/core/state/gov.go
@@ -0,0 +1,31 @@
+package state
+
+import (
+	"github.com/dexon-foundation/dexon/common"
+	"github.com/dexon-foundation/dexon/core/types"
+	"github.com/dexon-foundation/dexon/trie"
+)
+
+func GetGovState(statedb *StateDB, header *types.Header,
+	addr common.Address) (*types.GovState, error) {
+	proof, err := statedb.GetProof(addr)
+	if err != nil {
+		return nil, err
+	}
+
+	govState := &types.GovState{
+		BlockHash: header.Hash(),
+		Number:    header.Number,
+		Root:      header.Root,
+		Proof:     proof,
+	}
+
+	if t := statedb.StorageTrie(addr); t != nil {
+		it := trie.NewIterator(t.NodeIterator(nil))
+		for it.Next() {
+			govState.Storage = append(govState.Storage,
+				[2][]byte{it.Key, it.Value})
+		}
+	}
+	return govState, nil
+}