Add Telegram bot
This commit is contained in:
parent
dd64dd75ac
commit
274a09390a
72
bot.own
Normal file
72
bot.own
Normal file
@ -0,0 +1,72 @@
|
||||
use std, functional, types, regex
|
||||
|
||||
include "own-modules/telegram-bot/TelegramBot.own"
|
||||
include "config.own"
|
||||
include "DanbooruApi.own"
|
||||
|
||||
DANBOORU_REGEX = regex("https://danbooru[^0-9 ]+([0-9]{5,9})")
|
||||
bot = new TelegramBot(config.telegram.token)
|
||||
api = new DanbooruApi(config.danbooru.auth)
|
||||
|
||||
lastUpdateId = -1
|
||||
while true {
|
||||
sleep(6200)
|
||||
updates = bot.getUpdatesSync(lastUpdateId + 1, 50, 58)
|
||||
if (updates.isEmpty()) continue
|
||||
|
||||
lastUpdateId = stream(updates)
|
||||
.map(::processUpdate)
|
||||
.reduce(0, def(a, b) = a + b)
|
||||
}
|
||||
// Confirm updates
|
||||
bot.getUpdatesSync(lastUpdateId + 1, 1, 10)
|
||||
thread(def() {
|
||||
sleep(2000)
|
||||
exit(0)
|
||||
})
|
||||
|
||||
|
||||
def processUpdate(update) {
|
||||
if !arrayKeyExists("message", update) {
|
||||
return update.update_id
|
||||
}
|
||||
msg = update.message
|
||||
text = msg.text ?? ""
|
||||
if text.isEmpty() {
|
||||
return update.update_id
|
||||
}
|
||||
m = DANBOORU_REGEX.matcher(text)
|
||||
while m.find() {
|
||||
processPost(m.group(1), msg.chat.id, msg.message_id)
|
||||
}
|
||||
return update.update_id
|
||||
}
|
||||
|
||||
def processPost(id, chatId, msgId) {
|
||||
post = api.getPost(id)
|
||||
println "Got post: " + post
|
||||
if !post.isImage() return 0
|
||||
blacklistTags = config.danbooru.blacklistTags
|
||||
if post.containsTags(blacklistTags) return 0
|
||||
|
||||
caption = getCaption(post)
|
||||
reply = jsonencode({"message_id": msgId, "chat_id": chatId})
|
||||
bot.invoke("sendPhoto", {
|
||||
"chat_id": chatId,
|
||||
"reply_parameters": reply,
|
||||
"photo": post.getImageUrl(),
|
||||
"parse_mode": "html",
|
||||
"caption": caption,
|
||||
}, def(r) {
|
||||
if !r.contains("wrong file identifier") return 1
|
||||
bot.invoke("sendPhoto", {
|
||||
"chat_id": chatId,
|
||||
"reply_parameters": reply,
|
||||
"photo": post.getSampleImageUrl(),
|
||||
"parse_mode": "html",
|
||||
"caption": "small " + caption,
|
||||
}, def(r) = 1)
|
||||
})
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user