-
Notifications
You must be signed in to change notification settings - Fork 4
/
subscribe.go
112 lines (95 loc) · 2.53 KB
/
subscribe.go
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
package main
import (
"encoding/json"
"github.com/gorilla/websocket"
log "github.com/sirupsen/logrus"
"github.com/urfave/cli/v2"
neutrinodtypes "github.com/vulpemventures/neutrino-elements/pkg/neutrinod-types"
)
var (
emptyOnChainMsg = neutrinodtypes.OnChainEventResponse{}
emptyGeneralMsg = neutrinodtypes.GeneralMessageResponse{}
emptyErrorMsg = neutrinodtypes.MessageErrorResponse{}
)
var subscribeCmd = cli.Command{
Name: "subscribe",
Usage: "subscribes to neutrinod events related to provided wallet descriptor",
Action: subscribeAction,
Flags: []cli.Flag{
&cli.StringFlag{
Name: "descriptor",
Usage: "wallet descriptor",
Required: true,
},
&cli.IntFlag{
Name: "block_height",
Usage: "block height to watch from",
Required: true,
},
&cli.StringSliceFlag{
Name: "events",
Usage: "events to watch for:\n" +
" unspentUtxo -> unspent utxo\n" +
" spentUtxo -> spent utxo\n",
},
},
}
func subscribeAction(ctx *cli.Context) error {
conn, cleanup, err := getNeutrinodConnection()
if err != nil {
return err
}
defer cleanup()
descriptor := ctx.String("descriptor")
blockHeight := ctx.Int("block_height")
eventsType := ctx.StringSlice("events")
events := make([]neutrinodtypes.EventType, 0, len(eventsType))
for _, v := range eventsType {
events = append(events, neutrinodtypes.EventType(v))
}
req := neutrinodtypes.SubscriptionRequestWs{
ActionType: neutrinodtypes.Register,
EventTypes: events,
DescriptorWallet: descriptor,
StartBlockHeight: blockHeight,
}
reqBytes, err := json.Marshal(req)
if err != nil {
return err
}
if err := conn.WriteMessage(websocket.TextMessage, reqBytes); err != nil {
return err
}
for {
_, message, err := conn.ReadMessage()
if err != nil {
return err
}
onChainMsg := neutrinodtypes.OnChainEventResponse{}
if err := json.Unmarshal(message, &onChainMsg); err != nil {
log.Error(err.Error())
return err
}
generalMsg := neutrinodtypes.GeneralMessageResponse{}
if err := json.Unmarshal(message, &generalMsg); err != nil {
log.Error(err.Error())
return err
}
errorMsg := neutrinodtypes.MessageErrorResponse{}
if err := json.Unmarshal(message, &errorMsg); err != nil {
log.Error(err.Error())
return err
}
if onChainMsg != emptyOnChainMsg {
if onChainMsg.TxID != "" {
log.Infof("tx_id: %v", onChainMsg.TxID)
}
}
if generalMsg != emptyGeneralMsg {
log.Infoln(generalMsg.Message)
}
if errorMsg != emptyErrorMsg {
log.Errorf("error: %v", errorMsg.ErrorMessage)
}
}
}