Rewrite handlers management
Having everything under the one hood was not working. Doing this we can parse more information in a neater fashion
This commit is contained in:
parent
d63af22da1
commit
ce641cb233
@ -29,13 +29,13 @@ func main() {
|
|||||||
Token: *pageToken,
|
Token: *pageToken,
|
||||||
})
|
})
|
||||||
|
|
||||||
m.Handle(messenger.TextAction, func(m messenger.Message, r *messenger.Response) {
|
m.HandleMessage(func(m messenger.Message, r *messenger.Response) {
|
||||||
fmt.Printf("%v (Sent, %v)\n", m.Text, m.Time.Format(time.UnixDate))
|
fmt.Printf("%v (Sent, %v)\n", m.Text, m.Time.Format(time.UnixDate))
|
||||||
fmt.Println(r.Text("Hello, World!"))
|
fmt.Println(r.Text("Hello, World!"))
|
||||||
})
|
})
|
||||||
|
|
||||||
m.Handle(messenger.DeliveryAction, func(m messenger.Message, r *messenger.Response) {
|
m.HandleDelivery(func(d messenger.Delivery, r *messenger.Response) {
|
||||||
fmt.Println(m.Delivery.Watermark.Format(time.UnixDate))
|
fmt.Println(d.Watermark.Format(time.UnixDate))
|
||||||
})
|
})
|
||||||
|
|
||||||
fmt.Println("Serving messenger bot on localhost:8080")
|
fmt.Println("Serving messenger bot on localhost:8080")
|
||||||
|
@ -8,7 +8,6 @@ type Message struct {
|
|||||||
Time time.Time
|
Time time.Time
|
||||||
Text string
|
Text string
|
||||||
Seq int
|
Seq int
|
||||||
Delivery *Delivery
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type Delivery struct {
|
type Delivery struct {
|
||||||
|
52
messenger.go
52
messenger.go
@ -18,17 +18,18 @@ type MessengerOptions struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type MessageHandler func(Message, *Response)
|
type MessageHandler func(Message, *Response)
|
||||||
|
type DeliveryHandler func(Delivery, *Response)
|
||||||
|
|
||||||
type Messenger struct {
|
type Messenger struct {
|
||||||
mux *http.ServeMux
|
mux *http.ServeMux
|
||||||
handlers map[Action]MessageHandler
|
messageHandlers []MessageHandler
|
||||||
|
deliveryHandlers []DeliveryHandler
|
||||||
token string
|
token string
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(mo MessengerOptions) *Messenger {
|
func New(mo MessengerOptions) *Messenger {
|
||||||
m := &Messenger{
|
m := &Messenger{
|
||||||
mux: http.NewServeMux(),
|
mux: http.NewServeMux(),
|
||||||
handlers: make(map[Action]MessageHandler),
|
|
||||||
token: mo.Token,
|
token: mo.Token,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -41,8 +42,12 @@ func New(mo MessengerOptions) *Messenger {
|
|||||||
return m
|
return m
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Messenger) Handle(a Action, f MessageHandler) {
|
func (m *Messenger) HandleMessage(f MessageHandler) {
|
||||||
m.handlers[a] = f
|
m.messageHandlers = append(m.messageHandlers, f)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Messenger) HandleDelivery(f DeliveryHandler) {
|
||||||
|
m.deliveryHandlers = append(m.deliveryHandlers, f)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Messenger) Handler() http.Handler {
|
func (m *Messenger) Handler() http.Handler {
|
||||||
@ -78,30 +83,29 @@ func (m *Messenger) dispatch(r Receive) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if f := m.handlers[a]; f != nil {
|
resp := &Response{
|
||||||
message := Message{
|
|
||||||
Sender: info.Sender,
|
|
||||||
Recipient: info.Recipient,
|
|
||||||
Time: time.Unix(info.Timestamp, 0),
|
|
||||||
}
|
|
||||||
|
|
||||||
switch a {
|
|
||||||
case TextAction:
|
|
||||||
message.Text = info.Message.Text
|
|
||||||
case DeliveryAction:
|
|
||||||
message.Delivery = &Delivery{
|
|
||||||
Mids: info.Delivery.Mids,
|
|
||||||
Seq: info.Delivery.Seq,
|
|
||||||
Watermark: time.Unix(info.Delivery.Watermark, 0),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
response := &Response{
|
|
||||||
to: Recipient{info.Sender.ID},
|
to: Recipient{info.Sender.ID},
|
||||||
token: m.token,
|
token: m.token,
|
||||||
}
|
}
|
||||||
|
|
||||||
f(message, response)
|
switch a {
|
||||||
|
case TextAction:
|
||||||
|
for _, f := range m.messageHandlers {
|
||||||
|
f(Message{
|
||||||
|
Sender: info.Sender,
|
||||||
|
Recipient: info.Recipient,
|
||||||
|
Time: time.Unix(info.Timestamp, 0),
|
||||||
|
Text: info.Message.Text,
|
||||||
|
}, resp)
|
||||||
|
}
|
||||||
|
case DeliveryAction:
|
||||||
|
for _, f := range m.deliveryHandlers {
|
||||||
|
f(Delivery{
|
||||||
|
Mids: info.Delivery.Mids,
|
||||||
|
Seq: info.Delivery.Seq,
|
||||||
|
Watermark: time.Unix(info.Delivery.Watermark, 0),
|
||||||
|
}, resp)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user