-
Notifications
You must be signed in to change notification settings - Fork 0
/
start_command.go
106 lines (89 loc) · 2.72 KB
/
start_command.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
package command
import (
_ "embed"
"fmt"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/joomcode/errorx"
"github.com/mih-kopylov/our-spb-bot/internal/bot"
"github.com/mih-kopylov/our-spb-bot/internal/bot/service"
"github.com/mih-kopylov/our-spb-bot/internal/category"
"github.com/mih-kopylov/our-spb-bot/internal/info"
"github.com/mih-kopylov/our-spb-bot/internal/state"
"strings"
"text/template"
)
//go:embed start.html
var startTextTemplate string
const (
StartCommandName = "start"
)
type StartCommand struct {
states state.States
service *service.Service
info *info.Info
}
func NewStartCommand(states state.States, service *service.Service, info *info.Info) bot.Command {
return &StartCommand{
states: states,
service: service,
info: info,
}
}
func (c *StartCommand) Name() string {
return StartCommandName
}
func (c *StartCommand) Description() string {
return "Запустить бота"
}
func (c *StartCommand) Handle(message *tgbotapi.Message) error {
userState, err := c.states.GetState(message.Chat.ID)
if err != nil {
if errorx.IsOfType(err, state.ErrRateLimited) {
err = c.service.SendMessage(message.Chat, "Превышен лимит подключений к базе данных")
if err != nil {
return errorx.EnhanceStackTrace(err, "failed to send reply")
}
return nil
}
return errorx.EnhanceStackTrace(err, "failed to get user state")
}
if message.Chat.IsPrivate() {
userState.FullName = strings.TrimSpace(fmt.Sprintf("user / @%v %v %v", message.Chat.UserName, message.Chat.FirstName, message.Chat.LastName))
} else {
userState.FullName = strings.TrimSpace(fmt.Sprintf("%v / %v",
message.Chat.Type, message.Chat.Title))
}
userState.MessageHandlerName = ""
if userState.Categories == "" {
userState.Categories = string(category.DefaultCategoriesText)
}
err = c.states.SetState(userState)
if err != nil {
return errorx.EnhanceStackTrace(err, "failed to set user state")
}
parsedTemplate, err := template.New("start").Parse(startTextTemplate)
if err != nil {
return errorx.EnhanceStackTrace(err, "failed to parse template")
}
context := renderContext{
Version: c.info.Version,
Commit: c.info.Commit,
}
writer := strings.Builder{}
err = parsedTemplate.Execute(&writer, context)
if err != nil {
return errorx.EnhanceStackTrace(err, "failed to render template")
}
_, err = c.service.SendMessageCustom(message.Chat, writer.String(), func(reply *tgbotapi.MessageConfig) {
reply.ParseMode = tgbotapi.ModeHTML
reply.ReplyMarkup = tgbotapi.NewRemoveKeyboard(false)
})
if err != nil {
return errorx.EnhanceStackTrace(err, "failed to send reply")
}
return nil
}
type renderContext struct {
Version string
Commit string
}