/
middleware.go
72 lines (64 loc) · 1.49 KB
/
middleware.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
package tgtest
import (
"github.com/go-faster/errors"
"github.com/gotd/td/bin"
"github.com/gotd/td/tg"
)
// UnpackInvoke is a simple Handler middleware to unpack some Invoke*-like requests.
// Including:
//
// tg.InvokeWithLayerRequest
// tg.InitConnectionRequest
// tg.InvokeWithoutUpdatesRequest
func UnpackInvoke(next Handler) Handler {
return HandlerFunc(func(srv *Server, req *Request) error {
id, err := req.Buf.PeekID()
if err != nil {
return err
}
// TODO(tdakkota): handle more Invoke* requests.
var (
obj = peekIDObject{}
r bin.Decoder
)
for {
switch id {
case tg.InvokeWithLayerRequestTypeID:
r = &tg.InvokeWithLayerRequest{
Query: &obj,
}
// TODO(tdakkota): pass Layer to session.
case tg.InitConnectionRequestTypeID:
r = &tg.InitConnectionRequest{
Query: &obj,
}
// TODO(tdakkota): pass DeviceInfo to session.
case tg.InvokeWithoutUpdatesRequestTypeID:
r = &tg.InvokeWithoutUpdatesRequest{
Query: &obj,
}
// TODO(tdakkota): pass NoUpdates flag to session.
default:
return next.OnMessage(srv, req)
}
if err := r.Decode(req.Buf); err != nil {
return err
}
id = obj.TypeID
}
})
}
type peekIDObject struct {
TypeID uint32
}
func (t *peekIDObject) Decode(b *bin.Buffer) error {
id, err := b.PeekID()
if err != nil {
return errors.Wrap(err, "peek id")
}
t.TypeID = id
return nil
}
func (t *peekIDObject) Encode(*bin.Buffer) error {
return errors.New("peekIDObject must not be encoded")
}