Change State#to_s and add State#to_s_color

The State#to_s method has been modified to only display the direct class name
and not the full module name, and to display parameters. Also, to_s_color has
been added which uses tty color escapes to color its output according to the
state's status.

1 files changed, 18 insertions, 1 deletions

diff --git a/state.rb b/state.rb
index 0d36f80..b27e3f2 100644
--- a/state.rb
+++ b/state.rb
@@ -90,7 +90,24 @@ class State
 
 	# Puts the class name and parameters
 	def to_s
-		"#{self.class.name}#{params.inspect} (#{@status.to_s})"
+		"#{name} (#{@status.to_s})"
+	end
+
+	# The name of this state
+	def name
+		"#{self.class.name.split("::").last}#{params.inspect}"
+	end
+
+	# Prints a string that uses TTY color escapes
+	def to_s_color
+		colors = {
+			:up => "1;32",
+			:down => "1;31",
+			:rising => "0;32",
+			:dropping => "0;31",
+		}
+
+		"\e[%sm%s\e[0;49m" % [colors[status], name]
 	end
 
 	# Hold this state. Holding a state informs the system that you would like it