app.rb

require 'sinatra/base'
require 'sinatra/reloader'
require 'sinatra/flash'
require_relative 'lib/listing_repository'
require_relative 'lib/booking_repository'
require_relative 'lib/user_repository'
require_relative 'lib/database_connection'

DatabaseConnection.connect('bnb_database_test')

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

  configure :development do
    register Sinatra::Reloader
  end

  get '/' do
    return erb :index, layout: nil
  end
  
  get '/listings' do
    if session[:user_id] == nil
      return redirect '/'
    else
      user_repo = UserRepository.new
      @user = user_repo.find(session[:user_id])

      repo = ListingRepository.new
      @listings = repo.all
      return erb(:listings)
    end
  end

  get '/listings/new' do
    if session[:user_id] == nil
      return redirect '/'
    else
      user_repo = UserRepository.new
      @user = user_repo.find(session[:user_id])
      return erb(:new_listing)
    end
  end

  get '/listings/:id' do
    if session[:user_id] == nil
      return redirect '/'
    else

      user_repo = UserRepository.new
      @user = user_repo.find(session[:user_id])

      repo = ListingRepository.new
  
      @listing = repo.find(params[:id])
      listing_start_date = @listing.start_date
      @first_available_day = compare_today_to(listing_start_date)
      return erb(:listing)
    end
  end

  post '/listings/:id/booking' do
    if session[:user_id] == nil
      return redirect '/'
    else
      if invalid_request_parameters(params[:date])

        flash[:error] = "Please enter a valid date"
        redirect "/listings/#{params[:id]}"
      end
      
      repo = BookingRepository.new
      # checks the existing bookings and if a confirmed booking is found on the same date, it fails to book and flashes a message.
      existing_bookings = repo.find_by_listing(params[:id])
      
      existing_bookings.each do |booking|
        if booking.date == params[:date] &&  booking.confirmed == true
          flash[:error] = "That date is already booked. Please select another"
          redirect "/listings/#{params[:id]}"
        end
      end
    end

    @new_booking = Booking.new
    @new_booking.date = params[:date]
    @new_booking.confirmed = false
    @new_booking.listing_id = params[:id]
    @new_booking.user_id = session[:user_id]
       
    repo.create(@new_booking)

    user_repo = UserRepository.new
    @user = user_repo.find(session[:user_id])
    @listing_repo = ListingRepository.new
    @bookings = repo.find_by_user_id(session[:user_id])
    
    return erb(:your_bookings)
  end
  
  get '/your_bookings' do
    repo = BookingRepository.new

    user_repo = UserRepository.new
    @user = user_repo.find(session[:user_id])

    @listing_repo = ListingRepository.new
    @bookings = repo.find_by_user_id(session[:user_id])
    return erb(:your_bookings)
  end

  post '/listings/new' do
    params_array = [params[:name], params[:price], params[:description]]
    params_array.each do |param|
      if invalid_new_listing_parameters(param)

        flash[:error] = "Please fill all fields with valid inputs"
        redirect "/listings/new"
      end
    end

    repo = ListingRepository.new
    new_listing = Listing.new
  
    new_listing.name = params[:name]
    new_listing.price = params[:price]
    new_listing.description = params[:description]
    new_listing.start_date = params[:start_date]
    new_listing.end_date = params[:end_date]
    new_listing.user_id = session[:user_id]
   
    
    if dates_checker(new_listing)
      status 400
      flash[:date_error] = 'The end date must be after the start date.'
      return redirect '/listings/new'
    end

    repo.create(new_listing)
    status 200
    flash[:listing_success] = 'Listing successfully created!'
    return redirect '/listings'
  end

  get '/requests' do
    if session[:user_id] == nil
      return redirect '/'
    else
      user_repo = UserRepository.new
      @user = user_repo.find(session[:user_id])
      
      booking_repo = BookingRepository.new
      @user_repo = UserRepository.new
      @listing_repo = ListingRepository.new
      
      # we're creating a bookings array. This is storing a 
      # booking based on the session id of the current user

      listings = @listing_repo.find_by_user_id(session[:user_id])
      
      bookings = []
      listings.each do |listing| 
        bookings += booking_repo.find_by_listing(listing.id)
      end

      @unconfirmed_bookings = bookings.select do |booking|
        !booking.confirmed
      end

      @confirmed_bookings = bookings.select do |booking|
        booking.confirmed
      end

      erb(:requests)
    end
  end

  post '/requests/confirm/:id' do
    repo = BookingRepository.new
    booking = repo.find(params[:id])

    repo.confirm_booking(booking)
    redirect('/requests')
  end

  post '/signup' do

    listing_repo = ListingRepository.new
    @listings = listing_repo.all
    
    @user = User.new
    @user.email = params[:email]
    @user.password = params[:password]
    
    repo = UserRepository.new
    if @user.email.empty? || @user.password.empty?
      flash[:signup_error] = "Please enter valid email and password"
      redirect "/"
    elsif @user.email.empty? && @user.password.empty?
      flash[:signup_error] = "Please enter valid email and password"
      redirect "/"
    end
    
    repo.create(@user)
    flash[:success] = "Sign up success!"
    return redirect "/"
  end
  
  post '/login' do

    listing_repo = ListingRepository.new
    @listings = listing_repo.all
    
    email = params[:email]
    password = params[:password]
    
    if email.empty? || password.empty?
      flash[:login_error] = "Please enter valid email and password"
      redirect "/"
    elsif email.empty? && password.empty?
      flash[:login_error] = "Please enter valid email and password"
      redirect "/"
    end
    
    repo = UserRepository.new
    user = repo.find_by_email(email)
    
    sign_in_status = repo.password_correct?(user.password, password)
    
    if sign_in_status == true
      
      session[:user_id] = user.id

      @user = repo.find(user.id)
      return erb(:listings)
    else
      return erb :signup_fail, layout: nil
    end
  end
  
  post '/logout' do
    session.clear
    redirect '/'
  end

  private

  def invalid_request_parameters(params)
    # Are the params nil?
    return true if params == nil
  
    # Are they empty strings?
    return true if params == ""
  
    return true if params.include? '<script>'

    return true if Date.parse(params) == false

    return false
  end

  def invalid_new_listing_parameters(params)
    # Are the params nil?
    return true if params == nil
  
    # Are they empty strings?
    return true if params == ""
  
    return true if params.include? '<script>'

    return true if params.match /^\A\d{4}-\d{2}-\d{2}\z/

    return false
  end
  
  def compare_today_to(listing_start_date)
    today = Time.now.to_date.to_s
    listing_start_date < today ? today : listing_start_date
  end

  def dates_checker(new_listing)
    start_date_parts = new_listing.start_date.split('-')
    start_date_to_check = Time.new(*start_date_parts)
    
    end_date_parts = new_listing.end_date.split('-')
    end_date_to_check = Time.new(*end_date_parts)

    if start_date_to_check > end_date_to_check
      return true
    end
  end
end