Class: Butler::Remote::Connection

Inherits:
Object
  • Object
show all
Includes:
Log::Comfort
Defined in:
lib/butler/remote/connection.rb

Instance Attribute Summary collapse

Attributes included from Log::Comfort

#logger

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Log::Comfort

#debug, #error, #exception, #fail, #info, #log, #warn

Constructor Details

#initialize(server, socket) ⇒ Connection

Returns a new instance of Connection.



35
36
37
38
39
40
 
# File 'lib/butler/remote/connection.rb', line 35

def initialize(server, socket)
	@server = server
	@socket = socket
	@user   = nil
	@logger = nil
end

Instance Attribute Details

#user ⇒ Object (readonly)

Returns the value of attribute user.



33
34
35
 
# File 'lib/butler/remote/connection.rb', line 33

def user
  @user
end

Class Method Details

.open(server, socket) ⇒ Object 



25
26
27
28
29
 
# File 'lib/butler/remote/connection.rb', line 25

def self.open(server, socket)
	connection = new(server, socket)
	connection.start
	connection
end

.open_thread(server, socket) ⇒ Object 



16
17
18
19
20
21
22
23
 
# File 'lib/butler/remote/connection.rb', line 16

def self.open_thread(server, socket)
	connection = new(server, socket)
	Thread.new {
		connection.start
		connection.read_loop
	}
	connection
end

Instance Method Details

#close ⇒ Object 



108
109
110
 
# File 'lib/butler/remote/connection.rb', line 108

def close
	@socket.close unless @socket.closed?
end

#closed? ⇒ Boolean

Returns:

  • (Boolean) 


104
105
106
 
# File 'lib/butler/remote/connection.rb', line 104

def closed?
	@socket.closed?
end

#puts(*text) ⇒ Object 



94
95
96
 
# File 'lib/butler/remote/connection.rb', line 94

def puts(*text)
	@socket.puts(*text)
end

#quit(reason = nil) ⇒ Object 



98
99
100
101
102
 
# File 'lib/butler/remote/connection.rb', line 98

def quit(reason=nil)
	puts reason if reason
	close
	@server.disconnected(self)
end

#read_loop ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/butler/remote/connection.rb', line 71

def read_loop
	while line = @socket.gets
		line.chomp!
		p line
		break if line.empty?
		begin
			@server.dispatch(Message.new(@server, self, @user, line))

		# on these errors we got to get out of the loop
		rescue Errno::EPIPE, SocketError
			raise
		# all others should not put the read-thread in any danger
		rescue Exception => e
			exception(e)
		end
	end
rescue Exception => e
	exception(e)
ensure
	@server.disconnected(self)
	close
end

#start ⇒ Object 



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
 
# File 'lib/butler/remote/connection.rb', line 42

def start
	puts("Please identify yourself")
	tries    = 3
	access   = nil
	username = nil
	timeout(180) {
		tries.times { |i|
			@socket.print("Username: ")
			username = @socket.gets.chomp
			@socket.print("Password: ")
			password = @socket.gets.chomp
			access   = @server.(username, password)
			break if access
			sleep(i)
			puts("Failed to authenticate, please try again.")
		}
	}
	return quit("Failed to authenticate") unless access
	puts "You're logged in, welcome #{username}."
	puts "To terminate the session, enter an empty line."
	puts ""
	@user = User.new(username, access)
rescue Timeout::Error
	quit("Login timed out.")
rescue Exception => e
	exception(e)
	quit(nil)
end