bot.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

require 'xmpp4r'
require 'xmpp4r/muc/helper/mucclient'

require_relative 'sasl'

require_relative 'config'

module Lain
  class Bot
    Version = '0.1'

    def initialize
      $stderr.puts "Lain #{Version} starting..."

      @cl = Jabber::Client.new(Jabber::JID.new(Config::JID))
      @cl.connect
      @cl.auth(Config::Password)
      @cl.send(Jabber::Presence.new)

      @modules = {}

      $stderr.puts 'Loading modules...'

      Config::Modules.each { |mod|
        require_relative "modules/#{mod}"
        @modules[mod] = Modules.const_get(mod).new self
      }

      @mucs = {}

      Config::Rooms.each { |r|
        muc = Jabber::MUC::MUCClient.new(@cl)
        @mucs[r] = muc

        muc.add_message_callback { |msg|
          @modules.each { | _, mod |
            begin
              mod.on_message muc, msg
            rescue
            end
          }
        }

        $stderr.puts "Joining room `#{r}'..."

        muc.join("#{r}/#{Config::Nick}")
        muc.configure()
      }

      $stderr.puts 'Startup finished.'
    end

    def run
      @mainthread = Thread.current
      Thread.stop
    end

    def close
      @cl.close
    end

  end
end