main.rb

require 'sinatra'
require_relative 'routes/init'
require_relative 'models/init'
require_relative 'enterprise/main' if File.exist?("enterprise/main.rb")

class ASYD < Sinatra::Application
  configure do
    set :public_folder, Proc.new { File.join(root, "static/lib") }
    set :environment, :production unless $DBG == 1
    enable :sessions
  end

  helpers do
    def user
      if session[:username]
        User.first(:username => session[:username])
      else
        nil
      end
    end
    def timezone
      session[:timezone] ? session[:timezone] : "UTC"
    end
    def t(*args)
      I18n.t(*args)
    end
  end

  before do
    loc = request.env["HTTP_ACCEPT_LANGUAGE"] ? request.env["HTTP_ACCEPT_LANGUAGE"][0,2] : "en"
    I18n.locale = I18n.available_locales.map(&:to_s).include?(loc) ? loc : "en"
  end

  # Check if ASYD was installed or user is logged in before doing anything
  # Now also checks if ASYD must update
  before /^(?!\/(setup))(?!\/(login))(?!\/(logout))(?!\/(update))(?!\/(confirm_update))(^(?!\/(password\/request)))(^(?!\/(password\/reset)))(^(?!(\/private)?\/api\/))/ do
    if !File.directory? 'data'
      redirect '/setup'
    else
      auth = Rack::Auth::Basic::Request.new(env)
      if auth.provided? and auth.basic? and auth.credentials
        u = User.auth(auth.credentials[0], auth.credentials[1])
        if u
          session[:username] = u.username
        end
      end
      if !session[:username] or User.first(:username => session[:username]).nil? then
        redirect '/login'
      else
        if File.directory? 'data' and File.exist? 'installer/updater.rb' then
          require_relative 'installer/updater'
          actions = Updater.update_actions
          if actions.length > 0
            redirect '/update'
          else
            Updater.remove_installer_dir
          end
        end
      end
    end
  end
end