-
Notifications
You must be signed in to change notification settings - Fork 0
/
commands.go
126 lines (108 loc) · 2.79 KB
/
commands.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package telbot
import (
"errors"
"fmt"
"github.com/amin1024/xtelbot/conf"
"github.com/amin1024/xtelbot/core"
"github.com/amin1024/xtelbot/core/e"
"go.uber.org/zap"
tele "gopkg.in/telebot.v3"
"os"
"time"
)
// -----------------------------------------------------------------
func UserIdMiddleware(next tele.HandlerFunc) tele.HandlerFunc {
return func(ctx tele.Context) error {
u := ctx.Sender()
if u != nil {
ctx.Set("tid", uint64(u.ID))
} else {
ctx.Set("tid", 0)
}
return nil
}
}
func NewBotHandler() *BotHandler {
log := conf.NewLogger()
log.Info("Creating new bot")
token := os.Getenv("BOT_TOKEN")
if token == "" {
log.Fatal("BOT_TOKEN env variable not found")
}
pref := tele.Settings{
Token: token,
Poller: &tele.LongPoller{Timeout: 10 * time.Second},
}
bot, err := tele.NewBot(pref)
if err != nil {
log.Fatal(err)
return nil
}
//bot.Use(UserIdMiddleware)
h := BotHandler{
bot: bot,
userService: core.NewUserService(),
log: log,
}
bot.Handle("/hello", h.Hi)
bot.Handle("/start", h.Register)
bot.Handle("/usage", h.TrafficUsage)
bot.Handle("/sub", h.Sub)
return &h
}
// -----------------------------------------------------------------
type BotHandler struct {
bot *tele.Bot
userService *core.UserService
log *zap.SugaredLogger
}
func (b *BotHandler) Start() {
b.log.Info("Starting the telegram-bot")
b.bot.Start()
}
func (b *BotHandler) Hi(c tele.Context) error {
b.log.Info("Received: /Hello")
return c.Send("Hi bitch!")
}
func (b *BotHandler) Register(c tele.Context) error {
b.log.Info("Received: /start")
tid := uint64(c.Sender().ID)
username := c.Sender().Username
// Check if user is already registered
_, err := b.userService.Status(tid)
if err == nil {
return c.Send(msgAlreadyRegistered)
}
if !errors.Is(err, e.UserNotFound) { // Any error other than UserNotFound considered as 5xx
return c.Send(msgWtf)
}
// Register the user on bot and every available panel
err = b.userService.Register(tid, username, "")
if err != nil {
return c.Send(msgRegistrationFailed)
}
return c.Send(msgRegistrationSuccess)
}
func (b *BotHandler) TrafficUsage(c tele.Context) error {
tid := uint64(c.Sender().ID)
user, err := b.userService.Status(tid)
if errors.Is(err, e.UserNotFound) {
return c.Send(msgNotRegisteredYet)
}
if errors.Is(err, e.BaseError) {
return c.Send(msgWtf)
}
remaining := user.R.Package.TrafficAllowed - user.TrafficUsage
return c.Send(fmt.Sprintf(msgTraffic, user.TrafficUsage, remaining))
}
func (b *BotHandler) Sub(c tele.Context) error {
tid := uint64(c.Sender().ID)
user, err := b.userService.Status(tid)
if errors.Is(err, e.UserNotFound) {
return c.Send(msgNotRegisteredYet)
}
if errors.Is(err, e.BaseError) {
return c.Send(msgWtf)
}
return c.Send(fmt.Sprintf(msgTmpSubLink, user.UUID))
}