This repository has been archived on 2021-04-24. You can view files and clone it, but cannot push or open issues or pull requests.
Mikubot-2/plugins/gImages.lua

61 lines
1.5 KiB
Lua
Raw Normal View History

local doc = [[
/image <query>
Returns a randomized top result from Google Images. Safe search is enabled by default; use "/insfw" to disable it. NSFW results will not display an image preview.
2015-07-03 00:15:52 +02:00
]]
local triggers = {
'^/i[mage]*[nsfw]*[@'..bot.username..']*$',
'^/i[mage]*[nsfw]*[@'..bot.username..']* '
2015-07-03 00:15:52 +02:00
}
local action = function(msg)
2015-07-03 00:15:52 +02:00
local input = msg.text:input()
2015-07-03 00:15:52 +02:00
if not input then
if msg.reply_to_message and msg.reply_to_message.text then
input = msg.reply_to_message.text
else
sendReply(msg, doc)
return
end
2015-07-03 00:15:52 +02:00
end
2015-12-05 15:23:32 +01:00
local url = 'https://www.googleapis.com/customsearch/v1?&searchType=image&imgSize=xlarge&alt=json&num=8&start=1'
url = url .. '&key=0000000' -- KEY Get https://console.developers.google.com/apis/credentials
url = url .. '&cx=ABCD:000' -- CX Get https://cse.google.com/cse
if not string.match(msg.text, '^/i[mage]*nsfw') then
2015-12-05 15:23:32 +01:00
url = url .. '&safe=high'
end
2015-07-03 00:15:52 +02:00
url = url .. '&q=' .. URL.escape(input)
2015-07-03 00:15:52 +02:00
local jstr, res = HTTPS.request(url)
2015-07-03 00:15:52 +02:00
if res ~= 200 then
sendReply(msg, config.errors.connection)
2015-07-03 00:15:52 +02:00
return
end
local jdat = JSON.decode(jstr)
2015-12-05 15:23:32 +01:00
if jdat.searchInformation.totalResults == '0' then
sendReply(msg, config.errors.results)
2015-07-03 00:15:52 +02:00
return
end
2015-12-05 15:23:32 +01:00
local i = math.random(jdat.queries.request[1].count)
local result = jdat.items[i].link
2015-07-03 00:15:52 +02:00
if string.match(msg.text, '^/i[mage]*nsfw') then
sendReply(msg, result)
else
sendMessage(msg.chat.id, result, false, msg.message_id)
end
2015-07-03 00:15:52 +02:00
end
return {
action = action,
triggers = triggers,
doc = doc
}