61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package geth
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
whisper "github.com/ethereum/go-ethereum/whisper/whisperv2"
|
|
)
|
|
|
|
var (
|
|
whisperFilters []int
|
|
)
|
|
|
|
func onWhisperMessage(message *whisper.Message) {
|
|
SendSignal(SignalEnvelope{
|
|
Type: "whisper",
|
|
Event: WhisperMessageEvent{
|
|
Payload: string(message.Payload),
|
|
From: common.ToHex(crypto.FromECDSAPub(message.Recover())),
|
|
To: common.ToHex(crypto.FromECDSAPub(message.To)),
|
|
Sent: message.Sent.Unix(),
|
|
TTL: int64(message.TTL / time.Second),
|
|
Hash: common.ToHex(message.Hash.Bytes()),
|
|
},
|
|
})
|
|
}
|
|
|
|
func AddWhisperFilter(args whisper.NewFilterArgs) int {
|
|
whisperService, err := NodeManagerInstance().WhisperService()
|
|
if err != nil {
|
|
return -1
|
|
}
|
|
|
|
filter := whisper.Filter{
|
|
To: crypto.ToECDSAPub(common.FromHex(args.To)),
|
|
From: crypto.ToECDSAPub(common.FromHex(args.From)),
|
|
Topics: whisper.NewFilterTopics(args.Topics...),
|
|
Fn: onWhisperMessage,
|
|
}
|
|
|
|
id := whisperService.Watch(filter)
|
|
whisperFilters = append(whisperFilters, id)
|
|
return id
|
|
}
|
|
|
|
func RemoveWhisperFilter(idFilter int) {
|
|
whisperService, err := NodeManagerInstance().WhisperService()
|
|
if err != nil {
|
|
return
|
|
}
|
|
whisperService.Unwatch(idFilter)
|
|
}
|
|
|
|
func ClearWhisperFilters() {
|
|
for _, idFilter := range whisperFilters {
|
|
RemoveWhisperFilter(idFilter)
|
|
}
|
|
whisperFilters = nil
|
|
}
|