-
Notifications
You must be signed in to change notification settings - Fork 62
/
dam.go
98 lines (80 loc) · 1.99 KB
/
dam.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
// SPDX-License-Identifier: AGPL-3.0-only
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, version 3.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>
package dam
import (
"regexp"
"strings"
"unicode"
"github.com/trim21/errgo"
"github.com/bangumi/server/config"
)
type Dam struct {
nsfwWord *regexp.Regexp
disableWord *regexp.Regexp
bannedDomain *regexp.Regexp
}
func New(c config.AppConfig) (Dam, error) {
var cc Dam
var err error
if c.NsfwWord != "" {
cc.nsfwWord, err = regexp.Compile(c.NsfwWord)
if err != nil {
return Dam{}, errgo.Wrap(err, "nsfw_word")
}
}
if c.DisableWords != "" {
cc.disableWord, err = regexp.Compile(c.DisableWords)
if err != nil {
return Dam{}, errgo.Wrap(err, "disable_words")
}
}
if c.BannedDomain != "" {
cc.bannedDomain, err = regexp.Compile(c.BannedDomain)
if err != nil {
return Dam{}, errgo.Wrap(err, "banned_domain")
}
}
return cc, nil
}
func (d Dam) NeedReview(text string) bool {
if text == "" {
return false
}
if d.disableWord == nil {
return false
}
text = strings.ToLower(text)
return d.disableWord.MatchString(text)
}
func (d Dam) CensoredWords(text string) bool {
if d.disableWord != nil && d.disableWord.MatchString(text) {
return true
}
if d.bannedDomain != nil && d.bannedDomain.MatchString(text) {
return true
}
return false
}
func AllPrintableChar(text string) bool {
for _, c := range text {
switch c {
case '\n', '\t':
continue
}
if !unicode.IsPrint(c) {
return false
}
}
return true
}