9f760114bd
All help messages and many other things moved from markdown to html. Eventually, I'd like only things made from user input to use markdown. cats.lua, rmspic.lua, and dilbert.lua moved to sendPhoto with URL. xkcd.lua and apod.lua not moved to retain formatting. Probably a lot of other stuff that I forget about. I should commit more often.
77 lines
2.3 KiB
Lua
77 lines
2.3 KiB
Lua
-- 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')
|
|
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'
|
|
|
|
function bing:init(config)
|
|
assert(config.bing_api_key,
|
|
'bing.lua requires a Bing API key from http://datamarket.azure.com/dataset/bing/search.'
|
|
)
|
|
|
|
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 = [[/bing <query>
|
|
Returns the top web results from Bing.
|
|
Aliases: /g, /google]]
|
|
bing.doc = bing.doc:gsub('/', config.cmd_pat)
|
|
|
|
end
|
|
|
|
function bing:action(msg, config)
|
|
local input = utilities.input_from_msg(msg)
|
|
if not input then
|
|
utilities.send_reply(msg, bing.doc, 'html')
|
|
return
|
|
end
|
|
|
|
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(msg, config.errors.connection)
|
|
return
|
|
end
|
|
|
|
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(msg, config.errors.results)
|
|
return
|
|
end
|
|
|
|
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(msg.chat.id, output, true, nil, 'html')
|
|
end
|
|
|
|
return bing
|