astra
/
ddgimagebot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update log levels, add safe search toggle

main
Astra 2022-10-14 23:33:34 +00:00
parent a3060fe3ad
commit 2035967c0e
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.go
View File

@ -53,7 +53,7 @@ type Params struct {
func main() { func main() {
logger.Layout = time.Stamp logger.Layout = time.Stamp
logger.BitwiseLevel = logger.LogEverything logger.BitwiseLevel = logger.LogInfo | logger.LogWarning | logger.LogCritical
bot, err := tgbotapi.NewBotAPI(os.Getenv("BOT_TOKEN")) bot, err := tgbotapi.NewBotAPI(os.Getenv("BOT_TOKEN"))
if err != nil { if err != nil {
logger.Critical(err.Error()) logger.Critical(err.Error())
@ -71,7 +71,7 @@ func main() {
for update := range updates { for update := range updates {
if update.InlineQuery != nil { if update.InlineQuery != nil {
if update.InlineQuery.Query != "" { if update.InlineQuery.Query != "" {
logger.Info("Inline query %q from %s", update.InlineQuery.Query, update.InlineQuery.From) logger.Debug("Inline query %q from %s", update.InlineQuery.Query, update.InlineQuery.From)
var results []interface{} var results []interface{}
images := getImageList(update.InlineQuery.Query) images := getImageList(update.InlineQuery.Query)
@ -100,6 +100,13 @@ func main() {
} }
func getImageList(query string) []Results { func getImageList(query string) []Results {
safeSearch := "-1" // -1 = safe off, 1 = safe on
if len(query) >= 8 {
if strings.HasPrefix(query, "safe:on") {
safeSearch = "1"
query = query[8:]
}
}
params := &Params{Query: query} params := &Params{Query: query}
s := sling.New().Get(ddgURL) s := sling.New().Get(ddgURL)
s.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.163 Safari/537.36") s.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.163 Safari/537.36")
@ -132,7 +139,7 @@ func getImageList(query string) []Results {
Q: query, Q: query,
VQD: token, VQD: token,
F: ",,,,,", F: ",,,,,",
P: "-1", P: safeSearch,
S: "0", S: "0",
U: "bing", U: "bing",
} }