exec_awestruct.rb

require 'awestruct/version'
require 'awestruct/cli/generate'
require 'logger'
require 'json'
require 'optparse'
require 'shellwords'
require 'tilt'

options = {}

begin
OptionParser.new do |opts|
  opts.banner = "Usage: exec_awestruct.rb [options]"

  opts.on('--repo REPOSITORY', 'Repository to use') do |repo|
    options[:repo] = repo
  end

  opts.on('--profile PROFILE', 'Profile to build') do |profile|
    options[:profile] = profile
  end

  opts.on('--url BASE_URL', 'Base URL to use during generation') do |url|
    options[:url] = url
  end

  opts.on('--username USERNAME', 'username containing the respository') do |username|
    options[:username] = username
  end

  opts.on_tail('-h', '--help', 'show this message') do
    $stdout.puts opts
    exit(0)
  end
end.parse!
rescue OptionParser::MissingArgument => e
  puts e.to_str.capitalize.gsub('argument:', 'value for:')
  puts 'That option requires a value. Please try again.'
  exit
end

required_opts = [:repo, :profile, :url, :username]
required_missing = false

required_opts.each do |opt|
  if options.has_key?(opt)
    next
  else
    puts "Required option --#{opt} missing."
    required_missing = true
  end
end

if required_missing
  exit
end

base_repo_dir = (ENV['OPENSHIFT_DATA_DIR']) ?  File.join(ENV['OPENSHIFT_DATA_DIR'], 'repos', options[:username]) : "repos/#{options[:username]}"

Dir.chdir File.absolute_path(File.join(base_repo_dir, "#{options[:repo]}")) do
  error_log = StringIO.new ''
  $LOG = Logger.new(error_log)
  $LOG.level = Logger::ERROR

  engine = Awestruct::Engine.new(Awestruct::Config.new)
  engine.adjust_load_path
  engine.load_default_site_yaml
  engine.load_site_yaml options[:profile]
  engine.set_base_url "#{options[:url]}/preview/#{options[:repo]}", "#{options[:url]}/preview/#{options[:repo]}"
  engine.load_yamls
  engine.site.context_url = '_site' # try to hack this so we get the right URLs
  engine.load_pipeline
  engine.load_pages
  engine.execute_pipeline
  engine.configure_compass
  engine.set_urls engine.site.pages
  engine.build_page_index
  engine.generate_output

  source_to_output = {}
  engine.site.pages.each do |p|
    output_path = p.output_path unless source_to_output.include? p.relative_source_path
    output_mtime = File.mtime(File.join('_site', Shellwords.escape(p.output_path))).to_i
    source_mtime = File.mtime(File.join(Shellwords.escape(p.relative_source_path[1..-1]))).to_i
    content_type = if Awestruct::Handlers::TiltMatcher.new().match(p.relative_source_path)
                     Tilt[p.relative_source_path].default_mime_type
                   elsif p.output_extension =~ /js/
                     'application/javascript'
                   elsif p.output_extension =~ /css/
                     'text/css'
                   else
                     `file --mime-type -b #{'_site' + Shellwords.escape(p.output_path)}`.strip
                   end

    source_to_output[p.relative_source_path] = { :output_path => output_path, :output_mtime => output_mtime, :'content-type' => content_type }
    source_to_output['/_site' + p.output_path] = { :relative_source_path => p.relative_source_path, :source_mtime => source_mtime, :'content-type' => content_type }
  end

  error_log.rewind
  $stderr.puts error_log.readlines.join unless error_log.readlines.empty?
  File.open(File.join('_tmp', 'mapping.json'), 'w') do |f|
    f.puts JSON.dump source_to_output
  end
end