go-ethereum/dev_console.go

156 lines
3.4 KiB
Go
Raw Normal View History

package main
import (
2014-01-11 08:27:08 -06:00
"bufio"
"encoding/hex"
"errors"
"fmt"
2014-01-23 13:16:52 -06:00
"github.com/ethereum/eth-go"
"github.com/ethereum/ethchain-go"
2014-01-11 08:27:08 -06:00
"github.com/ethereum/ethdb-go"
"github.com/ethereum/ethutil-go"
"os"
"strings"
)
2014-01-09 09:19:44 -06:00
type Console struct {
2014-01-23 13:16:52 -06:00
db *ethdb.MemDatabase
trie *ethutil.Trie
ethereum *eth.Ethereum
}
2014-01-23 13:16:52 -06:00
func NewConsole(s *eth.Ethereum) *Console {
2014-01-11 08:27:08 -06:00
db, _ := ethdb.NewMemDatabase()
trie := ethutil.NewTrie(db, "")
2014-01-23 13:16:52 -06:00
return &Console{db: db, trie: trie, ethereum: s}
}
2014-01-09 09:19:44 -06:00
func (i *Console) ValidateInput(action string, argumentLength int) error {
2014-01-11 08:27:08 -06:00
err := false
var expArgCount int
2014-01-11 08:27:08 -06:00
switch {
case action == "update" && argumentLength != 2:
err = true
expArgCount = 2
case action == "get" && argumentLength != 1:
err = true
expArgCount = 1
case action == "dag" && argumentLength != 2:
err = true
expArgCount = 2
2014-01-14 14:48:16 -06:00
case action == "decode" && argumentLength != 1:
err = true
expArgCount = 1
case action == "encode" && argumentLength != 1:
err = true
expArgCount = 1
case action == "tx" && argumentLength != 2:
err = true
expArgCount = 2
2014-01-11 08:27:08 -06:00
}
2014-01-11 08:27:08 -06:00
if err {
return errors.New(fmt.Sprintf("'%s' requires %d args, got %d", action, expArgCount, argumentLength))
} else {
return nil
}
}
2014-01-14 14:48:16 -06:00
func (i *Console) PrintRoot() {
root := ethutil.Conv(i.trie.RootT)
if len(root.AsBytes()) != 0 {
fmt.Println(hex.EncodeToString(root.AsBytes()))
} else {
fmt.Println(i.trie.RootT)
}
}
2014-01-09 09:19:44 -06:00
func (i *Console) ParseInput(input string) bool {
2014-01-11 08:27:08 -06:00
scanner := bufio.NewScanner(strings.NewReader(input))
scanner.Split(bufio.ScanWords)
2014-01-11 08:27:08 -06:00
count := 0
var tokens []string
for scanner.Scan() {
count++
tokens = append(tokens, scanner.Text())
}
if err := scanner.Err(); err != nil {
fmt.Fprintln(os.Stderr, "reading input:", err)
}
2014-01-11 08:27:08 -06:00
if len(tokens) == 0 {
return true
}
2014-01-11 08:27:08 -06:00
err := i.ValidateInput(tokens[0], count-1)
if err != nil {
fmt.Println(err)
} else {
switch tokens[0] {
case "update":
2014-01-14 14:48:16 -06:00
i.trie.UpdateT(tokens[1], tokens[2])
2014-01-14 14:48:16 -06:00
i.PrintRoot()
2014-01-11 08:27:08 -06:00
case "get":
2014-01-14 14:48:16 -06:00
fmt.Println(i.trie.GetT(tokens[1]))
2014-01-11 08:27:08 -06:00
case "root":
2014-01-14 14:48:16 -06:00
i.PrintRoot()
2014-01-11 08:27:08 -06:00
case "rawroot":
2014-01-14 14:48:16 -06:00
fmt.Println(i.trie.RootT)
2014-01-11 08:27:08 -06:00
case "print":
i.db.Print()
case "dag":
2014-01-23 13:16:52 -06:00
fmt.Println(ethchain.DaggerVerify(ethutil.Big(tokens[1]), // hash
2014-01-11 08:27:08 -06:00
ethutil.BigPow(2, 36), // diff
ethutil.Big(tokens[2]))) // nonce
2014-01-14 14:48:16 -06:00
case "decode":
d, _ := ethutil.Decode([]byte(tokens[1]), 0)
fmt.Printf("%q\n", d)
case "encode":
fmt.Printf("%q\n", ethutil.Encode(tokens[1]))
case "tx":
tx := ethutil.NewTransaction(tokens[1], ethutil.Big(tokens[2]), []string{""})
2014-01-23 13:16:52 -06:00
i.ethereum.TxPool.QueueTransaction(tx)
2014-01-11 08:27:08 -06:00
case "exit", "quit", "q":
return false
case "help":
fmt.Printf("COMMANDS:\n" +
"\033[1m= DB =\033[0m\n" +
"update KEY VALUE - Updates/Creates a new value for the given key\n" +
"get KEY - Retrieves the given key\n" +
"root - Prints the hex encoded merkle root\n" +
"rawroot - Prints the raw merkle root\n" +
"\033[1m= Dagger =\033[0m\n" +
2014-01-14 14:48:16 -06:00
"dag HASH NONCE - Verifies a nonce with the given hash with dagger\n" +
2014-01-15 11:40:35 -06:00
"\033[1m= Encoding =\033[0m\n" +
2014-01-14 14:48:16 -06:00
"decode STR\n" +
"encode STR\n")
2014-01-11 08:27:08 -06:00
default:
fmt.Println("Unknown command:", tokens[0])
}
}
2014-01-11 08:27:08 -06:00
return true
}
2014-01-09 09:19:44 -06:00
func (i *Console) Start() {
2014-01-11 08:27:08 -06:00
fmt.Printf("Eth Console. Type (help) for help\n")
reader := bufio.NewReader(os.Stdin)
for {
fmt.Printf("eth >>> ")
str, _, err := reader.ReadLine()
if err != nil {
fmt.Println("Error reading input", err)
} else {
if !i.ParseInput(string(str)) {
return
}
}
}
}