package ethpipe import ( "bytes" "encoding/json" "fmt" "sync/atomic" "github.com/ethereum/eth-go/ethchain" "github.com/ethereum/eth-go/ethcrypto" "github.com/ethereum/eth-go/ethreact" "github.com/ethereum/eth-go/ethstate" "github.com/ethereum/eth-go/ethutil" ) type JSPipe struct { *Pipe } func NewJSPipe(eth ethchain.EthManager) *JSPipe { return &JSPipe{New(eth)} } func (self *JSPipe) BlockByHash(strHash string) *JSBlock { hash := ethutil.Hex2Bytes(strHash) block := self.obj.BlockChain().GetBlock(hash) return NewJSBlock(block) } func (self *JSPipe) GetBlockByNumber(num int32) *JSBlock { if num == -1 { return NewJSBlock(self.obj.BlockChain().CurrentBlock) } return NewJSBlock(self.obj.BlockChain().GetBlockByNumber(uint64(num))) } func (self *JSPipe) Key() *JSKey { return NewJSKey(self.obj.KeyManager().KeyPair()) } func (self *JSPipe) StateObject(addr string) *JSObject { object := &Object{self.World().safeGet(ethutil.Hex2Bytes(addr))} return NewJSObject(object) } func (self *JSPipe) PeerCount() int { return self.obj.PeerCount() } func (self *JSPipe) Peers() []JSPeer { var peers []JSPeer for peer := self.obj.Peers().Front(); peer != nil; peer = peer.Next() { p := peer.Value.(ethchain.Peer) // we only want connected peers if atomic.LoadInt32(p.Connected()) != 0 { peers = append(peers, *NewJSPeer(p)) } } return peers } func (self *JSPipe) IsMining() bool { return self.obj.IsMining() } func (self *JSPipe) IsListening() bool { return self.obj.IsListening() } func (self *JSPipe) CoinBase() string { return ethutil.Bytes2Hex(self.obj.KeyManager().Address()) } func (self *JSPipe) BalanceAt(addr string) string { return self.World().SafeGet(ethutil.Hex2Bytes(addr)).Balance.String() } func (self *JSPipe) NumberToHuman(balance string) string { b := ethutil.Big(balance) return ethutil.CurrencyToString(b) } func (self *JSPipe) StorageAt(addr, storageAddr string) string { storage := self.World().SafeGet(ethutil.Hex2Bytes(addr)).Storage(ethutil.Hex2Bytes(storageAddr)) return storage.BigInt().String() } func (self *JSPipe) TxCountAt(address string) int { return int(self.World().SafeGet(ethutil.Hex2Bytes(address)).Nonce) } func (self *JSPipe) IsContract(address string) bool { return len(self.World().SafeGet(ethutil.Hex2Bytes(address)).Code) > 0 } func (self *JSPipe) SecretToAddress(key string) string { pair, err := ethcrypto.NewKeyPairFromSec(ethutil.Hex2Bytes(key)) if err != nil { return "" } return ethutil.Bytes2Hex(pair.Address()) } type KeyVal struct { Key string `json:"key"` Value string `json:"value"` } 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) } 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) } } var keyPair *ethcrypto.KeyPair var err error if ethutil.IsHex(key) { keyPair, err = ethcrypto.NewKeyPairFromSec([]byte(ethutil.Hex2Bytes(key[2:]))) } else { keyPair, err = ethcrypto.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 tx *ethchain.Transaction ) if ethutil.IsHex(codeStr) { data = ethutil.Hex2Bytes(codeStr[2:]) } else { data = ethutil.Hex2Bytes(codeStr) } if contractCreation { tx = ethchain.NewContractCreationTx(value, gas, gasPrice, data) } else { tx = ethchain.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 { logger.Infof("Contract addr %x", tx.CreationAddress()) } return NewJSReciept(contractCreation, tx.CreationAddress(), tx.Hash(), keyPair.Address()), nil } func (self *JSPipe) CompileMutan(code string) string { data, err := self.Pipe.CompileMutan(code) if err != nil { return err.Error() } return ethutil.Bytes2Hex(data) } func (self *JSPipe) Watch(object map[string]interface{}) *JSFilter { return NewJSFilterFromMap(object, self.Pipe.obj) /*} else if str, ok := object.(string); ok { println("str") return NewJSFilterFromString(str, self.Pipe.obj) */ } func (self *JSPipe) Messages(object map[string]interface{}) string { filter := self.Watch(object) filter.Uninstall() return filter.Messages() } type JSFilter struct { eth ethchain.EthManager *ethchain.Filter quit chan bool BlockCallback func(*ethchain.Block) MessageCallback func(ethstate.Messages) } func NewJSFilterFromMap(object map[string]interface{}, eth ethchain.EthManager) *JSFilter { filter := &JSFilter{eth, ethchain.NewFilterFromMap(object, eth), make(chan bool), nil, nil} go filter.mainLoop() return filter } func NewJSFilterFromString(str string, eth ethchain.EthManager) *JSFilter { return nil } func (self *JSFilter) MessagesToJson(messages ethstate.Messages) string { var msgs []JSMessage for _, m := range messages { msgs = append(msgs, NewJSMessage(m)) } b, err := json.Marshal(msgs) if err != nil { return "{\"error\":" + err.Error() + "}" } return string(b) } func (self *JSFilter) Messages() string { return self.MessagesToJson(self.Find()) } func (self *JSFilter) mainLoop() { blockChan := make(chan ethreact.Event, 5) messageChan := make(chan ethreact.Event, 5) // Subscribe to events reactor := self.eth.Reactor() reactor.Subscribe("newBlock", blockChan) reactor.Subscribe("messages", messageChan) out: for { select { case <-self.quit: break out case block := <-blockChan: if block, ok := block.Resource.(*ethchain.Block); ok { if self.BlockCallback != nil { self.BlockCallback(block) } } case msg := <-messageChan: if messages, ok := msg.Resource.(ethstate.Messages); ok { if self.MessageCallback != nil { println("messages!") msgs := self.FilterMessages(messages) if len(msgs) > 0 { self.MessageCallback(msgs) } } } } } } func (self *JSFilter) Changed(object interface{}) { fmt.Printf("%T\n", object) } func (self *JSFilter) Uninstall() { self.quit <- true }