module Rake

end

module Rake::StX

end

module Rake::Stx::SCM

  # Not quite sure why following code

  #

  #     include RakeFileUtils

  #

  # does not make extra methods (like `sh`) defined in `extensions.rb`

  # visible here. To workaround, define them here. Ugly...

  module_function

  def sh(cmd, cwd: Dir.pwd, &block)

    return RakeFileUtils::sh(cmd, cwd: cwd, &block)

  end

  class CheckoutException < Exception

  end # class CheckoutException

  def self._check_type(type)

    if (type != :cvs and type != :svn and type != :git and type != :hg)

      raise CheckoutException.new("Unknown version control system type (#{type})")

    end

  end

  def self.update(type, repository, directory, *params)

    self._check_type(type)

    if params.size() > 0

      p = params.last

    else

      p = {}

    end

    root = p[:root] || BUILD_DIR

    wc = root / directory

    if (! File.exist? wc)

      self.checkout(type, repository, directory, *params)

      return

    end

    case type

      when :svn    then _update_svn(wc)

      when :cvs    then _update_cvs(wc)

      when :git    then _update_git(wc, repository, directory, *params)

      when :hg     then _update_hg(wc, repository, directory, *params)

    end

  end

  def self._update_hg(wc, repository = nil, directory = nil, *params)

    url = ''

    if repository != nil and directory != nil then

      if params.size() > 0

        p = params.last

      else

        p = {}

      end

      separator = p[:separator] || '.'

      url = "#{repository}/#{directory.gsub('/', separator)}"

    end

    end

  end

  def self._update_svn(wc)    

    if not sh %W{svn --non-interactive --trust-server-cert update}, cwd: wc

      raise CheckoutException.new("SVN: Cannot update #{wc}")

    end

  end

  def self._update_cvs(wc)

    if File.directory? wc

      if not sh %W{cvs -z 9 update -A -d}, cwd: wc

        raise CheckoutException.new("CVS: Cannot update #{wc}")

      end

    else

      if not sh %W{cvs -z 9 update -A #{File.basename(wc)}}, cwd: File.dirname(wc)

        raise CheckoutException.new("CVS: Cannot update #{wc}")

      end

    end

  end

  def self.checkout(type, repository, directory, *params)    

    self._check_type(type)

    if params.size() > 0

      p = params.last

    else

      p = {}

    end

    root = p[:root] || BUILD_DIR

    branch = p[:branch]

    if branch == nil

      if type == :svn

        branch = 'trunk'

      elsif type == :hg

        branch = 'default'

      end

    end

    wc = root / directory

    if (File.exist? wc)

      self.update(type, repository, directory, *params)

      return

    end

    if (not File.exists? File.dirname(wc))

      begin

        FileUtils.mkdir_p(File.dirname(wc))

      rescue => ex

        raise CheckoutException.new("Cannot create directory for working copy (#{ex})")

      end

    end

    case type

      when :svn    then _checkout_svn(repository, directory, branch, root, *params)

      when :cvs    then _checkout_cvs(repository, directory, branch, root, *params)

      when :git    then _checkout_git(repository, directory, branch, root, *params)

      when :hg     then _checkout_hg(repository, directory, branch, root, *params)      

    end

  end

  def self._checkout_svn(repository, directory, branch, root, *params)

    url = "#{repository}/#{directory}/#{branch}"

    if not sh %W{svn --non-interactive --trust-server-cert co #{url} #{directory}}, cwd: root

      raise CheckoutException.new("SVN: Cannot checkout from #{url}")

    end

  end

  def self._checkout_hg(repository, directory, branch, root, *params)

    if params.size() > 0

      p = params.last

    else

      p = {}

    end

    separator = p[:separator] || '.'

  end

  def self._checkout_cvs(repository, directory, branch, root, *params)    

    if not sh %W{cvs -z 9 -d #{repository} co #{directory}}, cwd: root

      raise CheckoutException.new("CVS: Cannot checkout #{directory}from #{repository}")

    end

  end

end # module Rake::Stx::SCM

def checkout(repo_name, directory, *params)

  # repository should be symbolic name

  repo = Rake::Stx::Configuration::Repository::find(repo_name)

  if not repo then

      error("checkout(): No repository found (#{repo_name})")

  end  

  if params.last.is_a? Hash 

    params.last[:separator] = repo.separator

  else

    params << {:separator => repo.separator}

  end    

  Rake::Stx::SCM.checkout(repo.type, repo.url, directory, *params)  

end

def update(repo_name, directory, *params)

  # repository should be symbolic name

  repo = Rake::Stx::Configuration::Repository::find(repo_name)

  if not repo then  

    error("update(): No repository found (#{repo_name})")

  end  

  Rake::Stx::SCM.update(repo.type, repo.url, directory, *params)

end

def cvs(repository, directory, *params)

  Rake::Stx::SCM.checkout(:cvs, repository, directory, *params)

end

def svn(repository, directory, *params)

  Rake::Stx::SCM.checkout(:svn, repository, directory, *params)

end

def hg(repository, directory, *params)

  Rake::Stx::SCM.checkout(:hg, repository, directory, *params)

end

def git(repository, directory, *params)

  Rake::Stx::SCM.checkout(:git, repository, directory, *params)

end