module Java::Commands

# File lib/buildr/java/commands.rb, line 133
def apt(*args)
  options = Hash === args.last ? args.pop : {}
  rake_check_options options, :compile, :source, :output, :classpath

  files = args.flatten.map(&:to_s).
    collect { |arg| File.directory?(arg) ? FileList["#{arg}/**/*.java"] : arg }.flatten
  cmd_args = [ trace?(:apt) ? '-verbose' : '-nowarn' ]
  if options[:compile]
    cmd_args << '-d' << options[:output].to_s
  else
    cmd_args << '-nocompile' << '-s' << options[:output].to_s
  end
  cmd_args << '-source' << options[:source] if options[:source]
  cp = classpath_from(options)
  cmd_args << '-classpath' << cp.join(File::PATH_SEPARATOR) unless cp.empty?
  cmd_args += files
  unless Buildr.application.options.dryrun
    info 'Running apt'
    trace (['apt'] + cmd_args).join(' ')
    Java.load
    ::Java::com.sun.tools.apt.Main.process(cmd_args.to_java(::Java::java.lang.String)) == 0 or
      fail 'Failed to process annotations, see errors above'
  end
end

# File lib/buildr/java/commands.rb, line 43
def java(*args, &block)
  options = Hash === args.last ? args.pop : {}
  options[:verbose] ||= trace?(:java)
  rake_check_options options, :classpath, :java_args, :properties, :name, :verbose, :dir

  name = options[:name]
  if name.nil?
    name = "java #{args.first}"
  end

  cmd_args = []
  if options[:dir]
    pwd = options[:dir]
    if Buildr::Util.win_os?
      # Ruby uses forward slashes regardless of platform,
      # unfortunately cd c:/some/path fails on Windows
      cmd_args << "cd /d \"#{pwd.gsub(%r{/}, '\\')}\" && "
    else
      cmd_args << "cd '#{pwd}' && "
    end
  end
  cmd_args << path_to_bin('java')
  cp = classpath_from(options)
  paths = cp.map do |c|
    path = File.directory?(c) && !c.end_with?('/') ? "#{c}/" : c.to_s
    Buildr::Util.win_os? ? "/#{path}" : path
  end
  manifest = Buildr::Packaging::Java::Manifest.new([{'Class-Path' => paths.join(" ")}])

  tjar = Tempfile.new(['javacmd', '.jar'])
  Zip::ZipOutputStream.open(tjar.path) do |zos|
    zos.put_next_entry('META-INF/MANIFEST.MF')
    zos.write manifest.to_s
    zos.write "\n"
  end
  tjar.close

  cmd_args << '-classpath' << tjar.path
  options[:properties].each { |k, v| cmd_args << "-D#{k}=#{v}" } if options[:properties]
  cmd_args += (options[:java_args] || (ENV['JAVA_OPTS'] || ENV['JAVA_OPTIONS']).to_s.split).flatten
  cmd_args += args.flatten.compact

  tmp = nil
  begin
      # Windows can't handle cmd lines greater than 2048/8192 chars.
      # If our cmd line is longer, we create a batch file and execute it instead.
    if Util.win_os? &&  cmd_args.map(&:inspect).join(' ').size > 2048
      # remove '-classpath' and the classpath itself from the cmd line.
      cp_i = cmd_args.index{|x| x.to_s =~ /^-classpath/}
      2.times do
        cmd_args.delete_at cp_i unless cp_i.nil?
      end
      # create tmp batch file.
      tmp = Tempfile.new(['starter', '.bat'])
      tmp.write "@echo off\n"
      tmp.write "SET CLASSPATH=#{cp.join(File::PATH_SEPARATOR).gsub(%r{/}, '\\')}\n"
      tmp.write cmd_args.map(&:inspect).join(' ')
      tmp.close
      # set new cmd line.
      cmd_args = [tmp.path]
    end
    
    unless Buildr.application.options.dryrun
      info "Running #{name}" if name && options[:verbose]
      block = lambda { |ok, res| fail "Failed to execute #{name}, see errors above" unless ok } unless block
      cmd_args = cmd_args.map(&:inspect).join(' ') if Util.win_os?
      sh(*cmd_args) do |ok, ps|
        block.call ok, ps
      end
    end
  ensure
    unless tmp.nil?
      tmp.close 
      tmp.unlink
    end
  end
end

# File lib/buildr/java/commands.rb, line 170
def javac(*args)
  options = Hash === args.last ? args.pop : {}
  rake_check_options options, :classpath, :sourcepath, :output, :javac_args, :name

  files = args.flatten.each { |f| f.invoke if f.respond_to?(:invoke) }.map(&:to_s).
    collect { |arg| File.directory?(arg) ? FileList["#{File.expand_path(arg)}/**/*.java"] : File.expand_path(arg) }.flatten
  name = options[:name] || Dir.pwd

  cmd_args = []
  cp = classpath_from(options)
  cmd_args << '-classpath' << cp.join(File::PATH_SEPARATOR) unless cp.empty?
  cmd_args << '-sourcepath' << [options[:sourcepath]].flatten.join(File::PATH_SEPARATOR) if options[:sourcepath]
  cmd_args << '-d' << File.expand_path(options[:output].to_s) if options[:output]
  cmd_args += options[:javac_args].flatten if options[:javac_args]
  cmd_args += files
  unless Buildr.application.options.dryrun
    mkdir_p options[:output] if options[:output]
    info "Compiling #{files.size} source files in #{name}"
    trace (['javac'] + cmd_args).join(' ')
    Java.load
    ::Java::com.sun.tools.javac.Main.compile(cmd_args.to_java(::Java::java.lang.String)) == 0 or
      fail 'Failed to compile, see errors above'
  end
end

# File lib/buildr/java/commands.rb, line 211
def javadoc(*args)
  options = Hash === args.last ? args.pop : {}
  fail "No output defined for javadoc" if options[:output].nil?
  options[:output] = File.expand_path(options[:output].to_s)
  cmd_args = [ '-d', options[:output], trace?(:javadoc) ? '-verbose' : '-quiet' ]
  options.reject { |key, value| [:output, :name, :sourcepath, :classpath].include?(key) }.
    each { |key, value| value.invoke if value.respond_to?(:invoke) }.
    each do |key, value|
      case value
      when true, nil
        cmd_args << "-#{key}"
      when false
        cmd_args << "-no#{key}"
      when Hash
        value.each { |k,v| cmd_args << "-#{key}" << k.to_s << v.to_s }
      else
        cmd_args += Array(value).map { |item| ["-#{key}", item.to_s] }.flatten
      end
    end
  [:sourcepath, :classpath].each do |option|
    options[option].to_a.flatten.tap do |paths|
      cmd_args << "-#{option}" << paths.flatten.map(&:to_s).join(File::PATH_SEPARATOR) unless paths.empty?
    end
  end
  files = args.each {|arg| arg.invoke if arg.respond_to?(:invoke)}.collect {|arg| arg.is_a?(Project) ? arg.compile.sources.collect{|dir| Dir["#{File.expand_path(dir.to_s)}/**/*.java"]} : File.expand_path(arg.to_s) }
  cmd_args += files.flatten.uniq.map(&:to_s)
  name = options[:name] || Dir.pwd
  unless Buildr.application.options.dryrun
    info "Generating Javadoc for #{name}"
    trace (['javadoc'] + cmd_args).join(' ')
    Java.load
    ::Java::com.sun.tools.javadoc.Main.execute(cmd_args.to_java(::Java::java.lang.String)) == 0 or
      fail 'Failed to generate Javadocs, see errors above'
  end
end

# File lib/buildr/java/commands.rb, line 265
def classpath_from(options)
  Buildr.artifacts(options[:classpath] || []).map(&:to_s).
    map { |t| task(t).invoke; File.expand_path(t) }
end

# File lib/buildr/java/commands.rb, line 253
def path_to_bin(name = nil)
  home = ENV['JAVA_HOME'] or fail 'Are we forgetting something? JAVA_HOME not set.'
  bin = Util.normalize_path(File.join(home, 'bin'))
  fail 'JAVA_HOME environment variable does not point to a valid JRE/JDK installation.' unless File.exist? bin
  Util.normalize_path(File.join(bin, name.to_s))
end