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/gSearch.lua

81 lines
1.9 KiB
Lua
Raw Normal View History

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
local triggers = {
'^/g[@'..bot.username..']*$',
'^/g[@'..bot.username..']* ',
'^/google[@'..bot.username..']*',
'^/gnsfw[@'..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()
if not input then
if msg.reply_to_message and msg.reply_to_message.text then
input = msg.reply_to_message.text
else
sendMessage(msg.chat.id, doc, true, msg.message_id, true)
return
end
2015-07-03 00:15:52 +02:00
end
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'
else
url = url .. '&rsz=4'
2015-07-03 00:15:52 +02:00
end
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)
local jstr, res = HTTPS.request(url)
2015-07-03 00:15:52 +02:00
if res ~= 200 then
sendReply(msg, config.errors.connection)
return
2015-07-03 00:15:52 +02:00
end
local jdat = JSON.decode(jstr)
if not jdat.responseData then
sendReply(msg, config.errors.connection)
return
end
if not jdat.responseData.results[1] then
sendReply(msg, config.errors.results)
return
2015-07-03 00:15:52 +02:00
end
local output = '*Google results for* _' .. input .. '_ *:*\n'
for i,v in ipairs(jdat.responseData.results) do
local title = jdat.responseData.results[i].titleNoFormatting:gsub('%[.+%]', ''):gsub('&amp;', '&')
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
sendMessage(msg.chat.id, output, true, nil, true)
2015-07-03 00:15:52 +02:00
end
return {
action = action,
triggers = triggers,
doc = doc,
command = command
}