module Zombie
  def self.setup
    return if defined?(@done)
    files.each do |path|
      path = "#{File.expand_path(path, root)}.rb"
      ast = File.read(path).to_ast
      zombify(ast, path)
      ToSource.to_source(ast.body)
    end
    @done = true
  end

  # Return library root directory
  #
  # @return [String]
  #
  # @api private
  #
  def self.root
    File.expand_path('../../../lib',__FILE__)
  end
  private_class_method :root


  # Replace Mutant with Zombie namespace
  #
  # @param [Rubinius::AST::Node]
  #
  # @api private
  #
  # @return [undefined]
  #
  def self.zombify(root, path)
    node = find_mutant(root)
    unless node
      raise "unable to find mutant in AST from: #{path.inspect}"
    end

    name = node.name

    node.name = Rubinius::AST::ModuleName.new(name.line, :Zombie)

    scope = node.body

    unless scope.kind_of?(Rubinius::AST::EmptyBody)
      node.body = Rubinius::AST::ModuleScope.new(scope.line, node.name, scope.body)
    end

    script = Rubinius::AST::Script.new(root)
    script.file = path

    # For some reason loading is not the same as eval...?
    # eval(root.to_source)

    ::Mutant::Loader.run(script)
  end
  private_class_method :zombify

  # Find mutant module in AST
  #
  # @param [Rubinius::AST::Node]
  #
  # @return [Rubinius::AST::Node]
  #
  def self.find_mutant(root)
    if is_mutant?(root)
      return root
    end

    unless root.kind_of?(Rubinius::AST::Block)
      raise "Cannot find mutant in: #{root.class}"
    end

    root.array.each do |node|
      return node if is_mutant?(node)
    end

    nil
  end
  private_class_method :find_mutant

  # Test if node is mutant module
  #
  # @param [Rubinius::AST::Node]
  #
  # @return [true]
  #   returns true if node is the mutant module
  #
  # @return [false]
  #   returns false otherwise
  #
  # @api private
  #
  def self.is_mutant?(node)
    node.kind_of?(Rubinius::AST::Module) && is_mutant_name?(node.name)
  end
  private_class_method :is_mutant?

  # Test if node is mutant module name
  #
  # @param [Rubinius::AST::ModuleName]
  #
  # @return [true]
  #   returns true if node is the mutant module name
  #
  # @return [false]
  #   returns false otherwise
  #
  # @api private
  #
  def self.is_mutant_name?(node)
    node.name == :Mutant
  end
  private_class_method :is_mutant_name?

  # Return all library files the mutant is made of.
  #
  # @return [Array<String>]
  #
  # @api private
  #
  # FIXME: 
  #  Yeah looks very ugly but im currently to exited to do a cleanup.
  #
  def self.files
    block = File.read("lib/mutant.rb").to_ast
    files = block.array.select do |node|
      node.class          == Rubinius::AST::SendWithArguments &&
      node.receiver.class == Rubinius::AST::Self              &&
      node.name           == :require
    end.map do |node|
      arguments = node.arguments.array
      raise unless arguments.one?
      argument = arguments.first
      raise unless argument.class == Rubinius::AST::StringLiteral
      argument.string
    end.select do |file|
      file =~ /\Amutant/
    end
  end
  private_class_method :files
end