require 'active_support'

site_url  = "http://yoursite.com"   # deployed site url for sitemap.xml generator

port      = "4000"     # preview project port eg. http://localhost:4000

site      = "site"     # compiled site directory

source    = "source" # source file directory

## -- Rsync Deploy config -- ##

ssh_user      = "user@host.com"    # for rsync deployment

document_root = "~/document_root/" # for rsync deployment

## ---- ##

## -- Github Pages deploy config -- ##

# Read http://pages.github.com for guidance

# If you're not using this, you can remove it

source_branch = "source" # this compiles to your deploy branch

deploy_branch = "gh-pages" # For user pages, use "master" for project pages use "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 via rsync"

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 :deploy_github => [:default, :clean_debug] do

  require 'git'