lib/vesta/helpers.rb in vesta-chat-0.1.0 vs lib/vesta/helpers.rb in vesta-chat-0.1.1
- old
+ new
@@ -1,68 +1,68 @@
-#
-# Helpers
-#
-module Vesta
- class Helpers
-
- Thread.abort_on_exception = true
-
- # Print banner
- def self.banner
- "\t\tRunning Vesta v#{Vesta::VERSION}"
- end
-
- # Do something in every x seconds
- def every(seconds)
- Thread.new do
- loop do
- sleep seconds
- yield
- end
- end
- end
-
- # Root folder
- def self.root
- "#{File.expand_path( File.dirname(File.dirname(File.dirname(__FILE__))) )}"
- end
-
- # Add peer to peer object [peers_object]
- def add_peer(peer,host,port,public_key)
- peer.push({host: host, port: port, public_key: public_key})
- peer
- end
-
- # Add message to [messages_object]
- def add_message(messages_object, from, message)
- messages_object.push({from: from, message: message})
- messages_object
- end
-
- # Print out the status
- def render_state
- puts "Node is running on " + Service::LHOST.to_s.magenta + ':'.colorize(:background => :white, :color => :magenta).bold + Service::LPORT.to_s.magenta.bold
- print "My Peers: ".upcase
- $PEERS.each do |peer|
- print "#{peer[:host]}:#{peer[:port]}".yellow.bold + ', '
- end
- end
-
- # Communicate with peers
- def communicate(index,host,port)
- communicate_response = Client.communicate(host, port, YAML.dump($PEERS))
- parsed_response = YAML.load(communicate_response)
- new_peer = parsed_response['peers']
- update_peers(new_peer)
- rescue Faraday::ConnectionFailed
- $PEERS.delete_at(index)
- end
-
- # Update peers
- def update_peers(new_peers)
- # if !new_peers.nil? # if new peers are not nil, then append them to $PEERS
- $PEERS = $PEERS | new_peers
- $PEERS.uniq
- # end
- end
- end
+#
+# Helpers
+#
+module Vesta
+ class Helpers
+
+ Thread.abort_on_exception = true
+
+ # Print banner
+ def self.banner
+ "\t\tRunning Vesta v#{Vesta::VERSION}"
+ end
+
+ # Do something in every x seconds
+ def every(seconds)
+ Thread.new do
+ loop do
+ sleep seconds
+ yield
+ end
+ end
+ end
+
+ # Root folder
+ def self.root
+ "#{File.expand_path( File.dirname(File.dirname(File.dirname(__FILE__))) )}"
+ end
+
+ # Add peer to peer object [peers_object]
+ def add_peer(peer,host,port,public_key)
+ peer.push({host: host, port: port, public_key: public_key})
+ peer
+ end
+
+ # Add message to [messages_object]
+ def add_message(messages_object, from, message)
+ messages_object.push({from: from, message: message})
+ messages_object
+ end
+
+ # Print out the status
+ def render_state
+ puts "Node is running on " + Service::LHOST.to_s.magenta + ':'.colorize(:background => :white, :color => :magenta).bold + Service::LPORT.to_s.magenta.bold
+ print "My Peers: ".upcase
+ $PEERS.each do |peer|
+ print "#{peer[:host]}:#{peer[:port]}".yellow.bold + ', '
+ end
+ end
+
+ # Communicate with peers
+ def communicate(index,host,port)
+ communicate_response = Client.communicate(host, port, YAML.dump($PEERS))
+ parsed_response = YAML.load(communicate_response)
+ new_peer = parsed_response['peers']
+ update_peers(new_peer)
+ rescue Faraday::ConnectionFailed
+ $PEERS.delete_at(index)
+ end
+
+ # Update peers
+ def update_peers(new_peers)
+ # if !new_peers.nil? # if new peers are not nil, then append them to $PEERS
+ $PEERS = $PEERS | new_peers
+ $PEERS.uniq
+ # end
+ end
+ end
end
\ No newline at end of file