Contents

# -*- coding: utf-8 -*-
# To change this template, choose Tools | Templates
# and open the template in the editor.

require 'func_filter'

class MethodFilter < Ripper::Filter
  attr_accessor :mclass, :method_str, :funcs_str, :func_filters

  def initialize(src, mclass)
    super src

    @method_str = ''
    @funcs_str = Array.new
    @mclass = mclass
    @func_filters = Array.new
    @is_method = false
    @is_func = false
    @is_do = false
  end

  def add_tok tok
    if @is_func
      @funcs_str[-1] += tok
    elsif @is_method
      @method_str += tok
    end
  end

  def on_default(event, tok, f)
    add_tok tok
  end

  def on_kw(tok, f)
    if tok == 'def'
      @is_method = true
      @is_func = false
    end

    if tok == 'do'
      if !@is_do
        @is_do = true
      end
    end

    if tok == 'end'
      if @is_do
        add_tok tok
        @is_do = false
      elsif @is_func
        add_tok tok
        @is_func = false
      elsif @is_method
        index = 0
        @mclass.funcs.each do |func|
          @func_filters << FuncFilterFactory.get(func, @funcs_str[index])
          @func_filters.last.parse
          index += 1
        end
        @is_method = false
      end
    end

    add_tok tok
  end

  def on_ident(tok, f)
    if !@method_name.nil? && MigrationDefs::FuncName.has_key?(tok)
      @func_type = tok
      @is_func = true
      @funcs_str << ''
    elsif !@func_type.nil?
      @func_class = @mclass.add_func(@func_type, tok)
      @func_type = nil
    elsif MigrationDefs::MethodName.include?(tok)
      @method_name = tok
    end
    add_tok tok
  end

  def on_nl(tok, f)
    @is_func = false if !@is_do
    Rails.logger.debug "on_nl2-----#{@is_do.inspect} / #{@is_func.to_s} / #{@is_method.to_s}-----"
  end

  def on_lbrase(tok, f)
    if !@is_func
      @is_func = true
    elsif !@is_do
      @is_do = true
    end
    add_tok tok
  end

  def on_rbrase(tok, f)
    if @is_do
      @is_do = false
    elsif @is_func
      @is_func = false
    end
    add_tok tok
  end

end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
visual_migrate-4.1.7.0	lib/method_filter.rb
visual_migrate-4.1.0.0	lib/method_filter.rb
visual_migrate-4.0.2.0	lib/method_filter.rb
visual_migrate-4.0.1.6	lib/method_filter.rb
visual_migrate-4.0.1.5	lib/method_filter.rb
visual_migrate-4.0.1.4	lib/method_filter.rb
visual_migrate-4.0.1.3	lib/method_filter.rb
visual_migrate-4.0.1.2	lib/method_filter.rb