add files
astravexton
commit
de8a92c967
|
|
@ -0,0 +1,18 @@
|
|||
FROM golang:alpine AS builder
|
||||
|
||||
WORKDIR /go/src/git.zio.sh/astra/ddgimagebot
|
||||
COPY . .
|
||||
|
||||
RUN apk update && \
|
||||
apk add --no-cache git bash && \
|
||||
go get -d -v ./... && \
|
||||
go install
|
||||
|
||||
FROM alpine:latest
|
||||
|
||||
COPY --from=builder /go/bin/ddgimagebot /usr/local/bin/ddgimagebot
|
||||
RUN apk add --no-cache bash
|
||||
|
||||
ENV BOT_TOKEN ""
|
||||
|
||||
CMD ["ddgimagebot"]
|
||||
|
|
@ -0,0 +1,15 @@
|
|||
module git.zio.sh/astra/ddgimagebot
|
||||
|
||||
go 1.19
|
||||
|
||||
require (
|
||||
git.zio.sh/astra/telegram-bot-api v1.0.1-0.20220824135549-c84603be8b0f
|
||||
github.com/astravexton/logger v0.2.2-0.20211216142523-c1e08a465709
|
||||
github.com/dghubble/sling v1.4.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/google/go-querystring v1.1.0 // indirect
|
||||
github.com/google/uuid v1.3.0 // indirect
|
||||
github.com/technoweenie/multipartstreamer v1.0.1 // indirect
|
||||
)
|
||||
|
|
@ -0,0 +1,24 @@
|
|||
git.zio.sh/astra/telegram-bot-api v1.0.1-0.20220824135549-c84603be8b0f h1:HD6LcxkXXFfzWUeYm5YsVNsoUZFVrd2/jgMGgiYWRMI=
|
||||
git.zio.sh/astra/telegram-bot-api v1.0.1-0.20220824135549-c84603be8b0f/go.mod h1:gEnVclR6CKL/hj3SZV2lmQzt4Ul0Chq8xkjBBKzdxOE=
|
||||
git.zio.sh/astra/telegram-bot-api v3.0.0+incompatible h1:OirhOPiMKlH9M8WCILFX69S43pAiAhr9rIxkCY7TVFs=
|
||||
git.zio.sh/astra/telegram-bot-api v3.0.0+incompatible/go.mod h1:kTm5G2y4tw4tlnUps6VTc5QwNEhdgTD+qErZhd+2W0o=
|
||||
github.com/astravexton/logger v0.2.1 h1:nYxhV6oCHIJzUzaPhawbAc51EEvJowIViVsB5EbdeJQ=
|
||||
github.com/astravexton/logger v0.2.1/go.mod h1:6uxVhLlWlvZn48W8F52dsMXt4yztlrInpRhdWiJ9RXc=
|
||||
github.com/astravexton/logger v0.2.2-0.20211216142523-c1e08a465709 h1:S+dBCax7wdg3ikWx5t6cClPKQIerrmD404qAzWhO2V4=
|
||||
github.com/astravexton/logger v0.2.2-0.20211216142523-c1e08a465709/go.mod h1:6BlG5E0U3CImXLmDtGpEwdoc8PHiiDv0kbvJBCbfiMY=
|
||||
github.com/dghubble/sling v1.4.0 h1:/n8MRosVTthvMbwlNZgLx579OGVjUOy3GNEv5BIqAWY=
|
||||
github.com/dghubble/sling v1.4.0/go.mod h1:0r40aNsU9EdDUVBNhfCstAtFgutjgJGYbO1oNzkMoM8=
|
||||
github.com/fatih/color v1.10.0/go.mod h1:ELkj/draVOlAH/xkhN6mQ50Qd0MPOk5AAr3maGEBuJM=
|
||||
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||
github.com/google/go-querystring v1.1.0 h1:AnCroh3fv4ZBgVIf1Iwtovgjaw/GiKJo8M8yD/fhyJ8=
|
||||
github.com/google/go-querystring v1.1.0/go.mod h1:Kcdr2DB4koayq7X8pmAG4sNG59So17icRSOU623lUBU=
|
||||
github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
|
||||
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/kris-nova/lolgopher v0.0.0-20210112022122-73f0047e8b65/go.mod h1:V0HF/ZBlN86HqewcDC/cVxMmYDiRukWjSrgKLUAn9Js=
|
||||
github.com/mattn/go-colorable v0.1.8/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=
|
||||
github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
|
||||
github.com/technoweenie/multipartstreamer v1.0.1 h1:XRztA5MXiR1TIRHxH2uNxXxaIkKQDeX7m2XsSOlQEnM=
|
||||
github.com/technoweenie/multipartstreamer v1.0.1/go.mod h1:jNVxdtShOxzAsukZwTSw6MDx5eUJoiEBsSvzDU9uzog=
|
||||
golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
|
|
@ -0,0 +1,152 @@
|
|||
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
|
||||
}
|
||||
Loading…
Reference in New Issue