125 lines
2.9 KiB
Lua
125 lines
2.9 KiB
Lua
do
|
|
local mime = require("mime")
|
|
|
|
local google_config = load_from_file('data/google.lua')
|
|
local cache = {}
|
|
|
|
--[[
|
|
local function send_request(url)
|
|
local t = {}
|
|
local options = {
|
|
url = url,
|
|
sink = ltn12.sink.table(t),
|
|
method = "GET"
|
|
}
|
|
local a, code, headers, status = http.request(options)
|
|
return table.concat(t), code, headers, status
|
|
end]]--
|
|
|
|
local function get_google_data(text)
|
|
--[[
|
|
local url = "http://ajax.googleapis.com/ajax/services/search/images?"
|
|
url = url.."v=1.0&rsz=5"
|
|
url = url.."&q="..URL.escape(text)
|
|
url = url.."&imgsz=small|medium|large"
|
|
--]]
|
|
|
|
if not google_config.cx then
|
|
print "cx not present in config file"
|
|
return nil
|
|
end
|
|
|
|
local url = "https://www.googleapis.com/customsearch/v1?"
|
|
url = url.."cx="..google_config.cx
|
|
url = url.."&searchType=image&safe=high"
|
|
url = url.."&q="..URL.escape(text)
|
|
|
|
if google_config.api_keys then
|
|
local i = math.random(#google_config.api_keys)
|
|
local api_key = google_config.api_keys[i]
|
|
if api_key then
|
|
url = url.."&key="..api_key
|
|
end
|
|
end
|
|
|
|
local res, code = https.request(url)
|
|
|
|
if code ~= 200 then
|
|
print("HTTP Error code:", code)
|
|
return nil
|
|
end
|
|
|
|
local google = json:decode(res)
|
|
return google
|
|
end
|
|
|
|
-- Returns only the useful google data to save on cache
|
|
local function simple_google_table(google)
|
|
local new_table = {}
|
|
new_table.responseData = {}
|
|
new_table.responseDetails = google.responseDetails
|
|
new_table.responseStatus = google.responseStatus
|
|
new_table.responseData.results = {}
|
|
local results = google.responseData.results
|
|
for k,result in pairs(results) do
|
|
new_table.responseData.results[k] = {}
|
|
new_table.responseData.results[k].unescapedUrl = result.unescapedUrl
|
|
new_table.responseData.results[k].url = result.url
|
|
end
|
|
return new_table
|
|
end
|
|
|
|
local function save_to_cache(query, data)
|
|
-- Saves result on cache
|
|
if string.len(query) <= 7 then
|
|
local text_b64 = mime.b64(query)
|
|
if not cache[text_b64] then
|
|
-- local simple_google = simple_google_table(data)
|
|
cache[text_b64] = data
|
|
end
|
|
end
|
|
end
|
|
|
|
local function process_google_data(google, receiver, query)
|
|
|
|
if not google or not google.items or #google.items == 0 then
|
|
local text = 'Image not found.'
|
|
send_msg(receiver, text, ok_cb, false)
|
|
return false
|
|
end
|
|
|
|
local i = math.random(#google.items)
|
|
local url = google.items[i].link
|
|
local old_timeout = https.TIMEOUT or 10
|
|
http.TIMEOUT = 10
|
|
send_photo_from_url(receiver, url)
|
|
http.TIMEOUT = old_timeout
|
|
|
|
save_to_cache(query, google)
|
|
|
|
end
|
|
|
|
function run(msg, matches)
|
|
local receiver = get_receiver(msg)
|
|
local text = matches[1]
|
|
local text_b64 = mime.b64(text)
|
|
local cached = cache[text_b64]
|
|
if cached then
|
|
process_google_data(cached, receiver, text)
|
|
else
|
|
local data = get_google_data(text)
|
|
process_google_data(data, receiver, text)
|
|
end
|
|
end
|
|
|
|
return {
|
|
description = "",
|
|
usage = "",
|
|
patterns = {
|
|
"^/img (.*)$"
|
|
},
|
|
run = run
|
|
}
|
|
|
|
end
|