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/otouto/plugins/bing.lua

79 lines
2.3 KiB
Lua
Raw Normal View History

-- Credit to Juan (tg:JuanPotato; gh:JuanPotato) for this plugin.
local bing = {}
local URL = require('socket.url')
local JSON = require('dkjson')
local mime = require('mime')
local https = require('ssl.https')
local ltn12 = require('ltn12')
2016-06-07 06:31:34 +02:00
local utilities = require('otouto.utilities')
bing.command = 'bing <query>'
bing.search_url = 'https://api.datamarket.azure.com/Data.ashx/Bing/Search/Web?Query=\'%s\'&$format=json'
2016-06-07 08:22:01 +02:00
function bing:init(config)
2016-08-14 04:46:18 +02:00
assert(config.bing_api_key,
'bing.lua requires a Bing API key from http://datamarket.azure.com/dataset/bing/search.'
)
2016-08-14 04:26:44 +02:00
2016-08-14 04:46:18 +02:00
bing.headers = { ["Authorization"] = "Basic " .. mime.b64(":" .. config.bing_api_key) }
bing.triggers = utilities.triggers(self.info.username, config.cmd_pat)
:t('bing', true):t('g', true):t('google', true).table
bing.doc = [[
2016-08-14 04:26:44 +02:00
/bing <query>
Returns the top web results from Bing.
Aliases: /g, /google
2016-08-14 04:46:18 +02:00
]]
bing.doc = bing.doc:gsub('/', config.cmd_pat)
2016-08-14 04:26:44 +02:00
end
2016-06-07 09:02:05 +02:00
function bing:action(msg, config)
2016-08-14 04:46:18 +02:00
local input = utilities.input_from_msg(msg)
if not input then
utilities.send_reply(self, msg, bing.doc, true)
return
end
2016-08-14 04:26:44 +02:00
2016-08-14 04:46:18 +02:00
local url = bing.search_url:format(URL.escape(input))
local resbody = {}
local _, code = https.request{
url = url,
headers = bing.headers,
sink = ltn12.sink.table(resbody),
}
if code ~= 200 then
utilities.send_reply(self, msg, config.errors.connection)
return
end
2016-08-14 04:26:44 +02:00
2016-08-14 04:46:18 +02:00
local data = JSON.decode(table.concat(resbody))
-- Four results in a group, eight in private.
local limit = msg.chat.type == 'private' and 8 or 4
-- No more results than provided.
limit = limit > #data.d.results and #data.d.results or limit
if limit == 0 then
utilities.send_reply(self, msg, config.errors.results)
return
end
2016-08-14 04:26:44 +02:00
2016-08-14 04:46:18 +02:00
local reslist = {}
for i = 1, limit do
table.insert(reslist, string.format(
'• <a href="%s">%s</a>',
utilities.html_escape(data.d.results[i].Url),
utilities.html_escape(data.d.results[i].Title)
))
end
local output = string.format(
'<b>Search results for</b> <i>%s</i><b>:</b>\n%s',
utilities.html_escape(input),
table.concat(reslist, '\n')
)
utilities.send_message(self, msg.chat.id, output, true, nil, 'html')
end
return bing