85 lines
3 KiB
Nim
85 lines
3 KiB
Nim
import telebot, std/[asyncdispatch, logging, options, strutils, random, with, os], norm/[model, sqlite]
|
|
|
|
type
|
|
CensoredData* = ref object of Model
|
|
ftype*: string
|
|
fhash*: string
|
|
fileid*: string
|
|
caption*: string
|
|
|
|
func NewCensoredData*(ftype = ""; fhash = ""; fileid = ""; caption = ""):
|
|
CensoredData = CensoredData(ftype: ftype, fhash: fhash, fileid: fileid, caption: caption)
|
|
|
|
let dbConn* = sqlite.open("censordata.db", "", "", "")
|
|
|
|
dbConn.createTables(NewCensoredData())
|
|
|
|
var L = newConsoleLogger(fmtStr="$levelname, [$time] ")
|
|
addHandler(L)
|
|
|
|
proc generate_hash(): string=
|
|
result = newString(7)
|
|
const charset = {'a' .. 'z', 'A' .. 'Z', '0' .. '9'}
|
|
for i in 0..6:
|
|
result[i] = sample(charset)
|
|
return result
|
|
|
|
const API_KEY = getEnv("TELEGRAM_TOKEN")
|
|
|
|
proc updateHandler(b: Telebot, u: Update): Future[bool] {.async, gcsafe.} =
|
|
let response = u.message.get
|
|
if response.text.isSome:
|
|
let message = response.text.get
|
|
if message == "/start":
|
|
discard await b.sendMessage(response.chat.id, "Hey, To create a censored post, you can share any album, video, photo, gif, sticker, etc. The messages could then be forwarded to any chat for them to view")
|
|
elif message.contains("/start"):
|
|
let deeplink = message.split(" ")
|
|
var TempData = @[NewCensoredData()]
|
|
dbConn.select(TempData, "fhash = ?", deeplink[1])
|
|
for i in TempData:
|
|
if i.ftype == "photo":
|
|
discard await b.sendPhoto(response.chat.id, i.fileid, i.caption)
|
|
if i.ftype == "document":
|
|
discard await b.sendDocument(response.chat.id, i.fileid, i.caption)
|
|
if i.ftype == "video":
|
|
discard await b.sendVideo(response.chat.id, i.fileid, caption=i.caption)
|
|
if i.ftype == "videonote":
|
|
discard await b.sendVideoNote(response.chat.id, i.fileid)
|
|
if i.ftype == "animation":
|
|
discard await b.sendAnimation(response.chat.id, i.fileid, caption=i.caption)
|
|
else:
|
|
var fileid = ""
|
|
var ftype = ""
|
|
var fcaption = ""
|
|
if response.caption.isSome:
|
|
fcaption = response.caption.get
|
|
if response.document.isSome:
|
|
fileid = response.document.get.fileId
|
|
ftype = "document"
|
|
elif response.video.isSome:
|
|
fileid = response.video.get.fileId
|
|
ftype = "video"
|
|
elif response.videoNote.isSome:
|
|
fileid = response.videoNote.get.fileId
|
|
ftype = "videonote"
|
|
elif response.animation.isSome:
|
|
fileid = response.animation.get.fileId
|
|
ftype = "animation"
|
|
elif response.photo.isSome:
|
|
fileid = response.photo.get[0].fileId
|
|
ftype = "photo"
|
|
elif response.mediaGroupId.isSome:
|
|
fileid = response.mediaGroupId.get
|
|
|
|
let filehash = generate_hash()
|
|
|
|
var CensoredRow = NewCensoredData(ftype, filehash, fileid, fcaption)
|
|
|
|
with dbConn:
|
|
insert CensoredRow
|
|
|
|
discard await b.sendMessage(response.chat.id, "Censored --> " & "tg://resolve?domain=botbotbotnotabot&start=" & filehash)
|
|
|
|
let bot = newTeleBot(API_KEY)
|
|
bot.onUpdate(updateHandler)
|
|
bot.poll(timeout=300)
|