64 lines
2.3 KiB
Lua
64 lines
2.3 KiB
Lua
-- The google plugin, just type:
|
||
-- !google <stuff to search>
|
||
|
||
local urltool = require("socket.url")
|
||
local json = require("json")
|
||
local socket = require("socket")
|
||
|
||
local htmlEntities = {
|
||
[" "] = " ",
|
||
[" "] = " ",
|
||
["""] = '"',
|
||
["""] = '"',
|
||
["'"] = "'",
|
||
["'"] = "'",
|
||
["&"] = "&",
|
||
["&"] = "&",
|
||
["<"] = "<",
|
||
["<"] = "<",
|
||
[">"] = ">",
|
||
[">"] = ">",
|
||
["¡"] = "¡",
|
||
["¡"] = "¡",
|
||
["´"] = "´",
|
||
["´"] = "´",
|
||
["Ñ"] = "Ñ",
|
||
["Ñ"] = "Ñ",
|
||
["ñ"] = "ñ",
|
||
["ñ"] = "ñ",
|
||
}
|
||
|
||
local function parseHtmlEntites(s)
|
||
local ret =string.gsub(s,"&.-;", function(input) return htmlEntities[input] or "?" end)
|
||
return ret
|
||
end
|
||
|
||
plugin.google = function (target, from, arg)
|
||
local search = urltool.escape(arg)
|
||
local result, error, header = socket.http.request( 'http://ajax.googleapis.com/ajax/services/search/web?v=1.0&q='..search.."&safe=off")
|
||
|
||
if error == 200 and result then
|
||
local jsonTable = json.decode(result)
|
||
if (jsonTable.responseData.results[1]) then
|
||
--unicode hax
|
||
content = string.gsub(urltool.unescape(jsonTable.responseData.results[1].content), "u(%x%x%x%x)",
|
||
function(c)
|
||
local status, result = pcall(string.char, "0x"..c)
|
||
if status then
|
||
return result
|
||
else
|
||
return "u"..c
|
||
end
|
||
end)
|
||
--parse html tags to irc tags
|
||
content = string.gsub(content,"(<b>)", "")
|
||
content = string.gsub(content,"(</b>)", "")
|
||
content = string.gsub(content,"(<u>)", " ")
|
||
content = string.gsub(content,"(</u>)", " ")
|
||
content = string.gsub(content,"(<i>)", " ")
|
||
content = string.gsub(content,"(</i>)", " ")
|
||
result = urltool.unescape(jsonTable.responseData.results[1].url).." "..content
|
||
irc.say(target, parseHtmlEntites(result))
|
||
end
|
||
end
|
||
end
|