root = Aura.method(:gem_root)
approot = Aura.method(:root)

$:.unshift *Dir[root["/vendor/*/lib"]]
require "sinatra/base"
require "sinatra/support"
require "rtopia"
require "sequel"
require "sinatra/content_for"
require "sinatra/sequel"
require "pistol"
require "json"
require "yaml"
require "shield"
require "ostruct"
require "fileutils"

# Helpers
require root['lib/core/hasharray']
require root['lib/core/object_ext']

# Aura proper
require root['lib/aura']
require root['lib/terra']
require root['lib/terra/ext']

if defined?(::Encoding)
  Encoding.default_external = 'utf-8'
end

# Class: Main
# The main Sinatra application.
#
# ## Description
#    This is the main application. It is also accessible as `Aura::App`.
#
class Main < Sinatra::Base
  set      :root, lambda { |*args| File.join(ENV['AURA_ROOT'], *args) }
  set      :root_path, lambda { |*args| root *args }
  set      :approot, lambda { |*args| File.join(ENV['APP_ROOT'] || root, *args) }

  set      :haml, :escape_html => true
  set      :haml, :escape_html => true, :ugly => true  if production?

  enable   :raise_errors

  use      Rack::Session::Cookie
  use      Rack::Deflater  if production?

  register Aura::Public
  register Aura::Rendering
  register Aura::Seeder
  register Sinatra::SequelExtension

  helpers  Rtopia
  helpers  Sinatra::ContentFor
  helpers  Sinatra::UserAgentHelpers
  helpers  Shield::Helpers

  # Load all, but load defaults first
  ( Dir[root('config/*.rb')].sort +
    Dir[approot('config/*.rb')].sort
  ).uniq.each { |f| load f }

  register Sinatra::CompassSupport

  set :login_success_message, nil

  set :multi_views,     [root('app/views')]
  set :app_files,       [ENV['APP_FILE']] + Dir[root('**/*.rb')] + Dir[approot('**/*.rb')]
  set :extensions_path, [root('extensions'), approot('extensions')]

  # Heroku - override the DB config with this env var.
  set :database_url, ENV['DATABASE_URL']  unless ENV['DATABASE_URL'].nil?

  add_public approot('public')

  def self.restart!
    require 'fileutils'
    FileUtils.touch approot('tmp', 'restart.txt')
  end

  # Connect to db
  database
end

# Initializers
require root['app/init/pistol.rb']
require root['app/init/admin.rb']
require root['app/init/sequel.rb']

# Bootstrap Aura
Dir[root['app/{models,helpers,routes}/**/*.rb']].each { |f| require f }

# Extensions
require root['app/init/extensions.rb']

# Load app/
if File.directory?(Main.approot('app'))
  Aura::Extension.new(Main.approot('app')).load!
end

Aura::App = Main

Main.set :port, ENV['PORT'].to_i  unless ENV['PORT'].nil?
Main.run!  if ENV['APP_FILE'] == $0