root/lang/ruby/bgpwatch/trunk/lib/notifier.rb @ 36519

#!/usr/local/bin/ruby
# coding: UTF-8
require 'rubygems'
require 'net/irc'
require 'kconv'

require 'modules'
require 'monkey-patch/net-irc-client'

#
# Notifier bridge module
#
module Notifier
  class BasicNotifier
    include Runnable

    # notify message via the notifier object.
    def notify(msg)
      puts "BasicNotifier says: #{msg}"
    end
  end # Notifier::BasicNotifier

  class IRCClient < BasicNotifier; end

  # 
  # IRC client
  # 
  class IRCClient::Client < Net::IRC::Client
    attr_accessor :channel, :msgq
    attr_accessor :owner

    def initialize(*args)
      super
      @channel = @opts.channel if !@opts.channel.nil?
      @msgq = @opts.msgq if !@opts.msgq.nil?
    end

    def on_idle
      return if @msgq.empty?
      msg = @msgq.shift
      post NOTICE, @channel, msg
    end

    def on_rpl_welcome(m)
      post JOIN, @channel
    end

    def on_privmsg(m)
      channel, message = m.to_a
      nick = m.prefix.nick

      return unless channel == @channel
      return unless /peerbot/ =~ message

      result = []
      rest = message.dup
      rest.gsub!(/AS(\d+)/i) do |s|
        asn = $1.to_i
        if resolver.has_key?(asn) then
          result.push "#{s.upcase}: #{resolver[asn]}"
        end
        ''
      end
      if !result.empty? then
        post NOTICE, @channel, result.join(', ')
      end
    rescue
    end

    def on_disconnected
    end

    private

    def resolver
      @owner.resolver
      # { 64512 => 'mera-chan' }
    end
  end # IRCClient::Client
  
  #
  # IRC Client. Client class of Notifier
  #
  class IRCClient < BasicNotifier
    extend Attributes
    attributes :server, :port, :nick, :realname, :channel, :charcode
    attr_reader :client, :thread, :msgq
    attr_accessor :owner

    def initialize
      init_attributes
      @client = Client.new(@server, @port, {
        :nick => @nick,
        :user => @nick,
        :real => @realname || @nick,
        :channel => @channel,
        :msgq => @msgq = Queue.new
      })
      @client.owner = self if @client.respond_to? :owner=

      # <XXX>
      @client.meta_def(:charconv_in) do |str|
        Kconv.kconv(str, Kconv::UTF8, @charcode || Kconv::AUTO)
      end
      @client.meta_def(:charconv_out) do |str|
        return str if @charcode.nil?
        Kconv.kconv(str, @charcode, Kconv::UTF8)
      end
      # </XXX>
      super
    end

    def run
      @thread = Thread.start { @client.start }
    end

    def shutdown
      @thread.kill
    end

    def notify(msg)
      @msgq << msg
    end

    def resolver
      @owner.resolver
    end
  end # Notifier::IRCClient

  #
  # Twitter Client. Client class of Notifier
  #
  class TwitterNotifier < BasicNotifier
    extend Attributes
    attributes :user, :password, :hash_tag
    attr_reader :twitter, :auth
    require 'twitter'

    def initialize
      init_attributes
      @auth = Twitter::HTTPAuth.new(@user, @password)
      @twitter = Twitter::Base.new(@auth)
      super
    end

    def notify(msg)
      msg += ' ' + @hash_tag unless @hash_tag.nil?
      begin
        @twitter.update(msg)
      rescue => e
        # log error
      end
    end
  end # Notifier::TwitterClient
end # Notifier