ircbot 0.0.2

data/lib/ircbot/core_ext/digest.rb

@@ -0,0 +1,14 @@
+class String
+  def digest (max, tail_adjust = '$')
+    euc = NKF::nkf('-e', self)
+    if euc .size <= max
+      return euc
+    end
+    euc = euc[0..max-1]
+    if euc[max-1] == 164
+      euc[max-1] = tail_adjust
+    end
+    euc
+  end
+end
+

data/lib/ircbot/core_ext/irc.rb

@@ -0,0 +1,61 @@
+# Extension to ruby-irc-lib
+
+module IRC
+  class AbnormalTerminated < Exception; end
+
+  class Connection
+    def sendNAMES (channel = nil)
+# p [:sendNAMES, Message(CMD_NAMES, nil, nil, channel)]
+      send(Message(CMD_NAMES, nil, nil, channel))
+    end
+  end
+
+  class Message
+    def create_hash! (client)
+      unless @hash
+	@hash = {}
+	@hash[:client] = client
+	@hash[:type]   = self.command
+	@hash[:prefix] = self.prefix
+	@hash[:from]   = User::parse(self.prefix).nick
+        @hash[:str]    = self .trailing
+        @hash[:string] = @hash[:str]
+	@hash[:to]     = to = self .params[0]
+	@hash[:timestamp] = Time .now
+
+	unless to
+	  case self .command
+	  when CMD_JOIN, CMD_QUIT
+	    @hash[:to] = to = @hash[:str]
+	  end
+	end
+
+        if to != nil and to == client .nick
+	  @hash[:to] = @hash[:from]
+        end
+	if client .is_a? Client
+	  @client = client
+	end
+      end
+    end
+
+    def [] (key)		# key must be a symbol
+      @hash[key]
+    end
+
+    def []= (key, val)
+      @hash[key] = val
+    end
+
+    def reply (string, to = nil)
+      self[:reply] = string
+      if @client and string
+	@client .speak(self, to)
+      end
+    end
+
+    def previous
+      @client && @client .previous_message(self)
+    end
+  end
+end

data/lib/ircbot/core_ext/rand-polimorphism.rb

@@ -0,0 +1,23 @@
+#!/usr/local/bin/ruby
+#
+# rand-extension.rb
+# Author:  
+# Created: Thu Nov  8 17:26:09 2001
+
+
+# rand ��¿����
+unless $__RAND_OVERRIDE__
+  $__RAND_OVERRIDE__ = true
+  alias :__rand__ :rand 
+  def rand (obj)
+    case obj
+    when Array
+      index = __rand__(obj .size)
+      obj[index]
+    when Range
+      rand(obj .to_a)
+    else
+      __rand__(obj)
+    end
+  end
+end

data/lib/ircbot/core_ext/writefile.rb

@@ -0,0 +1,45 @@
+#!/usr/local/bin/ruby
+#
+# writefile.rb
+# Author: maiha@wota.jp
+# Created: 2002/05/10 14:16:13
+# Changed: 2010/01/02 20:42:25
+
+=begin
+== usage
+  ensure directories of a path exist before open it.
+
+=== syntacs
+  File::open!(path, ...)
+
+== sample
+  Dir["/tmp/**/*"]			# => []
+  File .open!("/tmp/foo/ora", "w+")
+  Dir["/tmp/**/*"]			# => ["/tmp/foo", "/tmp/foo/ora"]
+
+== changes
+* 2002/06/11 19:45:32
+  change interface: open(...,'w!') => open!(...)
+=end
+
+def File.open! (path, *args, &block)
+  dirname = File.dirname(path) + '/'
+  unless File.directory?(dirname)
+    dirs = []
+    dirname.scan(/\//) {dirs << $`}
+    dirs.shift
+    dirs.each do |dir|
+      unless directory?(dir)
+        Dir.mkdir(dir)
+      end
+    end
+  end
+  open(path, *args, &block)
+end
+
+if $0 == __FILE__
+  path = "/tmp/a/b/c.txt"
+  File.open!(path, "w+") {|f| f.puts "[OK] Wrote to #{path}"}
+  puts File.read(path){}
+  File.unlink(path)
+end

data/lib/ircbot/framework.rb

@@ -0,0 +1,50 @@
+module Ircbot
+  mattr_accessor :load_paths
+  self.load_paths = {}
+
+  class InvalidFilename < SecurityError; end
+  class FileNotFound < SecurityError; end
+
+  class Recover < RuntimeError
+    attr_reader :wait
+
+    def initialize(wait = 300)
+      @wait = wait
+    end
+  end
+
+  class << self
+    def root
+      @root || File.expand_path(Dir.pwd)
+    end
+
+    def root=(value)
+      @root = value
+    end
+
+    def push_path(type, path, file_glob = "**/*.rb")
+      load_paths[type] = [Pathname(path), file_glob]
+    end
+
+    def dir_for(type)
+      load_paths[type][0] or
+        raise RuntimeError, "directory not found: #{type}"
+    end
+
+    def glob_for(type)
+      load_paths[type][1]
+    end
+
+    def path_for(type, name)
+      name = name.to_s
+      raise InvalidFilename, name if name =~ %r{\.\.|~|/}
+
+      path = dir_for(type) + name
+      path.readable_real? or
+        raise FileNotFound, name
+
+      return path
+    end
+  end
+end
+

data/lib/ircbot/ordered_hash.rb

@@ -0,0 +1,91 @@
+## -*- coding: utf-8 -*-
+# ordered_hash.rb
+# Author:  
+# Created: Tue Jul 24 18:52:49 2001
+
+class Ircbot::OrderedHash < Hash
+  def self.[] (*array)
+    size = array .size
+    if size % 2 == 0
+      hash = self .new
+      (size / 2) .times do |index|
+	key, value = array[index*2,2]
+	hash[key] = value
+      end
+      hash
+    else
+      super
+    end
+  end
+
+  def initialize(*args)
+    super
+    @store_order = []
+  end
+
+  def []= (key, value)
+    unless self .has_key? key
+      @store_order << key
+    end
+    super
+  end
+
+  def keys
+    @store_order
+  end
+
+  def each
+    keys .each do |key|
+      yield(key, self[key])
+    end
+  end
+  alias :each_pair :each
+
+  def delete (key)
+    @store_order .delete(key)
+    super
+  end
+
+  def each_value
+    keys .each do |key|
+      yield(self[key])
+    end
+  end
+
+  def collect
+    keys .collect do |key|
+      yield(key, self[key])
+    end
+  end
+
+  def to_a
+    keys .collect do |key|
+      [key, self[key]]
+    end
+  end
+
+  # $B@hF,$KDI2C$9$k!#(B
+  def unshift (key, val)
+    self[key] = val
+    @store_order .delete key
+    @store_order .unshift key
+    val
+  end
+
+  # $B;XDj$5$l$?%-!<$r;}$DMWAG$NE:;z$rJV$9!#(B(Array#index $BAjEv$N5!G=(B)
+  def index_of_key (value)
+    keys .index value
+  end
+
+  def position (value)
+    keys .each_with_index do |key, i|
+      return i if self[key] == value
+    end
+    return nil
+  end
+
+  def index_of_value (value)
+    position(value)
+  end
+end
+

data/lib/ircbot/reply_client.rb

@@ -0,0 +1,328 @@
+# -*- coding: euc-jp -*-
+
+require 'ircbot/config_client'
+
+module Ircbot
+  class ReplyClient < ConfigClient
+    DIGEST_MAX_SIZE = 400
+    REGEXP_REDIRECT = ">|��"
+
+    def myname_regexp_string
+      names = @mynames.join('|')
+      "(#{ names })"
+    end
+
+    def speak0 (to, obj, save = nil)
+      Thread.critical = true
+      message =
+        case obj
+        when String     ; obj
+        when Array
+          array = obj.collect {|item| item.inspect}.join(', ')
+          "(#{obj.size})[#{array}]"
+        when Regexp     ; "/#{obj.source}/"
+        else            ; obj.inspect
+        end
+
+      message.split(/\n+/).each do |line|
+	buffer = line.digest(DIGEST_MAX_SIZE)
+	unless buffer.empty?
+	  # ����������������
+	  @message_queue.push([buffer, to])
+
+          # ��å��������֥������Ȥ�������ơ�@messages �˻Ĥ�
+          if save
+            # :AnnaChan!~anna@219-106-253-19.cust.bit-drive.ne.jp PRIVMSG #bot :test\r\n"
+            str = ":#{@nick}!~bot@localhost PRIVMSG #{@channel} :#{buffer}\r\n"
+            msg = Message::parse(str)
+            msg.create_hash!(self)
+            msg[:id] = add_message(msg)
+          end
+	end
+      end
+      Thread.critical = false
+    end
+
+    def speak (msg, to = nil, log = nil)
+      speak0(to || msg[:to], msg[:reply], log)
+    end
+
+    def newAgent (name)
+      script = Ircbot.path_for(:cpi, "#{name}.cpi").read{}
+      eval(script, TOPLEVEL_BINDING)
+
+    rescue SecurityError => e
+      err = "#{e.class}: #{e}"
+      syslog(err, :error)
+      return err
+    end
+
+    # xxxAgent: Agent �� xxx ����
+    # ����: String (Agent �μ��̻�)
+    # ����: String (������)
+
+    # ����������Ȥο�����Ͽ
+    def registerAgent (name0)
+      						# ex) name0="daiben(12345)"
+      name, arg = name0.split(/\s|\(/, 2)	# ex) name="daiben"
+      arg = arg.to_s.delete(")").strip	# ex) arg="12345"
+
+      err    = nil
+      this   = "registerAgent"
+      agent  = @agents[name]
+      @agents_opt[name] = arg
+
+#      p [:debug, :registerAgent, name0, name, arg]
+
+      agent = newAgent(name)
+      case agent
+      when NilClass
+	err = "Agent plugin script return nil (will not be registerd)."
+      when String
+	err = agent
+      else
+	@agents[name] = AgentManager.new(name, agent, self, arg)
+      end
+
+      status = "#{this}(#{name})... #{err || 'OK.'}"
+      syslog(status, err ? :error : :normal)
+      return status
+    end
+
+    def startAgent (name0)
+      						# ex) name0="daiben(12345)"
+      name, arg = name0.split(/\s|\(/, 2)	# ex) name="daiben"
+      arg = arg.to_s.delete(")").strip	# ex) arg="12345"
+
+      err   = nil
+      this  = "startAgent"
+      agent = @agents[name]
+      arg   = @agents_opt[name] if arg.empty?
+
+      case agent
+      when NilClass
+	err = "no such agent."
+      when AgentManager
+	agent.start(arg)
+      else
+	err = "unknown agent type(#{agent.class})."
+      end
+
+      status = "#{this}(#{name})... #{err || 'OK.'}"
+      syslog(status, err ? :error : :normal)
+      return status
+    end
+
+    def removeAgent (name)
+      err    = nil
+      this   = "removeAgent"
+      agent  = @agents[name]
+
+      case agent
+      when NilClass
+	err = "no such agent."
+      when AgentManager
+	agent.stop
+	@agents.delete(name)
+      else
+	err = "unknown agent type(#{agent.class})."
+      end
+
+      status = "#{this}(#{name})... #{err || 'OK.'}"
+      syslog(status, err ? :error : :normal)
+      return status
+    end
+
+    # Agent �κƵ�ư
+    def restartAgent (name)
+      err    = nil
+      this   = "restartAgent"
+      agent  = @agents[name]
+
+      case agent
+      when NilClass
+	err = "no such agent."
+      when AgentManager
+	agent.stop
+	obj = newAgent(name)
+	case obj
+	when NilClass
+	  err = "Agent plugin script return nil (will not be registerd)."
+	when String
+	  err = agent
+	else
+          arg = @agents_opt[name]
+	  agent = AgentManager.new(name, obj, self, arg)
+	  @agents[name] = agent
+	  agent.start
+	end
+      else
+	err = "unknown agent type(#{agent.class})."
+      end
+
+      status = "#{this}(#{name})... #{err || 'OK.'}"
+      syslog(status, err ? :error : :normal)
+      return status
+    end
+
+    def stopAgent (name)
+      err   = nil
+      this  = "stopAgent"
+      agent = @agents[name]
+
+      case agent
+      when NilClass
+	err = "no such agent."
+      when AgentManager
+	agent.stop
+      else
+	err = "unknown agent type(#{agent.class})."
+      end
+
+      status = "#{this}(#{name})... #{err || 'OK.'}"
+      syslog(status, err ? :error : :normal)
+      return status
+    end
+
+    def do_help (msg)
+      target = msg[:arg].to_s.strip
+      if target.empty?
+	# �ܥåȤΥإ�ס�cpi�������֤�
+	alives = 0
+	names  = @agents.collect {|name, agent|
+	  if agent.alive?
+	    alives += 1
+	    name = "*#{name}"
+	  end
+	  name
+	}.join(', ')
+	msg[:reply] = "#{@help}\ncpi(#{alives}/#{@agents.size}): #{names}"
+	speak(msg)
+
+      else
+	# ���̤� cpi �Υإ�פ��֤���
+	target = target.delete('()[] ')
+	@agents.each_pair do |name, agent|
+	  next unless target == name
+	  next unless (result = agent.apply_methods(msg, :do_help))
+ 	  msg[:reply] = result
+ 	  speak(msg)
+ 	end
+      end
+    end
+
+    def parse_command (msg)
+      case msg[:type].to_s
+      when CMD_PRIVMSG
+	case msg[:str].to_s
+	when /^\s*#{myname_regexp_string}\s*(\.\s*[^\s\(]+)/o
+	  msg[:called]  = $1
+	  msg[:command] = $2
+	  msg[:arg]     = $'.to_s.strip
+	  msg[:command].gsub!(/^\./, '') if msg[:command]
+	when /(#{REGEXP_REDIRECT})\s*#{myname_regexp_string}\s*(\.\s*[^\s\(]+)?\s*$/o
+	  msg[:arg]     = $`.to_s.strip
+	  msg[:called]  = $2
+	  msg[:command] = $3
+	  msg[:command].gsub!(/^\./, '') if msg[:command]
+	end	  
+      end
+      return msg
+    end
+
+    def distributeMessage (msg)
+      # prepare accessors for dummy agents
+      msg.create_hash!(self)
+      msg[:id] = add_message(msg)
+      msg = parse_command(msg)
+
+# p [msg[:type], msg[:from], msg[:from].to_s.empty?, msg[:str]]
+
+      # �ȸ������ʥ�å�����(PING��)��̵��
+      if msg[:from].to_s.empty?
+	return nil
+      end
+
+      begin
+	# reserved commands
+        name = msg[:arg].to_s.sub(/\A\s*\(?\s*/, '')
+        name = name.sub(/\s*\)?\s*\Z/, '') unless name.include?('(')
+
+	case msg[:command].to_s
+	when /^restart$/
+	  names = name.split(/\s*,?\s+/)
+	  names = @agents.keys if names.empty?
+	  names.each do |name|
+	    msg.reply(restartAgent(name))
+	  end
+	  return nil
+	when /^register$/	; msg.reply(registerAgent(name)); return msg.reply(startAgent(name))
+	when /^remove$/		; return msg.reply(removeAgent(name))
+	when /^start$/		; return msg.reply(startAgent(name))
+	when /^stop$/		; return msg.reply(stopAgent(name))
+	when /^help$/		; return do_help(msg)
+	end
+
+	case msg[:type].to_s
+	when CMD_JOIN	; each_agent {|agent| agent.apply_methods(msg, :do_join)}
+	when CMD_PART	; each_agent {|agent| agent.apply_methods(msg, :do_part)}
+	when CMD_PING	; each_agent {|agent| agent.apply_methods(msg, :do_ping)}
+	when RPL_NAMREPLY
+#	  p [:debug, :get_CMD_NAMES]
+          each_agent {|agent| agent.apply_methods(msg, :do_names)}
+	when CMD_PRIVMSG
+
+	  # �����ν񤭽Ф�
+	  if @log
+	    do_log(msg)
+	  end
+
+	  # do_log �����˸ƤӽФ���(�����������Ѥ�����Τ���)
+	  message = nil
+	  catch(:reply) {
+	    if msg[:called]
+	      each_agent do |agent|
+		next unless (message = agent.apply_methods(msg, :do_command))
+		throw :reply
+	      end
+	    else
+	      each_agent do |agent|
+		next unless (message = agent.apply_methods(msg, :do_reply))
+		throw :reply
+	      end
+	    end
+	  }
+
+	  # Agent#do_log �θƤӽФ�
+	  each_agent do |agent|
+	    msg.reply(agent.apply_methods(msg, :do_log))
+	  end
+          
+          # ��ʬ��ȯ���⵭Ͽ
+          if message != nil then
+            dummy = {}
+            dummy[:str]  = message
+            dummy[:from] = @nick
+            dummy[:to]   = @channels
+            each_agent { |agent|
+              msg.reply(agent.apply_methods(dummy, :do_log))
+            }
+          end
+
+	  return msg.reply(message)
+	end
+      rescue Exception
+	syslog("distributeMessage: #{$!}(#{$@[0]})", :error)
+      end
+    end
+  end
+end
+
+
+if $0 == __FILE__
+  config = ARGV.shift or
+    raise "Specify your config file\nusage: #{$0} config/xxx.dat"
+  irc = IRC::ReplyClient.read_config(config)
+  irc.start
+end
+