app.rb

require 'sinatra/base'
require 'sinatra/reloader'
require 'sinatra/activerecord'
require 'bcrypt'
require_relative 'lib/space'
require_relative 'lib/user'
require_relative 'lib/availability'
require_relative 'lib/request'

class Application < Sinatra::Base
  register Sinatra::ActiveRecordExtension
  enable :sessions

  configure :development do
    register Sinatra::Reloader
  end

  get '/' do
    render_erb(:index)
  end

  post '/register' do
    # add user credentials to the database
    user = User.new(
      username: params[:username],
      firstname: params[:firstname],
      lastname: params[:lastname],
      email: params[:email],
      password: params[:password]
    )
    # add user to db
    user.save ? (redirect '/login') : 'Failed to add user!'
  end

  get '/login' do
    render_erb(:login)
  end

  post '/login' do
    user = User.find_by(username: params[:username])
    if user && user.authenticate(params[:password])
      session[:user_id] = user.id
      redirect '/spaces'
    else
      render_erb(:login)
    end
  end

  get '/logout' do
    # kill the session and redirect to the homepage
    session.clear
    redirect '/'
  end

  get '/spaces' do
    redirect_if_not_logged_in
    @spaces = Space.all
    render_erb(:spaces)
  end

  get '/create-space' do
    redirect_if_not_logged_in
    render_erb(:create_space)
  end

  post '/createspace' do
    user_id = session[:user_id]
    space = Space.new(
      name: params[:name], 
      description: params[:description], 
      price: params[:price], 
      address: params[:address]
    )
    space.save!
    book_in = params[:date_from]
    book_out = params[:date_to]
    
    availability = Availability.create(
      space_id: space.id,
      book_in: book_in,
      book_out: book_out
    )
    
    
    redirect '/spaces'
  end

  get '/requests' do
    redirect_if_not_logged_in
    # find spaces owned by user logged in
    spaces = Space.where("user_id = #{session[:user_id]}").ids
    # find the requests on those spaces:
    @requests_on_my_spaces = Request.where('space_id IN (?)', spaces)
    
    # users requests
    @users_requests = Request.where("user_id = #{session[:user_id]}")
    render_erb(:requests)
  end

  post '/request' do
    user_id = session[:user_id]
    space_id = session[:space_id]
    request = Request.create(
      space_id: space_id,
      user_id: user_id,
      book_in: params[:book_in],
      book_out: params[:book_out]
    )
    redirect '/spaces'
  end

  get '/book/:id' do
    redirect_if_not_logged_in
    space_id = params[:id]
    session[:space_id] = space_id # a bit of a hack
    @space = Space.find(space_id)
    @availabilities = Availability.where("space_id = #{space_id}")
    render_erb(:book)
  end

  helpers do
    def logged_in?
      !!session[:user_id]
    end

    def redirect_if_not_logged_in
      redirect '/login' unless logged_in?
    end

    def render_erb(template)
      @username = User.find(session[:user_id]).username unless !logged_in?
      erb template, :layout => !request.xhr?
    end
  end
end