# Copyright (c) 2010 The Mirah project authors. All Rights Reserved.
# All contributing project authors may be found in the NOTICE file.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require 'test_helper'
require 'stringio'
require 'fileutils'

module JVMCompiler
  import java.lang.System
  import java.io.PrintStream
  include Mirah

  def new_state
    state = Mirah::Util::CompilationState.new
    state.save_extensions = false
    state
  end


#  def create_transformer
#    state = Mirah::Util::CompilationState.new
#    state.save_extensions = false
#
#    transformer = Mirah::Transform::Transformer.new(state)
#    Java::MirahImpl::Builtin.initialize_builtins(transformer)
#    transformer
#  end


  def clear_tmp_files
    return unless @tmp_classes

    File.unlink(*@tmp_classes)
    @tmp_classes.clear
  end

  def compiler_type
    JVM::Compiler::JVMBytecode
  end

  def parse name, code, transformer
    AST.parse(code, name, true, transformer)
  end

 def infer_and_resolve_types ast, generator
   scoper, typer = generator.infer_asts(ast, true)
   ast
 end

 def parse_and_resolve_types name, code
   clear_tmp_files

   state = new_state

   generator = Mirah::Generator.new(state, compiler_type, false, false)
   transformer = Mirah::Transform::Transformer.new(state, generator.typer)

   #Java::MirahImpl::Builtin.initialize_builtins(transformer)

   ast = [AST.parse(code, name, true, transformer)]

   infer_and_resolve_types ast, generator

   ast
 end


  def generate_classes compiler_results
    classes = {}

    compiler_results.each do |result|
      bytes = result.bytes

      FileUtils.mkdir_p(File.dirname(result.filename))
      File.open(result.filename, 'wb') { |f| f.write(bytes) }

      classes[result.filename[0..-7]] = Mirah::Util::ClassLoader.binary_string bytes
    end

    loader = Mirah::Util::ClassLoader.new(JRuby.runtime.jruby_class_loader, classes)

    classes.keys.map do |name|
      cls = loader.load_class(name.tr('/', '.'))
      proxy = JavaUtilities.get_proxy_class(cls.name)
      @tmp_classes << "#{name}.class"
      proxy
    end
  end

  #def compile_ast  ast
  #  compiler = create_compiler
  #  compiler.compile(ast)
  #  compiler
  #end

  def compile(code, name = tmp_script_name)
    clear_tmp_files

    state = new_state

    generator = Mirah::Generator.new(state, compiler_type, false, false)
    transformer = Mirah::Transform::Transformer.new(state, generator.typer)

    #Java::MirahImpl::Builtin.initialize_builtins(transformer)

    ast = [AST.parse(code, name, true, transformer)]

    scoper, typer = generator.infer_asts(ast, true)
    compiler_results = generator.compiler.compile_asts(ast, scoper, typer)

    generate_classes compiler_results
  end

  def tmp_script_name
    "script" + System.nano_time.to_s
  end
end


module CommonAssertions
  import java.lang.System
  import java.io.PrintStream

  def assert_include(value, array, message=nil)
    message = build_message message, '<?> does not include <?>', array, value
    assert_block message do
      array.include? value
    end
  end

  def capture_output
    saved_output = System.out
    output = StringIO.new
    System.setOut(PrintStream.new(output.to_outputstream))
    begin
      yield
      output.rewind
      output.read
    ensure
      System.setOut(saved_output)
    end
  end

  def assert_output(expected, &block)
    assert_equal(expected, capture_output(&block))
  end

end

class Test::Unit::TestCase
  include JVMCompiler

  def setup
    @tmp_classes = []
  end

  def teardown
    #reset_type_factory
    clear_tmp_files
  end
end