75 lines
2.3 KiB
Lua
75 lines
2.3 KiB
Lua
|
local bImages = {}
|
||
|
|
|
||
|
|
local HTTPS = require('ssl.https')
|
||
|
|
HTTPS.timeout = 10
|
||
|
|
local URL = require('socket.url')
|
||
|
|
local JSON = require('dkjson')
|
||
|
|
local redis = (loadfile "./miku/redis.lua")()
|
||
|
|
local utilities = require('miku.utilities')
|
||
|
|
local bindings = require('miku.bindings')
|
||
|
|
|
||
|
|
function bImages:init(config)
|
||
|
|
if not cred_data.bing_search_key then
|
||
|
|
print('Missing config value: bing_search_key.')
|
||
|
|
print('bImages.lua will not be enabled.')
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
bImages.triggers = {"^/nil$"}
|
||
|
|
bImages.inline_triggers = {
|
||
|
|
"^b (.*)"
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
local apikey = cred_data.bing_search_key
|
||
|
|
local BASE_URL = 'https://api.cognitive.microsoft.com/bing/v5.0'
|
||
|
|
|
||
|
|
function bImages:getImages(query)
|
||
|
|
local url = BASE_URL..'/images/search?q='..URL.escape(query)..'&count=50&mkt=de-de'
|
||
|
|
local response_body = {}
|
||
|
|
local request_constructor = {
|
||
|
|
url = url,
|
||
|
|
method = "GET",
|
||
|
|
sink = ltn12.sink.table(response_body),
|
||
|
|
redirect = false,
|
||
|
|
headers = {
|
||
|
|
["Ocp-Apim-Subscription-Key"] = apikey
|
||
|
|
}
|
||
|
|
}
|
||
|
|
local ok, response_code, response_headers = HTTPS.request(request_constructor)
|
||
|
|
if not ok then return end
|
||
|
|
local images = JSON.decode(table.concat(response_body)).value
|
||
|
|
if not images[1] then return end
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
local results = '['
|
||
|
|
for n in pairs(images) do
|
||
|
|
if images[n].encodingFormat == 'jpeg' then -- Inline-Querys MUST use JPEG photos!
|
||
|
|
local photo_url = images[n].contentUrl
|
||
|
|
local thumb_url = images[n].thumbnailUrl
|
||
|
|
results = results..'{"type":"photo","id":"'..math.random(100000000000000000)..'","photo_url":"'..photo_url..'","thumb_url":"'..thumb_url..'","photo_width":'..images[n].width..',"photo_height":'..images[n].height..',"reply_markup":{"inline_keyboard":[[{"text":"Bing aufrufen","url":"'..images[n].webSearchUrl..'"},{"text":"Bild aufrufen","url":"'..photo_url..'"}]]}},'
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
local results = results:sub(0, -2)
|
||
|
|
local results = results..']'
|
||
|
|
cache_data('bImages', string.lower(query), results, 1209600, 'key')
|
||
|
|
return results
|
||
|
|
end
|
||
|
|
|
||
|
|
function bImages:inline_callback(inline_query, config, matches)
|
||
|
|
local query = matches[1]
|
||
|
|
local results = redis:get('telegram:cache:bImages:'..string.lower(query))
|
||
|
|
if not results then
|
||
|
|
results = bImages:getImages(query)
|
||
|
|
end
|
||
|
|
|
||
|
|
if not results then return end
|
||
|
|
utilities.answer_inline_query(self, inline_query, results, 3600)
|
||
|
|
end
|
||
|
|
|
||
|
|
function bImages:action()
|
||
|
|
end
|
||
|
|
|
||
|
|
return bImages
|