hangman plugin: rudimentary stats tracking along with some other enhancements

[user/henk/code/ruby/rbot.git] / data / rbot / plugins / autoop.rb

diff --git a/data/rbot/plugins/autoop.rb b/data/rbot/plugins/autoop.rb
index 9da5bd379d4224f076375fec9cab4f9310da9435..e910bcd91d07de2957bd1488711a68b518e28a79 100644 (file)
--- a/data/rbot/plugins/autoop.rb
+++ b/data/rbot/plugins/autoop.rb
@@ -1,56 +1,96 @@
 class AutoOP < Plugin
-    def help(plugin, topic="")
-        "perform autoop based on hostmask - usage: addop <hostmask> [channel channel ...], rmop <hostmask> [channel], list - list current ops. If you don't specify which channels, all channels are assumed"
-    end
-    
-    def join(m)
-      return if m.address?
+  Config.register Config::BooleanValue.new('autoop.on_nick',
+    :default => true,
+    :desc => "Determines if the bot should auto-op when someone changes nick and the new nick matches a listed netmask")
+
+  def help(plugin, topic="")
+    return "perform autoop based on hostmask - usage: add <hostmask> [channel channel ...], rm <hostmask> [channel], list - list current ops. If you don't specify which channels, all channels are assumed"
+  end
+
+  def join(m)
+    return if m.address?
+    @registry.each { |mask,channels|
+      if m.source.matches?(mask.to_irc_netmask(:server => m.server)) &&
+        (channels.empty? || channels.include?(m.channel.to_s))
+        @bot.mode(m.channel, "+o", m.source.nick)
+        return
+      end
+    }
+  end
+
+  def nick(m)
+    return if m.address?
+    return unless @bot.config['autoop.on_nick']
+    is_on = m.server.channels.inject(ChannelList.new) { |list, ch|
+      list << ch if ch.users.include?(m.source)
+      list
+    }
+    is_on.each { |channel|
+      ch = channel.to_s
       @registry.each { |mask,channels|
-        if(Irc.netmaskmatch(mask, m.source) &&
-            (channels.empty? || channels.include?(m.channel)))
-          @bot.mode(m.channel, "+o", m.sourcenick)
+        if m.source.matches?(mask.to_irc_netmask(:server => m.server)) &&
+          (channels.empty? || channels.include?(ch))
+          @bot.mode(ch, "+o", m.source.nick)
           return
         end
       }
-    end
+    }
+  end
 
-    def add(m, params)
+  def add(m, params)
+    if params[:channels].empty? || !@registry.has_key?(params[:mask])
+      # if the channels parameter is omitted (meaning all channels), or the
+      # hostmask isn't present in the registry, we just (over)write the channels
+      # in the registry
       @registry[params[:mask]] = params[:channels].dup
       m.okay
-    end
-
-    def rm(m, params)
-      unless @registry.has_key?(params[:mask])
-        m.reply @bot.lang.get('dunno')
-        return
-      end
-      if (params[:channels] && @registry[params[:mask]] != nil)
-        params[:channels].each do |c|
-          @registry[params[:mask]] = @registry[params[:mask]].reject {|ele| ele =~ /^#{c}$/i}
-        end
-      elsif(!@registry.delete(params[:mask]))
-        m.reply @bot.lang.get('dunno')
+    else
+      # otherwise, merge the channels with the ones existing in the registry
+      current_channels = @registry[params[:mask]]
+      if current_channels.empty?
+        m.reply "#{params[:mask]} is already being auto-opped on all channels"
       else
+        # merge the already set channels
+        @registry[params[:mask]] = (params[:channels] | current_channels).uniq
         m.okay
       end
     end
+  end
 
-    def list(m, params)
-      if(@registry.length)
-        @registry.each { |mask,channels|
-          m.reply "#{mask} in #{channels.empty? ? 'all channels' : channels.join(', ')}"
-        }
-      else
-        m.reply "No entrys"
+  def rm(m, params)
+    unless @registry.has_key?(params[:mask])
+      m.reply @bot.lang.get('dunno')
+      return
+    end
+    if (!params[:channels].empty? && @registry[params[:mask]] != nil)
+      params[:channels].each do |c|
+        @registry[params[:mask]] = @registry[params[:mask]].reject {|ele| ele =~ /^#{c}$/i}
+      end
+      if @registry[params[:mask]].empty?
+        @registry.delete(params[:mask])
       end
+    else
+      @registry.delete(params[:mask])
+    end
+    m.okay
+  end
+
+  def list(m, params)
+    debug @registry.length
+    if(@registry.length > 0)
+      @registry.each { |mask,channels|
+        m.reply "#{mask} in #{channels.empty? ? 'all channels' : channels.join(', ')}"
+      }
+    else
+      m.reply "No entries"
     end
+  end
 end
 
 plugin = AutoOP.new
 
 plugin.map 'autoop list', :action => 'list'
-plugin.map 'autoop add :mask *channels', :action => 'add'
-plugin.map 'autoop add :mask', :action => 'add'
-plugin.map 'autoop rm :mask *channels', :action => 'rm'
-plugin.map 'autoop rm :mask', :action => 'rm'
+plugin.map 'autoop add :mask [*channels]', :action => 'add'
+plugin.map 'autoop rm :mask [*channels]', :action => 'rm'
 
+plugin.default_auth('*',false)