]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/blobdiff - lib/rbot/botuser.rb
Rename the file core.rb to reflect the new botmoulde name
[user/henk/code/ruby/rbot.git] / lib / rbot / botuser.rb
index ea47fcbd7641b33ab51ae3966bb783fd63c8947e..5cb05f1734d09523ff5be54b04b03b5be10d5822 100644 (file)
@@ -9,19 +9,8 @@
 \r
 require 'singleton'\r
 \r
-module Irc\r
-\r
-  # This method raises a TypeError if _user_ is not of class User\r
-  #\r
-  def Irc.error_if_not_user(user)\r
-    raise TypeError, "#{user.inspect} must be of type Irc::User and not #{user.class}" unless user.class <= User\r
-  end\r
 \r
-  # This method raises a TypeError if _chan_ is not of class Chan\r
-  #\r
-  def Irc.error_if_not_channel(chan)\r
-    raise TypeError, "#{chan.inspect} must be of type Irc::User and not #{chan.class}" unless chan.class <= Channel\r
-  end\r
+module Irc\r
 \r
 \r
   # This module contains the actual Authentication stuff\r
@@ -31,6 +20,12 @@ module Irc
     BotConfig.register BotConfigStringValue.new( 'auth.password',\r
       :default => 'rbotauth', :wizard => true,\r
       :desc => 'Password for the bot owner' )\r
+    BotConfig.register BotConfigBooleanValue.new( 'auth.login_by_mask',\r
+      :default => 'false',\r
+      :desc => 'Set true if new botusers should allow logging in without a password when the user netmask is known')\r
+    BotConfig.register BotConfigBooleanValue.new( 'auth.login_auto',\r
+      :default => 'false',\r
+      :desc => 'Set true if new botusers should try to recognize IRC users without a need to manually login')\r
     # BotConfig.register BotConfigIntegerValue.new( 'auth.default_level',\r
     #   :default => 10, :wizard => true,\r
     #   :desc => 'The default level for new/unknown users' )\r
@@ -84,14 +79,33 @@ module Irc
         debug "Created command #{@command.inspect} with path #{@path.join(', ')}"\r
       end\r
 \r
-    end\r
+      # Returs self\r
+      def to_irc_auth_command\r
+        self\r
+      end\r
 \r
-    # This method raises a TypeError if _user_ is not of class User\r
-    #\r
-    def Irc.error_if_not_command(cmd)\r
-      raise TypeError, "#{cmd.inspect} must be of type Irc::Auth::Command and not #{cmd.class}" unless cmd.class <= Command\r
     end\r
 \r
+  end\r
+\r
+end\r
+\r
+\r
+class String\r
+\r
+  # Returns an Irc::Auth::Comand from the receiver\r
+  def to_irc_auth_command\r
+    Irc::Auth::Command.new(self)\r
+  end\r
+\r
+end\r
+\r
+\r
+module Irc\r
+\r
+\r
+  module Auth\r
+\r
 \r
     # This class describes a permission set\r
     class PermissionSet\r
@@ -109,8 +123,8 @@ module Irc
 \r
       # Sets the permission for command _cmd_ to _val_,\r
       #\r
-      def set_permission(cmd, val)\r
-        Irc::error_if_not_command(cmd)\r
+      def set_permission(str, val)\r
+        cmd = str.to_irc_auth_command\r
         case val\r
         when true, false\r
           @perm[cmd.command] = val\r
@@ -130,8 +144,8 @@ module Irc
       # Tells if command _cmd_ is permitted. We do this by returning\r
       # the value of the deepest Command#path that matches.\r
       #\r
-      def permit?(cmd)\r
-        Irc::error_if_not_command(cmd)\r
+      def permit?(str)\r
+        cmd = str.to_irc_auth_command\r
         allow = nil\r
         cmd.path.reverse.each { |k|\r
           if @perm.has_key?(k)\r
@@ -145,14 +159,16 @@ module Irc
     end\r
 \r
 \r
-    # This is the basic class for bot users: they have a username, a password, a\r
-    # 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 password,\r
+    # list of netmasks to match against, and a list of permissions.\r
     #\r
     class BotUser\r
 \r
       attr_reader :username\r
       attr_reader :password\r
       attr_reader :netmasks\r
+      attr_writer :login_by_mask\r
+      attr_writer :autologin\r
 \r
       # Create a new BotUser with given username\r
       def initialize(username)\r
@@ -160,6 +176,8 @@ module Irc
         @password = nil\r
         @netmasks = NetmaskList.new\r
         @perm = {}\r
+        @login_by_mask = Auth.manager.bot.config['auth.login_by_mask'] unless defined?(@login_by_mask)\r
+        @autologin = Auth.manager.bot.config['auth.login_auto'] unless defined?(@autologin)\r
       end\r
 \r
       # Inspection\r
@@ -168,7 +186,9 @@ module Irc
         str << " @username=#{@username.inspect}"\r
         str << " @netmasks=#{@netmasks.inspect}"\r
         str << " @perm=#{@perm.inspect}"\r
-        str\r
+        str << " @login_by_mask=#{@login_by_mask}"\r
+        str << " @autologin=#{@autologin}"\r
+        str << ">"\r
       end\r
 \r
       # Convert into a hash\r
@@ -177,16 +197,32 @@ module Irc
           :username => @username,\r
           :password => @password,\r
           :netmasks => @netmasks,\r
-          :perm => @perm\r
+          :perm => @perm,\r
+          :login_by_mask => @login_by_mask,\r
+          :autologin => @autologin\r
         }\r
       end\r
 \r
+      # Do we allow logging in without providing the password?\r
+      #\r
+      def login_by_mask?\r
+        @login_by_mask\r
+      end\r
+\r
+      # Do we allow automatic logging in?\r
+      #\r
+      def autologin?\r
+        @autologin\r
+      end\r
+\r
       # Restore from hash\r
       def from_hash(h)\r
         @username = h[:username] if h.has_key?(:username)\r
         @password = h[:password] if h.has_key?(:password)\r
         @netmasks = h[:netmasks] if h.has_key?(:netmasks)\r
         @perm = h[:perm] if h.has_key?(:perm)\r
+        @login_by_mask = h[:login_by_mask] if h.has_key?(:login_by_mask)\r
+        @autologin = h[:autologin] if h.has_key?(:autologin)\r
       end\r
 \r
       # This method sets the password if the proposed new password\r
@@ -217,12 +253,7 @@ module Irc
       def set_permission(cmd, val, chan="*")\r
         k = chan.to_s.to_sym\r
         @perm[k] = PermissionSet.new unless @perm.has_key?(k)\r
-        case cmd\r
-        when String\r
-          @perm[k].set_permission(Command.new(cmd), val)\r
-        else\r
-          @perm[k].set_permission(cmd, val)\r
-        end\r
+        @perm[k].set_permission(cmd, val)\r
       end\r
 \r
       # Resets the permission for command _cmd_ on channel _chan_\r
@@ -250,34 +281,26 @@ module Irc
       # Adds a Netmask\r
       #\r
       def add_netmask(mask)\r
-        case mask\r
-        when Netmask\r
-          @netmasks << mask\r
-        else\r
-          @netmasks << Netmask.new(mask)\r
-        end\r
+        @netmasks << mask.to_irc_netmask\r
       end\r
 \r
       # Removes a Netmask\r
       #\r
       def delete_netmask(mask)\r
-        case mask\r
-        when Netmask\r
-          m = mask\r
-        else\r
-          m << Netmask.new(mask)\r
-        end\r
+        m = mask.to_irc_netmask\r
         @netmasks.delete(m)\r
       end\r
 \r
       # Removes all <code>Netmask</code>s\r
+      #\r
       def reset_netmask_list\r
         @netmasks = NetmaskList.new\r
       end\r
 \r
       # This method checks if BotUser has a Netmask that matches _user_\r
-      def knows?(user)\r
-        Irc::error_if_not_user(user)\r
+      #\r
+      def knows?(usr)\r
+        user = usr.to_irc_user\r
         known = false\r
         @netmasks.each { |n|\r
           if user.matches?(n)\r
@@ -293,7 +316,7 @@ module Irc
       # 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
-        if password == @password\r
+        if password == @password or (@password.nil? and @login_by_mask and knows?(user))\r
           add_netmask(user) unless knows?(user)\r
           debug "#{user} logged in as #{self.inspect}"\r
           return true\r
@@ -326,11 +349,30 @@ module Irc
 \r
       include Singleton\r
 \r
+      # The default BotUser is named 'everyone', it doesn't allow autologin\r
+      # (meaningless) and it allows login-by-mask\r
+      #\r
       def initialize\r
+        @login_by_mask = true\r
+        @autologin = false\r
         super("everyone")\r
         @default_perm = PermissionSet.new\r
       end\r
 \r
+      # This method returns without changing anything\r
+      #\r
+      def login_by_mask=(val)\r
+        debug "Tried to change the login-by-mask for default bot user, ignoring"\r
+        return @login_by_mask\r
+      end\r
+\r
+      # This method returns without changing anything\r
+      #\r
+      def autologin=(val)\r
+        debug "Tried to change the autologin for default bot user, ignoring"\r
+        return\r
+      end\r
+\r
       # Sets the default permission for the default user (i.e. the ones\r
       # set by the BotModule writers) on all channels\r
       #\r
@@ -342,7 +384,11 @@ module Irc
       # default knows everybody\r
       #\r
       def knows?(user)\r
-        Irc::error_if_not_user(user)\r
+        return true if user.to_irc_user\r
+      end\r
+\r
+      # We always allow logging in as the default user\r
+      def login(user, password)\r
         return true\r
       end\r
 \r
@@ -379,6 +425,8 @@ module Irc
       include Singleton\r
 \r
       def initialize\r
+        @login_by_mask = false\r
+        @autologin = false\r
         super("owner")\r
       end\r
 \r
@@ -452,7 +500,7 @@ module Irc
         raise "Won't load with unsaved changes" if @has_changes and not forced\r
         reset_hashes\r
         ary.each { |x|\r
-          raise TypeError, "#{x} should be a Hash" unless x.class <= Hash\r
+          raise TypeError, "#{x} should be a Hash" unless x.kind_of?(Hash)\r
           u = x[:username]\r
           unless include?(u)\r
             create_botuser(u)\r
@@ -475,9 +523,8 @@ module Irc
 \r
       # Maps <code>Irc::User</code> to BotUser\r
       def irc_to_botuser(ircuser)\r
-        Irc::error_if_not_user(ircuser)\r
         # TODO check netmasks\r
-        return @botusers[ircuser] || everyone\r
+        @botusers[ircuser.to_irc_user] || everyone\r
       end\r
 \r
       # creates a new BotUser\r
@@ -495,32 +542,44 @@ module Irc
         @allbotusers.fetch(BotUser.sanitize_username(name).to_sym)\r
       end\r
 \r
-      # Logs Irc::User _ircuser_ in to BotUser _botusername_ with password _pwd_\r
+      # Logs Irc::User _user_ in to BotUser _botusername_ with password _pwd_\r
       #\r
       # raises an error if _botusername_ is not a known BotUser username\r
       #\r
       # It is possible to autologin by Netmask, on request\r
       #\r
-      def login(ircuser, botusername, pwd, bymask = false)\r
-        Irc::error_if_not_user(ircuser)\r
+      def login(user, botusername, pwd)\r
+        ircuser = user.to_irc_user\r
         n = BotUser.sanitize_username(botusername)\r
         k = n.to_sym\r
         raise "No such BotUser #{n}" unless include?(k)\r
         if @botusers.has_key?(ircuser)\r
+          return true if @botusers[ircuser].name = n\r
           # TODO\r
           # @botusers[ircuser].logout(ircuser)\r
         end\r
         bu = @allbotusers[k]\r
-        if bymask && bu.knows?(ircuser)\r
-          @botusers[ircuser] = bu\r
-          return true\r
-        elsif bu.login(ircuser, pwd)\r
+        if bu.login(ircuser, pwd)\r
           @botusers[ircuser] = bu\r
           return true\r
         end\r
         return false\r
       end\r
 \r
+      # Tries to auto-login Irc::User _user_ by looking at the known botusers that allow autologin\r
+      # and trying to login without a password\r
+      #\r
+      def autologin(user)\r
+        ircuser = user.to_irc_user\r
+        debug "Trying to autlogin #{ircuser}"\r
+        return @botusers[ircuser] if @botusers.has_key?(ircuser)\r
+        @allbotusers.each { |n, bu|\r
+          debug "Checking with #{n}"\r
+          return bu if bu.autologin? and login(ircuser, n)\r
+        }\r
+        return everyone\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
@@ -530,10 +589,11 @@ module Irc
       # * everyone on _chan_\r
       # * everyone on all channels\r
       #\r
-      def permit?(user, cmdtxt, chan=nil)\r
+      def permit?(user, cmdtxt, channel=nil)\r
         botuser = irc_to_botuser(user)\r
-        cmd = Command.new(cmdtxt)\r
+        cmd = cmdtxt.to_irc_auth_command\r
 \r
+        chan = channel\r
         case chan\r
         when User\r
           chan = "?"\r