From 0b705e79109fa004bf8956ef65288cf00a9f3312 Mon Sep 17 00:00:00 2001 From: Giuseppe Bilotta Date: Sat, 5 Aug 2006 00:18:40 +0000 Subject: Adjust auth framework to work with improved Irc framework and fix some issues in the Irc framework itself. Get rid of some of the debug clauses --- lib/rbot/botuser.rb | 92 +++++++++++++++++++++++++---------------------------- lib/rbot/irc.rb | 11 +++---- 2 files changed, 47 insertions(+), 56 deletions(-) (limited to 'lib/rbot') diff --git a/lib/rbot/botuser.rb b/lib/rbot/botuser.rb index b324a2a7..1c2ef929 100644 --- a/lib/rbot/botuser.rb +++ b/lib/rbot/botuser.rb @@ -9,19 +9,8 @@ require 'singleton' -module Irc - # This method raises a TypeError if _user_ is not of class User - # - def Irc.error_if_not_user(user) - raise TypeError, "#{user.inspect} must be of type Irc::User and not #{user.class}" unless user.kind_of?(User) - end - - # This method raises a TypeError if _chan_ is not of class Chan - # - def Irc.error_if_not_channel(chan) - raise TypeError, "#{chan.inspect} must be of type Irc::User and not #{chan.class}" unless chan.kind_of?(Channel) - end +module Irc # This module contains the actual Authentication stuff @@ -84,14 +73,33 @@ module Irc debug "Created command #{@command.inspect} with path #{@path.join(', ')}" end - end + # Returs self + def to_irc_auth_command + self + end - # This method raises a TypeError if _user_ is not of class User - # - def Irc.error_if_not_command(cmd) - raise TypeError, "#{cmd.inspect} must be of type Irc::Auth::Command and not #{cmd.class}" unless cmd.kind_of?(Command) end + end + +end + + +class String + + # Returns an Irc::Auth::Comand from the receiver + def to_irc_auth_command + Irc::Auth::Command.new(self) + end + +end + + +module Irc + + + module Auth + # This class describes a permission set class PermissionSet @@ -109,8 +117,8 @@ module Irc # Sets the permission for command _cmd_ to _val_, # - def set_permission(cmd, val) - Irc::error_if_not_command(cmd) + def set_permission(str, val) + cmd = str.to_irc_auth_command case val when true, false @perm[cmd.command] = val @@ -130,8 +138,8 @@ module Irc # Tells if command _cmd_ is permitted. We do this by returning # the value of the deepest Command#path that matches. # - def permit?(cmd) - Irc::error_if_not_command(cmd) + def permit?(str) + cmd = str.to_irc_auth_command allow = nil cmd.path.reverse.each { |k| if @perm.has_key?(k) @@ -217,12 +225,7 @@ module Irc def set_permission(cmd, val, chan="*") k = chan.to_s.to_sym @perm[k] = PermissionSet.new unless @perm.has_key?(k) - case cmd - when String - @perm[k].set_permission(Command.new(cmd), val) - else - @perm[k].set_permission(cmd, val) - end + @perm[k].set_permission(cmd, val) end # Resets the permission for command _cmd_ on channel _chan_ @@ -250,34 +253,26 @@ module Irc # Adds a Netmask # def add_netmask(mask) - case mask - when Netmask - @netmasks << mask - else - @netmasks << Netmask.new(mask) - end + @netmasks << mask.to_irc_netmask end # Removes a Netmask # def delete_netmask(mask) - case mask - when Netmask - m = mask - else - m << Netmask.new(mask) - end + m = mask.to_irc_netmask @netmasks.delete(m) end # Removes all Netmasks + # def reset_netmask_list @netmasks = NetmaskList.new end # This method checks if BotUser has a Netmask that matches _user_ - def knows?(user) - Irc::error_if_not_user(user) + # + def knows?(usr) + user = usr.to_irc_user known = false @netmasks.each { |n| if user.matches?(n) @@ -342,8 +337,7 @@ module Irc # default knows everybody # def knows?(user) - Irc::error_if_not_user(user) - return true + return true if user.to_irc_user end # Resets the NetmaskList @@ -475,9 +469,8 @@ module Irc # Maps Irc::User to BotUser def irc_to_botuser(ircuser) - Irc::error_if_not_user(ircuser) # TODO check netmasks - return @botusers[ircuser] || everyone + @botusers[ircuser.to_irc_user] || everyone end # creates a new BotUser @@ -501,8 +494,8 @@ module Irc # # It is possible to autologin by Netmask, on request # - def login(ircuser, botusername, pwd, bymask = false) - Irc::error_if_not_user(ircuser) + def login(user, botusername, pwd, bymask = false) + ircuser = user.to_irc_user n = BotUser.sanitize_username(botusername) k = n.to_sym raise "No such BotUser #{n}" unless include?(k) @@ -530,10 +523,11 @@ module Irc # * everyone on _chan_ # * everyone on all channels # - def permit?(user, cmdtxt, chan=nil) + def permit?(user, cmdtxt, channel=nil) botuser = irc_to_botuser(user) - cmd = Command.new(cmdtxt) + cmd = cmdtxt.to_irc_auth_command + chan = channel case chan when User chan = "?" diff --git a/lib/rbot/irc.rb b/lib/rbot/irc.rb index 69b5d239..5d3b01ec 100644 --- a/lib/rbot/irc.rb +++ b/lib/rbot/irc.rb @@ -478,8 +478,6 @@ module Irc # Empty +nick+, +user+ or +host+ are converted to the generic glob pattern # def initialize(str="", opts={}) - debug "String: #{str.inspect}, options: #{opts.inspect}" - # First of all, check for server/casemap option # init_server_or_casemap(opts) @@ -517,7 +515,7 @@ module Irc if self.class == Netmask return self if fits_with_server_and_casemap?(opts) end - return self.fullform.to_irc_netmask(opts) + return self.fullform.to_irc_netmask(server_and_casemap.merge(opts)) end # Converts the receiver into a User with the given (optional) @@ -525,7 +523,7 @@ module Irc # is needed (different casemap/server) # def to_irc_user(opts={}) - self.fullform.to_irc_user(opts) + self.fullform.to_irc_user(server_and_casemap.merge(opts)) end # Inspection of a Netmask reveals the server it's bound to (if there is @@ -659,6 +657,7 @@ module Irc end + class String # We keep extending String, this time adding a method that converts a @@ -701,7 +700,6 @@ module Irc # into a Netmask) provided that the given Netmask does not have globs. # def initialize(str="", opts={}) - debug "String: #{str.inspect}, options: #{opts.inspect}" super raise ArgumentError, "#{str.inspect} must not have globs (unescaped * or ?)" if nick.has_irc_glob? && nick != "*" raise ArgumentError, "#{str.inspect} must not have globs (unescaped * or ?)" if user.has_irc_glob? && user != "*" @@ -764,7 +762,7 @@ module Irc # def to_irc_user(opts={}) return self if fits_with_server_and_casemap?(opts) - return self.fullform.to_irc_user(opts) + return self.fullform.to_irc_user(server_and_casemap(opts)) end # We can replace everything at once with data from another User @@ -807,7 +805,6 @@ class String # String into an Irc::User object # def to_irc_user(opts={}) - debug "opts = #{opts.inspect}" Irc::User.new(self, opts) end -- cgit v1.2.3