require 'railroader/processors/base_processor'
require 'railroader/processors/alias_processor'
require 'railroader/processors/lib/module_helper'
require 'railroader/tracker/library'

# Process generic library and stores it in Tracker.libs
class Railroader::LibraryProcessor < Railroader::BaseProcessor
  include Railroader::ModuleHelper

  def initialize tracker
    super
    @file_name = nil
    @alias_processor = Railroader::AliasProcessor.new tracker
    @current_module = nil
    @current_class = nil
    @intializer_env = nil
  end

  def process_library src, file_name = nil
    @file_name = file_name
    process src
  end

  def process_class exp
    handle_class exp, @tracker.libs, Railroader::Library
  end

  def process_module exp
    handle_module exp, Railroader::Library
  end

  def process_defn exp
    if exp.method_name == :initialize
      @alias_processor.process_safely exp.body_list
      @initializer_env = @alias_processor.only_ivars
    elsif node_type? exp, :defn
      exp = @alias_processor.process_safely exp, @initializer_env
    else
      exp = @alias_processor.process exp
    end

    if @current_class
      exp.body = process_all! exp.body
      @current_class.add_method :public, exp.method_name, exp, @file_name
    elsif @current_module
      exp.body = process_all! exp.body
      @current_module.add_method :public, exp.method_name, exp, @file_name
    end

    exp
  end

  alias process_defs process_defn

  def process_call exp
    if process_call_defn? exp
      exp
    else
      process_default exp
    end
  end

  def process_iter exp
    res = process_default exp

    if node_type? res, :iter and call? exp.block_call # sometimes this changes after processing
      if exp.block_call.method == :included and (@current_module or @current_class)
        (@current_module || @current_class).options[:included] = res.block
      end
    end

    res
  end
end