app.rb

require 'sinatra'
require 'haml'
require 'authy'
require 'json'
require 'dotenv'

require_relative 'models/user'
require_relative 'helpers'
require_relative 'routes/signup'
require_relative 'routes/sessions'
require_relative 'routes/confirmation'

# Load environment configuration
Dotenv.load

# Set the environment after dotenv loads
# Default to production
environment = (ENV['APP_ENV'] || ENV['RACK_ENV'] || :production).to_sym
set :environment, environment

require 'bundler'
Bundler.require :default, environment

database_url = "sqlite3://#{Dir.pwd}/development.sqlite3"
DataMapper.setup(:default, database_url)
DataMapper.finalize
User.auto_upgrade!

module TwoFactorAuth
  class App < Sinatra::Base
    enable :sessions

    set :root, File.dirname(__FILE__)

    helpers Helpers

    register Routes::Signup
    register Routes::Sessions
    register Routes::Confirmation

    get '/' do
      haml :home
    end

    get '/protected' do
      authenticate!
      
      @user = User.first(id: current_user)
      haml :protected
    end
  end
end