Class: IRC::Setup

Inherits:
Object
  • Object
show all
Defined in:
lib/em-ruby-irc/IRC.rb,
lib/em-ruby-irc/default-handlers.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(bot, name, config) ⇒ Setup

Returns a new instance of Setup.



10
11
12
13
14
15
16
17
18
 
# File 'lib/em-ruby-irc/IRC.rb', line 10

def initialize(bot, name, config)
	@name = name
	@connection = nil
	@startup_handlers = Array.new
	@config = config
	@memcache = nil
	@bot = bot
	default_handlers
end

Instance Attribute Details

#bot ⇒ Object

Returns the value of attribute bot.



9
10
11
 
# File 'lib/em-ruby-irc/IRC.rb', line 9

def bot
  @bot
end

#config ⇒ Object (readonly)

Returns the value of attribute config.



8
9
10
 
# File 'lib/em-ruby-irc/IRC.rb', line 8

def config
  @config
end

#connection ⇒ Object

Returns the value of attribute connection.



9
10
11
 
# File 'lib/em-ruby-irc/IRC.rb', line 9

def connection
  @connection
end

#memcache ⇒ Object

Returns the value of attribute memcache.



9
10
11
 
# File 'lib/em-ruby-irc/IRC.rb', line 9

def memcache
  @memcache
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



8
9
10
 
# File 'lib/em-ruby-irc/IRC.rb', line 8

def name
  @name
end

#startup_handlers ⇒ Object

Returns the value of attribute startup_handlers.



9
10
11
 
# File 'lib/em-ruby-irc/IRC.rb', line 9

def startup_handlers
  @startup_handlers
end

Instance Method Details

#add_startup_handler(proc = nil, &handler) ⇒ Object 



20
21
22
 
# File 'lib/em-ruby-irc/IRC.rb', line 20

def add_startup_handler(proc=nil, &handler)
	startup_handlers << proc
end

#connect ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/em-ruby-irc/IRC.rb', line 28

def connect
	begin
		if not defined?(EventMachine::fork_reactor) or ENV["FORK"] == "false"
		  logger.warn("WARNING: Version of eventmachine does not support forking or you used 'FORK=false'.  If you specified multiple connections you will only connect to one.")
			EventMachine::run {
				begin
					self.connection = EventMachine::connect(config["server_address"], config["server_port"].to_i, IRC::Connection, :setup => self)
				rescue => err
					log_error(err)
				end
			}
		else
			logger.warn("Event machine supports forking, attempting to fork.")
			pid = EventMachine::fork_reactor {
				begin
					self.connection = EventMachine::connect(config["server_address"], config["server_port"].to_i, IRC::Connection, :setup => self)
				rescue => err
					log_error(err)
				end
			}

        File.open("#{self.name}.pid", 'w') {|f| f << pid}
        Process.detach(pid)
		end
	rescue => err
		log_error(err)
	end
end

#default_handlers ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'lib/em-ruby-irc/default-handlers.rb', line 3

def default_handlers
	logger.debug("Starting default handlers")
	begin
	@default_join_handler = Proc.new do |event|
		begin
			#Issues a who command on the channel when I join so I can generate the users hostnames
			event.connection.send_to_server "WHO #{event.message}" if event.from.downcase.chomp == event.connection.nickname.downcase.chomp
			#Make sure the bot has this channel in memory
			user = IRC::Utils.channel_user(event.connection, event.message, event.from, event.hostmask) #connection, channel, user
		rescue => err
			log_error(err)
		end
	end
	
	@default_names_reply_handler = Proc.new do |event|
		begin
			users = event.message.split
			users.each do |user|
				channel_user = IRC::Utils.channel_user(event.connection, event.mode, user) #connection, channel, user
			end
		rescue => err
			log_error(err)
		end
	end
	
	@default_part_handler = Proc.new do |event|
		begin
			if event.from.downcase.chomp == event.connection.nickname.downcase.chomp
				IRC::Utils.remove_channel(event.connection, event.channel)
			else						
				IRC::Utils.remove_channel_user(event.connection, event.channel, event.from)
			end
		rescue => err
			log_error(err)
		end
	end

	@default_who_reply_handler = Proc.new do |event|
		begin
			IRC::Utils.update_hostname(event.connection, event.stats[7], "#{event.stats[4]}@#{event.stats[5]}")
		rescue => err
			log_error(err)
		end
	end
	
	@default_nick_in_use_handler = Proc.new do |event|
		begin
			logger.debug("OMG NICKNAME IS BEING USED!")
			new_nickname = event.connection.nickname[0,14] + rand(9).to_s
			event.connection.send_to_server "NICK #{new_nickname}"
			event.connection.send_to_server "USER #{event.connection.username} 8 * :#{event.connection.realname}"
			event.connection.nickname = new_nickname
		rescue => err
			log_error(err)
		end
	end

	logger.debug("Registering handlers")
	IRC::Utils.add_handler('join', @default_join_handler, self)
	IRC::Utils.add_handler('namreply', @default_names_reply_handler, self)
	IRC::Utils.add_handler('part', @default_part_handler, self)
	IRC::Utils.add_handler('whoreply', @default_who_reply_handler, self)
	IRC::Utils.add_handler('nicknameinuse', @default_nick_in_use_handler, self)
	IRC::Utils.add_handler('ping', lambda {|event| event.connection.send_to_server("PONG #{event.message}") }, self)
	rescue => err
		log_error(err)
	end
end

#reset_startup_handlers ⇒ Object 



24
25
26
 
# File 'lib/em-ruby-irc/IRC.rb', line 24

def reset_startup_handlers
	startup_handlers.clear
end