require "rubygems"

require "bundler/setup"

require "stringex"

## -- Rsync Deploy config -- ##

# Be sure your public key is listed in your server's ~/.ssh/authorized_keys file

ssh_user       = "user@domain.com"

ssh_port       = "22"

document_root  = "~/website.com/"

rsync_delete   = true

deploy_default = "rsync"

# This will be configured for you when you run config_deploy

deploy_branch  = "gh-pages"

## -- Misc Configs -- ##

public_dir      = "public"    # compiled site directory

source_dir      = "source"    # source file directory

blog_index_dir  = 'source'    # directory for your blog's index page (if you put your index in source/blog/index.html, set this to 'source/blog')

deploy_dir      = "_deploy"   # deploy directory (for Github pages deployment)

stash_dir       = "_stash"    # directory to stash posts for speedy generation

posts_dir       = "_posts"    # directory for blog files

themes_dir      = ".themes"   # directory for blog files

new_post_ext    = "markdown"  # default new post file extension when using the new_post task

new_page_ext    = "markdown"  # default new page file extension when using the new_page task

server_port     = "4000"      # port for preview server eg. localhost:4000

desc "Initial setup for Octopress: copies the default theme into the path of Jekyll's generator. Rake install defaults to rake install[classic] to install a different theme run rake install[some_theme_name]"

task :install, :theme do |t, args|

  if File.directory?(source_dir) || File.directory?("sass")

    abort("rake aborted!") if ask("A theme is already installed, proceeding will overwrite existing files. Are you sure?", ['y', 'n']) == 'n'

  end

  # copy theme into working Jekyll directories

  theme = args.theme || 'classic'

  puts "## Copying "+theme+" theme into ./#{source_dir} and ./sass"

  mkdir_p source_dir

  cp_r "#{themes_dir}/#{theme}/source/.", source_dir

  mkdir_p "sass"

  cp_r "#{themes_dir}/#{theme}/sass/.", "sass"

  mkdir_p "#{source_dir}/#{posts_dir}"

  mkdir_p public_dir

end

#######################

# Working with Jekyll #

#######################

desc "Generate jekyll site"

task :generate do

  raise "### You haven't set anything up yet. First run `rake install` to set up an Octopress theme." unless File.directory?(source_dir)

  puts "## Generating Site with Jekyll"

  system "compass compile --css-dir #{source_dir}/stylesheets"

  system "jekyll"

end

desc "Watch the site and regenerate when it changes"

task :watch do

  raise "### You haven't set anything up yet. First run `rake install` to set up an Octopress theme." unless File.directory?(source_dir)

  puts "Starting to watch source with Jekyll and Compass."

  system "compass compile --css-dir #{source_dir}/stylesheets" unless File.exist?("#{source_dir}/stylesheets/screen.css")

  jekyllPid = Process.spawn({"OCTOPRESS_ENV"=>"preview"}, "jekyll --auto")

  compassPid = Process.spawn("compass watch")

  trap("INT") {

    [jekyllPid, compassPid].each { |pid| Process.kill(9, pid) rescue Errno::ESRCH }

    exit 0

  }

  [jekyllPid, compassPid].each { |pid| Process.wait(pid) }

end

desc "preview the site in a web browser"

task :preview do

  raise "### You haven't set anything up yet. First run `rake install` to set up an Octopress theme." unless File.directory?(source_dir)

  puts "Starting to watch source with Jekyll and Compass. Starting Rack on port #{server_port}"

  system "compass compile --css-dir #{source_dir}/stylesheets" unless File.exist?("#{source_dir}/stylesheets/screen.css")

  jekyllPid = Process.spawn({"OCTOPRESS_ENV"=>"preview"}, "jekyll --auto")

  compassPid = Process.spawn("compass watch")

  rackupPid = Process.spawn("rackup --port #{server_port}")

  trap("INT") {

    [jekyllPid, compassPid, rackupPid].each { |pid| Process.kill(9, pid) rescue Errno::ESRCH }

    exit 0

  }

  [jekyllPid, compassPid, rackupPid].each { |pid| Process.wait(pid) }

end

# usage rake new_post[my-new-post] or rake new_post['my new post'] or rake new_post (defaults to "new-post")

desc "Begin a new post in #{source_dir}/#{posts_dir}"

task :new_post, :title do |t, args|

  raise "### You haven't set anything up yet. First run `rake install` to set up an Octopress theme." unless File.directory?(source_dir)

  mkdir_p "#{source_dir}/#{posts_dir}"

  args.with_defaults(:title => 'new-post')

  title = args.title

  filename = "#{source_dir}/#{posts_dir}/#{Time.now.strftime('%Y-%m-%d')}-#{title.to_url}.#{new_post_ext}"

  if File.exist?(filename)

    abort("rake aborted!") if ask("#{filename} already exists. Do you want to overwrite?", ['y', 'n']) == 'n'

  end

  puts "Creating new post: #{filename}"

  open(filename, 'w') do |post|