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 )
2016-08-14 04:26:44 +02:00
assert ( config.google_api_key and config.google_cse_key ,
' gImages.lua requires a Google API key from http://console.developers.google.com and a Google Custom Search Engine key from http://cse.google.com/cse. '
)
2016-04-11 06:04:47 +02:00
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 '
2016-08-14 04:26:44 +02:00
gImages.search_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-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-08-14 04:26:44 +02:00
local input = utilities.input_from_msg ( msg )
2015-07-03 00:15:52 +02:00
if not input then
2016-08-14 04:26:44 +02:00
utilities.send_reply ( self , msg , gImages.doc , true )
return
2015-07-03 00:15:52 +02:00
end
2016-08-14 04:26:44 +02:00
local url = gImages.search_url
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