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

68 lines
1.3 KiB
Lua
Raw Normal View History

2015-07-03 00:15:52 +02:00
local PLUGIN = {}
PLUGIN.doc = [[
2015-07-08 09:38:04 +02:00
/images <query>
2015-07-03 00:15:52 +02:00
This command performs a Google Images search for the given query. One random top result is returned. Safe search is enabled by default; use '!insfw' to get potentially NSFW results.
]]
PLUGIN.triggers = {
2015-07-08 09:38:04 +02:00
'^/images?',
'^/img',
'^/i ',
'^/insfw'
2015-07-03 00:15:52 +02:00
}
PLUGIN.exts = {
'.png$',
'.jpg$',
'.jpeg$',
'.jpe$',
'.gif$'
}
function PLUGIN.action(msg)
local url = 'http://ajax.googleapis.com/ajax/services/search/images?v=1.0&rsz=8'
2015-07-08 09:38:04 +02:00
if not string.match(msg.text, '^/insfw ') then
2015-07-03 00:15:52 +02:00
url = url .. '&safe=active'
end
local input = get_input(msg.text)
if not input then
return send_msg(msg, PLUGIN.doc)
end
url = url .. '&q=' .. URL.escape(input)
local jstr, res = HTTP.request(url)
if res ~= 200 then
2015-07-03 04:04:00 +02:00
send_msg(msg, 'Connection error.')
2015-07-03 00:15:52 +02:00
return
end
local jdat = JSON.decode(jstr)
if #jdat.responseData.results < 1 then
2015-07-03 04:04:00 +02:00
send_msg(msg, 'No results found.')
2015-07-03 00:15:52 +02:00
return
end
is_real = false
while is_real == false do
local i = math.random(#jdat.responseData.results)
result_url = jdat.responseData.results[i].url
for i,v in pairs(PLUGIN.exts) do
if string.match(string.lower(result_url), v) then
is_real = true
end
end
end
send_message(msg.chat.id, result_url, false, msg.message_id)
end
return PLUGIN