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
 
# Github pages deploy config
# For github user pages, use "master"
# For github project pages use "gh-pages"
# If you're not using this, you can remove it
# Read http://pages.github.com for guidance
 
github_pages_branch = "gh-pages"
 
def ok_failed(condition)
  if (condition)
    puts "OK"
  else
    puts "FAILED"
  end
end
 
desc "generate website in output directory"
task :default => [:generate_site, :generate_style] do
  puts ">>> Site Generating Complete! <<<\n\n"
end
 
desc "list tasks"
task :list do
  puts "Tasks: #{(Rake::Task.tasks - [Rake::Task[:list]]).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}/test"].each { |f| rm_rf(f) }
end
 
desc "Generate styles only"
task :generate_style do
  puts ">>> Generating styles <<<"
  system "compass"
end
 
desc "Generate site files only"
task :generate_site => [:clean, :generate_style] do
  puts "\n\n>>> Generating site files <<<"
  system "jekyll --pygments"
  system "mv #{site}/atom.html #{site}/atom.xml"
end
 
def rebuild_site(relative)
  puts "\n"
  puts ">>> Change Detected to: #{relative} <<<"
  IO.popen('rake generate_site'){|io| print(io.readpartial(512)) until io.eof?}
  puts '>>> Update Complete <<<'
end
 
def rebuild_style(relative)
  puts "\n\n>>> Change Detected to: #{relative} <<<"
  IO.popen('rake generate_style'){|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 do
    path "#{File.dirname(__FILE__)}/#{source}" do
      update {|base, relative| rebuild_site(relative)}
      delete {|base, relative| rebuild_site(relative)}
      create {|base, relative| rebuild_site(relative)}
    end
    path "#{File.dirname(__FILE__)}/stylesheets" do
      glob '**/*.sass'
      update {|base, relative| rebuild_style(relative)}
      delete {|base, relative| rebuild_style(relative)}
      create {|base, relative| rebuild_style(relative)}
    end
  end
end
 
desc "generate and deploy website"
multitask :deploy_rsync => [:default, :clean_debug] do
  print ">>> Deploying website <<<"
  ok_failed system("rsync -avz --delete #{site}/ #{ssh_user}:#{document_root}")
end
 
desc "generate and deploy website to github user pages"
multitask :github_user_deploy => [:default, :clean_debug] do