68 lines
1.3 KiB
Lua
68 lines
1.3 KiB
Lua
|
local PLUGIN = {}
|
||
|
|
|
||
|
|
PLUGIN.doc = [[
|
||
|
|
!images <query>
|
||
|
|
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 = {
|
||
|
|
'^!images?',
|
||
|
|
'^!img',
|
||
|
|
'^!i ',
|
||
|
|
'^!insfw'
|
||
|
|
}
|
||
|
|
|
||
|
|
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'
|
||
|
|
|
||
|
|
if not string.match(msg.text, '^!insfw ') then
|
||
|
|
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
|
||
|
|
send_msg(target, 'Connection error.', ok_cb, false)
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
local jdat = JSON.decode(jstr)
|
||
|
|
|
||
|
|
if #jdat.responseData.results < 1 then
|
||
|
|
send_msg(target, 'No results found.', ok_cb, false)
|
||
|
|
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
|