]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/blob - lib/rbot/plugins.rb
removes bdb&tc registry; adds dbm registry
[user/henk/code/ruby/rbot.git] / lib / rbot / plugins.rb
1 #-- vim:sw=2:et
2 #++
3 #
4 # :title: rbot plugin management
5
6 require 'singleton'
7
8 module Irc
9 class Bot
10     Config.register Config::ArrayValue.new('plugins.blacklist',
11       :default => [], :wizard => false, :requires_rescan => true,
12       :desc => "Plugins that should not be loaded")
13     Config.register Config::ArrayValue.new('plugins.whitelist',
14       :default => [], :wizard => false, :requires_rescan => true,
15       :desc => "Only whitelisted plugins will be loaded unless the list is empty")
16 module Plugins
17   require 'rbot/messagemapper'
18
19 =begin rdoc
20   BotModule is the base class for the modules that enhance the rbot
21   functionality. Rather than subclassing BotModule, however, one should
22   subclass either CoreBotModule (reserved for system modules) or Plugin
23   (for user plugins).
24
25   A BotModule interacts with Irc events by defining one or more of the following
26   methods, which get called as appropriate when the corresponding Irc event
27   happens.
28
29   map(template, options)::
30   map!(template, options)::
31      map is the new, cleaner way to respond to specific message formats without
32      littering your plugin code with regexps, and should be used instead of
33      #register() and #privmsg() (see below) when possible.
34
35      The difference between map and map! is that map! will not register the new
36      command as an alternative name for the plugin.
37
38      Examples:
39
40        plugin.map 'karmastats', :action => 'karma_stats'
41
42        # while in the plugin...
43        def karma_stats(m, params)
44          m.reply "..."
45        end
46
47        # the default action is the first component
48        plugin.map 'karma'
49
50        # attributes can be pulled out of the match string
51        plugin.map 'karma for :key'
52        plugin.map 'karma :key'
53
54        # while in the plugin...
55        def karma(m, params)
56          item = params[:key]
57          m.reply 'karma for #{item}'
58        end
59
60        # you can setup defaults, to make parameters optional
61        plugin.map 'karma :key', :defaults => {:key => 'defaultvalue'}
62
63        # the default auth check is also against the first component
64        # but that can be changed
65        plugin.map 'karmastats', :auth => 'karma'
66
67        # maps can be restricted to public or private message:
68        plugin.map 'karmastats', :private => false
69        plugin.map 'karmastats', :public => false
70
71      See MessageMapper#map for more information on the template format and the
72      allowed options.
73
74   listen(UserMessage)::
75                          Called for all messages of any type. To
76                          differentiate them, use message.kind_of? It'll be
77                          either a PrivMessage, NoticeMessage, KickMessage,
78                          QuitMessage, PartMessage, JoinMessage, NickMessage,
79                          etc.
80
81   ctcp_listen(UserMessage)::
82                          Called for all messages that contain a CTCP command.
83                          Use message.ctcp to get the CTCP command, and
84                          message.message to get the parameter string. To reply,
85                          use message.ctcp_reply, which sends a private NOTICE
86                          to the sender.
87
88   message(PrivMessage)::
89                          Called for all PRIVMSG. Hook on this method if you
90                          need to handle PRIVMSGs regardless of whether they are
91                          addressed to the bot or not, and regardless of
92
93   privmsg(PrivMessage)::
94                          Called for a PRIVMSG if the first word matches one
95                          the plugin #register()ed for. Use m.plugin to get
96                          that word and m.params for the rest of the message,
97                          if applicable.
98
99   unreplied(PrivMessage)::
100                          Called for a PRIVMSG which has not been replied to.
101
102   notice(NoticeMessage)::
103                          Called for all Notices. Please notice that in general
104                          should not be replied to.
105
106   kick(KickMessage)::
107                          Called when a user (or the bot) is kicked from a
108                          channel the bot is in.
109
110   invite(InviteMessage)::
111                          Called when the bot is invited to a channel.
112
113   join(JoinMessage)::
114                          Called when a user (or the bot) joins a channel
115
116   part(PartMessage)::
117                          Called when a user (or the bot) parts a channel
118
119   quit(QuitMessage)::
120                          Called when a user (or the bot) quits IRC
121
122   nick(NickMessage)::
123                          Called when a user (or the bot) changes Nick
124   modechange(ModeChangeMessage)::
125                          Called when a User or Channel mode is changed
126   topic(TopicMessage)::
127                          Called when a user (or the bot) changes a channel
128                          topic
129
130   welcome(WelcomeMessage)::
131                          Called when the welcome message is received on
132                          joining a server succesfully.
133
134   motd(MotdMessage)::
135                          Called when the Message Of The Day is fully
136                          recevied from the server.
137
138   connect::              Called when a server is joined successfully, but
139                          before autojoin channels are joined (no params)
140
141   set_language(String)::
142                          Called when the user sets a new language
143                          whose name is the given String
144
145   save::                 Called when you are required to save your plugin's
146                          state, if you maintain data between sessions
147
148   cleanup::              called before your plugin is "unloaded", prior to a
149                          plugin reload or bot quit - close any open
150                          files/connections or flush caches here
151 =end
152
153   class BotModule
154     # the associated bot
155     attr_reader :bot
156
157     # the plugin registry
158     attr_reader :registry
159
160     # the message map handler
161     attr_reader :handler
162
163     # Initialise your bot module. Always call super if you override this method,
164     # as important variables are set up for you:
165     #
166     # @bot::
167     #   the rbot instance
168     # @registry::
169     #   the botmodule's registry, which can be used to store permanent data
170     #   (see Registry::Accessor for additional documentation)
171     #
172     # Other instance variables which are defined and should not be overwritten
173     # byt the user, but aren't usually accessed directly, are:
174     #
175     # @manager::
176     #   the plugins manager instance
177     # @botmodule_triggers::
178     #   an Array of words this plugin #register()ed itself for
179     # @handler::
180     #   the MessageMapper that handles this plugin's maps
181     #
182     def initialize
183       @manager = Plugins::manager
184       @bot = @manager.bot
185       @priority = nil
186
187       @botmodule_triggers = Array.new
188
189       @handler = MessageMapper.new(self)
190       @registry = Registry::Accessor.new(@bot, self.class.to_s.gsub(/^.*::/, ""))
191
192       @manager.add_botmodule(self)
193       if self.respond_to?('set_language')
194         self.set_language(@bot.lang.language)
195       end
196     end
197
198     # Changing the value of @priority directly will cause problems,
199     # Please use priority=.
200     def priority
201       @priority ||= 1
202     end
203
204     # Returns the symbol :BotModule
205     def botmodule_class
206       :BotModule
207     end
208
209     # Method called to flush the registry, thus ensuring that the botmodule's permanent
210     # data is committed to disk
211     #
212     def flush_registry
213       # debug "Flushing #{@registry}"
214       @registry.flush
215     end
216
217     # Method called to cleanup before the plugin is unloaded. If you overload
218     # this method to handle additional cleanup tasks, remember to call super()
219     # so that the default cleanup actions are taken care of as well.
220     #
221     def cleanup
222       # debug "Closing #{@registry}"
223       @registry.close
224     end
225
226     # Handle an Irc::PrivMessage for which this BotModule has a map. The method
227     # is called automatically and there is usually no need to call it
228     # explicitly.
229     #
230     def handle(m)
231       @handler.handle(m)
232     end
233
234     # Signal to other BotModules that an even happened.
235     #
236     def call_event(ev, *args)
237       @bot.plugins.delegate('event_' + ev.to_s.gsub(/[^\w\?!]+/, '_'), *(args.push Hash.new))
238     end
239
240     # call-seq: map(template, options)
241     #
242     # This is the preferred way to register the BotModule so that it
243     # responds to appropriately-formed messages on Irc.
244     #
245     def map(*args)
246       do_map(false, *args)
247     end
248
249     # call-seq: map!(template, options)
250     #
251     # This is the same as map but doesn't register the new command
252     # as an alternative name for the plugin.
253     #
254     def map!(*args)
255       do_map(true, *args)
256     end
257
258     # Auxiliary method called by #map and #map!
259     def do_map(silent, *args)
260       @handler.map(self, *args)
261       # register this map
262       map = @handler.last
263       name = map.items[0]
264       self.register name, :auth => nil, :hidden => silent
265       @manager.register_map(self, map)
266       unless self.respond_to?('privmsg')
267         def self.privmsg(m) #:nodoc:
268           handle(m)
269         end
270       end
271     end
272
273     # Sets the default auth for command path _cmd_ to _val_ on channel _chan_:
274     # usually _chan_ is either "*" for everywhere, public and private (in which
275     # case it can be omitted) or "?" for private communications
276     #
277     def default_auth(cmd, val, chan="*")
278       case cmd
279       when "*", ""
280         c = nil
281       else
282         c = cmd
283       end
284       Auth::defaultbotuser.set_default_permission(propose_default_path(c), val)
285     end
286
287     # Gets the default command path which would be given to command _cmd_
288     def propose_default_path(cmd)
289       [name, cmd].compact.join("::")
290     end
291
292     # Return an identifier for this plugin, defaults to a list of the message
293     # prefixes handled (used for error messages etc)
294     def name
295       self.class.to_s.downcase.sub(/^#<module:.*?>::/,"").sub(/(plugin|module)?$/,"")
296     end
297
298     # Just calls name
299     def to_s
300       name
301     end
302
303     # Intern the name
304     def to_sym
305       self.name.to_sym
306     end
307
308     # Return a help string for your module. For complex modules, you may wish
309     # to break your help into topics, and return a list of available topics if
310     # +topic+ is nil. +plugin+ is passed containing the matching prefix for
311     # this message - if your plugin handles multiple prefixes, make sure you
312     # return the correct help for the prefix requested
313     def help(plugin, topic)
314       "no help"
315     end
316
317     # Register the plugin as a handler for messages prefixed _cmd_.
318     #
319     # This can be called multiple times for a plugin to handle multiple message
320     # prefixes.
321     #
322     # This command is now superceded by the #map() command, which should be used
323     # instead whenever possible.
324     #
325     def register(cmd, opts={})
326       raise ArgumentError, "Second argument must be a hash!" unless opts.kind_of?(Hash)
327       who = @manager.who_handles?(cmd)
328       if who
329         raise "Command #{cmd} is already handled by #{who.botmodule_class} #{who}" if who != self
330         return
331       end
332       if opts.has_key?(:auth)
333         @manager.register(self, cmd, opts[:auth])
334       else
335         @manager.register(self, cmd, propose_default_path(cmd))
336       end
337       @botmodule_triggers << cmd unless opts.fetch(:hidden, false)
338     end
339
340     # Default usage method provided as a utility for simple plugins. The
341     # MessageMapper uses 'usage' as its default fallback method.
342     #
343     def usage(m, params = {})
344       if params[:failures].respond_to? :find
345         friendly = params[:failures].find do |f|
346           f.kind_of? MessageMapper::FriendlyFailure
347         end
348         if friendly
349           m.reply friendly.friendly
350           return
351         end
352       end
353       m.reply(_("incorrect usage, ask for help using '%{command}'") % {:command => "#{@bot.nick}: help #{m.plugin}"})
354     end
355
356     # Define the priority of the module.  During event delegation, lower
357     # priority modules will be called first.  Default priority is 1
358     def priority=(prio)
359       if @priority != prio
360         @priority = prio
361         @bot.plugins.mark_priorities_dirty
362       end
363     end
364
365     # Directory name to be joined to the botclass to access data files. By
366     # default this is the plugin name itself, but may be overridden, for
367     # example by plugins that share their datafiles or for backwards
368     # compatibilty
369     def dirname
370       name
371     end
372
373     # Filename for a datafile built joining the botclass, plugin dirname and
374     # actual file name
375     def datafile(*fname)
376       @bot.path dirname, *fname
377     end
378   end
379
380   # A CoreBotModule is a BotModule that provides core functionality.
381   #
382   # This class should not be used by user plugins, as it's reserved for system
383   # plugins such as the ones that handle authentication, configuration and basic
384   # functionality.
385   #
386   class CoreBotModule < BotModule
387     def botmodule_class
388       :CoreBotModule
389     end
390   end
391
392   # A Plugin is a BotModule that provides additional functionality.
393   #
394   # A user-defined plugin should subclass this, and then define any of the
395   # methods described in the documentation for BotModule to handle interaction
396   # with Irc events.
397   #
398   class Plugin < BotModule
399     def botmodule_class
400       :Plugin
401     end
402   end
403
404   # Singleton to manage multiple plugins and delegate messages to them for
405   # handling
406   class PluginManagerClass
407     include Singleton
408     attr_reader :bot
409     attr_reader :botmodules
410     attr_reader :maps
411
412     # This is the list of patterns commonly delegated to plugins.
413     # A fast delegation lookup is enabled for them.
414     DEFAULT_DELEGATE_PATTERNS = %r{^(?:
415       connect|names|nick|
416       listen|ctcp_listen|privmsg|unreplied|
417       kick|join|part|quit|
418       save|cleanup|flush_registry|
419       set_.*|event_.*
420     )$}x
421
422     def initialize
423       @botmodules = {
424         :CoreBotModule => [],
425         :Plugin => []
426       }
427
428       @names_hash = Hash.new
429       @commandmappers = Hash.new
430       @maps = Hash.new
431
432       # modules will be sorted on first delegate call
433       @sorted_modules = nil
434
435       @delegate_list = Hash.new { |h, k|
436         h[k] = Array.new
437       }
438
439       @core_module_dirs = []
440       @plugin_dirs = []
441
442       @failed = Array.new
443       @ignored = Array.new
444
445       bot_associate(nil)
446     end
447
448     def inspect
449       ret = self.to_s[0..-2]
450       ret << ' corebotmodules='
451       ret << @botmodules[:CoreBotModule].map { |m|
452         m.name
453       }.inspect
454       ret << ' plugins='
455       ret << @botmodules[:Plugin].map { |m|
456         m.name
457       }.inspect
458       ret << ">"
459     end
460
461     # Reset lists of botmodules
462     def reset_botmodule_lists
463       @botmodules[:CoreBotModule].clear
464       @botmodules[:Plugin].clear
465       @names_hash.clear
466       @commandmappers.clear
467       @maps.clear
468       @failures_shown = false
469       mark_priorities_dirty
470     end
471
472     # Associate with bot _bot_
473     def bot_associate(bot)
474       reset_botmodule_lists
475       @bot = bot
476     end
477
478     # Returns the botmodule with the given _name_
479     def [](name)
480       @names_hash[name.to_sym]
481     end
482
483     # Returns +true+ if _cmd_ has already been registered as a command
484     def who_handles?(cmd)
485       return nil unless @commandmappers.has_key?(cmd.to_sym)
486       return @commandmappers[cmd.to_sym][:botmodule]
487     end
488
489     # Registers botmodule _botmodule_ with command _cmd_ and command path _auth_path_
490     def register(botmodule, cmd, auth_path)
491       raise TypeError, "First argument #{botmodule.inspect} is not of class BotModule" unless botmodule.kind_of?(BotModule)
492       @commandmappers[cmd.to_sym] = {:botmodule => botmodule, :auth => auth_path}
493     end
494
495     # Registers botmodule _botmodule_ with map _map_. This adds the map to the #maps hash
496     # which has three keys:
497     #
498     # botmodule:: the associated botmodule
499     # auth:: an array of auth keys checked by the map; the first is the full_auth_path of the map
500     # map:: the actual MessageTemplate object
501     #
502     #
503     def register_map(botmodule, map)
504       raise TypeError, "First argument #{botmodule.inspect} is not of class BotModule" unless botmodule.kind_of?(BotModule)
505       @maps[map.template] = { :botmodule => botmodule, :auth => [map.options[:full_auth_path]], :map => map }
506     end
507
508     def add_botmodule(botmodule)
509       raise TypeError, "Argument #{botmodule.inspect} is not of class BotModule" unless botmodule.kind_of?(BotModule)
510       kl = botmodule.botmodule_class
511       if @names_hash.has_key?(botmodule.to_sym)
512         case self[botmodule].botmodule_class
513         when kl
514           raise "#{kl} #{botmodule} already registered!"
515         else
516           raise "#{self[botmodule].botmodule_class} #{botmodule} already registered, cannot re-register as #{kl}"
517         end
518       end
519       @botmodules[kl] << botmodule
520       @names_hash[botmodule.to_sym] = botmodule
521       mark_priorities_dirty
522     end
523
524     # Returns an array of the loaded plugins
525     def core_modules
526       @botmodules[:CoreBotModule]
527     end
528
529     # Returns an array of the loaded plugins
530     def plugins
531       @botmodules[:Plugin]
532     end
533
534     # Returns a hash of the registered message prefixes and associated
535     # plugins
536     def commands
537       @commandmappers
538     end
539
540     # Tells the PluginManager that the next time it delegates an event, it
541     # should sort the modules by priority
542     def mark_priorities_dirty
543       @sorted_modules = nil
544     end
545
546     # Makes a string of error _err_ by adding text _str_
547     def report_error(str, err)
548       ([str, err.inspect] + err.backtrace).join("\n")
549     end
550
551     # This method is the one that actually loads a module from the
552     # file _fname_
553     #
554     # _desc_ is a simple description of what we are loading (plugin/botmodule/whatever)
555     #
556     # It returns the Symbol :loaded on success, and an Exception
557     # on failure
558     #
559     def load_botmodule_file(fname, desc=nil)
560       # create a new, anonymous module to "house" the plugin
561       # the idea here is to prevent namespace pollution. perhaps there
562       # is another way?
563       plugin_module = Module.new
564       # each plugin uses its own textdomain, we bind it automatically here
565       bindtextdomain_to(plugin_module, "rbot-#{File.basename(fname, '.rb')}")
566
567       desc = desc.to_s + " " if desc
568
569       begin
570         plugin_string = IO.read(fname)
571         debug "loading #{desc}#{fname}"
572         plugin_module.module_eval(plugin_string, fname)
573         return :loaded
574       rescue Exception => err
575         # rescue TimeoutError, StandardError, NameError, LoadError, SyntaxError => err
576         error report_error("#{desc}#{fname} load failed", err)
577         bt = err.backtrace.select { |line|
578           line.match(/^(\(eval\)|#{fname}):\d+/)
579         }
580         bt.map! { |el|
581           el.gsub(/^\(eval\)(:\d+)(:in `.*')?(:.*)?/) { |m|
582             "#{fname}#{$1}#{$3}"
583           }
584         }
585         msg = err.to_s.gsub(/^\(eval\)(:\d+)(:in `.*')?(:.*)?/) { |m|
586           "#{fname}#{$1}#{$3}"
587         }
588         begin
589           newerr = err.class.new(msg)
590         rescue ArgumentError => aerr_in_err
591           # Somebody should hang the ActiveSupport developers by their balls
592           # with barbed wire. Their MissingSourceFile extension to LoadError
593           # _expects_ a second argument, breaking the usual Exception interface
594           # (instead, the smart thing to do would have been to make the second
595           # parameter optional and run the code in the from_message method if
596           # it was missing).
597           # Anyway, we try to cope with this in the simplest possible way. On
598           # the upside, this new block can be extended to handle other similar
599           # idiotic approaches
600           if err.class.respond_to? :from_message
601             newerr = err.class.from_message(msg)
602           else
603             raise aerr_in_err
604           end
605         rescue NoMethodError => nmerr_in_err
606           # Another braindead extension to StandardError, OAuth2::Error,
607           # doesn't get a string as message, but a response
608           if err.respond_to? :response
609             newerr = err.class.new(err.response)
610           else
611             raise nmerr_in_err
612           end
613         end
614         newerr.set_backtrace(bt)
615         return newerr
616       end
617     end
618     private :load_botmodule_file
619
620     # add one or more directories to the list of directories to
621     # load core modules from
622     def add_core_module_dir(*dirlist)
623       @core_module_dirs += dirlist
624       debug "Core module loading paths: #{@core_module_dirs.join(', ')}"
625     end
626
627     # add one or more directories to the list of directories to
628     # load plugins from
629     def add_plugin_dir(*dirlist)
630       @plugin_dirs += dirlist
631       debug "Plugin loading paths: #{@plugin_dirs.join(', ')}"
632     end
633
634     def clear_botmodule_dirs
635       @core_module_dirs.clear
636       @plugin_dirs.clear
637       debug "Core module and plugin loading paths cleared"
638     end
639
640     def scan_botmodules(opts={})
641       type = opts[:type]
642       processed = Hash.new
643
644       case type
645       when :core
646         dirs = @core_module_dirs
647       when :plugins
648         dirs = @plugin_dirs
649
650         @bot.config['plugins.blacklist'].each { |p|
651           pn = p + ".rb"
652           processed[pn.intern] = :blacklisted
653         }
654
655         whitelist = @bot.config['plugins.whitelist'].map { |p|
656           p + ".rb"
657         }
658       end
659
660       dirs.each do |dir|
661         next unless FileTest.directory?(dir)
662         d = Dir.new(dir)
663         d.sort.each do |file|
664           next unless file =~ /\.rb$/
665           next if file =~ /^\./
666
667           case type
668           when :plugins
669             if !whitelist.empty? && !whitelist.include?(file)
670               @ignored << {:name => file, :dir => dir, :reason => :"not whitelisted" }
671               next
672             elsif processed.has_key?(file.intern)
673               @ignored << {:name => file, :dir => dir, :reason => processed[file.intern]}
674               next
675             end
676
677             if(file =~ /^(.+\.rb)\.disabled$/)
678               # GB: Do we want to do this? This means that a disabled plugin in a directory
679               #     will disable in all subsequent directories. This was probably meant
680               #     to be used before plugins.blacklist was implemented, so I think
681               #     we don't need this anymore
682               processed[$1.intern] = :disabled
683               @ignored << {:name => $1, :dir => dir, :reason => processed[$1.intern]}
684               next
685             end
686           end
687
688           begin
689             did_it = load_botmodule_file("#{dir}/#{file}", "plugin")
690           rescue Exception => e
691             error e
692             did_it = e
693           end
694
695           case did_it
696           when Symbol
697             processed[file.intern] = did_it
698           when Exception
699             @failed << { :name => file, :dir => dir, :reason => did_it }
700           end
701         end
702       end
703     end
704
705     # load plugins from pre-assigned list of directories
706     def scan
707       @failed.clear
708       @ignored.clear
709       @delegate_list.clear
710
711       scan_botmodules(:type => :core)
712       scan_botmodules(:type => :plugins)
713
714       debug "finished loading plugins: #{status(true)}"
715       (core_modules + plugins).each { |p|
716        p.methods.grep(DEFAULT_DELEGATE_PATTERNS).each { |m|
717          @delegate_list[m.intern] << p
718        }
719       }
720       mark_priorities_dirty
721     end
722
723     # call the save method for each active plugin
724     def save
725       delegate 'flush_registry'
726       delegate 'save'
727     end
728
729     # call the cleanup method for each active plugin
730     def cleanup
731       delegate 'cleanup'
732       reset_botmodule_lists
733     end
734
735     # drop all plugins and rescan plugins on disk
736     # calls save and cleanup for each plugin before dropping them
737     def rescan
738       save
739       cleanup
740       scan
741     end
742
743     def status(short=false)
744       output = []
745       if self.core_length > 0
746         if short
747           output << n_("%{count} core module loaded", "%{count} core modules loaded",
748                     self.core_length) % {:count => self.core_length}
749         else
750           output <<  n_("%{count} core module: %{list}",
751                      "%{count} core modules: %{list}", self.core_length) %
752                      { :count => self.core_length,
753                        :list => core_modules.collect{ |p| p.name}.sort.join(", ") }
754         end
755       else
756         output << _("no core botmodules loaded")
757       end
758       # Active plugins first
759       if(self.length > 0)
760         if short
761           output << n_("%{count} plugin loaded", "%{count} plugins loaded",
762                        self.length) % {:count => self.length}
763         else
764           output << n_("%{count} plugin: %{list}",
765                        "%{count} plugins: %{list}", self.length) %
766                    { :count => self.length,
767                      :list => plugins.collect{ |p| p.name}.sort.join(", ") }
768         end
769       else
770         output << "no plugins active"
771       end
772       # Ignored plugins next
773       unless @ignored.empty? or @failures_shown
774         if short
775           output << n_("%{highlight}%{count} plugin ignored%{highlight}",
776                        "%{highlight}%{count} plugins ignored%{highlight}",
777                        @ignored.length) %
778                     { :count => @ignored.length, :highlight => Underline }
779         else
780           output << n_("%{highlight}%{count} plugin ignored%{highlight}: use %{bold}%{command}%{bold} to see why",
781                        "%{highlight}%{count} plugins ignored%{highlight}: use %{bold}%{command}%{bold} to see why",
782                        @ignored.length) %
783                     { :count => @ignored.length, :highlight => Underline,
784                       :bold => Bold, :command => "help ignored plugins"}
785         end
786       end
787       # Failed plugins next
788       unless @failed.empty? or @failures_shown
789         if short
790           output << n_("%{highlight}%{count} plugin failed to load%{highlight}",
791                        "%{highlight}%{count} plugins failed to load%{highlight}",
792                        @failed.length) %
793                     { :count => @failed.length, :highlight => Reverse }
794         else
795           output << n_("%{highlight}%{count} plugin failed to load%{highlight}: use %{bold}%{command}%{bold} to see why",
796                        "%{highlight}%{count} plugins failed to load%{highlight}: use %{bold}%{command}%{bold} to see why",
797                        @failed.length) %
798                     { :count => @failed.length, :highlight => Reverse,
799                       :bold => Bold, :command => "help failed plugins"}
800         end
801       end
802       output.join '; '
803     end
804
805     # return list of help topics (plugin names)
806     def helptopics
807       rv = status
808       @failures_shown = true
809       rv
810     end
811
812     def length
813       plugins.length
814     end
815
816     def core_length
817       core_modules.length
818     end
819
820     # return help for +topic+ (call associated plugin's help method)
821     def help(topic="")
822       case topic
823       when /fail(?:ed)?\s*plugins?.*(trace(?:back)?s?)?/
824         # debug "Failures: #{@failed.inspect}"
825         return _("no plugins failed to load") if @failed.empty?
826         return @failed.collect { |p|
827           _('%{highlight}%{plugin}%{highlight} in %{dir} failed with error %{exception}: %{reason}') % {
828               :highlight => Bold, :plugin => p[:name], :dir => p[:dir],
829               :exception => p[:reason].class, :reason => p[:reason],
830           } + if $1 && !p[:reason].backtrace.empty?
831                 _('at %{backtrace}') % {:backtrace => p[:reason].backtrace.join(', ')}
832               else
833                 ''
834               end
835         }.join("\n")
836       when /ignored?\s*plugins?/
837         return _('no plugins were ignored') if @ignored.empty?
838
839         tmp = Hash.new
840         @ignored.each do |p|
841           reason = p[:loaded] ? _('overruled by previous') : _(p[:reason].to_s)
842           ((tmp[p[:dir]] ||= Hash.new)[reason] ||= Array.new).push(p[:name])
843         end
844
845         return tmp.map do |dir, reasons|
846           # FIXME get rid of these string concatenations to make gettext easier
847           s = reasons.map { |r, list|
848             list.map { |_| _.sub(/\.rb$/, '') }.join(', ') + " (#{r})"
849           }.join('; ')
850           "in #{dir}: #{s}"
851         end.join('; ')
852       when /^(\S+)\s*(.*)$/
853         key = $1
854         params = $2
855
856         # Let's see if we can match a plugin by the given name
857         (core_modules + plugins).each { |p|
858           next unless p.name == key
859           begin
860             return p.help(key, params)
861           rescue Exception => err
862             #rescue TimeoutError, StandardError, NameError, SyntaxError => err
863             error report_error("#{p.botmodule_class} #{p.name} help() failed:", err)
864           end
865         }
866
867         # Nope, let's see if it's a command, and ask for help at the corresponding botmodule
868         k = key.to_sym
869         if commands.has_key?(k)
870           p = commands[k][:botmodule]
871           begin
872             return p.help(key, params)
873           rescue Exception => err
874             #rescue TimeoutError, StandardError, NameError, SyntaxError => err
875             error report_error("#{p.botmodule_class} #{p.name} help() failed:", err)
876           end
877         end
878       end
879       return false
880     end
881
882     def sort_modules
883       @sorted_modules = (core_modules + plugins).sort do |a, b|
884         a.priority <=> b.priority
885       end || []
886
887       @delegate_list.each_value do |list|
888         list.sort! {|a,b| a.priority <=> b.priority}
889       end
890     end
891
892     # call-seq: delegate</span><span class="method-args">(method, m, opts={})</span>
893     # <span class="method-name">delegate</span><span class="method-args">(method, opts={})
894     #
895     # see if each plugin handles _method_, and if so, call it, passing
896     # _m_ as a parameter (if present). BotModules are called in order of
897     # priority from lowest to highest.
898     #
899     # If the passed _m_ is a BasicUserMessage and is marked as #ignored?, it
900     # will only be delegated to plugins with negative priority. Conversely, if
901     # it's a fake message (see BotModule#fake_message), it will only be
902     # delegated to plugins with positive priority.
903     #
904     # Note that _m_ can also be an exploded Array, but in this case the last
905     # element of it cannot be a Hash, or it will be interpreted as the options
906     # Hash for delegate itself. The last element can be a subclass of a Hash, though.
907     # To be on the safe side, you can add an empty Hash as last parameter for delegate
908     # when calling it with an exploded Array:
909     #   @bot.plugins.delegate(method, *(args.push Hash.new))
910     #
911     # Currently supported options are the following:
912     # :above ::
913     #   if specified, the delegation will only consider plugins with a priority
914     #   higher than the specified value
915     # :below ::
916     #   if specified, the delegation will only consider plugins with a priority
917     #   lower than the specified value
918     #
919     def delegate(method, *args)
920       # if the priorities order of the delegate list is dirty,
921       # meaning some modules have been added or priorities have been
922       # changed, then the delegate list will need to be sorted before
923       # delegation.  This should always be true for the first delegation.
924       sort_modules unless @sorted_modules
925
926       opts = {}
927       opts.merge(args.pop) if args.last.class == Hash
928
929       m = args.first
930       if BasicUserMessage === m
931         # ignored messages should not be delegated
932         # to plugins with positive priority
933         opts[:below] ||= 0 if m.ignored?
934         # fake messages should not be delegated
935         # to plugins with negative priority
936         opts[:above] ||= 0 if m.recurse_depth > 0
937       end
938
939       above = opts[:above]
940       below = opts[:below]
941
942       # debug "Delegating #{method.inspect}"
943       ret = Array.new
944       if method.match(DEFAULT_DELEGATE_PATTERNS)
945         debug "fast-delegating #{method}"
946         m = method.to_sym
947         debug "no-one to delegate to" unless @delegate_list.has_key?(m)
948         return [] unless @delegate_list.has_key?(m)
949         @delegate_list[m].each { |p|
950           begin
951             prio = p.priority
952             unless (above and above >= prio) or (below and below <= prio)
953               ret.push p.send(method, *args)
954             end
955           rescue Exception => err
956             raise if err.kind_of?(SystemExit)
957             error report_error("#{p.botmodule_class} #{p.name} #{method}() failed:", err)
958           end
959         }
960       else
961         debug "slow-delegating #{method}"
962         @sorted_modules.each { |p|
963           if(p.respond_to? method)
964             begin
965               # debug "#{p.botmodule_class} #{p.name} responds"
966               prio = p.priority
967               unless (above and above >= prio) or (below and below <= prio)
968                 ret.push p.send(method, *args)
969               end
970             rescue Exception => err
971               raise if err.kind_of?(SystemExit)
972               error report_error("#{p.botmodule_class} #{p.name} #{method}() failed:", err)
973             end
974           end
975         }
976       end
977       return ret
978       # debug "Finished delegating #{method.inspect}"
979     end
980
981     # see if we have a plugin that wants to handle this message, if so, pass
982     # it to the plugin and return true, otherwise false
983     def privmsg(m)
984       debug "Delegating privmsg #{m.inspect} with pluginkey #{m.plugin.inspect}"
985       return unless m.plugin
986       k = m.plugin.to_sym
987       if commands.has_key?(k)
988         p = commands[k][:botmodule]
989         a = commands[k][:auth]
990         # We check here for things that don't check themselves
991         # (e.g. mapped things)
992         debug "Checking auth ..."
993         if a.nil? || @bot.auth.allow?(a, m.source, m.replyto)
994           debug "Checking response ..."
995           if p.respond_to?("privmsg")
996             begin
997               debug "#{p.botmodule_class} #{p.name} responds"
998               p.privmsg(m)
999             rescue Exception => err
1000               raise if err.kind_of?(SystemExit)
1001               error report_error("#{p.botmodule_class} #{p.name} privmsg() failed:", err)
1002             end
1003             debug "Successfully delegated #{m.inspect}"
1004             return true
1005           else
1006             debug "#{p.botmodule_class} #{p.name} is registered, but it doesn't respond to privmsg()"
1007           end
1008         else
1009           debug "#{p.botmodule_class} #{p.name} is registered, but #{m.source} isn't allowed to call #{m.plugin.inspect} on #{m.replyto}"
1010         end
1011       else
1012         debug "Command #{k} isn't handled"
1013       end
1014       return false
1015     end
1016
1017     # delegate IRC messages, by delegating 'listen' first, and the actual method
1018     # afterwards. Delegating 'privmsg' also delegates ctcp_listen and message
1019     # as appropriate.
1020     def irc_delegate(method, m)
1021       delegate('listen', m)
1022       if method.to_sym == :privmsg
1023         delegate('ctcp_listen', m) if m.ctcp
1024         delegate('message', m)
1025         privmsg(m) if m.address? and not m.ignored?
1026         delegate('unreplied', m) unless m.replied
1027       else
1028         delegate(method, m)
1029       end
1030     end
1031   end
1032
1033   # Returns the only PluginManagerClass instance
1034   def Plugins.manager
1035     return PluginManagerClass.instance
1036   end
1037
1038 end
1039 end
1040 end