aboutsummaryrefslogtreecommitdiffstats
path: root/xeth/whisper.go
blob: 72e1ee04f46d7ec74dbda8ebc75a4f7a5353c9eb (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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package xeth

import (
    "errors"
    "time"

    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/logger"
    "github.com/ethereum/go-ethereum/whisper"
)

var qlogger = logger.NewLogger("XSHH")

type Whisper struct {
    *whisper.Whisper
}

func NewWhisper(w *whisper.Whisper) *Whisper {
    return &Whisper{w}
}

func (self *Whisper) Post(payload string, to, from string, topics []string, priority, ttl uint32) error {
    if priority == 0 {
        priority = 1000
    }

    if ttl == 0 {
        ttl = 100
    }

    pk := crypto.ToECDSAPub(common.FromHex(from))
    if key := self.Whisper.GetIdentity(pk); key != nil || len(from) == 0 {
        msg := whisper.NewMessage(common.FromHex(payload))
        envelope, err := msg.Seal(time.Duration(priority*100000), whisper.Opts{
            Ttl:    time.Duration(ttl) * time.Second,
            To:     crypto.ToECDSAPub(common.FromHex(to)),
            From:   key,
            Topics: whisper.TopicsFromString(topics...),
        })

        if err != nil {
            return err
        }

        if err := self.Whisper.Send(envelope); err != nil {
            return err
        }
    } else {
        return errors.New("unmatched pub / priv for seal")
    }

    return nil
}

func (self *Whisper) NewIdentity() string {
    key := self.Whisper.NewIdentity()

    return common.ToHex(crypto.FromECDSAPub(&key.PublicKey))
}

func (self *Whisper) HasIdentity(key string) bool {
    return self.Whisper.HasIdentity(crypto.ToECDSAPub(common.FromHex(key)))
}

// func (self *Whisper) RemoveIdentity(key string) bool {
//  return self.Whisper.RemoveIdentity(crypto.ToECDSAPub(common.FromHex(key)))
// }

func (self *Whisper) Watch(opts *Options) int {
    filter := whisper.Filter{
        To:     crypto.ToECDSAPub(common.FromHex(opts.To)),
        From:   crypto.ToECDSAPub(common.FromHex(opts.From)),
        Topics: whisper.TopicsFromString(opts.Topics...),
    }

    var i int
    filter.Fn = func(msg *whisper.Message) {
        opts.Fn(NewWhisperMessage(msg))
    }

    i = self.Whisper.Watch(filter)

    return i
}

func (self *Whisper) Messages(id int) (messages []WhisperMessage) {
    msgs := self.Whisper.Messages(id)
    messages = make([]WhisperMessage, len(msgs))
    for i, message := range msgs {
        messages[i] = NewWhisperMessage(message)
    }

    return
}

type Options struct {
    To     string
    From   string
    Topics []string
    Fn     func(msg WhisperMessage)
}

type WhisperMessage struct {
    ref     *whisper.Message
    Payload string `json:"payload"`
    To      string `json:"to"`
    From    string `json:"from"`
    Sent    int64  `json:"sent"`
}

func NewWhisperMessage(msg *whisper.Message) WhisperMessage {
    return WhisperMessage{
        ref:     msg,
        Payload: common.ToHex(msg.Payload),
        From:    common.ToHex(crypto.FromECDSAPub(msg.Recover())),
        To:      common.ToHex(crypto.FromECDSAPub(msg.To)),
        Sent:    msg.Sent,
    }
}