plugeth/ethpipe/js_pipe.go

265 lines
6.3 KiB
Go
Raw Normal View History

package ethpipe
import (
2014-08-20 11:05:26 +00:00
"bytes"
"encoding/json"
"sync/atomic"
2014-10-31 09:59:17 +00:00
"github.com/ethereum/go-ethereum/chain"
2014-10-31 11:37:43 +00:00
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/ethstate"
"github.com/ethereum/go-ethereum/ethutil"
)
type JSPipe struct {
*Pipe
}
2014-10-31 09:59:17 +00:00
func NewJSPipe(eth chain.EthManager) *JSPipe {
return &JSPipe{New(eth)}
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) BlockByHash(strHash string) *JSBlock {
hash := ethutil.Hex2Bytes(strHash)
2014-10-20 09:53:11 +00:00
block := self.obj.ChainManager().GetBlock(hash)
return NewJSBlock(block)
}
2014-08-20 14:40:19 +00:00
func (self *JSPipe) BlockByNumber(num int32) *JSBlock {
2014-08-20 11:36:54 +00:00
if num == -1 {
2014-10-20 09:53:11 +00:00
return NewJSBlock(self.obj.ChainManager().CurrentBlock)
2014-08-20 11:36:54 +00:00
}
2014-10-20 09:53:11 +00:00
return NewJSBlock(self.obj.ChainManager().GetBlockByNumber(uint64(num)))
2014-08-20 11:36:54 +00:00
}
2014-08-20 14:40:19 +00:00
func (self *JSPipe) Block(v interface{}) *JSBlock {
if n, ok := v.(int32); ok {
return self.BlockByNumber(n)
} else if str, ok := v.(string); ok {
return self.BlockByHash(str)
} else if f, ok := v.(float64); ok { // Don't ask ...
return self.BlockByNumber(int32(f))
}
return nil
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) Key() *JSKey {
return NewJSKey(self.obj.KeyManager().KeyPair())
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) StateObject(addr string) *JSObject {
object := &Object{self.World().safeGet(ethutil.Hex2Bytes(addr))}
return NewJSObject(object)
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) PeerCount() int {
return self.obj.PeerCount()
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) Peers() []JSPeer {
var peers []JSPeer
for peer := self.obj.Peers().Front(); peer != nil; peer = peer.Next() {
2014-10-31 09:59:17 +00:00
p := peer.Value.(chain.Peer)
// we only want connected peers
if atomic.LoadInt32(p.Connected()) != 0 {
peers = append(peers, *NewJSPeer(p))
}
}
return peers
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) IsMining() bool {
return self.obj.IsMining()
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) IsListening() bool {
return self.obj.IsListening()
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) CoinBase() string {
return ethutil.Bytes2Hex(self.obj.KeyManager().Address())
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) NumberToHuman(balance string) string {
b := ethutil.Big(balance)
return ethutil.CurrencyToString(b)
}
func (self *JSPipe) StorageAt(addr, storageAddr string) string {
2014-08-20 07:59:09 +00:00
storage := self.World().SafeGet(ethutil.Hex2Bytes(addr)).Storage(ethutil.Hex2Bytes(storageAddr))
2014-08-21 19:06:42 +00:00
return ethutil.Bytes2Hex(storage.Bytes())
}
func (self *JSPipe) BalanceAt(addr string) string {
return self.World().SafeGet(ethutil.Hex2Bytes(addr)).Balance().String()
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) TxCountAt(address string) int {
return int(self.World().SafeGet(ethutil.Hex2Bytes(address)).Nonce)
}
func (self *JSPipe) CodeAt(address string) string {
return ethutil.Bytes2Hex(self.World().SafeGet(ethutil.Hex2Bytes(address)).Code)
}
func (self *JSPipe) IsContract(address string) bool {
return len(self.World().SafeGet(ethutil.Hex2Bytes(address)).Code) > 0
}
func (self *JSPipe) SecretToAddress(key string) string {
2014-10-31 11:37:43 +00:00
pair, err := crypto.NewKeyPairFromSec(ethutil.Hex2Bytes(key))
if err != nil {
return ""
}
return ethutil.Bytes2Hex(pair.Address())
}
func (self *JSPipe) Execute(addr, value, gas, price, data string) (string, error) {
ret, err := self.ExecuteObject(&Object{
self.World().safeGet(ethutil.Hex2Bytes(addr))},
ethutil.Hex2Bytes(data),
ethutil.NewValue(value),
ethutil.NewValue(gas),
ethutil.NewValue(price),
)
return ethutil.Bytes2Hex(ret), err
}
type KeyVal struct {
Key string `json:"key"`
Value string `json:"value"`
}
2014-08-18 08:17:45 +00:00
func (self *JSPipe) EachStorage(addr string) string {
var values []KeyVal
object := self.World().SafeGet(ethutil.Hex2Bytes(addr))
object.EachStorage(func(name string, value *ethutil.Value) {
value.Decode()
values = append(values, KeyVal{ethutil.Bytes2Hex([]byte(name)), ethutil.Bytes2Hex(value.Bytes())})
})
valuesJson, err := json.Marshal(values)
if err != nil {
return ""
}
return string(valuesJson)
}
2014-08-20 11:05:26 +00:00
func (self *JSPipe) ToAscii(str string) string {
padded := ethutil.RightPadBytes([]byte(str), 32)
return "0x" + ethutil.Bytes2Hex(padded)
}
func (self *JSPipe) FromAscii(str string) string {
if ethutil.IsHex(str) {
str = str[2:]
}
return string(bytes.Trim(ethutil.Hex2Bytes(str), "\x00"))
}
func (self *JSPipe) FromNumber(str string) string {
if ethutil.IsHex(str) {
str = str[2:]
}
return ethutil.BigD(ethutil.Hex2Bytes(str)).String()
}
func (self *JSPipe) Transact(key, toStr, valueStr, gasStr, gasPriceStr, codeStr string) (*JSReceipt, error) {
var hash []byte
var contractCreation bool
if len(toStr) == 0 {
contractCreation = true
} else {
// Check if an address is stored by this address
addr := self.World().Config().Get("NameReg").StorageString(toStr).Bytes()
if len(addr) > 0 {
hash = addr
} else {
hash = ethutil.Hex2Bytes(toStr)
}
}
2014-10-31 11:37:43 +00:00
var keyPair *crypto.KeyPair
var err error
if ethutil.IsHex(key) {
2014-10-31 11:37:43 +00:00
keyPair, err = crypto.NewKeyPairFromSec([]byte(ethutil.Hex2Bytes(key[2:])))
} else {
2014-10-31 11:37:43 +00:00
keyPair, err = crypto.NewKeyPairFromSec([]byte(ethutil.Hex2Bytes(key)))
}
if err != nil {
return nil, err
}
var (
value = ethutil.Big(valueStr)
gas = ethutil.Big(gasStr)
gasPrice = ethutil.Big(gasPriceStr)
data []byte
2014-10-31 09:59:17 +00:00
tx *chain.Transaction
)
if ethutil.IsHex(codeStr) {
data = ethutil.Hex2Bytes(codeStr[2:])
} else {
data = ethutil.Hex2Bytes(codeStr)
}
if contractCreation {
2014-10-31 09:59:17 +00:00
tx = chain.NewContractCreationTx(value, gas, gasPrice, data)
} else {
2014-10-31 09:59:17 +00:00
tx = chain.NewTransactionMessage(hash, value, gas, gasPrice, data)
}
acc := self.obj.StateManager().TransState().GetOrNewStateObject(keyPair.Address())
tx.Nonce = acc.Nonce
acc.Nonce += 1
self.obj.StateManager().TransState().UpdateStateObject(acc)
tx.Sign(keyPair.PrivateKey)
self.obj.TxPool().QueueTransaction(tx)
if contractCreation {
2014-10-02 15:03:15 +00:00
logger.Infof("Contract addr %x", tx.CreationAddress(self.World().State()))
}
2014-10-02 15:03:15 +00:00
return NewJSReciept(contractCreation, tx.CreationAddress(self.World().State()), tx.Hash(), keyPair.Address()), nil
}
2014-08-17 10:41:52 +00:00
func (self *JSPipe) PushTx(txStr string) (*JSReceipt, error) {
2014-10-31 09:59:17 +00:00
tx := chain.NewTransactionFromBytes(ethutil.Hex2Bytes(txStr))
self.obj.TxPool().QueueTransaction(tx)
2014-10-02 15:03:15 +00:00
return NewJSReciept(tx.CreatesContract(), tx.CreationAddress(self.World().State()), tx.Hash(), tx.Sender()), nil
}
2014-08-17 10:41:52 +00:00
func (self *JSPipe) CompileMutan(code string) string {
data, err := self.Pipe.CompileMutan(code)
if err != nil {
return err.Error()
}
return ethutil.Bytes2Hex(data)
}
2014-08-18 08:17:45 +00:00
2014-09-13 22:13:23 +00:00
func ToJSMessages(messages ethstate.Messages) *ethutil.List {
2014-08-18 08:17:45 +00:00
var msgs []JSMessage
for _, m := range messages {
msgs = append(msgs, NewJSMessage(m))
}
2014-09-13 22:13:23 +00:00
return ethutil.NewList(msgs)
2014-08-20 07:59:09 +00:00
}