require 'rubygems'
require 'rake'
require 'rdoc/task'
require 'date'

#############################################################################
#
# Helper functions
#
#############################################################################

def name
  @name ||= Dir['*.gemspec'].first.split('.').first
end

def version
  line = File.read("lib/#{name}.rb")[/^\s*VERSION\s*=\s*.*/]
  line.match(/.*VERSION\s*=\s*['"](.*)['"]/)[1]
end

def date
  Date.today.to_s
end

def rubyforge_project
  name
end

def gemspec_file
  "#{name}.gemspec"
end

def gem_file
  "#{name}-#{version}.gem"
end

def replace_header(head, header_name)
  head.sub!(/(\.#{header_name}\s*= ').*'/) { "#{$1}#{send(header_name)}'"}
end

#############################################################################
#
# Standard tasks
#
#############################################################################

task :default => :test

require 'rake/testtask'
Rake::TestTask.new(:test) do |test|
  file_list = FileList['test/**/test_*.rb']
  file_list = file_list.exclude("test/test_god_system.rb")
  test.test_files = file_list
  test.libs << 'lib' << 'test'
  test.verbose = true
end

require 'rake/testtask'
Rake::TestTask.new(:system_test) do |test|
  test.libs << 'lib' << 'test'
  test.pattern = 'test/**/test_god_system.rb'
  test.verbose = true
end

desc "Generate RCov test coverage and open in your browser"
task :coverage do
  require 'rcov'
  sh "rm -fr coverage"
  sh "rcov test/test_*.rb"
  sh "open coverage/index.html"
end

require 'rdoc/task'
Rake::RDocTask.new do |rdoc|
  rdoc.rdoc_dir = 'rdoc'
  rdoc.title = "#{name} #{version}"
  rdoc.rdoc_files.include('README*')
  rdoc.rdoc_files.include('lib/**/*.rb')
end

desc "Open an irb session preloaded with this library"
task :console do
  sh "irb -rubygems -r ./lib/#{name}.rb"
end

#############################################################################
#
# Custom tasks (add your own tasks here)
#
#############################################################################

desc "Generate and view the site locally"
task :site do
  # Generate the dynamic parts of the site.
  puts "Generating dynamic..."
  require 'gollum'
  wiki = Gollum::Wiki.new('.', :base_path => '/doc')
  html = wiki.page('god', 'HEAD').formatted_data.gsub("\342\200\231", "'")
  template = File.read('./site/index.template.html')
  index = template.sub("{{ content }}", html)
  File.open('./site/index.html', 'w') do |f|
    f.write(index)
  end

  puts "Done. Opening in browser..."
  sh "open site/index.html"
end

desc "Commit the local site to the gh-pages branch and deploy"
task :site_release do
  # Ensure the gh-pages dir exists so we can generate into it.
  puts "Checking for gh-pages dir..."
  unless File.exist?("./gh-pages")
    puts "No gh-pages directory found. Run the following commands first:"
    puts "  `git clone git@github.com:mojombo/god gh-pages"
    puts "  `cd gh-pages"
    puts "  `git checkout gh-pages`"
    exit(1)
  end

  # Copy the rest of the site over.
  puts "Copying static..."
  sh "cp -R site/* gh-pages/"

  # Commit the changes
  sha = `git log`.match(/[a-z0-9]{40}/)[0]
  sh "cd gh-pages && git add . && git commit -m 'Updating to #{sha}.' && git push"
  puts 'Done.'
end

#############################################################################
#
# Packaging tasks
#
#############################################################################

desc "Create tag v#{version} and build and push #{gem_file} to Rubygems"
task :release => :build do
  unless `git branch` =~ /^\* master$/
    puts "You must be on the master branch to release!"
    exit!
  end
  sh "git commit --allow-empty -a -m 'Release #{version}'"
  sh "git tag v#{version}"
  sh "git push origin master"
  sh "git push origin v#{version}"
  sh "gem push pkg/#{name}-#{version}.gem"
end

desc "Build #{gem_file} into the pkg directory"
task :build => :gemspec do
  sh "mkdir -p pkg"
  sh "gem build #{gemspec_file}"
  sh "mv #{gem_file} pkg"
end

desc "Generate #{gemspec_file}"
task :gemspec do
  # read spec file and split out manifest section
  spec = File.read(gemspec_file)
  head, manifest, tail = spec.split("  # = MANIFEST =\n")

  # replace name version and date
  replace_header(head, :name)
  replace_header(head, :version)
  replace_header(head, :date)
  #comment this out if your rubyforge_project has a different name
  replace_header(head, :rubyforge_project)

  # determine file list from git ls-files
  files = `git ls-files`.
    split("\n").
    sort.
    reject { |file| file =~ /^\./ }.
    reject { |file| file =~ /^(rdoc|pkg|examples|ideas|init|site)/ }.
    map { |file| "    #{file}" }.
    join("\n")

  # piece file back together and write
  manifest = "  s.files = %w[\n#{files}\n  ]\n"
  spec = [head, manifest, tail].join("  # = MANIFEST =\n")
  File.open(gemspec_file, 'w') { |io| io.write(spec) }
  puts "Updated #{gemspec_file}"
end