-
-
Notifications
You must be signed in to change notification settings - Fork 118
/
config.go
72 lines (60 loc) · 1.66 KB
/
config.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 config contains config service implementation for tgtest server.
package config
import (
"context"
"github.com/gotd/td/bin"
"github.com/gotd/td/tg"
"github.com/gotd/td/tgtest"
"github.com/gotd/td/tgtest/services"
)
// Service is a Telegram config service.
type Service struct {
cfg *tg.Config
cdnCfg *tg.CDNConfig
}
// NewService creates new Service.
func NewService(cfg *tg.Config, cdnCfg *tg.CDNConfig) *Service {
return &Service{cfg: cfg, cdnCfg: cdnCfg}
}
func (c *Service) HelpGetCDNConfig(ctx context.Context, req *tg.HelpGetCDNConfigRequest) (*tg.CDNConfig, error) {
cfg := c.cdnCfg
return cfg, nil
}
func (c *Service) HelpGetConfig(ctx context.Context, dc int, req *tg.HelpGetConfigRequest) (*tg.Config, error) {
cfg := *c.cfg
cfg.ThisDC = dc
return &cfg, nil
}
// OnMessage implements tgtest.Handler.
func (c *Service) OnMessage(server *tgtest.Server, req *tgtest.Request) error {
id, err := req.Buf.PeekID()
if err != nil {
return err
}
var (
decode bin.Decoder
result bin.Encoder
)
switch id {
case tg.HelpGetCDNConfigRequestTypeID:
cfg := c.cdnCfg
decode = &tg.HelpGetCDNConfigRequest{}
result = cfg
case tg.HelpGetConfigRequestTypeID:
cfg := *c.cfg
cfg.ThisDC = req.DC
decode = &tg.HelpGetConfigRequest{}
result = &cfg
default:
return services.ErrMethodNotImplemented
}
if err := decode.Decode(req.Buf); err != nil {
return err
}
return server.SendResult(req, result)
}
// Register registers service handlers.
func (c *Service) Register(dispatcher *tgtest.Dispatcher) {
dispatcher.HandleFunc(tg.HelpGetCDNConfigRequestTypeID, c.OnMessage)
dispatcher.HandleFunc(tg.HelpGetConfigRequestTypeID, c.OnMessage)
}