Class: ClientWrapper

Inherits:
Object
  • Object
show all
Defined in:
lib/accu-net.rb

Overview

The wrapper for TCPClient.

Instance Method Summary collapse

Constructor Details

#initialize(port = 2000, ip = "127.0.0.1") {|_self, @client| ... } ⇒ ClientWrapper

Returns a new instance of ClientWrapper.

Yields:

  • (_self, @client)

Yield Parameters:

  • _self (ClientWrapper) —

    the object that the method was called on



133
134
135
136
137
138
 
# File 'lib/accu-net.rb', line 133

def initialize(port=2000,ip="127.0.0.1")
	@client = TCPSocket.open(ip, port)
	@password,@mode = "password",:double
	yield self,@client if block_given?
	@client.close
end

Instance Method Details

#listen {|msg| ... } ⇒ Object

Listens for a generic network communication and yields to a passed block (if any).

Yields:

  • (msg) 


142
143
144
145
146
 
# File 'lib/accu-net.rb', line 142

def listen()
	msg = @client.gets.chomp
	yield msg if block_given?
	return msg
end

#listen_code ⇒ Object

Listens for a network code.



159
160
161
162
163
164
165
166
167
 
# File 'lib/accu-net.rb', line 159

def listen_code()
	listen { |msg|
		if msg.slice(0..1) == "C:" then
			return msg.slice(2..msg.length).to_i
		else
			return msg.slice(0..1)
		end
	}
end

#listen_encrypted ⇒ Object

Listen for an encrypted connection.



170
171
172
173
174
175
176
177
 
# File 'lib/accu-net.rb', line 170

def listen_encrypted()
	if listen_code == 3 then
		password,source = EncryptDecrypt.decrypt(@password,((listen_lines).join),@mode)
		return source
	else
		raise "Sent code was not 3!"
	end
end

#listen_lines ⇒ Object

Listens for multiple lines, appends them together and stops at END.



150
151
152
153
154
155
156
 
# File 'lib/accu-net.rb', line 150

def listen_lines()
	lines = []
	while line = @client.gets.chomp and line != "END" do
		lines << line
	end
	lines
end

#send(text) ⇒ Object

Sends a plaintext message.



180
181
182
 
# File 'lib/accu-net.rb', line 180

def send(text)
	@client.puts text
end

#send_code(code) ⇒ Object

Sends a code.



203
204
205
 
# File 'lib/accu-net.rb', line 203

def send_code(code)
	send("C:#{code}")
end

#send_encrypted(lines) ⇒ Object

Send over an encrypted multiline message.



195
196
197
198
199
200
 
# File 'lib/accu-net.rb', line 195

def send_encrypted(lines)
	secret,text = EncryptDecrypt.encrypt(@password,lines,@mode)
	send_code(3)
	send(text)
	send("END")
end

#send_lines(lines) ⇒ Object

Sends multiple lines and appends END to end the send.



186
187
188
189
190
191
192
 
# File 'lib/accu-net.rb', line 186

def send_lines(lines)
	array = lines.split("\n")
	array.each do |line|
		send(line)
	end
	send("END")
end

#set_password(string) ⇒ Object

Set encryption password.



208
209
210
211
 
# File 'lib/accu-net.rb', line 208

def set_password(string)
	warn "Argument 1 is not a string!  Continuing anyway." if not string.is_a? String
	@password = string.to_s
end