2016-01-08 14:44:37 +01:00
|
|
|
local command = 'google <query>'
|
|
|
|
local doc = [[```
|
|
|
|
/google <query>
|
|
|
|
Returns four (if group) or eight (if private message) results from Google. Safe search is enabled by default, use "/gnsfw" to disable it.
|
|
|
|
Alias: /g
|
|
|
|
```]]
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local triggers = {
|
2016-01-08 04:30:12 +01:00
|
|
|
'^/g[@'..bot.username..']*$',
|
|
|
|
'^/g[@'..bot.username..']* ',
|
|
|
|
'^/google[@'..bot.username..']*',
|
|
|
|
'^/gnsfw[@'..bot.username..']*'
|
2015-07-03 00:15:52 +02:00
|
|
|
}
|
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local action = function(msg)
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local input = msg.text:input()
|
|
|
|
if not input then
|
|
|
|
if msg.reply_to_message and msg.reply_to_message.text then
|
|
|
|
input = msg.reply_to_message.text
|
|
|
|
else
|
2016-01-08 14:44:37 +01:00
|
|
|
sendMessage(msg.chat.id, doc, true, msg.message_id, true)
|
2015-11-25 03:22:04 +01:00
|
|
|
return
|
|
|
|
end
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local url = 'https://ajax.googleapis.com/ajax/services/search/web?v=1.0'
|
|
|
|
|
|
|
|
if msg.from.id == msg.chat.id then
|
2015-07-03 00:15:52 +02:00
|
|
|
url = url .. '&rsz=8'
|
2015-11-25 03:22:04 +01:00
|
|
|
else
|
|
|
|
url = url .. '&rsz=4'
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
if not string.match(msg.text, '^/g[oogle]*nsfw') then
|
|
|
|
url = url .. '&safe=active'
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
|
|
|
url = url .. '&q=' .. URL.escape(input)
|
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local jstr, res = HTTPS.request(url)
|
2015-07-03 00:15:52 +02:00
|
|
|
if res ~= 200 then
|
2015-11-25 03:22:04 +01:00
|
|
|
sendReply(msg, config.errors.connection)
|
|
|
|
return
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
|
|
|
local jdat = JSON.decode(jstr)
|
2016-01-08 14:44:37 +01:00
|
|
|
if not jdat.responseData then
|
|
|
|
sendReply(msg, config.errors.connection)
|
|
|
|
return
|
|
|
|
end
|
|
|
|
if not jdat.responseData.results[1] then
|
2015-11-25 03:22:04 +01:00
|
|
|
sendReply(msg, config.errors.results)
|
|
|
|
return
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
2016-01-12 11:22:28 +01:00
|
|
|
local output = '*Google results for* _' .. input .. '_ *:*\n'
|
2015-11-25 03:22:04 +01:00
|
|
|
for i,v in ipairs(jdat.responseData.results) do
|
2016-01-08 14:44:37 +01:00
|
|
|
local title = jdat.responseData.results[i].titleNoFormatting:gsub('%[.+%]', ''):gsub('&', '&')
|
|
|
|
if title:len() > 48 then
|
|
|
|
title = title:sub(1, 45) .. '...'
|
|
|
|
end
|
|
|
|
local url = jdat.responseData.results[i].unescapedUrl
|
|
|
|
if url:find('%)') then
|
|
|
|
output = output .. '• ' .. title .. '\n' .. url:gsub('_', '\\_') .. '\n'
|
|
|
|
else
|
|
|
|
output = output .. '• [' .. title .. '](' .. url .. ')\n'
|
|
|
|
end
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
2016-01-08 14:44:37 +01:00
|
|
|
sendMessage(msg.chat.id, output, true, nil, true)
|
2015-07-03 00:15:52 +02:00
|
|
|
|
|
|
|
end
|
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
return {
|
|
|
|
action = action,
|
|
|
|
triggers = triggers,
|
2016-01-08 14:44:37 +01:00
|
|
|
doc = doc,
|
|
|
|
command = command
|
2015-11-25 03:22:04 +01:00
|
|
|
}
|