non-interactive option
- add -y flag for non-interactive use - refactor main - output to logfile (not ideal..) but not to all ethutil loggers for privacy
This commit is contained in:
parent
f4c13f8656
commit
32b09d652d
|
@ -17,12 +17,14 @@ var ImportKey string
|
||||||
var ExportKey bool
|
var ExportKey bool
|
||||||
var LogFile string
|
var LogFile string
|
||||||
var DataDir string
|
var DataDir string
|
||||||
|
var NonInteractive bool
|
||||||
|
|
||||||
func Init() {
|
func Init() {
|
||||||
flag.BoolVar(&StartConsole, "c", false, "debug and testing console")
|
flag.BoolVar(&StartConsole, "c", false, "debug and testing console")
|
||||||
flag.BoolVar(&StartMining, "m", false, "start dagger mining")
|
flag.BoolVar(&StartMining, "m", false, "start dagger mining")
|
||||||
flag.BoolVar(&ShowGenesis, "g", false, "prints genesis header and exits")
|
flag.BoolVar(&ShowGenesis, "g", false, "prints genesis header and exits")
|
||||||
//flag.BoolVar(&UseGui, "gui", true, "use the gui")
|
//flag.BoolVar(&UseGui, "gui", true, "use the gui")
|
||||||
|
flag.BoolVar(&NonInteractive, "y", false, "non-interactive mode (say yes to confirmations)")
|
||||||
flag.BoolVar(&UseUPnP, "upnp", false, "enable UPnP support")
|
flag.BoolVar(&UseUPnP, "upnp", false, "enable UPnP support")
|
||||||
flag.BoolVar(&UseSeed, "seed", true, "seed peers")
|
flag.BoolVar(&UseSeed, "seed", true, "seed peers")
|
||||||
flag.BoolVar(&GenAddr, "genaddr", false, "create a new priv/pub key")
|
flag.BoolVar(&GenAddr, "genaddr", false, "create a new priv/pub key")
|
||||||
|
|
|
@ -30,13 +30,27 @@ func RegisterInterupts(s *eth.Ethereum) {
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func confirm(message string) bool {
|
||||||
|
fmt.Println(message, "Are you sure? (y/n)")
|
||||||
|
var r string
|
||||||
|
fmt.Scanln(&r)
|
||||||
|
for ; ; fmt.Scanln(&r) {
|
||||||
|
if r == "n" || r == "y" {
|
||||||
|
break
|
||||||
|
} else {
|
||||||
|
fmt.Printf("Yes or no?", r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r == "y"
|
||||||
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
Init()
|
Init()
|
||||||
|
|
||||||
runtime.GOMAXPROCS(runtime.NumCPU())
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
||||||
|
|
||||||
// set logger
|
// set logger
|
||||||
var logger *log.Logger
|
var logSys *log.Logger
|
||||||
flags := log.LstdFlags
|
flags := log.LstdFlags
|
||||||
|
|
||||||
if LogFile != "" {
|
if LogFile != "" {
|
||||||
|
@ -46,12 +60,13 @@ func main() {
|
||||||
}
|
}
|
||||||
defer logfile.Close()
|
defer logfile.Close()
|
||||||
log.SetOutput(logfile)
|
log.SetOutput(logfile)
|
||||||
logger = log.New(logfile, "", flags)
|
logSys = log.New(logfile, "", flags)
|
||||||
} else {
|
} else {
|
||||||
logger = log.New(os.Stdout, "", flags)
|
logSys = log.New(os.Stdout, "", flags)
|
||||||
}
|
}
|
||||||
ethutil.ReadConfig(DataDir)
|
ethutil.ReadConfig(DataDir)
|
||||||
ethutil.Config.Log.AddLogSystem(logger)
|
logger := ethutil.Config.Log
|
||||||
|
logger.AddLogSystem(logSys)
|
||||||
|
|
||||||
ethchain.InitFees()
|
ethchain.InitFees()
|
||||||
ethutil.Config.Seed = UseSeed
|
ethutil.Config.Seed = UseSeed
|
||||||
|
@ -64,67 +79,42 @@ func main() {
|
||||||
}
|
}
|
||||||
ethereum.Port = OutboundPort
|
ethereum.Port = OutboundPort
|
||||||
|
|
||||||
if GenAddr {
|
// bookkeeping tasks
|
||||||
fmt.Println("This action overwrites your old private key. Are you sure? (y/n)")
|
switch {
|
||||||
|
case GenAddr:
|
||||||
var r string
|
if NonInteractive || confirm("This action overwrites your old private key.") {
|
||||||
fmt.Scanln(&r)
|
|
||||||
for ; ; fmt.Scanln(&r) {
|
|
||||||
if r == "n" || r == "y" {
|
|
||||||
break
|
|
||||||
} else {
|
|
||||||
fmt.Printf("Yes or no?", r)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if r == "y" {
|
|
||||||
utils.CreateKeyPair(true)
|
utils.CreateKeyPair(true)
|
||||||
}
|
}
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
} else {
|
case len(ImportKey) > 0:
|
||||||
if len(ImportKey) > 0 {
|
if NonInteractive || confirm("This action overwrites your old private key.") {
|
||||||
fmt.Println("This action overwrites your old private key. Are you sure? (y/n)")
|
|
||||||
var r string
|
|
||||||
fmt.Scanln(&r)
|
|
||||||
for ; ; fmt.Scanln(&r) {
|
|
||||||
if r == "n" || r == "y" {
|
|
||||||
break
|
|
||||||
} else {
|
|
||||||
fmt.Printf("Yes or no?", r)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if r == "y" {
|
|
||||||
mnemonic := strings.Split(ImportKey, " ")
|
mnemonic := strings.Split(ImportKey, " ")
|
||||||
if len(mnemonic) == 24 {
|
if len(mnemonic) == 24 {
|
||||||
fmt.Println("Got mnemonic key, importing.")
|
logSys.Println("Got mnemonic key, importing.")
|
||||||
key := ethutil.MnemonicDecode(mnemonic)
|
key := ethutil.MnemonicDecode(mnemonic)
|
||||||
utils.ImportPrivateKey(key)
|
utils.ImportPrivateKey(key)
|
||||||
} else if len(mnemonic) == 1 {
|
} else if len(mnemonic) == 1 {
|
||||||
fmt.Println("Got hex key, importing.")
|
logSys.Println("Got hex key, importing.")
|
||||||
utils.ImportPrivateKey(ImportKey)
|
utils.ImportPrivateKey(ImportKey)
|
||||||
} else {
|
} else {
|
||||||
fmt.Println("Did not recognise format, exiting.")
|
logSys.Println("Did not recognise format, exiting.")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
case len(ImportKey) == 0:
|
||||||
} else {
|
|
||||||
utils.CreateKeyPair(false)
|
utils.CreateKeyPair(false)
|
||||||
}
|
fallthrough
|
||||||
}
|
case ExportKey:
|
||||||
|
|
||||||
if ExportKey {
|
|
||||||
key := ethutil.Config.Db.GetKeys()[0]
|
key := ethutil.Config.Db.GetKeys()[0]
|
||||||
fmt.Printf("%x\n", key.PrivateKey)
|
logSys.Println(fmt.Sprintf("prvk: %x\n", key.PrivateKey))
|
||||||
|
os.Exit(0)
|
||||||
|
case ShowGenesis:
|
||||||
|
logSys.Println(ethereum.BlockChain().Genesis())
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
if ShowGenesis {
|
// client
|
||||||
fmt.Println(ethereum.BlockChain().Genesis())
|
logger.Infoln(fmt.Sprintf("Starting Ethereum v%s", ethutil.Config.Ver))
|
||||||
os.Exit(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
ethutil.Config.Log.Infoln(fmt.Sprintf("Starting Ethereum v%s", ethutil.Config.Ver))
|
|
||||||
|
|
||||||
// Set the max peers
|
// Set the max peers
|
||||||
ethereum.MaxPeers = MaxPeer
|
ethereum.MaxPeers = MaxPeer
|
||||||
|
@ -144,13 +134,13 @@ func main() {
|
||||||
ethereum.Start()
|
ethereum.Start()
|
||||||
|
|
||||||
if StartMining {
|
if StartMining {
|
||||||
ethutil.Config.Log.Infoln("Miner started")
|
logger.Infoln("Miner started")
|
||||||
|
|
||||||
// Fake block mining. It broadcasts a new block every 5 seconds
|
// Fake block mining. It broadcasts a new block every 5 seconds
|
||||||
go func() {
|
go func() {
|
||||||
|
|
||||||
if StartMining {
|
if StartMining {
|
||||||
ethutil.Config.Log.Infoln("Miner started")
|
logger.Infoln("Miner started")
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
data, _ := ethutil.Config.Db.Get([]byte("KeyRing"))
|
data, _ := ethutil.Config.Db.Get([]byte("KeyRing"))
|
||||||
|
|
Loading…
Reference in New Issue