119 lines
3.1 KiB
Lua
119 lines
3.1 KiB
Lua
|
-- This is a proprietary plugin, property of Andreas Bielawski, (c) 2015 <andi (dot) b (at) outlook (dot) de>
|
||
|
|
-- DO NOT USE WITHOUT PERMISSION
|
||
|
|
|
||
|
|
do
|
||
|
|
|
||
|
|
local _blacklist
|
||
|
|
|
||
|
|
local function getNSFWImage(text)
|
||
|
|
local apikey = cred_data.google_apikey
|
||
|
|
local cseid = cred_data.google_cse_id
|
||
|
|
local url = 'https://www.googleapis.com/customsearch/v1?cx='..cseid..'&key='..apikey..'&searchType=image&num=10&fields=items(link)&q='..URL.escape(text)
|
||
|
|
local res, code = https.request(url)
|
||
|
|
if code ~= 200 then return nil end
|
||
|
|
|
||
|
|
local google = json:decode(res).items
|
||
|
|
return google
|
||
|
|
end
|
||
|
|
|
||
|
|
local function is_blacklisted(msg)
|
||
|
|
local var = false
|
||
|
|
for v,word in pairs(_blacklist) do
|
||
|
|
if string.find(string.lower(msg), string.lower(word)) then
|
||
|
|
print("Wort steht auf der Blacklist!")
|
||
|
|
var = true
|
||
|
|
break
|
||
|
|
end
|
||
|
|
end
|
||
|
|
return var
|
||
|
|
end
|
||
|
|
|
||
|
|
local function show_blacklist()
|
||
|
|
if not _blacklist[1] then
|
||
|
|
return "Keine Wörter geblacklisted!\nBlackliste welche mit !imgblacklist add [Wort]"
|
||
|
|
else
|
||
|
|
local sort_alph = function( a,b ) return a < b end
|
||
|
|
table.sort( _blacklist, sort_alph )
|
||
|
|
local blacklist = "Folgende Wörter stehen auf der Blacklist:\n"
|
||
|
|
for v,word in pairs(_blacklist) do
|
||
|
|
blacklist = blacklist..'- '..word..'\n'
|
||
|
|
end
|
||
|
|
return blacklist
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
local function cache_nsfw_image(results, text)
|
||
|
|
local cache = {}
|
||
|
|
for v in pairs(results) do
|
||
|
|
table.insert(cache, results[v].link)
|
||
|
|
end
|
||
|
|
cache_data('img_google_nsfw', string.lower(text), cache, 1209600, 'set')
|
||
|
|
end
|
||
|
|
|
||
|
|
function run(msg, matches)
|
||
|
|
local receiver = get_receiver(msg)
|
||
|
|
local text = matches[1]
|
||
|
|
if matches[2] then word = string.lower(matches[2]) end
|
||
|
|
|
||
|
|
_blacklist = redis:smembers("telegram:img_blacklist")
|
||
|
|
|
||
|
|
print ('Checking if search contains blacklisted words: '..text)
|
||
|
|
if is_blacklisted(text) then
|
||
|
|
return "Vergiss es ._."
|
||
|
|
end
|
||
|
|
|
||
|
|
local hash = 'telegram:cache:img_google_nsfw:'..string.lower(text)
|
||
|
|
local results = redis:smembers(hash)
|
||
|
|
if not results[1] then
|
||
|
|
print('doing web request')
|
||
|
|
results = getNSFWImage(text)
|
||
|
|
if not results then
|
||
|
|
return "Kein Bild gefunden!"
|
||
|
|
end
|
||
|
|
cache_nsfw_image(results, text)
|
||
|
|
end
|
||
|
|
-- Random image from table
|
||
|
|
local i = math.random(#results)
|
||
|
|
local url = nil
|
||
|
|
|
||
|
|
local failed = true
|
||
|
|
local nofTries = 0
|
||
|
|
while failed and nofTries < #results do
|
||
|
|
if not results[i].link then
|
||
|
|
url = results[i]
|
||
|
|
else
|
||
|
|
url = results[i].link
|
||
|
|
end
|
||
|
|
print("Bilder-URL: ", url)
|
||
|
|
|
||
|
|
if string.ends(url, ".gif") then
|
||
|
|
failed = not send_document_from_url(receiver, url, nil, nil, true)
|
||
|
|
return 'Source: '..url
|
||
|
|
elseif string.ends(url, ".jpg") or string.ends(url, ".jpeg") or string.ends(url, ".png") then
|
||
|
|
failed = not send_photo_from_url(receiver, url, nil, nil, true)
|
||
|
|
return 'Source: '..url
|
||
|
|
end
|
||
|
|
|
||
|
|
nofTries = nofTries + 1
|
||
|
|
i = i+1
|
||
|
|
if i > #results then
|
||
|
|
i = 1
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
if failed then
|
||
|
|
return "Fehler beim Herunterladen eines Bildes."
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
return {
|
||
|
|
description = "Sucht Bild mit Google-API und versendet es",
|
||
|
|
usage = {
|
||
|
|
"/img2 [Suchbegriff]"
|
||
|
|
},
|
||
|
|
patterns = {
|
||
|
|
"^/img2 (.*)$"
|
||
|
|
},
|
||
|
|
run = run
|
||
|
|
}
|
||
|
|
end
|