36 lines
608 B
Ruby
36 lines
608 B
Ruby
require_relative "../common/String.rb"
|
|
|
|
class CodeSet
|
|
attr_accessor :code
|
|
|
|
def initialize(array_code = nil)
|
|
if (array_code)
|
|
if (array_code.any? { |code| code < 0 or code > 6 })
|
|
raise ArgumentError, "Code must be between 1 and 6"
|
|
end
|
|
|
|
@code = array_code
|
|
else
|
|
@code = Array.new(4)
|
|
end
|
|
end
|
|
|
|
def to_s
|
|
output = ""
|
|
|
|
for cell in @code
|
|
colorcode = cell + 31
|
|
output += "\e[#{colorcode}m#{cell.to_s}\e[0m\n"
|
|
end
|
|
|
|
output
|
|
end
|
|
|
|
def put_codes(array_code)
|
|
@code << array_code
|
|
end
|
|
end
|
|
|
|
thing = CodeSet.new([1,2,3,4])
|
|
print thing.to_s
|