From 86cfddab5bbe0cf4e39bc2b44216cfdab477d602 Mon Sep 17 00:00:00 2001
From: jsvisa <delweng@gmail.com>
Date: Mon, 11 Sep 2023 17:49:07 +0800
Subject: [PATCH] cmd/utils: add --dev.alloc flag

Signed-off-by: jsvisa <delweng@gmail.com>
---
 cmd/utils/flags.go | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/cmd/utils/flags.go b/cmd/utils/flags.go
index 9743a7b9ca..b1f7f3ff18 100644
--- a/cmd/utils/flags.go
+++ b/cmd/utils/flags.go
@@ -21,6 +21,7 @@ import (
 	"context"
 	"crypto/ecdsa"
 	"encoding/hex"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"math"
@@ -173,6 +174,12 @@ var (
 		Value:    11500000,
 		Category: flags.DevCategory,
 	}
+	DeveloperAllocFlag = &cli.PathFlag{
+		Name:      "dev.alloc",
+		Usage:     "Genesis account allocation file to use in developer mode",
+		TakesFile: true,
+		Category:  flags.DevCategory,
+	}
 
 	IdentityFlag = &cli.StringFlag{
 		Name:     "identity",
@@ -1876,8 +1883,18 @@ func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *ethconfig.Config) {
 		}
 		log.Info("Using developer account", "address", developer.Address)
 
+		var alloc core.GenesisAlloc
+		if allocPath := ctx.Path(DeveloperAllocFlag.Name); allocPath != "" {
+			data, err := os.ReadFile(allocPath)
+			if err != nil {
+				Fatalf("Failed to read developer genesis alloc file: %v", err)
+			}
+			if err := json.Unmarshal(data, &alloc); err != nil {
+				Fatalf("Failed to parse developer genesis alloc file: %v", err)
+			}
+		}
 		// Create a new developer genesis block or reuse existing one
-		cfg.Genesis = core.DeveloperGenesisBlock(ctx.Uint64(DeveloperGasLimitFlag.Name), developer.Address)
+		cfg.Genesis = core.DeveloperGenesisBlock(ctx.Uint64(DeveloperGasLimitFlag.Name), developer.Address, &alloc)
 		if ctx.IsSet(DataDirFlag.Name) {
 			chaindb := tryMakeReadOnlyDatabase(ctx, stack)
 			if rawdb.ReadCanonicalHash(chaindb, 0) != (common.Hash{}) {