1 # Plugin for the Ruby IRC bot (http://linuxbrit.co.uk/rbot/)
3 # Create mini plugins on IRC.
5 # Scripts are little Ruby programs that run in the context of the script plugin. You
6 # can create them directly in an IRC channel, and invoke them just like normal rbot plugins.
8 # (c) 2006 Mark Kretschmann <markey@web.de>
9 # Licensed under GPL V2.
12 Command = Struct.new( "Command", :code, :nick, :created, :channel )
15 class ScriptPlugin < Plugin
19 if @registry.has_key?(:commands)
20 @commands = @registry[:commands]
28 @registry[:commands] = @commands
32 def help( plugin, topic="" )
34 "Scripts are little Ruby programs that run in the context of the script plugin. You can access @bot (class IrcBot), m (class PrivMessage), user (class String, either the first argument, or if missing the sourcenick), and args (class Array, an array of arguments). Example: 'script add greet m.reply( 'Hello ' + user )'. Invoke the script just like a plugin: '<botnick>: greet'."
36 "Create mini plugins on IRC. 'script add <name> <code>' => Create script named <name> with the Ruby program <code>. 'script list' => Show a list of all known scripts. 'script show <name>' => Show the source code for <name>. 'script del <name>' => Delete the script <name>."
42 name = m.message.split.first
44 if m.address? and @commands.has_key?( name )
45 code = @commands[name].code.dup.untaint
47 # Convenience variables, can be accessed by scripts:
48 args = m.message.split
50 user = args.empty? ? m.sourcenick : args.first
53 # TODO allow different safe levels for different botusers
59 m.reply( "Script '#{name}' crapped out :(" )
67 def handle_eval( m, params )
68 code = params[:code].to_s.dup.untaint
70 # TODO allow different safe levels for different botusers
74 m.reply( "Script '#{name}' crapped out :(" )
81 def handle_echo( m, params )
82 code = params[:code].to_s.dup.untaint
84 # TODO allow different safe levels for different botusers
86 m.reply eval( code ).to_s
88 m.reply( "Script '#{name}' crapped out :(" )
95 def handle_add( m, params, force = false )
97 if !force and @commands.has_key?( name )
98 m.reply( "#{m.sourcenick}: #{name} already exists. Use 'add -f' if you really want to overwrite it." )
102 code = params[:code].to_s
104 created = Time.new.strftime '%Y/%m/%d %H:%m'
107 command = Command.new( code, nick, created, channel )
108 @commands[name] = command
114 def handle_add_force( m, params )
115 handle_add( m, params, true )
119 def handle_del( m, params )
121 unless @commands.has_key?( name )
122 m.reply( "Script does not exist." ); return
125 @commands.delete( name )
130 def handle_list( m, params )
131 if @commands.length == 0
132 m.reply( "No scripts available." ); return
136 cmds = @commands.keys.sort
137 num_pages = cmds.length / cmds_per_page + 1
138 page = params[:page].to_i
140 page = [page, num_pages].min
141 str = cmds[(page-1)*cmds_per_page, cmds_per_page].join(', ')
143 m.reply "Available scripts (page #{page}/#{num_pages}): #{str}"
147 def handle_show( m, params )
149 unless @commands.has_key?( name )
150 m.reply( "Script does not exist." ); return
153 cmd = @commands[name]
154 m.reply( "#{cmd.code} [#{cmd.nick}, #{cmd.created} in #{cmd.channel}]" )
160 plugin = ScriptPlugin.new
161 plugin.register( "script" )
162 plugin.default_auth( 'edit', false )
163 plugin.default_auth( 'eval', false )
164 plugin.default_auth( 'echo', false )
166 plugin.map 'script add -f :name *code', :action => 'handle_add_force', :auth_path => 'edit'
167 plugin.map 'script add :name *code', :action => 'handle_add', :auth_path => 'edit'
168 plugin.map 'script del :name', :action => 'handle_del', :auth_path => 'edit'
169 plugin.map 'script eval *code', :action => 'handle_eval'
170 plugin.map 'script echo *code', :action => 'handle_echo'
171 plugin.map 'script list :page', :action => 'handle_list', :defaults => { :page => '1' }
172 plugin.map 'script show :name', :action => 'handle_show'