-
Notifications
You must be signed in to change notification settings - Fork 0
/
kuji_bot.go
138 lines (105 loc) · 2.88 KB
/
kuji_bot.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
package main
import (
"context"
"encoding/json"
"fmt"
"io/ioutil"
"log"
"os"
"time"
"github.com/antchfx/htmlquery"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// Show struct contains show id
type Show struct {
Link string
}
// SensitiveData struct contains all secret keys for proxy and Telegram
type SensitiveData struct {
BotToken string `json:"token"`
User string `json:"user"`
Pass string `json:"pass"`
Port int `json:"port"`
Host string `json:"host"`
MongoURI string `json:"mongostring"`
MongoDB string `json:"mongodb"`
MongoCollection string `json:"mongocollection"`
}
func createMongoConnection(mongoURI string) *mongo.Client {
clientOptions := options.Client().ApplyURI(mongoURI)
client, err := mongo.Connect(context.TODO(), clientOptions)
if err != nil {
log.Fatal(err)
}
err = client.Ping(context.TODO(), nil)
if err != nil {
log.Fatal(err)
}
return client
}
func botWork(token string) *tgbotapi.BotAPI {
bot, err := tgbotapi.NewBotAPI(token)
if err != nil {
log.Panic(err)
}
bot.Debug = true
return bot
}
func getSensitiveData(jsonPath string) SensitiveData {
jsonFile, err := os.Open(jsonPath)
if err != nil {
log.Fatal(err)
}
defer jsonFile.Close()
byteValue, _ := ioutil.ReadAll(jsonFile)
var data SensitiveData
json.Unmarshal(byteValue, &data)
return data
}
func check(url string, id int64, bot *tgbotapi.BotAPI, collection *mongo.Collection) {
doc, err := htmlquery.LoadURL(url)
if err != nil {
panic(err)
}
for _, n := range htmlquery.Find(doc, "//div[contains(@class, 'bgimg_first_hover')]") {
a := htmlquery.FindOne(n, "//div[contains(@class, 'bgimg_first_hover')]")
showImageLink := htmlquery.SelectAttr(a, "data-original")
filter := bson.D{{"link", showImageLink}}
var result Show
err = collection.FindOne(context.TODO(), filter).Decode(&result)
if err != nil {
text := fmt.Sprintf("New show: %s", showImageLink)
msg := tgbotapi.NewMessage(id, text)
bot.Send(msg)
show := Show{
showImageLink}
_, err := collection.InsertOne(context.TODO(), show)
if err != nil {
log.Fatal(err)
}
}
}
}
func main() {
var id int64 = -388648621
var dataPath string
if _, err := os.Stat("local_sensitive_data.json"); err == nil {
dataPath = "local_sensitive_data.json"
} else if os.IsNotExist(err) {
dataPath = "sensitive_data.json"
}
data := getSensitiveData(dataPath)
bot := botWork(data.BotToken)
mongoClient := createMongoConnection(data.MongoURI)
collection := mongoClient.Database(data.MongoDB).Collection(data.MongoCollection)
for {
for i := 1; i < 7; i++ {
url := fmt.Sprintf("https://standupstore.ru/page/%d/", i)
check(url, id, bot, collection)
}
time.Sleep(time.Second * 55)
}
}