/
dialog-alert-invite.go
81 lines (65 loc) · 2 KB
/
dialog-alert-invite.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
package telegram
import (
"database/sql"
"fmt"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/paulheg/alaaarm/pkg/dialog"
"github.com/paulheg/alaaarm/pkg/models"
"github.com/skip2/go-qrcode"
)
func (t *Telegram) newAlertInviteDialog() *dialog.Dialog {
const GROUP_CONTEXT_KEY = "Group"
return t.newSelectAlertDialog().
Append(t.newSelectDialog("Group", "Private", "group_private_question",
// on group
func(u Update, ctx dialog.ValueStore) (dialog.Status, error) {
ctx.Set(GROUP_CONTEXT_KEY, true)
return dialog.Next, nil
},
func(u Update, ctx dialog.ValueStore) (dialog.Status, error) {
ctx.Set(GROUP_CONTEXT_KEY, false)
return dialog.Next, nil
})).
Chain(failable(func(u Update, ctx dialog.ValueStore) (dialog.Status, error) {
alert, ok := ctx.Value(ALERT_CONTEXT_KEY).(models.Alert)
if !ok {
return dialog.Reset, errContextDataMissing
}
groupInvite, ok := ctx.Value(GROUP_CONTEXT_KEY).(bool)
if !ok {
return dialog.Reset, errContextDataMissing
}
invite, err := t.repository.GetInviteByAlertID(alert.ID)
if err == sql.ErrNoRows {
invite, err = t.repository.CreateInvite(alert)
if err != nil {
return dialog.Reset, err
}
} else if err != nil {
return dialog.Reset, err
}
inviteURL := t.invitationURL(invite, groupInvite)
qrBytes, err := qrcode.Encode(inviteURL, qrcode.Low, 200)
if err != nil {
return dialog.Reset, err
}
text, _ := u.Dictionary.Lookup("invitation_link_generated")
msg := tgbotapi.NewPhoto(u.ChatID, tgbotapi.FileBytes{
Name: "invite_qr.png",
Bytes: qrBytes,
})
msg.Caption = fmt.Sprintf(text, inviteURL)
_, err = t.bot.Send(msg)
if err != nil {
return dialog.Reset, err
}
return dialog.Success, nil
}))
}
func (t *Telegram) invitationURL(invite models.Invite, group bool) string {
param := "start"
if group {
param = "startgroup"
}
return fmt.Sprintf("https://t.me/%s?%s=%s", t.bot.Self.UserName, param, invite.Token)
}