2015-12-05 16:03:43 +01:00
|
|
|
-- You need a Google API key and a Google Custom Search Engine set up to use this, in config.google_api_key and config.google_cse_key, respectively.
|
2016-04-12 05:55:46 +02:00
|
|
|
-- You must also sign up for the CSE in the Google Developer Console, and enable image results.
|
2015-12-05 16:03:43 +01:00
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
local gImages = {}
|
|
|
|
|
|
|
|
local HTTPS = require('ssl.https')
|
|
|
|
local URL = require('socket.url')
|
2016-04-15 21:07:23 +02:00
|
|
|
local JSON = require('dkjson')
|
2016-06-07 06:31:34 +02:00
|
|
|
local utilities = require('otouto.utilities')
|
2016-04-11 06:04:47 +02:00
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
function gImages:init(config)
|
|
|
|
if not config.google_api_key then
|
2016-04-11 06:04:47 +02:00
|
|
|
print('Missing config value: google_api_key.')
|
|
|
|
print('gImages.lua will not be enabled.')
|
|
|
|
return
|
2016-05-27 02:26:30 +02:00
|
|
|
elseif not config.google_cse_key then
|
2016-04-11 06:04:47 +02:00
|
|
|
print('Missing config value: google_cse_key.')
|
|
|
|
print('gImages.lua will not be enabled.')
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2016-05-27 05:28:44 +02:00
|
|
|
gImages.triggers = utilities.triggers(self.info.username, config.cmd_pat):t('image', true):t('i', true):t('insfw', true).table
|
2016-07-25 11:03:35 +02:00
|
|
|
gImages.doc = config.cmd_pat .. [[image <query>
|
|
|
|
Returns a randomized top result from Google Images. Safe search is enabled by default; use "]] .. config.cmd_pat .. [[insfw" to disable it. NSFW results will not display an image preview.
|
|
|
|
Alias: ]] .. config.cmd_pat .. 'i'
|
2015-12-05 16:03:43 +01:00
|
|
|
end
|
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
gImages.command = 'image <query>'
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
function gImages:action(msg, config)
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2016-04-08 23:12:02 +02:00
|
|
|
local input = utilities.input(msg.text)
|
2015-07-03 00:15:52 +02:00
|
|
|
if not input then
|
2015-11-25 03:22:04 +01:00
|
|
|
if msg.reply_to_message and msg.reply_to_message.text then
|
|
|
|
input = msg.reply_to_message.text
|
2015-07-15 08:15:23 +02:00
|
|
|
else
|
2016-05-29 19:08:39 +02:00
|
|
|
utilities.send_message(self, msg.chat.id, gImages.doc, true, msg.message_id, true)
|
2015-11-25 03:22:04 +01:00
|
|
|
return
|
2015-07-15 08:15:23 +02:00
|
|
|
end
|
2015-07-03 00:15:52 +02:00
|
|
|
end
|
|
|
|
|
2016-05-27 02:26:30 +02:00
|
|
|
local url = 'https://www.googleapis.com/customsearch/v1?&searchType=image&imgSize=xlarge&alt=json&num=8&start=1&key=' .. config.google_api_key .. '&cx=' .. config.google_cse_key
|
2015-11-25 03:22:04 +01:00
|
|
|
|
2016-05-27 05:28:44 +02:00
|
|
|
if not string.match(msg.text, '^'..config.cmd_pat..'i[mage]*nsfw') then
|
2015-12-05 15:23:32 +01:00
|
|
|
url = url .. '&safe=high'
|
2015-11-25 03:22:04 +01:00
|
|
|
end
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
url = url .. '&q=' .. URL.escape(input)
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2015-11-25 03:22:04 +01:00
|
|
|
local jstr, res = HTTPS.request(url)
|
2015-07-03 00:15:52 +02:00
|
|
|
if res ~= 200 then
|
2016-05-27 02:26:30 +02:00
|
|
|
utilities.send_reply(self, msg, config.errors.connection)
|
2015-07-03 00:15:52 +02:00
|
|
|
return
|
|
|
|
end
|
|
|
|
|
|
|
|
local jdat = JSON.decode(jstr)
|
2015-12-05 15:23:32 +01:00
|
|
|
if jdat.searchInformation.totalResults == '0' then
|
2016-05-27 02:26:30 +02:00
|
|
|
utilities.send_reply(self, msg, config.errors.results)
|
2015-07-03 00:15:52 +02:00
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2015-12-05 15:23:32 +01:00
|
|
|
local i = math.random(jdat.queries.request[1].count)
|
2016-02-25 10:42:13 +01:00
|
|
|
local img_url = jdat.items[i].link
|
|
|
|
local img_title = jdat.items[i].title
|
|
|
|
local output = '[' .. img_title .. '](' .. img_url .. ')'
|
2016-01-17 19:39:00 +01:00
|
|
|
|
2015-07-03 00:15:52 +02:00
|
|
|
|
2016-02-25 10:42:13 +01:00
|
|
|
if msg.text:match('nsfw') then
|
2016-05-29 19:08:39 +02:00
|
|
|
utilities.send_reply(self, '*NSFW*\n'..msg, output)
|
2015-11-25 03:22:04 +01:00
|
|
|
else
|
2016-05-29 19:08:39 +02:00
|
|
|
utilities.send_message(self, msg.chat.id, output, false, nil, true)
|
2015-11-25 03:22:04 +01:00
|
|
|
end
|
2015-07-03 00:15:52 +02:00
|
|
|
|
|
|
|
end
|
|
|
|
|
2016-04-11 06:04:47 +02:00
|
|
|
return gImages
|