Merge Upstream

miku/bot.lua

@@ -113,11 +113,6 @@ function bot:on_callback_receive(callback, msg, config) -- whenever a new callba
   -- vardump(msg)
   -- vardump(callback)
 
-  if msg.date < os.time() - 3600 then -- Do not process old messages.
-    utilities.answer_callback_query(callback, 'Nachricht älter als eine Stunde, bitte sende den Befehl selbst noch einmal.', true)
-    return
-  end
-  
   if not callback.data:find(':') then
     utilities.answer_callback_query(callback, 'Ungültiger CallbackQuery: Kein Parameter.')
 	return
@@ -125,36 +120,40 @@ function bot:on_callback_receive(callback, msg, config) -- whenever a new callba
 
   -- Check if user is blocked
   local user_id = callback.from.id
-  local chat_id = msg.chat.id
   if redis:get('blocked:'..user_id) then
     utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
 	return
   end
  
-  -- Check if user is banned
-  local banned = redis:get('banned:'..chat_id..':'..user_id)
-  if banned then
-    utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
-	return
-  end
+  -- Check if user is banned, not working for Callbacks from InlineQuerys!
+  if msg then
+    local chat_id = msg.chat.id
+    local banned = redis:get('banned:'..chat_id..':'..user_id)
+    if banned then
+      utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
+	  return
+    end
   
-  -- Check if whitelist is enabled and user/chat is whitelisted
-  local whitelist = redis:get('whitelist:enabled')
-  if whitelist and not is_sudo(callback, config) then
-	local hash = 'whitelist:user#id'..user_id
-	local allowed = redis:get(hash) or false
-	if not allowed then
-      if msg.chat.type == 'group' or msg.chat.type == 'supergroup' then
-        local allowed = redis:get('whitelist:chat#id'.. chat_id)
-	    if not allowed then
+    -- Check if whitelist is enabled and user/chat is whitelisted
+    local whitelist = redis:get('whitelist:enabled')
+    if whitelist and not is_sudo(callback, config) then
+	  local hash = 'whitelist:user#id'..user_id
+	  local allowed = redis:get(hash) or false
+	  if not allowed then
+        if msg.chat.type == 'group' or msg.chat.type == 'supergroup' then
+          local allowed = redis:get('whitelist:chat#id'.. chat_id)
+	      if not allowed then
+	        utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
+		    return
+	      end
+	    else
 	      utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
 		  return
 	    end
-	  else
-	    utilities.answer_callback_query(callback, 'Du darfst den Bot nicht nutzen!', true)
-		return
 	  end
-	end
+    end
+
+    msg = utilities.enrich_message(msg)
   end
 
   local called_plugin = callback.data:match('(.*):.*')
@@ -162,12 +161,18 @@ function bot:on_callback_receive(callback, msg, config) -- whenever a new callba
 
   print('Callback Query "'..param..'" für Plugin "'..called_plugin..'" ausgelöst von '..callback.from.first_name..' ('..callback.from.id..')')
 
-  msg = utilities.enrich_message(msg)
-
   for n=1, #self.plugins do
     local plugin = self.plugins[n]
 	if plugin.name == called_plugin then
-	  if is_plugin_disabled_on_chat(plugin.name, msg) then utilities.answer_callback_query(callback, 'Plugin wurde in diesem Chat deaktiviert.') return end
+      
+      -- Check if plugin is disabled on this chat
+      if msg then
+	    if is_plugin_disabled_on_chat(plugin.name, msg) then
+          utilities.answer_callback_query(callback, 'Plugin wurde in diesem Chat deaktiviert.')
+          return
+        end
+      end
+      
       if plugin.callback then
 	    plugin:callback(callback, msg, self, config, param)
         return

miku/plugins/cats.lua

@@ -23,8 +23,8 @@ function cats:init(config)
 *]]..config.cmd_pat..[[kitty* _gif_: Postet eine zufällige, animierte Katze]]
 end
 
-
 local apikey = cred_data.cat_apikey or "" -- apply for one here: http://thecatapi.com/api-key-registration.html
+local BASE_URL = 'http://thecatapi.com/api/images/get'
 
 function cats:inline_callback(inline_query, config, matches)
   if matches[1] == 'gif' then
@@ -59,15 +59,53 @@ function cats:inline_callback(inline_query, config, matches)
   utilities.answer_inline_query(inline_query, results, 30)
 end
 
-function cats:action(msg, config, matches)
-  if matches[1] == 'gif' then
-    local url = 'http://thecatapi.com/api/images/get?type=gif&apikey='..apikey
-	local file = download_to_file(url, 'miau.gif')
-    utilities.send_document(msg.chat.id, file, nil, msg.message_id)
+function cats:get_cat(gif)
+  if gif then
+    local url = BASE_URL..'?type=gif&apikey='..apikey
+    file = download_to_file(url, 'miau.gif')
   else
-    local url = 'http://thecatapi.com/api/images/get?type=jpg,png&apikey='..apikey
-	local file = download_to_file(url, 'miau.png')
-    utilities.send_photo(msg.chat.id, file, nil, msg.message_id)
+    local url = BASE_URL..'?type=jpg,png&apikey='..apikey
+    file = download_to_file(url, 'miau.png')
+  end
+  return file
+end
+
+function cats:callback(callback, msg, self, config, input)
+  utilities.answer_callback_query(callback, 'Miau!')
+  utilities.send_typing(msg.chat.id, 'upload_photo')
+  if string.isempty(input) then
+    local file = cats:get_cat()
+    if not file then
+      utilities.answer_callback_query(callback, 'Beim Herunterladen ist ein Fehler aufgetreten :(', true)
+      return
+    end
+	utilities.send_photo(msg.chat.id, file, nil, msg.message_id, '{"inline_keyboard":[[{"text":"Nochmal!","callback_data":"cats:"}]]}')
+  else
+    local file = cats:get_cat(true)
+    if not file then
+      utilities.answer_callback_query(callback, 'Beim Herunterladen ist ein Fehler aufgetreten :(', true)
+      return
+    end
+	utilities.send_document(msg.chat.id, file, nil, msg.message_id, '{"inline_keyboard":[[{"text":"Nochmal!","callback_data":"cats:gif"}]]}')
+  end
+end
+
+function cats:action(msg, config, matches)
+  utilities.send_typing(msg.chat.id, 'upload_photo')
+  if matches[1] == 'gif' then
+    local file = cats:get_cat(true)
+    if not file then
+      utilities.send_reply(msg, config.errors.connection)
+      return
+    end
+    utilities.send_document(msg.chat.id, file, nil, msg.message_id, '{"inline_keyboard":[[{"text":"Nochmal!","callback_data":"cats:gif"}]]}')
+  else
+    local file = cats:get_cat()
+    if not file then
+      utilities.send_reply(msg, config.errors.connection)
+      return
+    end
+    utilities.send_photo(msg.chat.id, file, nil, msg.message_id, '{"inline_keyboard":[[{"text":"Nochmal!","callback_data":"cats:"}]]}')
   end
 end