Rakefile

# encoding: utf-8
require 'rake/clean'
require 'rdoc'
require 'rdoc/encoding'
require 'rdoc/markup/to_html'
require 'rdoc/options'
require 'uri'
require 'nokogiri'
require 'kramdown'


def cleanup(html, fragment)
  if fragment
    html_doc = Nokogiri::HTML.fragment(html)
    html_doc.xpath("h1").first.remove # Removes Heading
    html_doc.xpath("p").first.remove # Removes badges
    html_doc.to_html
  else
    html_doc = Nokogiri::HTML(html)
    html_doc.xpath("//h1").first.remove # Removes Sinatra Heading
    html_doc.xpath("//ul").first.remove # Removes the ToC in Markdown
    toc_header = html_doc.xpath("//h2").first
    toc_header.remove if toc_header.inner_html == "Table of Contents"
    html_doc.to_html
  end
end


def readme(pattern = "%s", &block)
  return readme(pattern).each(&block) if block_given?

  %w[
   README
  ].map do |extension|
    pattern % extension
  end
end

def contrib(pattern = "%s", &block)
  return contrib(pattern).each(&block) if block_given?

  %w[
    sinatra-config-file
    sinatra-multi-route
    sinatra-content-for
    sinatra-namespace
    sinatra-cookies
    sinatra-reloader
    sinatra-respond-with
    sinatra-extension
    sinatra-streaming
    sinatra-json
    sinatra-link-header
    sinatra-webdav
    sinatra-runner
    sinatra-test-helpers
    sinatra-required-params
    sinatra-custom-logger
    sinatra-capture
    sinatra-engine-tracking
  ].map do |extension|
    pattern % extension
  end
end

def protection(pattern = "%s", &block)
  return protection(pattern).each(&block) if block_given?

  %w[
    rack-protection-escaped-params
    rack-protection-frame-options
    rack-protection-strict-transport
    rack-protection-http-origin
    rack-protection-json-csrf
    rack-protection-remote-token
    rack-protection-authenticity-token
    rack-protection-path-traversal
    rack-protection-cookie-tossing
    rack-protection-content-security-policy
    rack-protection-xss-header
    rack-protection-remote-referrer
    rack-protection-form-token
    rack-protection-ip-spoofing
    rack-protection-session-hijacking
  ].map do |extension|
    pattern % extension
  end
end

# generates Table of Contents
def with_toc(src)
  toc = "<div class='toc'>\n"
  last_level = 1
  src = src.gsub(/<h(\d)>(.*)<\/h\d>/) do |line|
    level, heading = $1.to_i, $2.strip
    aname = URI.escape(heading)
    toc << "\t"*last_level << "<ol class='level-#{level-1}'>\n" if level > last_level
    toc << "\t"*level << "</ol>"*(last_level - level) << "\n" if level < last_level
    toc << "\t"*level << "<li><a href='##{aname}'>#{heading}</a></li>\n"
    last_level = level
    "<a name='#{aname}'></a>\n" << line
  end
  toc << "\t" << "</ol>"*(last_level-1) << "\n</div>\n"
  toc + src
end

task :default => ['_sinatra', :build]

desc "Build outdated static files and API docs"
task :build => [:pull, 'build:static']

desc "Build outdated static files"
task 'build:static' =>
  readme("_includes/%s.html") +
  contrib("_includes/%s.html") +
  protection("_includes/%s.html") +
  ["_includes/sinatra-contrib-readme.html"] +
  ["_includes/rack-protection-readme.html"]

desc "Build anything that's outdated and stage changes for next commit"
task :regen => [:build] do
  sh 'git add api _includes'
  puts "\nPrebuilt files regenerated and staged in your index. Commit with:"
  puts "  git commit -m 'Regen prebuilt files'"
end

desc 'Pull in the latest from the sinatra repo'
task :pull => ['pull:sinatra']

directory "_sinatra" do
  puts 'Cloning sinatra repo'
  sh "git clone git@github.com:sinatra/sinatra.git _sinatra"
end

desc 'Pull in the latest from the sinatra repo'
task 'pull:sinatra' => "_sinatra" do
    puts 'Pulling sinatra.git'
    sh "cd _sinatra && git pull &>/dev/null || true"
end

readme("_sinatra/%s.md") { |fn| file fn => '_sinatra' }
file 'AUTHORS' => '_sinatra'

readme do |fn|
  file "_includes/#{fn}.html" => ["_sinatra/#{fn}.md", "Rakefile"] do |f|
    readme_to_html(f.name, "_sinatra/#{fn}.md")
  end

  file "_includes/sinatra-contrib-readme.html" => ["_sinatra/sinatra-contrib/README.md"] do |f|
    readme_to_html(f.name, "_sinatra/sinatra-contrib/README.md", false, true)
  end

  file "_includes/rack-protection-readme.html" => ["_sinatra/rack-protection/README.md"] do |f|
    readme_to_html(f.name, "_sinatra/rack-protection/README.md", false, true)
  end
end

def readme_to_html(fname, path, with_toc = false, fragment = false)
  markdown_string = File.read(path).
    encode('UTF-16le', :invalid => :replace, :replace => "").
    encode("UTF-8")
  markdown_string.gsub!(/```(\s?(\w+\n))?/) do |match|
    match =~ /```\s?\n/ ? "~~~~~\n" : match.sub(/```\s?/, "~~~~")
  end
  markdown = Kramdown::Document.new(markdown_string,
    :fenced_code_blocks => true,
    :syntax_highlighter => :rouge,
    :auto_ids => true)

  html = cleanup(markdown.to_html, fragment)
  File.open(fname, 'w') do |io|
    if with_toc
      io.write with_toc(html)
    else
      io.write html
    end
  end
end

desc 'Build protection docs'
task 'build:protection_docs' do
  puts 'Building rack-protection docs'
  sh "cd _sinatra/rack-protection && rake doc"
end


protection("_sinatra/rack-protection/doc/%s.rdoc") { |fn| file fn => '_sinatra/rack-protection' }

protection do |fn|
  file "_includes/#{fn}.html" => ["build:protection_docs", "_sinatra/rack-protection/doc/#{fn}.rdoc", "Rakefile"] do |f|
    path_to_file = "_sinatra/rack-protection/doc/#{fn}"

    begin
      file_to_convert = File.read(path_to_file + ".rdoc")
    rescue
      file_to_convert = File.read(path_to_file + ".md")
    end

    html = RDoc::Markup::ToHtml.new(RDoc::Options.new).convert(file_to_convert)
    File.open(f.name, 'wb') { |io| io.write html }
  end
end

desc 'Build contrib docs'
task 'build:contrib_docs' do
  puts 'Building sinatra-contrib docs'
  sh "cd _sinatra/sinatra-contrib && rake doc"
end

contrib("_sinatra/sinatra-contrib/doc/%s.rdoc") { |fn| file fn => '_sinatra/sinatra-contrib' }

contrib do |fn|
  file "_includes/#{fn}.html" => ["build:contrib_docs", "_sinatra/sinatra-contrib/doc/#{fn}.rdoc", "Rakefile"] do |f|
    path_to_file = "_sinatra/sinatra-contrib/doc/#{fn}"

    begin
      file_to_convert = File.read(path_to_file + ".rdoc")
    rescue
      file_to_convert = File.read(path_to_file + ".md")
    end

    html = RDoc::Markup::ToHtml.new(RDoc::Options.new).convert(file_to_convert)
    File.open(f.name, 'wb') { |io| io.write html }
  end
end

desc 'Rebuild site under _site with Jekyll'
task :jekyll do
  rm_rf '_site'
  sh 'jekyll build'
end

desc 'Start the Jekyll server on http://localhost:4000/'
task :server do
  rm_rf '_site'
  puts 'jekyll serve --watch'
  exec 'jekyll serve --watch'
end

CLEAN.include('_site', "_includes/*.html").exclude('_includes/navbar.html', '_includes/head.html')
CLOBBER.include "_sinatra/sinatra-contrib", "_sinatra/rack-protection", "_sinatra"