#!/usr/bin/env ruby

require File.expand_path(File.dirname(__FILE__) + "/../lib/stasis")
gem "slop", "3.3.2"
require 'slop'

slop = Slop.parse :help => true, :optional_arguments => true do
  on :d, :development, "Development mode", :as => Integer
  on :o, :only, "Only generate specific files (comma-separated)", :as => Array
  on :p, :public, "Public directory path"
  on :m, :mime_types, "Additional Mime Types", :as => Array
end

exit if slop.help?
options = slop.to_hash
options[:mime_types] = Hash[*options[:mime_types]]

# Autoload Gemfile if present in pwd
begin
  require "bundler/setup"
  Bundler.require
rescue; end

if slop.development?
  require 'stasis/dev_mode'
  Stasis::DevMode.new(Dir.pwd, options)
elsif slop.only? && slop.public?
  Stasis.new(Dir.pwd, slop[:public], options).render(*slop[:only])
elsif slop.only?
  Stasis.new(Dir.pwd, options).render(*slop[:only])
elsif slop.public?
  Stasis.new(Dir.pwd, slop[:public], options).render(*(slop[:only].to_a))
else
  Stasis.new(Dir.pwd, options).render
end