app.rb

require 'rubygems'
require 'sinatra'
require 'active_record'
require './models/user'
require './models/tweet'
require './models/follow'
require 'json'

# require 'sinatra/active_record'

# configure :production do
#   require 'newrelic_rpm'
# end

# ActiveRecord::Base.logger = Logger.new(STDOUT)

configure do
    enable :sessions
    env = ENV["SINATRA_ENV"] || "development"
    databases = YAML.load(ERB.new(File.read("config/database.yml")).result)
    ActiveRecord::Base.establish_connection(databases[env])
    
end 

helpers do
    def username
        session[:username] ? session[:username] : 'Login'
    end
end

before '/user/profile' do
    if !session[:username] then
        session[:previous_url] = request.path
        @error = 'Must be logged into to visit ' + request.path
        halt erb(:login_form)
    end
end

before '/tweet' do
    if !session[:username] and session[:username] != @current_user.username then
        session[:previous_url] = request.path
        @error = 'Must be logged into to visit ' + request.path
        halt erb(:login_form)
    end
end


get '/' do
    if session[:username]
        erb :timeline
    else
        erb :main
    end
end

get '/login' do 
    response.headers['Cache-Control'] = 'public, max-age=31536000'
    erb :login_form
end

get '/loaderio-2aee583d8cfdd329307e880970966129/' do
  "loaderio-2aee583d8cfdd329307e880970966129"
end

get '/loaderio-100947118e9efbf25b07a3bcb3568bb1/' do
  "loaderio-100947118e9efbf25b07a3bcb3568bb1"
end

post '/login/attempt' do
    @current_user = User.find_by(username: params[:username], password: params[:password])
    if !@current_user
        redirect to '/login'
    else
        session[:username] = @current_user.username
        session[:name] = @current_user.name
        session[:email] = @current_user.email
        session[:id] = @current_user.id
    end
    where_user_came_from = session[:previous_url] || '/'
    redirect to where_user_came_from 
end

get '/logout' do
    session.clear
    erb "<div class='alert alert-message'>Logged out</div>"
end


get '/user/profile' do
    response.headers['Cache-Control'] = 'public, max-age=500'
    erb :profile
end

delete '/delete/tweet' do
    tweet = Tweet.find_by_id(params[:id])
    if tweet
        tweet.destroy
        tweet.to_json
    else
        error 404, {:error => "tweet not found"}.to_json
    end
end

get '/user/register' do
    response.headers['Cache-Control'] = 'public, max-age=31536000'
    erb :register
end

post '/user/register/attempt' do
    @current_user = User.new(name: params[:name], username: params[:username], email: params[:email], password: params[:password])
    if @current_user.save
        session[:username] = params[:username]
        session[:name] = params[:name]
        session[:email] = params[:email]
        session[:id] = User.where(username: params[:username]).take.id
        redirect to "/user/profile"
    else
        erb :register
    end
end

#tweet
post '/tweet' do
    tweet = Tweet.new(text: params[:tweet], owner: session[:id])
    tweet.save!
    redirect to '/'
end

get '/tweets/all' do
    erb :alltweets
end

get '/user/:username' do
    if User.exists?(username: params[:username])
        erb :user
    else
        status 404
        erb :pagenotfound
    end
end

post '/follow' do
    Follow.new(user_id: params[:followId], follower_id: session[:id]).save
    where_user_came_from = "/user/#{params[:followName]}" || '/'
    redirect to where_user_came_from 
end

post '/unfollow' do
    Follow.where(user_id: params[:followId], follower_id: session[:id]).take.delete
    where_user_came_from = "/user/#{params[:followName]}" || '/'
    redirect to where_user_came_from 
end

get '/search' do
    erb :search_results
end

get '/test_reset' do
    if User.exists?(username: 'test_user')
        test_user = User.where(:username => 'test_user').take
        Tweet.destroy_all(owner: test_user.id)
        Follow.destroy_all(follower_id: test_user.id)
        User.destroy(test_user.id)
    end
    User.new(name: 'test_user', username: 'test_user', email: 'test_user', password: 'test_user').save
end

get '/test_tweet' do
    test_user = User.where(:username => 'test_user').take
    tweet = Tweet.new(text: "Test tweet created at #{Time.now}", owner: test_user.id).save
end

get '/test_follow' do
    if User.exists?(username: 'test_user')
        test_user_id = User.where(:username => 'test_user').take.id
        rand_user = User.order("RANDOM()").first
        if rand_user.username == 'test_user'
            rand_user = User.order("RANDOM()").first
        end
        if Follow.exists?(user_id: rand_user.id, follower_id: test_user_id)
            Follow.where(user_id: rand_user.id, follower_id: test_user_id).take.delete
        else
            Follow.new(user_id: rand_user.id, follower_id: test_user_id).save
        end
    end
end

get '/test_user' do
    erb :test_user
end

post '/main_page_recent_tweets' do
    content_type :json
    tweets = Tweet.includes(:user).last(100)
    tweets_hash = Hash.new
    tweets.each do |t|
        if !t.owner.nil?
            tweets_hash[t.id] = "<a href='/user/#{t.user.username}'>@#{t.user.username}</a>: #{t.text} <br /><br />"
        end
    end
    tweets_hash.to_json
end

post '/test_user_recent_tweets_timeline' do
    content_type :json
    user = User.where(:username => 'test_user').take
    followed_user_ids = Follow.select(:user_id).where(:follower_id => user.id).pluck(:user_id) 
    followed_user_ids.push(user.id) 
    tweets = Tweet.includes(:user).where(:owner => followed_user_ids).last(100).reverse 
    tweets_hash = Hash.new
    tweets.each do |t|
        if !t.owner.nil?
            tweets_hash[t.id] = "<a href='/user/#{t.user.username}'>@#{t.user.username}</a>: #{t.text} <br /><br />"
        end
    end
    tweets_hash.to_json
end