From 1613da91cab4dfa422f73c2615cd39da97fbda3f Mon Sep 17 00:00:00 2001
From: rigelrozanski <rigel.rozanski@gmail.com>
Date: Wed, 11 Oct 2017 14:06:57 -0400
Subject: [PATCH] genesis account coin denom flag

---
 server/commands/init.go  | 37 +++++++++++++++++++++----------------
 server/commands/start.go |  4 ++--
 2 files changed, 23 insertions(+), 18 deletions(-)

diff --git a/server/commands/init.go b/server/commands/init.go
index aaa5aba0b..0935463ec 100644
--- a/server/commands/init.go
+++ b/server/commands/init.go
@@ -18,24 +18,29 @@ import (
 	cmn "github.com/tendermint/tmlibs/common"
 )
 
-// InitCmd - node initialization command
-var InitCmd = &cobra.Command{
-	Use:   "init [address]",
-	Short: "Initialize genesis files for a blockchain",
-	RunE:  initCmd,
-}
-
-//nolint - flags
 var (
+	// InitCmd - node initialization command
+	InitCmd = GetInitCmd("mycoin")
+
+	//nolint - flags
 	FlagChainID = "chain-id" //TODO group with other flags or remove? is this already a flag here?
+	FlagDenom   = "denom"    //TODO group with other flags or remove? is this already a flag here?
 	FlagOption  = "option"
 	FlagStatic  = "static"
 )
 
-func init() {
-	InitCmd.Flags().String(FlagChainID, "test_chain_id", "Chain ID")
-	InitCmd.Flags().StringSliceP(FlagOption, "p", []string{}, "Genesis option in the format <app>/<option>/<value>")
-	InitCmd.Flags().Bool(FlagStatic, false, "use a static private validator")
+// GetInitCmd - get the node initialization command, with a custom genesis account denom
+func GetInitCmd(defaultDenom string) *cobra.Command {
+	initCmd := &cobra.Command{
+		Use:   "init [address]",
+		Short: "Initialize genesis files for a blockchain",
+		RunE:  initCmd,
+	}
+	initCmd.Flags().String(FlagChainID, "test_chain_id", "Chain ID")
+	initCmd.Flags().String(FlagDenom, defaultDenom, "Coin denomination for genesis account")
+	initCmd.Flags().StringSliceP(FlagOption, "p", []string{}, "Genesis option in the format <app>/<option>/<value>")
+	initCmd.Flags().Bool(FlagStatic, false, "use a static private validator")
+	return initCmd
 }
 
 // returns 1 iff it set a file, otherwise 0 (so we can add them)
@@ -110,7 +115,7 @@ func initCmd(cmd *cobra.Command, args []string) error {
 		privValJSON = string(pvBytes)
 	}
 
-	genesis := GetGenesisJSON(pubkey, viper.GetString(FlagChainID),
+	genesis := GetGenesisJSON(pubkey, viper.GetString(FlagChainID), viper.GetString(FlagDenom),
 		userAddr, optionsStr)
 	return CreateGenesisValidatorFiles(cfg, genesis, privValJSON, cmd.Root().Name())
 }
@@ -164,7 +169,7 @@ func CreateGenesisValidatorFiles(cfg *config.Config, genesis, privVal, appName s
 // GetGenesisJSON returns a new tendermint genesis with Basecoin app_options
 // that grant a large amount of "mycoin" to a single address
 // TODO: A better UX for generating genesis files
-func GetGenesisJSON(pubkey, chainID, addr string, options string) string {
+func GetGenesisJSON(pubkey, chainID, denom, addr string, options string) string {
 	return fmt.Sprintf(`{
   "app_hash": "",
   "chain_id": "%s",
@@ -184,7 +189,7 @@ func GetGenesisJSON(pubkey, chainID, addr string, options string) string {
       "address": "%s",
       "coins": [
         {
-          "denom": "mycoin",
+          "denom": "%s",
           "amount": 9007199254740992
         }
       ]
@@ -193,5 +198,5 @@ func GetGenesisJSON(pubkey, chainID, addr string, options string) string {
       "coin/issuer", {"app": "sigs", "addr": "%s"}%s
     ]
   }
-}`, chainID, pubkey, addr, addr, options)
+}`, chainID, pubkey, addr, denom, addr, options)
 }
diff --git a/server/commands/start.go b/server/commands/start.go
index 3804d13e4..494660300 100644
--- a/server/commands/start.go
+++ b/server/commands/start.go
@@ -30,8 +30,8 @@ var StartCmd = &cobra.Command{
 	RunE:  startCmd,
 }
 
-// InitTickStartCmd - initialize a command as the start command with tick
-func InitTickStartCmd(tick app.Ticker) *cobra.Command {
+// GetTickStartCmd - initialize a command as the start command with tick
+func GetTickStartCmd(tick app.Ticker) *cobra.Command {
 	startCmd := &cobra.Command{
 		Use:   "start",
 		Short: "Start this full node",