nadoka 0.8.0

data/ndk/server_state.rb

@@ -0,0 +1,324 @@
+#
+# Copyright (c) 2004-2005 SASADA Koichi <ko1 at atdot.net>
+#
+# This program is free software with ABSOLUTELY NO WARRANTY.
+# You can re-distribute and/or modify this program under
+# the same terms of the Ruby's license.
+#
+#
+# $Id$
+# Create : K.S. 04/04/20 10:42:27
+#
+
+module Nadoka
+  class NDK_State
+    class ChannelState
+      def initialize name
+        @name  = name
+        @topic = nil
+        @member= {}
+
+        # member data
+        # { "nick" => "mode", ... }
+      end
+
+      attr_accessor :topic
+      attr_reader   :member, :name
+
+      # get members nick array
+      def members
+        @member.keys
+      end
+
+      # get user's mode
+      def mode nick
+        @member[nick]
+      end
+      
+      def names
+        @member.map{|nick, mode|
+          prefix = if /o/ =~ mode
+            '@'
+          elsif /v/ =~ mode
+            '+'
+          else
+            ''
+          end
+          prefix + nick
+        }
+      end
+      
+      def state
+        '='
+      end
+
+      def clear_members
+        @member = {}
+      end
+      
+      #####################################
+      
+      def on_join nick, mode=''
+        @member[nick] = mode
+      end
+      
+      def on_part nick
+        @member.delete nick
+      end
+      
+      def on_nick nick, newnick
+        if @member.has_key? nick
+          @member[newnick] = @member[nick]
+          @member.delete nick
+        end
+      end
+      
+      def on_kick nick
+        if @member.has_key? nick
+          @member.delete nick
+        end
+      end
+      
+      MODE_WITH_NICK_ARG = 'ov'
+      MODE_WITH_ARGS     = 'klbeI'
+      MODE_WITHOUT_ARGS  = 'aimnqpsrt'
+      
+      def on_mode nick, args
+        while mode = args.shift
+          modes = mode.split(//)
+          flag  = modes.shift
+          modes.each{|m|
+            if MODE_WITH_NICK_ARG.include? m
+              chg_mode args.shift, flag, m
+            elsif MODE_WITH_ARGS.include? m
+              args.shift
+            elsif MODE_WITHOUT_ARGS.include? m
+              # ignore
+            end
+          }
+        end
+      end
+
+      def chg_mode nick, flag, mode
+        if @member.has_key? nick
+          if flag == '+'
+            @member[nick] += mode
+          elsif flag == '-'
+            @member[nick].gsub!(mode, '')
+          end
+        end
+      end
+
+      def to_s
+        str = ''
+        @member.each{|k, v|
+          str << "#{k}: #{v}, "
+        }
+        str
+      end
+    end
+    
+    def initialize manager
+      @manager = manager
+      @config  = nil
+      @logger  = nil
+      
+      @current_nick     = nil
+      @original_nick    = nil
+      @try_nick         = nil
+      
+      @current_channels = {}
+    end
+    attr_reader :current_channels
+    attr_reader :current_nick
+    attr_accessor :original_nick
+    attr_writer :logger, :config
+    
+    def nick=(n)
+      @try_nick     = nil
+      @current_nick = n
+    end
+    
+    def nick
+      @current_nick
+    end
+
+    def nick_succ fail_nick
+      if @try_nick
+        if @try_nick.length == fail_nick
+          @try_nick.succ!
+        else
+          @try_nick = fail_nick[0..-2] + '0'
+        end
+      else
+        @try_nick = fail_nick + '0'
+      end
+    end
+    
+    def channels
+      @current_channels.keys
+    end
+
+    def channel_raw_names
+      @current_channels.map{|k, cs| cs.name}
+    end
+
+    # need canonicalized channel name
+    def channel_users ch
+      if @current_channels.has_key? ch
+        @current_channels[ch].members
+      else
+        []
+      end
+    end
+    
+    # need canonicalized channel name
+    def channel_user_mode ch, user
+      if channel_users(ch).include?(user)
+        @current_channels[ch].mode(user)
+      else
+        ''
+      end
+    end
+    
+    def canonical_channel_name ch
+      @config.canonical_channel_name ch
+    end
+
+    def clear_channels_member
+      @current_channels.each{|ch, cs|
+        cs.clear_members
+      }
+    end
+    
+    #
+    def on_join user, rch
+      ch = canonical_channel_name(rch)
+      if user == nick
+        chs = @current_channels[ch] = ChannelState.new(rch)
+      else
+        if @current_channels.has_key? ch
+          @current_channels[ch].on_join(user)
+        end
+      end
+    end
+    
+    def on_part user, rch
+      ch = canonical_channel_name(rch)
+      if user == nick
+        @current_channels.delete ch
+      else
+        if @current_channels.has_key? ch
+          @current_channels[ch].on_part user
+        end
+      end
+    end
+    
+    def on_nick user, newnick, msg
+      if user == nick
+        @current_nick = newnick
+        @try_nick     = nil
+      end
+      # logging
+      @current_channels.each{|ch, chs|
+        if chs.on_nick user, newnick
+          @config.logger.logging_nick ch, chs.name, user, newnick, msg
+        end
+      }
+    end
+    
+    def on_quit user, qmsg, msg
+      if user == nick
+        @current_channels = {} # clear
+      else
+        # logging
+        @current_channels.each{|ch, chs|
+          if chs.on_part(user)
+            @config.logger.logging_quit ch, chs.name, user, qmsg, msg
+            @manager.invoke_event :invoke_bot, :quit_from_channel, chs.name, user, qmsg
+          end
+        }
+      end
+    end
+    
+    def on_mode user, rch, args
+      ch = canonical_channel_name(rch)
+      if @current_channels.has_key? ch
+        @current_channels[ch].on_mode user, args
+      end
+    end
+
+    def on_kick kicker, rch, user, comment
+      ch = canonical_channel_name(rch)
+      if user == nick
+        @current_channels.delete ch
+      else
+        if @current_channels.has_key? ch
+          @current_channels[ch].on_kick user
+        end
+      end
+    end
+
+    def on_topic user, rch, topic
+      ch = canonical_channel_name(rch)
+
+      if @current_channels.has_key? ch
+        @current_channels[ch].topic = topic
+      end
+    end
+    
+    def on_332 rch, topic
+      ch = canonical_channel_name(rch)
+
+      if @current_channels.has_key? ch
+        @current_channels[ch].topic = topic
+      end
+    end
+    
+    # RPL_NAMREPLY
+    # ex) :lalune 353 test_ndk = #nadoka :test_ndk ko1_nmdk
+    # 
+    def on_353 rch, users
+      ch = canonical_channel_name(rch)
+
+      if @current_channels.has_key? ch
+        chs = @current_channels[ch]
+        users.split(/ /).each{|e|
+          /^([\@\+])?(.+)/ =~ e
+          case $1
+          when '@'
+            mode = 'o'
+          when '+'
+            mode = 'v'
+          else
+            mode = ''
+          end
+          chs.on_join $2, mode
+        }
+
+        # change initial mode
+        if @config.channel_info[ch] &&
+           (im = @config.channel_info[ch][:initial_mode]) &&
+           chs.members.size == 1
+          @manager.send_to_server Cmd.mode(rch, im)
+        end
+      end
+    end
+
+    def safe_channel? ch
+      ch[0] == ?!
+    end
+    
+    # ERR_NOSUCHCHANNEL
+    # ex) :NadokaProgram 403 simm !hoge :No such channel
+    def on_403 ch
+      if safe_channel?(ch) && ch[1] != ?!
+        if @config.channel_info[ch] && @config.channel_info[ch][:auto_create]
+          @manager.join_to_channel( "!" + ch)
+        end
+      end
+    end
+    
+  end
+end
+

data/ndk/version.rb

@@ -0,0 +1,44 @@
+#
+# Copyright (c) 2004-2005 SASADA Koichi <ko1 at atdot.net>
+#
+# This program is free software with ABSOLUTELY NO WARRANTY.
+# You can re-distribute and/or modify this program under
+# the same terms of the Ruby's license.
+#
+# 
+# $Id$
+# Create : K.S. 04/04/27 16:35:40
+#
+
+module Nadoka
+  VERSION  = '0.8.0'
+  NDK_Version  = VERSION.dup
+  NDK_Created  = Time.now
+
+  if File.directory?(File.expand_path('../../.git', __FILE__))
+    NDK_Version.concat("+git")
+  end
+  if /trunk/ =~ '$HeadURL$'
+    NDK_Version.concat('-trunk')
+    rev = '-'
+    $LOAD_PATH.each{|path|
+      path = path + '/ChangeLog'
+      if FileTest.exist?(path)
+        if /^\# ChangeLog of Nadoka\(\$Rev: (\d+) \$\)$/ =~ open(path){|f| s = f.gets}
+          rev = "rev: #{$1}"
+          break
+        end
+      end
+    }
+    NDK_Version.concat(" (#{rev})")
+  end
+  
+  def self.version
+    "Nadoka Ver.#{NDK_Version}" +
+    " with Ruby #{RUBY_VERSION} (#{RUBY_RELEASE_DATE}) [#{RUBY_PLATFORM}]"
+  end
+end
+
+if __FILE__ == $0
+  puts Nadoka::NDK_Version
+end

data/plugins/autoawaybot.nb

@@ -0,0 +1,66 @@
+# -*-ruby-*-
+#
+# Copyright (c) 2004-2005 SASADA Koichi <ko1 at atdot.net>
+#
+# This program is free software with ABSOLUTELY NO WARRANTY.
+# You can re-distribute and/or modify this program under
+# the same terms of the Ruby's license.
+#
+#
+# This bot is created by
+#   akira yamada <akira at arika.org>
+#
+# $Id$
+#
+
+=begin
+
+== Abstract
+
+Auto away if no action
+
+== Configuration
+
+BotConfig = [
+{
+  :name      => :AutoAwayBot,
+  :threshold => 15*60,        # sec
+  :message   => 'Away',
+}
+]
+
+=end
+
+class AutoAwayBot < Nadoka::NDK_Bot
+  def bot_initialize
+    @threshold = @bot_config.fetch(:threshold, 15*60).to_i
+    @message   = @bot_config.fetch(:message, 'Away')
+    @lastseen  = Time.now
+    @in_away   = false
+  end
+
+  def on_client_privmsg client, ch, msg
+    if @threshold > 0
+      @lastseen = Time.now
+    end
+    if @in_away
+      @manager.send_to_server Nadoka::Cmd.away()
+    end
+  end
+
+  def on_timer time
+    if @threshold > 0 && Time.now - @lastseen > @threshold && !@in_away
+      @manager.send_to_server Nadoka::Cmd.away(@message)
+    end
+  end
+
+  def on_rpl_unaway *arg
+    @in_away = false
+  end
+
+  def on_rpl_nowaway *arg
+    @in_away = true
+  end
+end
+
+

data/plugins/autodumpbot.nb

@@ -0,0 +1,227 @@
+# -*-ruby-*-
+#
+# Copyright (c) 2004-2005 SASADA Koichi <ko1 at atdot.net>
+#
+# This program is free software with ABSOLUTELY NO WARRANTY.
+# You can re-distribute and/or modify this program under
+# the same terms of the Ruby's license.
+#
+#
+# This bot is created by
+#   akira yamada <akira at arika.org>
+#
+# $Id$
+#
+
+=begin
+
+== Abstract
+
+Auto save configuration
+
+== Configuration
+
+add these program to nadokarc(or your configuration file)
+
+  begin
+    dump_file = File.join(Log_dir, Setting_name + '-channel_info.dump')
+    channel_info = {}
+    begin
+      File.open(dump_file, 'r') {|i| channel_info = Marshal.load(i)}
+    rescue TypeError
+      File.open(dump_file, 'r') {|i| channel_info = YAML.load(i.read)}
+    end
+
+    Channel_info = channel_info
+  rescue
+    Channel_info = {
+    }
+  end
+
+more detail, see [nadoka:162]
+
+=end
+
+
+class AutoDumpBot < Nadoka::NDK_Bot
+  def bot_initialize
+    @auto_dump = @bot_config.fetch(:auto_dump, true)
+    @default_dump_style = @bot_config.fetch(:dump_style, 'yaml')
+    @default_timing = @bot_config.fetch(:dump_timing, :startup)
+
+    @cinfo = {}
+    @manager.state.channels.each do |ch|
+      @manager.send_to_server Nadoka::Cmd.mode(ch)
+    end
+  end
+
+  def canonical_channel_name ch
+    @manager.state.canonical_channel_name ch
+  end
+
+  def on_join prefix, ch
+    if @manager.state.current_nick == prefix.nick
+      dch = canonical_channel_name(ch)
+      @cinfo[dch] ||= {}
+      @cinfo[dch][:name] = ch
+      @cinfo[dch][:mode] ||= []
+
+      @manager.send_to_server Nadoka::Cmd.mode(ch)
+    end
+  end
+
+  def on_part prefix, ch, msg=''
+    if @manager.client_count > 0 && @manager.state.current_nick == prefix.nick
+      @cinfo.delete(canonical_channel_name(ch))
+    end
+  end
+
+  def on_client_logout count, client
+    if count == 1 && @auto_dump
+      dump_cinfo
+    end
+  end
+  
+  MODE_WITH_NICK_ARG = 'ov'
+  MODE_WITH_ARGS     = 'klbeI'
+  MODE_WITHOUT_ARGS  = 'aimnqpsrt'
+
+  def on_mode prefix, nick, ch, *args
+    dch = canonical_channel_name(ch)
+    if @manager.state.current_channels.has_key? dch
+      @cinfo[dch] ||= {}
+      @cinfo[dch][:name] = ch
+      @cinfo[dch][:mode] ||= []
+
+      while mode = args.shift
+	modes = mode.split(//)
+	flag  = modes.shift
+	modes.each{|m|
+	  if MODE_WITH_NICK_ARG.include? m
+	    #chg_cinfo dch, flag, m, args.shift 
+	  elsif MODE_WITH_ARGS.include? m
+	    chg_cinfo dch, flag, m, args.shift 
+	  elsif MODE_WITHOUT_ARGS.include? m
+	    chg_cinfo dch, flag, m, nil
+	  end
+	}
+      end
+    end
+  end
+  
+  def on_rpl_channelmodeis prefix, nick, ch, *arg
+    on_mode(prefix, nick, ch, *arg)
+  end
+
+  def on_nadoka_command client, command, *params
+    if command == 'dump'
+      dump_cinfo(params.shift)
+      raise ::Nadoka::NDK_BotSendCancel
+    end
+  end
+
+  def chg_cinfo ch, flag, mode, arg
+    @cinfo[ch][:mode] ||= []
+    @cinfo[ch][:mode].delete_if do |m|
+      m[0, 1] == mode
+    end
+    if flag == '+'
+      if arg
+	@cinfo[ch][:mode] << mode + ' ' + arg
+      else
+	@cinfo[ch][:mode] << mode
+      end
+    end
+  end
+
+  def dump_cinfo(style = nil)
+    style = @default_dump_style unless style
+
+    channel_info = {}
+    @cinfo.keys.each do |ch|
+      name = @cinfo[ch][:name]
+      if @state.current_channels.include?(ch)
+	name = @state.current_channels[ch].name
+      end
+      channel_info[name] = cinfo = {}
+      config = @manager.instance_eval {@config}
+
+      if config.default_channels.detect {|tch| canonical_channel_name(tch) == ch}
+      	cinfo[:timing] = :startup
+      elsif config.login_channels.detect {|tch| canonical_channel_name(tch) == ch}
+  	cinfo[:timing] = :login
+      else
+      	cinfo[:timing] = @default_timing
+      end
+      if @cinfo[ch][:mode].include?('i')
+	cinfo[:timing] = :startup
+      elsif !cinfo.include?(:timing)
+	cinfo[:timing] = :login
+      end
+
+      @cinfo[ch][:mode].each do |m|
+	if m[0] == ?k
+	  cinfo[:key] = m[2 .. -1]
+	end
+      end
+
+      m1, m2 = @cinfo[ch][:mode].partition {|m| m.size == 1}
+      imode = nil
+      unless m1.empty?
+	imode ||= '+'
+       	imode << m1.join('')
+      end
+      unless m2.empty?
+	imode ||= '+'
+	imode << m2.join(' +')
+      end
+      cinfo[:initial_mode] = imode
+    end
+
+    begin
+      dump_file = File.join(@config.log_dir,
+	@config.setting_name + '-channel_info.dump')
+      File.open(dump_file, 'wb') do |f|
+	f.chmod(0600)
+	dump_channel_info(channel_info, style, f)
+      end
+      @logger.slog "current channel information was dumped (#{style})"
+    rescue Exception => e
+      @logger.slog "current channel information could not be dumped"
+      @logger.slog e.message
+    end
+  end
+
+  def dump_channel_info(channel_info, style, port)
+    if style == 'marshal'
+      Marshal.dump(channel_info, port)
+
+    elsif style == 'yaml'
+      require 'yaml'
+      YAML.dump(channel_info, port)
+
+    elsif style == 'text'
+      port.puts "Channel_info = {"
+      channel_info.each do |ch, info|
+	port.puts "  #{ch.dump} => {"
+	info.each do |tag, value|
+	  if tag.kind_of?(Symbol)
+	    if value.kind_of?(Symbol)
+	      port.puts "    :#{tag.to_s} => :#{value.to_s},"
+	    else
+	      port.puts "    :#{tag.to_s} => #{value.to_s.dump},"
+	    end
+	  end
+	end
+	port.puts "  },"
+      end
+      port.puts "}"
+
+    else
+      raise RuntimeError, "unsupported dump style: #{style.dump}"
+    end
+  end
+
+end
+
+# vim:filetype=ruby: