class MethodParser

# File lib/method_parser.rb, line 13
def initialize(s)
  a = scan Parser::CurrentRuby.parse(s)
  a2 = flatten_down(a)
  @to_a = nestify(a2)
end

# File lib/method_parser.rb, line 19
def to_xml
  
  xml = RexleBuilder.new
  
  a = xml.doc do
    @to_a.each do |x|
      container, name, children = x
      xml.send(container.to_s.sub(/^c/,'k').to_sym, name: name.to_s) do
        children.each do |y|
          _, name, scope, args = y[0..2].map(&:to_s) << y[-1]
          xml.send(y.first, name: name, scope: scope, args: args.join(', '))
        end
      end
    end
  end
  
  Rexle.new(a).xml pretty: true
  
end

# File lib/method_parser.rb, line 79
def flatten_down(a3)

  r = a3.map do |x|
    x2 = (x[0].is_a?(Array) and x.length == 1) ? x.flatten(1) : x
    x2.is_a?(Array) ? flatten_down(x2) : x2
  end

  r.length == 1 ? r.flatten(1) : r
end

# File lib/method_parser.rb, line 89
def nestify(a2)

  scope = :public
  a2.inject([]) do |r, x|

    case x.first
      when :class
        r << [:class, x.last, []]
      else
        x.each do |mx|

          case mx.first
          when :def

            method = mx[1]                    
            r.last.last << 
              [:def, method, method == :initialize ? :private : scope, mx[2]]
          when :private
            scope = :private
          when :protected
            scope = :protected
          end
        end      
    end
    r
  end

end

# File lib/method_parser.rb, line 41
def scan(x)

  r = x.children.map do |y| 
    if y.is_a? Parser::AST::Node then
      list = scan(y) 
      if list.length > 0 then
        list
      else
        nil
      end
    end
  end.compact

  a2 = case x.type

    when :def
      [x.type, x.to_a.first, x.children[1].children.map{|x| x.children[0]}]
    when :class
      [x.type, x.to_a.first.to_a[1]]

    when :send

      y = x.children.last

      if  y == :private or y == :protected then
        y
      else
        []
      end

    else
      []
  end

  [a2.length > 0 ? a2 : nil, r.length > 0 ? r : nil].compact

end