konn-rupircd 0.6.0 → 0.6.1

data/lib/rupircd/channel.rb

@@ -0,0 +1,418 @@
+=begin
+= rupircd -- Ruby Pseudo IRC Deamon
+
+  ver 0.6 2009-08-11T23:45:52+09:00
+  
+  Copyright (c) 2009 Hiromi Ishii
+  
+  You can redistribute it and/or modify it under the same term as Ruby.
+=end
+
+require 'rupircd/utils'
+require 'rupircd/message'
+
+module IRCd
+
+class ChannelMode
+  include Utils
+  attr_accessor :bans, :topic_op_only, :ops, :invite, :voices, :anony, :moderate, :n,
+                :quiet, :private, :secret, :reops, :key, :user_max, :excepts,
+                :dont_need_invite, :creators
+  
+  def initialize(server, chname)
+    @chname = chname
+    @server = server
+    @bans = []
+    @topic_op_only = false
+    @creators = []
+    @ops = []
+    @invite = false
+    @voices = []
+    @anony = false
+    @moderate = false
+    @n = false
+    @quiet = false
+    @private = false
+    @secret = false
+    @reops = []
+    @key = ""
+    @user_max = -1
+    @excepts = []
+    @invite_onlys = []
+  end
+
+  def set_flags(who, flags, param)
+    args = param.dup
+    toggle = flags[0] == ?+
+    rpl = [flags[0,1]]
+    flags = flags[1..3]
+    flags.each_byte{|b|
+      rpl.push b.chr
+      case b
+      when ?n
+        @n = toggle
+      when ?i
+        @invite = toggle
+      when ?a
+        @anony = toggle
+      when ?m
+        @moderate = toggle
+      when ?p
+        @secret = false if toggle
+        @private = toggle
+      when ?s
+        @private = false if toggle
+        @secret = toggle
+      when ?k
+        key = param.shift
+        if toggle
+          @key = key
+        elsif key == @key
+          @key = ""
+        end
+      when ?l
+        @user_max = param.shift.to_i
+      when ?r
+        if toggle
+          @reops << mask_to_regex(param.shift)
+        else
+          @reops.delete(mask_to_regex(param.shift))
+        end
+      when ?t
+        @topic_op_only = toggle
+      when ?o
+        user = @server.user_from_nick(param.shift)
+        if toggle
+          @ops << user
+          @ops.compact!
+          @ops.uniq!
+        else
+          @ops.delete(user)
+        end
+      when ?v
+        user = @server.user_from_nick(param.shift)
+        if toggle
+          @voices << user
+          @voices.uniq!
+          @voices.compact!
+        else
+          @voices.delete(user)
+        end
+      when ?b, ?e, ?I
+        masks = param.shift
+        mask_list = []
+        list_msg = ""
+        endof_msg = ""
+        what = ""
+        case b
+        when ?b
+          mask_list = @bans
+          list_msg = "367"
+          endof_msg = "368"
+          what = "ban"
+        when ?e
+          mask_list = @excepts
+          list_msg = "348"
+          endof_msg = "349"
+          what = "exception"
+        when ?I
+          mask_list = @invite_onlys
+          list_msg = "346"
+          endof_msg = "347"
+          what = "invite"
+        end
+        if masks
+          if toggle
+            mask_list << mask_to_regex(masks)
+          else
+            mask_list.delete(mask_to_regex(masks))
+          end
+        else
+          rpl.pop
+          mask_list.each{|msk|
+            @server.send_server_message(who, list_msg, @chname, msk)
+          }
+          @server.send_server_message(who, endof_msg, @chname, "End of channel #{what} list")
+        end
+      else
+        @server.send_server_message(who, "472", rpl.pop, "is unknown mode char to me for #@chname")
+      end
+    }
+    [rpl.join(""), args].flatten if rpl.size > 1
+  end
+
+  def proc_mask(toggle, rpl, param, mask_list, reply_msg, endof_list, name)
+    masks = param.shift
+    if masks
+      if toggle
+        mask_list << mask_to_regex(masks)
+      else
+        mask_list.delete(mask_to_regex(masks))
+      end
+    else
+      rpl.pop
+      mask_list.each{|msk|
+        @server.send_server_message(who, reply_msg, @chname, msk)
+      }
+      @server.send_server_message(who, endof_list, @chname, "End of channel #{name} list")
+    end
+  end
+  private :proc_mask
+
+  def get_flags
+    params = []
+    flg = "+"
+    flg << "n" if @n
+    flg << "i" if @invite
+    flg << "a" if @anony
+    flg << "m" if @moderate
+    flg << "q" if @quiet
+    flg << "p" if @private
+    flg << "s" if @secret
+    flg << "k" unless @key.empty?
+    flg << "t" if @topic_op_only
+    if @user_max > -1
+      flg << "l"
+      params << @user_max
+    end
+    params.unshift flg
+  end
+
+  def add_op(hoge)
+    @ops.push hoge
+  end
+
+  def unregister(who)
+    @ops.delete(who)
+    @voices.delete(who)
+  end
+
+  def op?(who)
+    @ops.include?(who)
+  end
+
+  def voiced?(who)
+    @voices.include?(who)
+  end
+
+  def can_change_topic?(who)
+    !@topic_op_only || op?(who)
+  end
+
+  def can_talk?(who)
+    !@moderate || op?(who) || @voices.include?(who)
+  end
+
+  def can_invite?(who)
+    !@invite || op?(who)
+  end
+
+  def banned?(who)
+    @bans.any?{|mask|
+      who.to_s =~ mask
+    }
+  end
+
+end
+
+class Channel
+  attr_reader :name, :members, :mode
+
+  def initialize(server, who, chname)
+    @server = server
+    @mode = ChannelMode.new(server, chname)
+    @name = chname
+    @topic = ""
+    @members = []
+    @invited = []
+  end
+
+  def join(who, key="")
+    if @mode.banned?(who)
+      ["474", @name, "Cannot join channel (+b)"]
+    elsif @mode.invite && !@invited.include?(who)
+      ["473", @name, "Cannot join channel (+i)"]
+    elsif !@mode.key.empty? && @mode.key != key
+      ["475", @name, "Cannot join channel (+k)"]
+    elsif @mode.user_max >= @members.size
+      ["471", @name, "Cannot join channel (+l)"]
+    elsif !joined?(who)
+      who.joined_channels.push self
+      @members.unshift who
+      
+      @mode.add_op who if @members.size == 1
+      sends = []
+      sends << get_topic(who)
+      sends += names(who)
+      @server.send_client_message(who, who, Command::JOIN, @name)
+      @server.handle_reply(who, sends)
+      send_to_other_members(who, Command::JOIN, @name)
+      return nil
+    end
+  end
+
+  def kick(who, by, msg="no reason")
+    if !@members.include?(by)
+      ["442", @name, "You're not on that channel"]
+    elsif !@members.include?(who)
+      ["441", who.nick, @name, "They aren't on that channel"]
+    elsif @mode.op?(by)
+      send_to_members(by, Command::KICK, @name, who.nick, msg)
+      unregister(who)
+      nil
+    else
+      ["482", @name, "You're not channel operator"]
+    end
+  end
+
+  def send_to_members(user, command, *args)
+    @members.each{|mem|
+      @server.send_client_message(mem, user, command, *args)
+    }
+  end
+
+  def send_to_other_members(user, command, *args)
+    (@members-[user]).each{|mem|
+      @server.send_client_message(mem, user, command, *args)
+    }
+  end
+
+  def joined?(who)
+    @members.include?(who)
+  end
+
+  def invite(from, who)
+    if !joined?(from)
+      return ["442", @name, "You're not on that channel"]
+    elsif joined?(who)
+      return ["443", who.nick, "is already on channel"]
+    elsif !@mode.can_invite?(from)
+      return ["482", @name, "You're not channel operator"]
+    else 
+      @invited.push who
+      @server.send_client_message(who, from, Command::INVITE, who.nick, @name)
+      return ["341", who.nick, @name]
+    end
+  end
+
+  def unregister(usr)
+    @members.delete(usr)
+    @mode.unregister(usr)
+    usr.joined_channels.delete(self)
+  end
+
+  def handle_mode(who, params)
+    if params.empty?
+      ["324", @name, *@mode.get_flags]
+    else
+      flags, *targets = params
+      flags = "+" + flags if flags[0,1] != "+" && flags[0,1] != "-"
+      if @mode.op?(who)
+        rpl = @mode.set_flags(who, flags, targets)
+        send_to_members(who, Command::MODE, @name, *rpl) if rpl
+        nil
+      elsif !joined?(who)
+        return ["482", @name, "You're not channel operator"]
+      else
+        return ["442", @name, "You're not on that channel"]
+      end
+    end
+  end
+
+  def part(who, message="")
+    if joined?(who)
+      send_to_members(who, Command::PART, @name, message)
+      unregister(who)
+      return nil
+    else
+      return ["442", @name, "You're not on that channel"]
+    end
+  end
+
+  def set_topic(who, str)
+    if @mode.can_change_topic?(who)
+      @topic = str
+      send_to_members(who, Command::TOPIC, @name, @topic)
+      return nil
+    elsif !@members.include?(who)
+      return ["442", @name, "You're not on that channel"]
+    else
+      return ["482", @name, "You're not channel operator"]
+    end
+  end
+
+  def get_topic(user, text=false)
+    return nil if @mode.secret && !joined?(user)
+    return @topic if text
+    if @topic.empty?
+      ["331", @name, "No topic is set"]
+    else
+      ["332", @name, @topic]
+    end
+  end
+
+  def privmsg(who, msg)
+    if @mode.n && !@members.include?(who)
+      return ["404", @name, "Cannot send to channel"]
+    elsif @mode.can_talk?(who)
+      send_to_other_members(who, Command::PRIVMSG, @name, msg)
+      nil
+    else
+      return ["404", @name, "Cannot send to channel"]
+    end
+  end
+
+  def notice(who, msg)
+    if @mode.can_talk?(who)
+      send_to_other_members(who, Command::NOTICE, @name, msg)
+    end
+    return nil
+  end
+
+  def visible?(user)
+    !@mode.secret || joined?(user)
+  end
+
+  def names(user)
+    return nil if @mode.secret && !joined?(user)
+    mems = @members.dup
+    unless @members.include?(user)
+      mems = mems.find_all{|usr|
+        !usr.invisible
+      }
+    end
+    memlist = mems.map{|us|
+      if @mode.op?(us)
+        pr = "@"
+      else
+        pr = ""
+      end
+      pr+us.nick
+    }.join(" ")
+    sends = []
+    unless memlist.empty?
+      prefix = if @mode.secret
+        "@"
+      elsif @mode.private
+        "*"
+      else
+        "="
+      end
+      sends << ["353", prefix, @name, memlist]
+    end
+    sends << ["366", @name, "End of NAMES list"]
+    sends
+  end
+
+end
+
+class NoModeChannel < Channel
+  def handle_mode(user, *ar)
+    ["477", @name, "Channel doesn't support modes"]
+  end
+end
+
+class SafeChannel < Channel
+end # 仕様 がよく分から ない ので実装出来なくても しよう がない
+
+end

data/lib/rupircd/conf.rb

@@ -0,0 +1,26 @@
+=begin
+= rupircd -- Ruby Pseudo IRC Deamon
+
+  ver 0.6 2009-08-11T23:45:52+09:00
+  
+  Copyright (c) 2009 Hiromi Ishii
+  
+  You can redistribute it and/or modify it under the same term as Ruby.
+=end
+
+module IRCd
+  class FileConf
+    attr_reader :conf
+    attr_accessor :path
+
+    def initialize(path)
+      @path = path
+      @conf = {}
+    end
+
+    def load
+      @conf = eval(open(@path){|f| f.read })
+    end
+
+  end
+end

data/lib/rupircd/message.rb

@@ -0,0 +1,254 @@
+=begin
+
+= rice - Ruby Irc interfaCE
+
+  $Id: irc.rb,v 1.9 2001/06/13 10:22:24 akira Exp $
+
+  Copyright (c) 2001 akira yamada <akira@ruby-lang.org>
+  You can redistribute it and/or modify it under the same term as Ruby.
+
+=end
+
+module IRCd
+  class Error < StandardError; end
+  class UnknownCommand < Error
+    attr_reader :cmd
+    def initialize(cmd)
+      @cmd = cmd
+      super("Unknown command: #@cmd")
+    end
+  end
+  class NotEnoughParameter < Error; end
+
+  class Message
+
+    def self.parse(msg)
+      prefix = nil
+      command = ""
+      tmp = []
+      if msg[0] == ?:
+        prefix, command, *tmp = msg[1..-1].split(" ")
+      else
+        command, *tmp = msg.split(" ")
+      end
+      command
+      params = []
+      eoc = false
+      tmp.each_with_index{|ar,ind|
+        if eoc
+          params[-1] += ' ' + ar
+        elsif ar[0] == ?:
+          eoc = true
+          params << ar[1..-1]
+        else
+          params << ar
+        end
+      }
+      self.build(prefix, command, params)
+    end
+
+    def self.build(prefix, command, params)
+      cmd = command
+      cmd.upcase! if String === cmd
+      if Command::Commands.include?(cmd)
+        Command::Commands[cmd.upcase].new(prefix, cmd, params)
+      elsif Reply::Replies.include?(cmd)
+        Reply::Replies[cmd].new(prefix, cmd, params)
+      else
+        raise UnknownCommand, command
+      end
+    end
+
+    def initialize(prefix, command, params)
+      @prefix  = prefix
+      @command = command
+      @params  = params
+    end
+    attr_reader :prefix, :command, :params
+
+    def to_s
+      str = ''
+      if @prefix && !@prefix.to_s.empty?
+        str << ':'
+        str << @prefix
+        str << ' '
+      end
+
+      str << @command
+
+      if @params
+        f = false
+        @params.each_with_index do |param, ind|
+          param = param.to_s
+          str << ' '
+          if !f && (param.empty? || /[: ]/ =~ param)
+            str << ':'
+            str << param
+            f = true
+          elsif !f && ind == params.size - 1
+            str << ':'
+            str << param
+          else
+            str << param
+          end
+        end
+      end
+
+      str << "\x0D\x0A"
+
+      str
+    end
+
+
+    def to_a
+      [@prefix, @command, @params]
+    end
+
+    def inspect
+      sprintf('#<%s:0x%x prefix:%s command:%s params:%s>',
+              self.class, self.object_id, @prefix, @command, @params.inspect)
+    end
+
+  end # Message
+
+
+  module Command
+    class Command < Message
+    end # Command
+
+    Commands = {}
+    %w(PASS NICK USER OPER MODE SERVICE QUIT SQUIT CLOSE
+       JOIN PART TOPIC NAMES LIST INVITE KICK
+       PRIVMSG NOTICE MOTD LUSERS VERSION STATS LINKS
+       TIME CONNECT TRACE ADMIN INFO SERVLIST SQUERY 
+       WHO WHOIS WHOWAS KILL PING PONG ERROR
+       AWAY REHASH DIE RESTART SUMMON USERS WALLOPS USERHOST ISON
+    ).each do |cmd|
+      eval <<E
+      class #{cmd} < Command
+      end
+      Commands['#{cmd}'] = #{cmd}
+
+      def #{cmd.downcase}(*params)
+        #{cmd}.new(nil, '#{cmd}', params)
+      end
+      module_function :#{cmd.downcase}
+E
+    end
+
+    # XXX:
+    class PRIVMSG
+      def to_s
+        str = ''
+        if @prefix
+          str << ':'
+          str << @prefix
+          str << ' '
+        end
+
+        str << @command
+
+        str << ' '
+        str << @params[0]
+
+        str << ' :'
+        str << @params[1..-1].join(' ')
+
+        str << "\x0D\x0A"
+        str
+      end
+    end
+  end # Command
+
+  module Reply
+    class Reply < Message
+    end
+
+    class CommandResponse < Reply
+    end
+
+    class ErrorReply < Reply
+    end
+
+    Replies = {}
+    %w(001,RPL_WELCOME 002,RPL_YOURHOST 003,RPL_CREATED
+       004,RPL_MYINFO 005,RPL_BOUNCE
+       302,RPL_USERHOST 303,RPL_ISON 301,RPL_AWAY
+       305,RPL_UNAWAY 306,RPL_NOWAWAY 311,RPL_WHOISUSER
+       312,RPL_WHOISSERVER 313,RPL_WHOISOPERATOR
+       317,RPL_WHOISIDLE 318,RPL_ENDOFWHOIS
+       319,RPL_WHOISCHANNELS 314,RPL_WHOWASUSER
+       369,RPL_ENDOFWHOWAS 321,RPL_LISTSTART
+       322,RPL_LIST 323,RPL_LISTEND 325,RPL_UNIQOPIS
+       324,RPL_CHANNELMODEIS 331,RPL_NOTOPIC
+       332,RPL_TOPIC 341,RPL_INVITING 342,RPL_SUMMONING
+       346,RPL_INVITELIST 347,RPL_ENDOFINVITELIST
+       348,RPL_EXCEPTLIST 349,RPL_ENDOFEXCEPTLIST
+       351,RPL_VERSION 352,RPL_WHOREPLY 315,RPL_ENDOFWHO
+       353,RPL_NAMREPLY 366,RPL_ENDOFNAMES 364,RPL_LINKS
+       365,RPL_ENDOFLINKS 367,RPL_BANLIST 368,RPL_ENDOFBANLIST
+       371,RPL_INFO 374,RPL_ENDOFINFO 375,RPL_MOTDSTART
+       372,RPL_MOTD 376,RPL_ENDOFMOTD 381,RPL_YOUREOPER
+       382,RPL_REHASHING 383,RPL_YOURESERVICE 391,RPL_TIM
+       392,RPL_ 393,RPL_USERS 394,RPL_ENDOFUSERS 395,RPL_NOUSERS
+       200,RPL_TRACELINK 201,RPL_TRACECONNECTING 
+       202,RPL_TRACEHANDSHAKE 203,RPL_TRACEUNKNOWN
+       204,RPL_TRACEOPERATOR 205,RPL_TRACEUSER 206,RPL_TRACESERVER
+       207,RPL_TRACESERVICE 208,RPL_TRACENEWTYPE 209,RPL_TRACECLASS
+       210,RPL_TRACERECONNECT 261,RPL_TRACELOG 262,RPL_TRACEEND
+       211,RPL_STATSLINKINFO 212,RPL_STATSCOMMANDS 219,RPL_ENDOFSTATS
+       242,RPL_STATSUPTIME 243,RPL_STATSOLINE 221,RPL_UMODEIS
+       234,RPL_SERVLIST 235,RPL_SERVLISTEND 251,RPL_LUSERCLIENT
+       252,RPL_LUSEROP 253,RPL_LUSERUNKNOWN 254,RPL_LUSERCHANNELS
+       255,RPL_LUSERME 256,RPL_ADMINME 257,RPL_ADMINLOC1
+       258,RPL_ADMINLOC2 259,RPL_ADMINEMAIL 263,RPL_TRYAGAIN
+       401,ERR_NOSUCHNICK 402,ERR_NOSUCHSERVER 403,ERR_NOSUCHCHANNEL
+       404,ERR_CANNOTSENDTOCHAN 405,ERR_TOOMANYCHANNELS
+       406,ERR_WASNOSUCHNICK 407,ERR_TOOMANYTARGETS
+       408,ERR_NOSUCHSERVICE 409,ERR_NOORIGIN 411,ERR_NORECIPIENT
+       412,ERR_NOTEXTTOSEND 413,ERR_NOTOPLEVEL 414,ERR_WILDTOPLEVEL
+       415,ERR_BADMASK 421,ERR_UNKNOWNCOMMAND 422,ERR_NOMOTD
+       423,ERR_NOADMININFO 424,ERR_FILEERROR 431,ERR_NONICKNAMEGIVEN
+       432,ERR_ERRONEUSNICKNAME 433,ERR_NICKNAMEINUSE
+       436,ERR_NICKCOLLISION 437,ERR_UNAVAILRESOURCE
+       441,ERR_USERNOTINCHANNEL 442,ERR_NOTONCHANNEL
+       443,ERR_USERONCHANNEL 444,ERR_NOLOGIN 445,ERR_SUMMONDISABLED
+       446,ERR_USERSDISABLED 451,ERR_NOTREGISTERED
+       461,ERR_NEEDMOREPARAMS 462,ERR_ALREADYREGISTRED
+       463,ERR_NOPERMFORHOST 464,ERR_PASSWDMISMATCH
+       465,ERR_YOUREBANNEDCREEP 466,ERR_YOUWILLBEBANNED
+       467,ERR_KEYSE 471,ERR_CHANNELISFULL 472,ERR_UNKNOWNMODE
+       473,ERR_INVITEONLYCHAN 474,ERR_BANNEDFROMCHAN 
+       475,ERR_BADCHANNELKEY 476,ERR_BADCHANMASK 477,ERR_NOCHANMODES
+       478,ERR_BANLISTFULL 481,ERR_NOPRIVILEGES 482,ERR_CHANOPRIVSNEEDED
+       483,ERR_CANTKILLSERVER 484,ERR_RESTRICTED 
+       485,ERR_UNIQOPPRIVSNEEDED 491,ERR_NOOPERHOST
+       501,ERR_UMODEUNKNOWNFLAG 502,ERR_USERSDONTMATCH
+       231,RPL_SERVICEINFO 232,RPL_ENDOFSERVICES
+       233,RPL_SERVICE 300,RPL_NONE 316,RPL_WHOISCHANOP
+       361,RPL_KILLDONE 362,RPL_CLOSING 363,RPL_CLOSEEND 
+       373,RPL_INFOSTART 384,RPL_MYPORTIS 213,RPL_STATSCLINE
+       214,RPL_STATSNLINE 215,RPL_STATSILINE 216,RPL_STATSKLINE
+       217,RPL_STATSQLINE 218,RPL_STATSYLINE 240,RPL_STATSVLINE
+       241,RPL_STATSLLINE 244,RPL_STATSHLINE 244,RPL_STATSSLINE
+       246,RPL_STATSPING 247,RPL_STATSBLINE 250,RPL_STATSDLINE
+       492,ERR_NOSERVICEHOST
+    ).each do |num_cmd|
+      num, cmd = num_cmd.split(',', 2)
+      eval <<E
+      class #{cmd} < #{if num[0] == ?0 || num[0] == ?2 || num[0] == ?3
+                        'CommandResponse'
+                       elsif num[0] == ?4 || num[0] == ?5
+                        'ErrorReply'
+                       end}
+      end
+      Replies['#{num}'] = #{cmd}
+
+      def #{cmd.downcase}(*params)
+        #{cmd}.new(nil, '#{cmd}', params)
+      end
+      module_function :#{cmd.downcase}
+E
+    end
+  end # Reply
+end # IRCd