require 'fileutils'
require 'tempfile'
require 'rscm/revision'
require 'rscm/base'
require 'yaml'

class Time
  def to_rfc2822
    utc.strftime("%a, %d %b %Y %H:%M:%S +0000")
  end
end

module RSCM
  # The RSCM StarTeam class requires that the following software be installed:
  #
  # * Java Runtime (1.4.2)
  # * StarTeam SDK
  # * Apache Ant (http://ant.apache.org/)
  #
  class StarTeam < Base
    #register self

    ann :description => "User name"
    attr_accessor :user_name

    ann :description => "Password"
    attr_accessor :password

    ann :description => "Server name"
    attr_accessor :server_name

    ann :description => "Server port"
    attr_accessor :server_port

    ann :description => "Project name"
    attr_accessor :project_name

    ann :description => "View name"
    attr_accessor :view_name

    ann :description => "Folder name"
    attr_accessor :folder_name

    def initialize(user_name="", password="", server_name="", server_port="", project_name="", view_name="", folder_name="")
      @user_name, @password, @server_name, @server_port, @project_name, @view_name, @folder_name = user_name, password, server_name, server_port, project_name, view_name, folder_name
    end

    def name
      "StarTeam"
    end
    
    def revisions(checkout_dir, from_identifier=Time.epoch, to_identifier=Time.infinity, &proc)
      # just assuming it is a Time for now, may support labels later.
      # the java class really wants rfc822 and not rfc2822, but this works ok anyway.
      from = from_identifier.to_rfc2822
      to = to_identifier.to_rfc2822      

      revisions = java("getRevisions(\"#{from}\";\"#{to}\")", &proc)
      raise "revisions must be of type #{Revisions.name} - was #{revisions.class.name}" unless revisions.is_a?(::RSCM::Revisions)

      # Just a little sanity check
      if(revisions.latest)
        latetime = revisions.latest.time
        if(latetime < from_identifier || to_identifier < latetime)
          raise "Latest time (#{latetime}) is not within #{from_identifier}-#{to_identifier}"
        end
      end
      revisions
    end

    def checkout(checkout_dir, to_identifier, &proc)
      # TODO: Take the to_identifier arg into consideration
      files = java("checkout(\"#{checkout_dir}\")", &proc)
      files
    end
    
  private
  
    def cmd
      rscm_jar = File.expand_path(File.dirname(__FILE__) + "../../../../ext/rscm.jar")
      starteam_jars = Dir["#{ENV['RSCM_STARTEAM']}/Lib/*jar"].join(File::PATH_SEPARATOR)
      ant_jars = Dir["#{ENV['ANT_HOME']}/lib/*jar"].join(File::PATH_SEPARATOR)
      classpath = "#{rscm_jar}#{File::PATH_SEPARATOR}#{ant_jars}#{File::PATH_SEPARATOR}#{starteam_jars}"

      "java -Djava.library.path=\"#{ENV['RSCM_STARTEAM']}#{File::SEPARATOR}Lib\" -classpath \"#{classpath}\" org.rubyforge.rscm.Main"
    end

    def java(m, &proc)
      raise "The RSCM_STARTEAM environment variable must be defined and point to the StarTeam SDK directory" unless ENV['RSCM_STARTEAM']
      raise "The ANT_HOME environment variable must be defined and point to the Ant installation directory" unless ENV['ANT_HOME']

      clazz = "org.rubyforge.rscm.starteam.StarTeam"
      ctor_args = "#{@user_name};#{@password};#{@server_name};#{@server_port};#{@project_name};#{@view_name};#{@folder_name}"

#     Uncomment if you're not Aslak - to run against a bogus java class.
#      clazz = "org.rubyforge.rscm.TestScm"
#      ctor_args = "hubba;bubba"

      command = "new #{clazz}(#{ctor_args}).#{m}"
      tf = Tempfile.new("rscm_starteam")
      tf.puts(command)
      tf.close 
      cmdline = "#{cmd} #{tf.path}"
      IO.popen(cmdline) do |io|
        yaml_source = io
        if(block_given?)
          yaml_source = ""
          io.each_line do |line|
            yield line
            yaml_source << line << "\n"
          end
        else
        end
        YAML::load(yaml_source)
      end
    end

  end
end