Implement module system and DDate module

3 files changed, 75 insertions, 22 deletions

diff --git a/lain.rb b/lain.rb
index 802d378..4f0301a 100644
--- a/lain.rb
+++ b/lain.rb
@@ -2,30 +2,53 @@
 
 
 require 'xmpp4r'
-require 'xmpp4r/muc/helper/simplemucclient'
+require 'xmpp4r/muc/helper/mucclient'
 
 require_relative 'sasl'
 
 require_relative 'config'
 
-include Jabber
-
-
-cl = Jabber::Client.new(Jabber::JID.new(LainConfig::JID))
-cl.connect
-cl.auth(LainConfig::Password)
-cl.send(Presence.new)
-
-mainthread = Thread.current
-
-m = Jabber::MUC::MUCClient.new(cl)
-
-m.add_message_callback { |x|
-  puts x
-}
-
-LainConfig::Rooms.each { |r| m.join("#{r}/#{LainConfig::Nick}") }
-
-Thread.stop
-
-cl.close
+module Lain
+  class Bot
+    def initialize
+      @cl = Jabber::Client.new(Jabber::JID.new(Config::JID))
+      @cl.connect
+      @cl.auth(Config::Password)
+      @cl.send(Jabber::Presence.new)
+
+      @muc = Jabber::MUC::MUCClient.new(@cl)
+
+      @modules = {}
+
+      @muc.add_message_callback { |msg|
+        @modules.each { | _, mod |
+          begin
+            mod.on_message @muc, msg
+          rescue
+          end
+        }
+      }
+
+      Config::Modules.each { |mod|
+        require_relative "modules/#{mod}"
+        @modules[mod] = Modules.const_get(mod).new self
+      }
+
+      Config::Rooms.each { |r| @muc.join("#{r}/#{Config::Nick}") }
+    end
+
+    def run
+      @mainthread = Thread.current
+      Thread.stop
+    end
+
+    def close
+      @cl.close
+    end
+
+  end
+end
+
+bot = Lain::Bot.new
+bot.run
+bot.close
diff --git a/module_base.rb b/module_base.rb
new file mode 100644
index 0000000..500fd20
--- /dev/null
+++ b/module_base.rb
@@ -0,0 +1,11 @@
+module Lain
+  module Modules
+    class Base
+      def initialize(lain)
+      end
+
+      def on_message(muc, message)
+      end
+    end
+  end
+end
diff --git a/modules/DDate.rb b/modules/DDate.rb
new file mode 100644
index 0000000..1402648
--- /dev/null
+++ b/modules/DDate.rb
@@ -0,0 +1,19 @@
+require 'xmpp4r/message'
+
+require_relative '../module_base'
+
+module Lain
+  module Modules
+    class DDate < Base
+      def on_message(muc, message)
+        return unless message.type == :groupchat
+        return unless /!ddate\b/ =~ message.body
+
+        p = IO.popen("ddate")
+        while (line = p.gets)
+          muc.send(Jabber::Message.new(message.to, line.chomp))
+        end
+      end
+    end
+  end
+end