Merge branch 'soend-master'

bot/bot.lua

@@ -34,14 +34,18 @@ function msg_valid(msg)
     print("Not valid, msg from us")
     return false
   end
+  
+  -- Before bot was started
   if msg.date < now then
     print("Not valid, old msg")
     return false
   end
+  
   if msg.unread == 0 then
     print("Not valid, readed")
     return false
   end
+
   return true
 end
 
@@ -158,6 +162,7 @@ function create_config( )
       "location",
       "media",
       "plugins",
+      "channels",
       "set",
       "stats",
       "time",
@@ -165,7 +170,8 @@ function create_config( )
       "weather",
       "xkcd",
       "youtube" },
-    sudo_users = {our_id}  
+    sudo_users = {our_id},
+    disabled_channels = {}
   }
   serialize_to_file(config, './data/config.lua')
   print ('saved config into ./data/config.lua')

plugins/channels.lua

@@ -0,0 +1,80 @@
+-- Checks if bot was disabled on specific chat
+local function is_channel_disabled( receiver )
+	if not _config.disabled_channels then
+		return false
+	end
+
+	if _config.disabled_channels[receiver] == nil then
+		return false
+	end
+
+  return _config.disabled_channels[receiver]
+end
+
+local function enable_channel(receiver)
+	if not _config.disabled_channels then
+		_config.disabled_channels = {}
+	end
+
+	if _config.disabled_channels[receiver] == nil then
+		return 'Channel isn\'t disabled'
+	end
+	
+	_config.disabled_channels[receiver] = false
+
+	save_config()
+	return "Channel reenabled"
+end
+
+local function disable_channel( receiver )
+	if not _config.disabled_channels then
+		_config.disabled_channels = {}
+	end
+	
+	_config.disabled_channels[receiver] = true
+
+	save_config()
+	return "Channel disabled"
+end
+
+local function pre_process(msg)
+	local receiver = get_receiver(msg)
+	
+	-- If is sudo can reeanble the channel
+	if is_sudo(msg) then
+	  if msg.text == "!channel enable" then
+	    enable_channel(receiver)
+	  end
+	end
+
+  if is_channel_disabled(receiver) then
+  	msg.text = ""
+  end
+
+	return msg
+end
+
+local function run(msg, matches)
+	local receiver = get_receiver(msg)
+	-- Enable a channel
+	if matches[1] == 'enable' then
+		return enable_channel(receiver)
+	end
+	-- Disable a channel
+	if matches[1] == 'disable' then
+		return disable_channel(receiver)
+	end
+end
+
+return {
+	description = "Plugin to manage channels. Enable or disable channel.", 
+	usage = {
+		"!channel enable: enable current channel",
+		"!channel disable: disable current channel" },
+	patterns = {
+		"^!channel? (enable)",
+		"^!channel? (disable)" }, 
+	run = run,
+	privileged = true,
+	pre_process = pre_process
+}