121 lines
4.0 KiB
Lua
121 lines
4.0 KiB
Lua
local pixabay = {}
|
|
|
|
local https = require('ssl.https')
|
|
local json = require('dkjson')
|
|
local utilities = require('otouto.utilities')
|
|
local redis = (loadfile "./otouto/redis.lua")()
|
|
|
|
function pixabay:init(config)
|
|
if not cred_data.pixabay_apikey then
|
|
print('Missing config value: pixabay_apikey.')
|
|
print('pixabay.lua will not be enabled.')
|
|
return
|
|
end
|
|
|
|
pixabay.triggers = {
|
|
"^/pix(id) (%d+)",
|
|
"^/pix (.*)$",
|
|
"(pixabay.com).*%-(%d+)"
|
|
}
|
|
pixabay.doc = [[*
|
|
]]..config.cmd_pat..[[pix* _<Suchbegriff>_: Sendet lizenzfreies Bild]]
|
|
end
|
|
|
|
pixabay.command = 'pix <Suchbegriff>'
|
|
|
|
local BASE_URL = 'https://pixabay.com/api'
|
|
local apikey = cred_data.pixabay_apikey
|
|
|
|
function pixabay:get_pixabay_directlink(id)
|
|
local url = BASE_URL..'/?key='..apikey..'&lang=de&id='..id
|
|
local b,c = https.request(url)
|
|
if c ~= 200 then return nil end
|
|
local data = json.decode(b)
|
|
if data.totalHits == 0 then return 'NOPIX' end
|
|
|
|
local webformatURL = data.hits[1].webformatURL
|
|
local image_url = string.gsub(webformatURL, '_640.jpg', '_960.jpg')
|
|
|
|
-- Link to full, high resolution image
|
|
local preview_url = data.hits[1].previewURL
|
|
local image_code = string.sub(preview_url, 59)
|
|
local image_code = string.sub(image_code, 0, -9)
|
|
local full_url = 'https://pixabay.com/de/photos/download/'..image_code..'.jpg'
|
|
|
|
local user = data.hits[1].user
|
|
local tags = data.hits[1].tags
|
|
local page_url = data.hits[1].pageURL
|
|
|
|
-- cache this shit
|
|
local hash = 'telegram:cache:pixabay:'..id
|
|
print('Caching data in '..hash..' with timeout 1209600')
|
|
redis:hset(hash, 'image_url', image_url)
|
|
redis:hset(hash, 'full_url', full_url)
|
|
redis:hset(hash, 'page_url', page_url)
|
|
redis:hset(hash, 'user', user)
|
|
redis:hset(hash, 'tags', tags)
|
|
redis:expire(hash, 1209600) -- 1209600 = two weeks
|
|
|
|
return image_url, full_url, page_url, user, tags
|
|
end
|
|
|
|
function pixabay:get_pixabay(term)
|
|
local count = 70 -- how many pictures should be returned (3 to 200) NOTE: more pictures = higher load time
|
|
local url = BASE_URL..'/?key='..apikey..'&lang=de&safesearch=true&per_page='..count..'&image_type=photo&q='..term
|
|
local b,c = https.request(url)
|
|
if c ~= 200 then return nil end
|
|
local photo = json.decode(b)
|
|
if photo.totalHits == 0 then return 'NOPIX' end
|
|
local photos = photo.hits
|
|
-- truly randomize
|
|
math.randomseed(os.time())
|
|
-- random max json table size
|
|
local i = math.random(#photos)
|
|
|
|
local webformatURL = photos[i].webformatURL
|
|
local image_url = string.gsub(webformatURL, '_640.jpg', '_960.jpg')
|
|
|
|
-- Link to full, high resolution image
|
|
local preview_url = photos[i].previewURL
|
|
local image_code = string.sub(preview_url, 59)
|
|
local image_code = string.sub(image_code, 0, -9)
|
|
local full_url = 'https://pixabay.com/de/photos/download/'..image_code..'.jpg'
|
|
|
|
local user = photos[i].user
|
|
local tags = photos[i].tags
|
|
local page_url = photos[i].pageURL
|
|
|
|
return image_url, full_url, page_url, user, tags
|
|
end
|
|
|
|
function pixabay:action(msg, config, matches)
|
|
local term = matches[1]
|
|
if matches[2] then
|
|
if redis:exists("telegram:cache:pixabay:"..matches[2]) == true then -- if cached
|
|
local hash = 'telegram:cache:pixabay:'..matches[2]
|
|
url = redis:hget(hash, 'image_url')
|
|
full_url = redis:hget(hash, 'full_url')
|
|
page_url = redis:hget(hash, 'page_url')
|
|
user = redis:hget(hash, 'user')
|
|
tags = redis:hget(hash, 'tags')
|
|
else
|
|
url, full_url, page_url, user, tags = pixabay:get_pixabay_directlink(matches[2])
|
|
end
|
|
else
|
|
url, full_url, page_url, user, tags = pixabay:get_pixabay(term)
|
|
end
|
|
|
|
if url == 'NOPIX' then
|
|
utilities.send_reply(self, msg, config.errors.results)
|
|
return
|
|
else
|
|
utilities.send_typing(self, msg.chat.id, 'upload_photo')
|
|
local file = download_to_file(url)
|
|
local text = '"'..tags..'" von '..user
|
|
utilities.send_photo(self, msg.chat.id, file, text, msg.message_id, '{"inline_keyboard":[[{"text":"Seite aufrufen","url":"'..page_url..'"},{"text":"Volles Bild (Login notwendig)","url":"'..full_url..'"}]]}')
|
|
return
|
|
end
|
|
end
|
|
|
|
return pixabay
|