aboutsummaryrefslogtreecommitdiffstats
path: root/ethpipe/js_pipe.go
diff options
context:
space:
mode:
Diffstat (limited to 'ethpipe/js_pipe.go')
-rw-r--r--ethpipe/js_pipe.go334
1 files changed, 334 insertions, 0 deletions
diff --git a/ethpipe/js_pipe.go b/ethpipe/js_pipe.go
new file mode 100644
index 000000000..b32e94a10
--- /dev/null
+++ b/ethpipe/js_pipe.go
@@ -0,0 +1,334 @@
+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) BlockByNumber(num int32) *JSBlock {
+ if num == -1 {
+ return NewJSBlock(self.obj.BlockChain().CurrentBlock)
+ }
+
+ return NewJSBlock(self.obj.BlockChain().GetBlockByNumber(uint64(num)))
+}
+
+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
+}
+
+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))
+ }
+
+ // Return an empty array instead of "null"
+ if len(msgs) == 0 {
+ return "[]"
+ }
+
+ 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
+}