require 'active_support'
 
## -- CHANGE FOR YOUR PROJECT -- ##
site_url      = "http://yoursite.com"   # deployed site url
ssh_user      = "user@host.com"    # for rsync deployment
document_root = "~/document_root/" # for rsync deployment
## ---- ##
 
port = "4000"     # preview project port eg. http://localhost:4000
site = "site"     # compiled site directory
source = "source" # source file directory
 
def ok_failed(condition)
  if (condition)
    puts "OK"
  else
    puts "FAILED"
  end
end
 
desc "list tasks"
task :default do
  puts "Tasks: #{(Rake::Task.tasks - [Rake::Task[:default]]).to_sentence}"
  puts "(type rake -T for more detail)\n\n"
end
 
desc "remove files in output directory"
task :clean do
  puts "Removing output..."
  Dir["#{site}/*"].each { |f| rm_rf(f) }
end
 
task :clean_debug do
  puts "Removing debug pages..."
  Dir["#{site}/debug"].each { |f| rm_rf(f) }
end
 
desc "generate website in output directory"
task :generate => :clean do
  puts "Generating website..."
  system "jekyll"
  Dir["#{site}/stylesheets/*.sass"].each { |f| rm_rf(f) }
  system "compass"
  system "mv #{site}/atom.html #{site}/atom.xml"
end
 
def rebuild_site(relative)
  puts ">>> Change Detected to: #{relative} <<<"
  IO.popen('rake generate'){|io| print(io.readpartial(512)) until io.eof?}
  puts '>>> Update Complete <<<'
end
 
desc "Watch the site and regenerate when it changes"
task :watch do
  require 'fssm'
  puts ">>> Watching for Changes <<<"
  FSSM.monitor("#{File.dirname(__FILE__)}/#{source}", '**/*') do
    update {|base, relative| rebuild_site(relative)}
    delete {|base, relative| rebuild_site(relative)}
    create {|base, relative| rebuild_site(relative)}
  end
end
 
desc "generate and deploy website"
multitask :deploy => [:generate, :clean_debug] do
  print "Deploying website..."
  ok_failed system("rsync -avz --delete #{site}/ #{ssh_user}:#{document_root}")
end
 
desc "start up an instance of serve on the output files"
task :start_serve => :stop_serve do
  cd "#{site}" do
    print "Starting serve..."
    ok_failed system("serve #{port} > /dev/null 2>&1 &")
  end
end
 
desc "stop all instances of serve"
task :stop_serve do
  pid = `ps auxw | awk '/bin\\/serve\\ #{port}/ { print $2 }'`.strip
  if pid.empty?
    puts "Serve is not running"
  else
    print "Stoping serve..."
    ok_failed system("kill -9 #{pid}")
  end
end
 
desc "preview the site in a web browser"
multitask :preview => [:generate, :start_serve] do
  system "open http://localhost:#{port}"
end
 
 
desc "Build an XML sitemap of all html files."
task :sitemap => :generate do
  html_files = FileList.new("#{site}/**/*.html").map{|f| f[("#{site}".size)..-1]}.map do |f|
    if f.ends_with?("index.html")
      f[0..(-("index.html".size + 1))]
    else
      f
    end
  end.sort_by{|f| f.size}