plugin(salut): use registry for storage see #42

[user/henk/code/ruby/rbot.git] / data / rbot / plugins / note.rb

diff --git a/data/rbot/plugins/note.rb b/data/rbot/plugins/note.rb
index 0c1683a6e0b813c4f0f25673cd8ba3b808df1465..e0a7be293a05a192375e4aed7e03cebe72ad347c 100644 (file)
--- a/data/rbot/plugins/note.rb
+++ b/data/rbot/plugins/note.rb
@@ -9,10 +9,30 @@
 # License:: MIT license
 
 class NotePlugin < Plugin
+
   Note = Struct.new('Note', :time, :from, :private, :text)
 
-  def help(plugin, topic="")
-    "note <nick> <string> => stores a note (<string>) for <nick>"
+  Config.register Config::BooleanValue.new 'note.private_message',
+    :default => false,
+    :desc => 'Send all notes in private messages instead of channel messages.'
+
+  def initialize
+    super
+    return if @registry.length < 1
+    debug 'Checking registry for old-formatted notes...'
+    n = 0
+    @registry.keys.each do |key|
+      unless key == key.downcase
+        @registry[key.downcase] = @registry[key] + (@registry[key.downcase] || [])
+        @registry.delete key
+        n += 1
+      end
+    end
+    debug "#{n} entries converted and merged."
+  end
+
+  def help(plugin, topic='')
+    'note <nick> <string> => stores a note (<string>) for <nick>'
   end
 
   def message(m)
@@ -24,16 +44,19 @@ class NotePlugin < Plugin
       pub = []
       priv = []
       @registry[nick].each do |n|
-        s = "[#{n.time.strftime('%H:%M')}] <#{n.from}> #{n.text}"
-        (n.private ? priv : pub).push s
+        s = "[#{n.time.strftime('%b-%e %H:%M')}] <#{n.from}> #{n.text}"
+        if n.private or @bot.config['note.private_message']
+          priv << s
+        else
+          pub << s
+        end
       end
-      if !pub.empty?
+      unless pub.empty?
         @bot.say m.replyto, "#{m.sourcenick}, you have notes! " +
           pub.join(' ')
       end
-
-      if !priv.empty?
-        @bot.say m.sourcenick, "you have notes! " + priv.join(' ')
+      unless priv.empty?
+        @bot.say m.sourcenick, 'you have notes! ' + priv.join(' ')
       end
       @registry.delete nick
     rescue Exception => e