hector 1.0.0

data/lib/hector.rb

@@ -0,0 +1,42 @@
+require "digest/sha1"
+require "eventmachine"
+require "fileutils"
+require "socket"
+require "yaml"
+
+require "hector/errors"
+
+require "hector/concerns/authentication"
+require "hector/concerns/keep_alive"
+require "hector/concerns/presence"
+
+require "hector/commands/away"
+require "hector/commands/join"
+require "hector/commands/mode"
+require "hector/commands/names"
+require "hector/commands/nick"
+require "hector/commands/notice"
+require "hector/commands/part"
+require "hector/commands/ping"
+require "hector/commands/pong"
+require "hector/commands/privmsg"
+require "hector/commands/quit"
+require "hector/commands/realname"
+require "hector/commands/topic"
+require "hector/commands/who"
+require "hector/commands/whois"
+
+require "hector/channel"
+require "hector/connection"
+require "hector/deference"
+require "hector/heartbeat"
+require "hector/identity"
+require "hector/logging"
+require "hector/request"
+require "hector/response"
+require "hector/server"
+require "hector/session"
+require "hector/service"
+require "hector/user_session"
+require "hector/version"
+require "hector/yaml_identity_adapter"

data/lib/hector/boot.rb

@@ -0,0 +1,62 @@
+require "pathname"
+require "logger"
+
+module Hector
+  class << self
+    attr_accessor :lib, :root
+
+    def start
+      raise LoadError, "please specify HECTOR_ROOT" unless Hector.root
+      load_defaults
+      load_application
+    end
+
+    def load_root
+      if root = ENV["HECTOR_ROOT"]
+        Hector.root = Pathname.new(File.expand_path(root))
+      else
+        Hector.root = find_application_root_from(Dir.pwd)
+      end
+    end
+
+    def load_defaults
+      log_path = Hector.root.join("log/hector.log")
+      Hector.logger = Logger.new(log_path)
+      Hector.logger.datetime_format = "%Y-%m-%d %H:%M:%S"
+
+      identities_path = Hector.root.join("config/identities.yml")
+      Hector::Identity.adapter = Hector::YamlIdentityAdapter.new(identities_path)
+    end
+
+    def load_application
+      $:.unshift Hector.root.join("lib")
+      load Hector.root.join("init.rb")
+    end
+
+    def find_application_root_from(working_directory)
+      dir = Pathname.new(working_directory)
+      dir = dir.parent while dir != dir.parent && dir.basename.to_s !~ /\.hect$/
+      dir == dir.parent ? nil : dir
+    end
+  end
+end
+
+Hector.lib = Pathname.new(File.expand_path(File.dirname(__FILE__) + "/.."))
+Hector.load_root
+
+if Hector.root
+  vendor_lib = Hector.root.join("vendor/hector/lib")
+  Hector.lib = vendor_lib if vendor_lib.exist?
+end
+
+$:.unshift Hector.lib
+
+begin
+  require "hector"
+rescue LoadError => e
+  if require "rubygems"
+    retry
+  else
+    raise e
+  end
+end

data/lib/hector/channel.rb

@@ -0,0 +1,106 @@
+module Hector
+  class Channel
+    attr_reader :name, :topic, :user_sessions, :created_at
+
+    class << self
+      def find(name)
+        channels[normalize(name)]
+      end
+
+      def find_all_for_session(session)
+        channels.values.find_all do |channel|
+          channel.has_session?(session)
+        end
+      end
+
+      def create(name)
+        new(name).tap do |channel|
+          channels[normalize(name)] = channel
+        end
+      end
+
+      def find_or_create(name)
+        find(name) || create(name)
+      end
+
+      def delete(name)
+        channels.delete(name)
+      end
+
+      def normalize(name)
+        if name =~ /^#\w[\w-]{0,15}$/u
+          name.downcase
+        else
+          raise NoSuchChannel, name
+        end
+      end
+
+      def reset!
+        @channels = nil
+      end
+
+      protected
+        def channels
+          @channels ||= {}
+        end
+    end
+
+    def initialize(name)
+      @name = name
+      @user_sessions = []
+      @created_at = Time.now
+    end
+
+    def broadcast(command, *args)
+      catch :stop do
+        Session.broadcast_to(sessions, command, *args)
+      end
+    end
+
+    def change_topic(session, topic)
+      @topic = { :body => topic, :nickname => session.nickname, :time => Time.now }
+    end
+
+    def channel?
+      true
+    end
+
+    def deliver(message_type, session, options)
+      if has_session?(session)
+        broadcast(message_type, name, options.merge(:except => session))
+      else
+        raise CannotSendToChannel, name
+      end
+    end
+
+    def destroy
+      self.class.delete(name)
+    end
+
+    def has_session?(session)
+      sessions.include?(session)
+    end
+
+    def join(session)
+      return if has_session?(session)
+      user_sessions.push(session)
+    end
+
+    def nicknames
+      user_sessions.map { |session| session.nickname }
+    end
+
+    def part(session)
+      user_sessions.delete(session)
+      destroy if user_sessions.empty?
+    end
+
+    def services
+      Service.all
+    end
+
+    def sessions
+      services + user_sessions
+    end
+  end
+end

data/lib/hector/commands/away.rb

@@ -0,0 +1,16 @@
+module Hector
+  module Commands
+    module Away
+      def on_away
+        away_message = request.args.first
+        if away_message and !away_message.empty?
+          @away_message = away_message
+          respond_with("306", :text => "You have been marked as being away")
+        else
+          @away_message = nil
+          respond_with("305", :text => "You are no longer marked as being away")
+        end
+      end
+    end
+  end
+end

data/lib/hector/commands/join.rb

@@ -0,0 +1,16 @@
+module Hector
+  module Commands
+    module Join
+      def on_join
+        request.args.first.split(",").each do |channel_name|
+          channel = Channel.find_or_create(channel_name)
+          if channel.join(self)
+            channel.broadcast(:join, :source => source, :text => channel.name)
+            respond_to_topic(channel)
+            respond_to_names(channel)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/hector/commands/mode.rb

@@ -0,0 +1,31 @@
+module Hector
+  module Commands
+    module Mode
+      def on_mode
+        subject = find(request.args.first)
+
+        if subject.channel?
+          if requesting_modes?
+            respond_with("324", nickname, subject.name, "+", :source => Hector.server_name)
+            respond_with("329", nickname, subject.name, subject.created_at.to_i, :source => Hector.server_name)
+          elsif requesting_bans?
+            respond_with("368", nickname, subject.name, :text => "End of Channel Ban List", :source => Hector.server_name)
+          end
+        else
+          if requesting_modes?
+            respond_with("221", nickname, "+", :source => Hector.server_name)
+          end
+        end
+      end
+
+      private
+        def requesting_modes?
+          request.args.length == 1
+        end
+
+        def requesting_bans?
+          request.args.length == 2 && request.args.last[/^\+?b$/]
+        end
+    end
+  end
+end

data/lib/hector/commands/names.rb

@@ -0,0 +1,16 @@
+module Hector
+  module Commands
+    module Names
+      def on_names
+        channel = Channel.find(request.args.first)
+        respond_to_names(channel)
+      end
+
+      def respond_to_names(channel)
+        responses = Response.apportion_text(channel.nicknames, "353", nickname, "=", channel.name, :source => Hector.server_name)
+        responses.each { |response| respond_with(response) }
+        respond_with("366", nickname, channel.name, :source => Hector.server_name, :text => "End of /NAMES list.")
+      end
+    end
+  end
+end

data/lib/hector/commands/nick.rb

@@ -0,0 +1,11 @@
+module Hector
+  module Commands
+    module Nick
+      def on_nick
+        old_nickname = nickname
+        rename(request.args.first)
+        broadcast(:nick, nickname, :source => old_nickname)
+      end
+    end
+  end
+end

data/lib/hector/commands/notice.rb

@@ -0,0 +1,10 @@
+module Hector
+  module Commands
+    module Notice
+      def on_notice
+        touch_presence
+        find(request.args.first).deliver(:notice, self, :source => source, :text => request.text)
+      end
+    end
+  end
+end

data/lib/hector/commands/part.rb

@@ -0,0 +1,11 @@
+module Hector
+  module Commands
+    module Part
+      def on_part
+        channel = Channel.find(request.args.first)
+        channel.broadcast(:part, channel.name, :source => source, :text => request.text)
+        channel.part(self)
+      end
+    end
+  end
+end

data/lib/hector/commands/ping.rb

@@ -0,0 +1,9 @@
+module Hector
+  module Commands
+    module Ping
+      def on_ping
+        respond_with(:pong, :source => Hector.server_name, :text => request.text)
+      end
+    end
+  end
+end

data/lib/hector/commands/pong.rb

@@ -0,0 +1,9 @@
+module Hector
+  module Commands
+    module Pong
+      def on_pong
+        @received_pong = true
+      end
+    end
+  end
+end

data/lib/hector/commands/privmsg.rb

@@ -0,0 +1,16 @@
+module Hector
+  module Commands
+    module Privmsg
+      def on_privmsg
+        touch_presence
+        subject = find(request.args.first)
+
+        subject.deliver(:privmsg, self, :source => source, :text => request.text)
+
+        if !subject.channel? and subject.away?
+          respond_with("301", subject.nickname, :text => subject.away_message)
+        end
+      end
+    end
+  end
+end

data/lib/hector/commands/quit.rb

@@ -0,0 +1,10 @@
+module Hector
+  module Commands
+    module Quit
+      def on_quit
+        @quit_message = "Quit: #{request.text}"
+        connection.close_connection(true)
+      end
+    end
+  end
+end

data/lib/hector/commands/realname.rb

@@ -0,0 +1,10 @@
+module Hector
+  module Commands
+    module Realname
+      def on_realname
+        @realname = request.text
+        broadcast("352", :$nickname, "*", who, :source => Hector.server_name)
+      end
+    end
+  end
+end

data/lib/hector/commands/topic.rb

@@ -0,0 +1,26 @@
+module Hector
+  module Commands
+    module Topic
+      def on_topic
+        channel = Channel.find(request.args.first)
+
+        if request.args.length > 1
+          topic = request.text
+          channel.change_topic(self, topic)
+          channel.broadcast(:topic, channel.name, :source => source, :text => topic)
+        else
+          respond_to_topic(channel)
+        end
+      end
+
+      def respond_to_topic(channel)
+        if topic = channel.topic
+          respond_with("332", nickname, channel.name, :source => Hector.server_name, :text => topic[:body])
+          respond_with("333", nickname, channel.name, topic[:nickname], topic[:time].to_i, :source => Hector.server_name)
+        else
+          respond_with("331", nickname, channel.name, :source => Hector.server_name, :text => "No topic is set.")
+        end
+      end
+    end
+  end
+end

data/lib/hector/commands/who.rb

@@ -0,0 +1,29 @@
+module Hector
+  module Commands
+    module Who
+      def on_who
+        name = request.args.first
+
+        if destination = destination_klass_for(name).find(name)
+          sessions_for_who(destination).each do |session|
+            respond_with("352", nickname, name, session.who, :source => Hector.server_name)
+          end
+        end
+
+        respond_with("315", nickname, name, :source => Hector.server_name, :text => "End of /WHO list.")
+      end
+
+      def sessions_for_who(destination)
+        if destination.respond_to?(:sessions)
+          destination.sessions
+        else
+          [destination]
+        end
+      end
+
+      def who
+        "#{username} #{Hector.server_name} #{Hector.server_name} #{nickname} H :0 #{realname}"
+      end
+    end
+  end
+end