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:
zelig 2014-04-16 14:57:51 +01:00
parent f4c13f8656
commit 32b09d652d
2 changed files with 50 additions and 58 deletions

View File

@ -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")

View File

@ -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"))