/
blacklists_db.go
100 lines (89 loc) · 2.95 KB
/
blacklists_db.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
package db
import (
"context"
"strings"
log "github.com/sirupsen/logrus"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
)
// BlacklistSettings Flood Control struct for user
type BlacklistSettings struct {
ChatId int64 `bson:"_id,omitempty" json:"_id,omitempty"`
Action string `bson:"action,omitempty" json:"action,omitempty"`
Triggers []string `bson:"triggers,omitempty" json:"triggers,omitempty"`
Reason string `bson:"reason,omitempty" json:"reason,omitempty"`
}
// check Chat Blacklists Settings, used to get data before performing any operation
func checkBlacklistSetting(chatID int64) (blSrc *BlacklistSettings) {
defaultBlacklistSrc := &BlacklistSettings{
ChatId: chatID,
Action: "none",
Triggers: make([]string, 0),
Reason: "Automated Blacklisted word %s",
}
errS := findOne(blacklistsColl, bson.M{"_id": chatID}).Decode(&blSrc)
if errS == mongo.ErrNoDocuments {
blSrc = defaultBlacklistSrc
err := updateOne(blacklistsColl, bson.M{"_id": chatID}, defaultBlacklistSrc)
if err != nil {
log.Errorf("[Database][GetBlacklistSettings]: %v ", err)
}
} else if errS != nil {
log.Errorf("[Database][GetBlacklistSettings]: %v - %d", errS, chatID)
blSrc = defaultBlacklistSrc
}
return blSrc
}
func AddBlacklist(chatId int64, trigger string) {
blSrc := checkBlacklistSetting(chatId)
blSrc.Triggers = append(blSrc.Triggers, strings.ToLower(trigger))
err := updateOne(blacklistsColl, bson.M{"_id": chatId}, blSrc)
if err != nil {
log.Errorf("[Database] AddBlacklist: %v - %d", err, chatId)
}
}
func RemoveBlacklist(chatId int64, trigger string) {
blSrc := checkBlacklistSetting(chatId)
blSrc.Triggers = removeStrfromStr(blSrc.Triggers, strings.ToLower(trigger))
err := updateOne(blacklistsColl, bson.M{"_id": chatId}, blSrc)
if err != nil {
log.Errorf("[Database] RemoveBlacklist: %v - %d", err, chatId)
}
}
func RemoveAllBlacklist(chatId int64) {
blSrc := checkBlacklistSetting(chatId)
blSrc.Triggers = make([]string, 0)
err := updateOne(blacklistsColl, bson.M{"_id": chatId}, blSrc)
if err != nil {
log.Errorf("[Database] RemoveBlacklist: %v - %d", err, chatId)
}
}
func SetBlacklistAction(chatId int64, action string) {
blSrc := checkBlacklistSetting(chatId)
blSrc.Action = strings.ToLower(action)
err := updateOne(blacklistsColl, bson.M{"_id": chatId}, blSrc)
if err != nil {
log.Errorf("[Database] ChangeBlacklistAction: %v - %d", err, chatId)
}
}
func GetBlacklistSettings(chatId int64) *BlacklistSettings {
return checkBlacklistSetting(chatId)
}
func LoadBlacklistsStats() (blacklistTriggers, blacklistChats int64) {
var BlacklistStriuct []*BlacklistSettings
cursor := findAll(blacklistsColl, bson.M{})
defer func(cursor *mongo.Cursor, ctx context.Context) {
err := cursor.Close(ctx)
if err != nil {
log.Error(err)
}
}(cursor, bgCtx)
for _, i := range BlacklistStriuct {
lenBl := len(i.Triggers)
blacklistTriggers += int64(lenBl)
if lenBl > 0 {
blacklistChats++
}
}
return
}