Implement Parser#command_type

Author: leocassarani

lib/parser.rb

@@ -1,4 +1,8 @@
 class Parser
+  class A_COMMAND; end
+  class C_COMMAND; end
+  class L_COMMAND; end
+
   def initialize(input)
     @lines = split_lines(input)
   end
@@ -8,12 +12,22 @@ def has_more_commands?
   end
 
   def advance
-    lines.shift
+    @current = lines.shift
+  end
+
+  def command_type
+    if current.start_with?('@')
+      A_COMMAND
+    elsif current.start_with?('(')
+      L_COMMAND
+    else
+      C_COMMAND
+    end
   end
 
   private
 
-  attr_reader :lines
+  attr_reader :lines, :current
 
   def split_lines(input)
     input