plugeth/ui/qt/qwhisper/whisper.go

99 lines
2.1 KiB
Go
Raw Normal View History

2014-12-12 21:24:41 +00:00
package qwhisper
import (
"fmt"
2014-12-12 21:24:41 +00:00
"time"
2014-12-16 18:55:57 +00:00
"unsafe"
2014-12-12 21:24:41 +00:00
2014-12-12 21:38:54 +00:00
"github.com/ethereum/go-ethereum/crypto"
2014-12-12 21:29:29 +00:00
"github.com/ethereum/go-ethereum/ethutil"
2014-12-12 21:24:41 +00:00
"github.com/ethereum/go-ethereum/whisper"
"gopkg.in/qml.v1"
2014-12-12 21:24:41 +00:00
)
func fromHex(s string) []byte {
if len(s) > 1 {
return ethutil.Hex2Bytes(s[2:])
}
return nil
}
func toHex(b []byte) string { return "0x" + ethutil.Bytes2Hex(b) }
2014-12-16 18:55:57 +00:00
type Watch struct {
}
func (self *Watch) Arrived(v unsafe.Pointer) {
fmt.Println(v)
}
2014-12-12 21:24:41 +00:00
type Whisper struct {
*whisper.Whisper
view qml.Object
2014-12-16 18:55:57 +00:00
watches map[int]*Watch
2014-12-12 21:24:41 +00:00
}
func New(w *whisper.Whisper) *Whisper {
2014-12-16 18:55:57 +00:00
return &Whisper{w, nil, make(map[int]*Watch)}
2014-12-12 21:24:41 +00:00
}
func (self *Whisper) SetView(view qml.Object) {
self.view = view
}
func (self *Whisper) Post(data string, to, from string, topics []string, pow, ttl uint32) {
2014-12-12 21:24:41 +00:00
msg := whisper.NewMessage(fromHex(data))
envelope, err := msg.Seal(time.Duration(pow), whisper.Opts{
Ttl: time.Duration(ttl),
To: crypto.ToECDSAPub(fromHex(to)),
From: crypto.ToECDSA(fromHex(from)),
2014-12-16 18:55:57 +00:00
Topics: whisper.TopicsFromString(topics...),
2014-12-12 21:24:41 +00:00
})
if err != nil {
fmt.Println(err)
2014-12-12 21:24:41 +00:00
// handle error
return
}
2014-12-12 21:38:54 +00:00
if err := self.Whisper.Send(envelope); err != nil {
fmt.Println(err)
2014-12-12 21:24:41 +00:00
// handle error
return
}
}
func (self *Whisper) NewIdentity() string {
return toHex(self.Whisper.NewIdentity().D.Bytes())
}
2014-12-15 16:28:51 +00:00
func (self *Whisper) HasIdentity(key string) bool {
2014-12-12 21:24:41 +00:00
return self.Whisper.HasIdentity(crypto.ToECDSA(fromHex(key)))
}
2014-12-16 18:55:57 +00:00
func (self *Whisper) Watch(opts map[string]interface{}) *Watch {
2014-12-12 21:24:41 +00:00
filter := filterFromMap(opts)
2014-12-12 21:38:54 +00:00
filter.Fn = func(msg *whisper.Message) {
2014-12-16 18:55:57 +00:00
fmt.Println(msg)
2014-12-12 21:24:41 +00:00
}
2014-12-16 18:55:57 +00:00
i := self.Whisper.Watch(filter)
self.watches[i] = &Watch{}
return self.watches[i]
2014-12-12 21:24:41 +00:00
}
2014-12-12 21:38:54 +00:00
func filterFromMap(opts map[string]interface{}) (f whisper.Filter) {
2014-12-12 21:24:41 +00:00
if to, ok := opts["to"].(string); ok {
2014-12-12 21:38:54 +00:00
f.To = crypto.ToECDSA(fromHex(to))
2014-12-12 21:24:41 +00:00
}
if from, ok := opts["from"].(string); ok {
2014-12-12 21:38:54 +00:00
f.From = crypto.ToECDSAPub(fromHex(from))
2014-12-12 21:24:41 +00:00
}
2014-12-16 18:55:57 +00:00
if topicList, ok := opts["topics"].(*qml.List); ok {
var topics []string
topicList.Convert(&topics)
f.Topics = whisper.TopicsFromString(topics...)
}
2014-12-12 21:24:41 +00:00
2014-12-12 21:38:54 +00:00
return
2014-12-12 21:24:41 +00:00
}