diff --git a/bot/bot.lua b/bot/bot.lua index eccd4d6..bcd3a36 100644 --- a/bot/bot.lua +++ b/bot/bot.lua @@ -59,7 +59,6 @@ function send_image_from_url (msg) end function is_image_url(text) - print ('IS image ' .. text ..'?') last = string.get_last_word(text) extension = string.get_extension_from_filename(last) if extension == 'jpg' or extension == 'png' then @@ -102,6 +101,12 @@ function do_action(msg) return end + if string.starts(msg.text, '!rae') then + text = msg.text:sub(6,-1) + meaning = getDulcinea(text) + send_msg(receiver, meaning, ok_cb, false) + end + if string.starts(msg.text, '!9gag') then url, title = get_9GAG() file_path = download_to_file(url) @@ -271,14 +276,39 @@ function get_fortunes_uc3m() return b end -function getGoogleImage(text) - local api = "https://ajax.googleapis.com/ajax/services/search/images?v=1.0&q=" - text = URL.escape(text) +function getDulcinea( text ) + local api = "http://dulcinea.herokuapp.com/api/?query=" b = http.request(api..text) - local google = json:decode(b) - math.randomseed(os.time()) - i = math.random(#google.responseData.results) - return google.responseData.results[i].url + dulcinea = json:decode(b) + while dulcinea.type == "multiple" do + text = dulcinea.response[1].id + b = http.request(api..text) + dulcinea = json:decode(b) + end + local text = "" + vardump(dulcinea) + for i = 1, 5, 1 do + text = text .. dulcinea.response[i].word .."\n" + text = text .. dulcinea.response[i].meanings[1].meaning .."\n" + text = text .. "\n" + end + return text + +end + +function getGoogleImage(text) + text = URL.escape(text) + for i = 1, 5, 1 do -- Try 5 times + local api = "https://ajax.googleapis.com/ajax/services/search/images?v=1.0&rsz=8&q=" + b = http.request(api..text) + local google = json:decode(b) + + if (google.responseStatus == 200) then -- OK + math.randomseed(os.time()) + i = math.random(#google.responseData.results) -- Random image from results + return google.responseData.results[i].url + end + end end function get_9GAG()