153 lines
3.6 KiB
Go
153 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
tgbotapi "git.zio.sh/astra/telegram-bot-api"
|
|
"github.com/astravexton/logger"
|
|
"github.com/dghubble/sling"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const ddgURL = "https://duckduckgo.com"
|
|
|
|
// ImgResult ...
|
|
type ImgResult struct {
|
|
Ads interface{} `json:"ads"`
|
|
Next string `json:"next"`
|
|
Query string `json:"query"`
|
|
QueryEncoded string `json:"queryEncoded"`
|
|
ResponseType string `json:"response_type"`
|
|
Results []Results `json:"results"`
|
|
Vqd struct {
|
|
Testing string `json:"testing"`
|
|
} `json:"vqd"`
|
|
}
|
|
|
|
type Results struct {
|
|
Height int `json:"height"`
|
|
Image string `json:"image"`
|
|
Source string `json:"source"`
|
|
Thumbnail string `json:"thumbnail"`
|
|
Title string `json:"title"`
|
|
URL string `json:"url"`
|
|
Width int `json:"width"`
|
|
}
|
|
|
|
type Params struct {
|
|
Query string `url:"q,omitempty"`
|
|
L string `url:"l,omitempty"`
|
|
O string `url:"o,omitempty"`
|
|
Q string `url:"q,omitempty"`
|
|
VQD string `url:"vqd,omitempty"`
|
|
F string `url:"f,omitempty"`
|
|
P string `url:"p,omitempty"`
|
|
S string `url:"s,omitempty"`
|
|
U string `url:"u,omitempty"`
|
|
}
|
|
|
|
func main() {
|
|
logger.Layout = time.Stamp
|
|
logger.BitwiseLevel = logger.LogEverything
|
|
bot, err := tgbotapi.NewBotAPI(os.Getenv("BOT_TOKEN"))
|
|
if err != nil {
|
|
logger.Critical(err.Error())
|
|
}
|
|
me, err := bot.GetMe()
|
|
if err != nil {
|
|
logger.Critical(err.Error())
|
|
}
|
|
logger.Info("Authorized as %s", me.UserName)
|
|
|
|
u := tgbotapi.NewUpdate(0)
|
|
u.Timeout = 60
|
|
updates := bot.GetUpdatesChan(u)
|
|
|
|
for update := range updates {
|
|
if update.InlineQuery != nil {
|
|
if update.InlineQuery.Query != "" {
|
|
logger.Info("Inline query %q from %s", update.InlineQuery.Query, update.InlineQuery.From)
|
|
var results []interface{}
|
|
|
|
images := getImageList(update.InlineQuery.Query)
|
|
for c, image := range images {
|
|
if c == 50 {
|
|
break
|
|
}
|
|
result := &tgbotapi.InlineQueryResultPhoto{
|
|
Type: "photo",
|
|
ID: uuid.New().String(),
|
|
URL: image.Image,
|
|
ThumbURL: image.Thumbnail,
|
|
}
|
|
results = append(results, result)
|
|
}
|
|
inlineConf := tgbotapi.InlineConfig{
|
|
InlineQueryID: update.InlineQuery.ID,
|
|
IsPersonal: false,
|
|
CacheTime: 1,
|
|
Results: results,
|
|
}
|
|
bot.Request(inlineConf)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
func getImageList(query string) []Results {
|
|
params := &Params{Query: query}
|
|
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.QueryStruct(params)
|
|
|
|
req, err := s.Request()
|
|
if err != nil {
|
|
logger.Critical(err.Error())
|
|
}
|
|
|
|
client := &http.Client{}
|
|
client.Timeout = time.Second * 3
|
|
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
logger.Critical(err.Error())
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
body, _ := io.ReadAll(res.Body)
|
|
bodyContents := string(body)
|
|
|
|
r := regexp.MustCompile(`vqd=([\d-]+)`)
|
|
token := strings.ReplaceAll(r.FindString(bodyContents), "vqd=", "")
|
|
|
|
params = &Params{
|
|
L: "us-en",
|
|
O: "json",
|
|
Q: query,
|
|
VQD: token,
|
|
F: ",,,,,",
|
|
P: "-1",
|
|
S: "0",
|
|
U: "bing",
|
|
}
|
|
s = s.Get(ddgURL).Path("/i.js").QueryStruct(params)
|
|
s.Set("Authority", "duckduckgo.com")
|
|
s.Set("Accept", "application/json, text/javascript, */*; q=0.01")
|
|
s.Set("Cookie", "bc=1; at=-1; au=-1; av=1; ax=-1; ak=-1; ao=-1; ap=-1; 1=-1; 5=1; aq=-1; ay=b; ah=uk-en; l=wt-wt; p=-2")
|
|
s.Set("Referer", "https://duckduckgo.com/")
|
|
|
|
imageList := &ImgResult{}
|
|
_, err = s.ReceiveSuccess(imageList)
|
|
if err != nil {
|
|
logger.Critical(err.Error())
|
|
}
|
|
|
|
return imageList.Results
|
|
}
|