Add items for variable assignment and brackets

Author: paulgoetze

lib/code_breaker/parser.rb

@@ -19,10 +19,29 @@ def run
     private
 
     def parse(input)
-      parsed_ast = ::Parser::CurrentRuby.parse(input)
-      nodes = parsed_ast.loc.node.children[1].children
+      ast = ::Parser::CurrentRuby.parse(input)
+
+      if variable_assignment?(ast)
+        parse_variable_assignment(ast)
+      else
+        parse_statement(ast)
+      end
+    end
+
+    def variable_assignment?(ast)
+      ast.loc.node.type == :lvasgn
+    end
+
+    def parse_variable_assignment(ast)
+      nodes = ast.loc.node.children[1].children
       result = parse_nodes(nodes)
-      cleanup(result)
+      cleanup(result).flatten(1).unshift(ast.loc.node.children[0], :'=')
+    end
+
+    def parse_statement(ast)
+      nodes = ast.loc.node.children
+      result = parse_nodes(nodes)
+      cleanup(result).flatten(1)
     end
 
     def parse_nodes(nodes)
@@ -32,17 +51,21 @@ def parse_nodes(nodes)
     end
 
     def cleanup(nodes)
-      result = nodes.map do |node|
+      nodes.map do |node|
         if node.kind_of?(Symbol)
           node
-        elsif node.kind_of?(Array) && node.length == 1
-          node[0].class
         elsif node.kind_of?(Array)
-          node[0].nil? ? node[1].to_s.constantize : cleanup(node)
+          if node.length > 1
+            node[0].nil? ? node[1].to_s.constantize : cleanup(node)
+          elsif node[0].kind_of?(Array)
+            cleanup(node)
+          else
+            node[0].class
+          end
+        else
+          node.class
         end
       end
-
-      result.flatten
     end
   end
 end

spec/code_breaker/parser_spec.rb

@@ -0,0 +1,67 @@
+require 'spec_helper'
+
+describe CodeBreaker::Parser do
+
+  let(:code_snippet) { 'sum = "2".to_i + 3' }
+  let(:output) { [:sum, :'=', String, :to_i, :+, Fixnum] }
+
+  subject { CodeBreaker::Parser.new(code_snippet) }
+
+  it { is_expected.to respond_to :run }
+  it { is_expected.to respond_to :input }
+  it { is_expected.to respond_to :output }
+
+  describe 'input' do
+    it 'returns the code snippet the parser was instanciated with' do
+      expect(subject.input).to eq code_snippet
+    end
+  end
+
+  describe 'output' do
+    it 'is an alias method for #run' do
+      expect(subject.method(:output)).to eq subject.method(:run)
+    end
+  end
+
+  describe '#run' do
+    it 'returns an Array' do
+      expect(subject.run).to be_an Array
+    end
+
+    context 'for code snippets defining a variable' do
+      it 'returns the called classes and methods of the right hand side' do
+        code_snippet = 'number = 1 + 2.3 - Rational(2,3) * Complex(0.4, 0.2)'
+        expected_result = [:number, :'=', Fixnum, :+, Float, :-, Rational, :*, Complex]
+
+        actual_result = CodeBreaker::Parser.new(code_snippet).run
+        expect(actual_result).to eq expected_result
+      end
+
+      it 'returns a single item Array for only one statement' do
+        code_snippet = '"3.14159265359"'
+        expected_result = [String]
+
+        actual_result = CodeBreaker::Parser.new(code_snippet).run
+        expect(actual_result).to eq expected_result
+      end
+
+      it 'returns a nested Array for nested statements' do
+        code_snippet = 'number = 1.2 * (Rational(2) - 1)'
+        expected_result = [:number, :'=', Float, :*, [Rational, :-, Fixnum]]
+
+        actual_result = CodeBreaker::Parser.new(code_snippet).run
+        expect(actual_result).to eq expected_result
+      end
+    end
+
+    context 'for code snippets not assigning a variable' do
+      it 'returns the called classes and methods' do
+        code_snippet = '1.2 * (Rational(2) - 1)'
+        expected_result = [Float, :*, [Rational, :-, Fixnum]]
+
+        actual_result = CodeBreaker::Parser.new(code_snippet).run
+        expect(actual_result).to eq expected_result
+      end
+    end
+  end
+end