nim-censor-bot/src/nim_censor_bot.nim

61 lines
2 KiB
Nim
Raw Normal View History

import telebot, std/[asyncdispatch, logging, options, strutils, random, with], norm/[model, sqlite]
type
CENSORED_DATA* = ref object of Model
ftype*: string
fhash*: string
fileid*: string
caption*: string
func newCENSORED_DATA*(ftype = ""; fhash = ""; fileid = ""; caption = ""):
CENSORED_DATA = CENSORED_DATA(ftype: ftype, fhash: fhash, fileid: fileid, caption: caption)
let dbConn* = sqlite.open("censordata.db", "", "", "")
dbConn.createTables(newCENSORED_DATA())
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
# remember to strip your secret key to avoid HTTP error
const API_KEY = ""
proc updateHandler(b: Telebot, u: Update): Future[bool] {.async, gcsafe.} =
var response = u.message.get
if response.document.isSome:
let filehash = generate_hash()
var censordoc = newCENSORED_DATA("document", filehash, response.document.get.fileId, "")
with dbConn:
insert censordoc
if response.video.isSome:
let filehash = generate_hash()
var censorvid = newCENSORED_DATA("video", filehash, response.video.get.fileId, "")
with dbConn:
insert censorvid
if response.videoNote.isSome:
let filehash = generate_hash()
var censorvidnote = newCENSORED_DATA("videonote", filehash, response.videoNote.get.fileId, "")
with dbConn:
insert censorvidnote
if response.animation.isSome:
let filehash = generate_hash()
var censoranimation = newCENSORED_DATA("animation", filehash, response.animation.get.fileId, "")
with dbConn:
insert censoranimation
if response.photo.isSome:
let filehash = generate_hash()
var censorphoto = newCENSORED_DATA("photo", filehash, response.photo.get[0].fileId, "")
with dbConn:
insert censorphoto
let bot = newTeleBot(API_KEY)
bot.onUpdate(updateHandler)
bot.poll(timeout=300)