aboutsummaryrefslogtreecommitdiffstats
path: root/ui/qt/qwhisper/whisper.go
blob: 8f05c06953ae37fe973d9d50d97a478aef082982 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package qwhisper

import (
    "fmt"
    "time"
    "unsafe"

    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/ethutil"
    "github.com/ethereum/go-ethereum/whisper"
    "gopkg.in/qml.v1"
)

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) }

type Watch struct {
}

func (self *Watch) Arrived(v unsafe.Pointer) {
    fmt.Println(v)
}

type Whisper struct {
    *whisper.Whisper
    view qml.Object

    watches map[int]*Watch
}

func New(w *whisper.Whisper) *Whisper {
    return &Whisper{w, nil, make(map[int]*Watch)}
}

func (self *Whisper) SetView(view qml.Object) {
    self.view = view
}

func (self *Whisper) Post(data string, to, from string, topics []string, pow, ttl uint32) {
    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)),
        Topics: whisper.TopicsFromString(topics...),
    })
    if err != nil {
        fmt.Println(err)
        // handle error
        return
    }

    if err := self.Whisper.Send(envelope); err != nil {
        fmt.Println(err)
        // handle error
        return
    }
}

func (self *Whisper) NewIdentity() string {
    return toHex(self.Whisper.NewIdentity().D.Bytes())
}

func (self *Whisper) HasIdentity(key string) bool {
    return self.Whisper.HasIdentity(crypto.ToECDSA(fromHex(key)))
}

func (self *Whisper) Watch(opts map[string]interface{}) *Watch {
    filter := filterFromMap(opts)
    filter.Fn = func(msg *whisper.Message) {
        fmt.Println(msg)
    }
    i := self.Whisper.Watch(filter)
    self.watches[i] = &Watch{}

    return self.watches[i]
}

func filterFromMap(opts map[string]interface{}) (f whisper.Filter) {
    if to, ok := opts["to"].(string); ok {
        f.To = crypto.ToECDSA(fromHex(to))
    }
    if from, ok := opts["from"].(string); ok {
        f.From = crypto.ToECDSAPub(fromHex(from))
    }
    if topicList, ok := opts["topics"].(*qml.List); ok {
        var topics []string
        topicList.Convert(&topics)
        f.Topics = whisper.TopicsFromString(topics...)
    }

    return
}