]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/blobdiff - lib/rbot/botuser.rb
New Auth Framework: BotUser transiency is now checked with #transient?
[user/henk/code/ruby/rbot.git] / lib / rbot / botuser.rb
index 615b18d69117ae72c8e6759334a7594f2ca99c8c..88da4da48593c36ef44d91e6fdd18ce9ec789c31 100644 (file)
@@ -8,6 +8,7 @@
 # License:: GPLv2\r
 \r
 require 'singleton'\r
+require 'set'\r
 \r
 \r
 module Irc\r
@@ -27,6 +28,9 @@ module Irc
     BotConfig.register BotConfigBooleanValue.new( 'auth.autologin',\r
       :default => 'true',\r
       :desc => _('Set false to prevent new botusers from recognizing IRC users without a need to manually login'))\r
+    BotConfig.register BotConfigBooleanValue.new( 'auth.autouser',\r
+      :default => 'false',\r
+      :desc => _('Set true to allow new botusers to be created automatically'))\r
     # BotConfig.register BotConfigIntegerValue.new( 'auth.default_level',\r
     #   :default => 10, :wizard => true,\r
     #   :desc => 'The default level for new/unknown users' )\r
@@ -177,8 +181,35 @@ module Irc
     end\r
 \r
 \r
-    # This is the basic class for bot users: they have a username, a password,\r
-    # a list of netmasks to match against, and a list of permissions.\r
+    # This is the basic class for bot users: they have a username, a\r
+    # password, a list of netmasks to match against, and a list of\r
+    # permissions. A BotUser can be marked as 'transient', usually meaning\r
+    # it's not intended for permanent storage. Transient BotUsers have lower\r
+    # priority than nontransient ones for autologin purposes.\r
+    #\r
+    # To initialize a BotUser, you pass a _username_ and an optional\r
+    # hash of options. Currently, only two options are recognized:\r
+    #\r
+    # transient:: true or false, determines if the BotUser is transient or\r
+    #             permanent (default is false, permanent BotUser).\r
+    #\r
+    #             Transient BotUsers are initialized by prepending an\r
+    #             asterisk (*) to the username, and appending a sanitized\r
+    #             version of the object_id. The username can be empty.\r
+    #             A random password is generated.\r
+    #\r
+    #             Permanent Botusers need the username as is, and no\r
+    #             password is generated.\r
+    #\r
+    # masks::     an array of Netmasks to initialize the NetmaskList. This\r
+    #             list is used as-is for permanent BotUsers.\r
+    #\r
+    #             Transient BotUsers will alter the list elements which are\r
+    #             Irc::User by globbing the nick and any initial nonletter\r
+    #             part of the ident.\r
+    #\r
+    #             The masks option is optional for permanent BotUsers, but\r
+    #             obligatory (non-empty) for transients.\r
     #\r
     class BotUser\r
 \r
@@ -188,24 +219,66 @@ module Irc
       attr_reader :perm\r
       attr_writer :login_by_mask\r
       attr_writer :autologin\r
-      attr_accessor :transient\r
+      attr_writer :transient\r
+\r
+      # Checks if the BotUser is transient\r
+      def transient?\r
+        @transient\r
+      end\r
+\r
+      # Checks if the BotUser is permanent (not transient)\r
+      def permanent?\r
+        !@permanent\r
+      end\r
+\r
+      # Sets if the BotUser is permanent or not\r
+      def permanent=(bool)\r
+        @transient=!bool\r
+      end\r
 \r
       # Create a new BotUser with given username\r
       def initialize(username, options={})\r
         opts = {:transient => false}.merge(options)\r
         @transient = opts[:transient]\r
-        @username = @transient ? "*" : ""\r
-        @username << BotUser.sanitize_username(username)\r
-        @password = nil\r
+\r
+        if @transient\r
+          @username = "*"\r
+          @username << BotUser.sanitize_username(username) if username and not username.to_s.empty?\r
+          @username << BotUser.sanitize_username(object_id)\r
+          reset_password\r
+          @login_by_mask=true\r
+          @autologin=true\r
+        else\r
+          @username = BotUser.sanitize_username(username)\r
+          @password = nil\r
+          reset_login_by_mask\r
+          reset_autologin\r
+        end\r
+\r
         @netmasks = NetmaskList.new\r
+        if opts.key?(:masks) and opts[:masks]\r
+          masks = opts[:masks]\r
+          masks = [masks] unless masks.respond_to?(:each)\r
+          masks.each { |m|\r
+            mask = m.to_irc_netmask\r
+            if @transient and User === m\r
+              mask.nick = "*"\r
+              mask.host = m.host.dup\r
+              mask.user = "*" + m.user.sub(/^\w?[^\w]+/,'')\r
+            end\r
+            add_netmask(mask) unless mask.to_s == "*"\r
+          }\r
+        end\r
+        raise "must provide a usable mask for transient BotUser #{@username}" if @transient and @netmasks.empty?\r
+\r
         @perm = {}\r
-        reset_login_by_mask\r
-        reset_autologin\r
       end\r
 \r
       # Inspection\r
       def inspect\r
-        str = "<#{self.class}:#{'0x%08x' % self.object_id}:"\r
+        str = "<#{self.class}:#{'0x%08x' % self.object_id}"\r
+        str << " (transient)" if @transient\r
+        str << ":"\r
         str << " @username=#{@username.inspect}"\r
         str << " @netmasks=#{@netmasks.inspect}"\r
         str << " @perm=#{@perm.inspect}"\r
@@ -356,7 +429,7 @@ module Irc
       # It returns true or false depending on whether the password\r
       # is right. If it is, the Netmask of the user is added to the\r
       # list of acceptable Netmask unless it's already matched.\r
-      def login(user, password)\r
+      def login(user, password=nil)\r
         if password == @password or (password.nil? and (@login_by_mask || @autologin) and knows?(user))\r
           add_netmask(user) unless knows?(user)\r
           debug "#{user} logged in as #{self.inspect}"\r
@@ -382,15 +455,6 @@ module Irc
 \r
     end\r
 \r
-    # A TransientBotUser is a BotUser that is not intended\r
-    # for permanent storage.\r
-    #\r
-    class TransientBotUser < BotUser\r
-      def initialize\r
-        super(object_id, :transient => true)\r
-      end\r
-    end\r
-\r
     # This is the default BotUser: it's used for all users which haven't\r
     # identified with the bot\r
     #\r
@@ -557,6 +621,7 @@ module Irc
         [everyone, botowner].each { |x|\r
           @allbotusers[x.username.to_sym] = x\r
         }\r
+        @transients = Set.new\r
       end\r
 \r
       def load_array(ary, forced)\r
@@ -580,7 +645,7 @@ module Irc
 \r
       def save_array\r
         @allbotusers.values.map { |x|\r
-          x.transient ? nil : x.to_hash\r
+          x.transient? ? nil : x.to_hash\r
         }.compact\r
       end\r
 \r
@@ -647,9 +712,29 @@ module Irc
           debug "Checking with #{n}"\r
           return bu if bu.autologin? and login(ircuser, n)\r
         }\r
+        # Check with transient users\r
+        @transients.each { |bu|\r
+          return bu if bu.login(ircuser)\r
+        }\r
+        # Finally, create a transient if we're set to allow it\r
+        if @bot.config['auth.autouser']\r
+          bu = create_transient_botuser(ircuser)\r
+          return bu\r
+        end\r
         return everyone\r
       end\r
 \r
+      # Creates a new transient BotUser associated with Irc::User _user_,\r
+      # automatically logging him in\r
+      #\r
+      def create_transient_botuser(user)\r
+        ircuser = user.to_irc_user\r
+        bu = BotUser.new(ircuser, :transient => true, :masks => ircuser)\r
+        bu.login(ircuser)\r
+        @transients << bu\r
+        return bu\r
+      end\r
+\r
       # Checks if User _user_ can do _cmd_ on _chan_.\r
       #\r
       # Permission are checked in this order, until a true or false\r