# myapp.rb
require 'sinatra'
require 'sinatra/partial'
require 'haml'
require 'pg'
require './config'

def connect_db
  return PG::Connection.open(:dbname=>$appsettings[:dbname],
                             :host=>$appsettings[:dbhost],
                             :user=>$appsettings[:dbuser],
                             :password=>$appsettings[:dbpassword])
end

def check_user(db, user)
  result = nil
  db.prepare 'check_user', 'SELECT exists(SELECT 1 FROM reviewers WHERE reviewer_name=$1)'
  if not user.nil?
    v = db.exec_prepared('check_user', [user])
    if v.first()['exists'] == 't'
      result = user
    end
  end
  return result
end

use Rack::Session::Pool, :expire_after => 2592000

get '/' do
  if params[:cur_user]
    session[:cur_user] = params[:cur_user]
  end

  short_name = params.find do |x| x[1] == nil end
  db = connect_db()
  cur_user = check_user(db, session[:cur_user])
  if short_name.nil?
    restaurants = db.exec 'SELECT restaurant_name,short_name FROM restaurants ORDER BY restaurant_name ASC'
    haml :index, :locals => {:restaurants => restaurants, :cur_user => cur_user}
  else
    short_name = short_name[0]
    db.prepare 'get_restaurant_name', 'SELECT restaurant_name FROM restaurants WHERE short_name=$1'
    db.prepare 'get_reviews', 'SELECT * FROM get_reviews($1)'

    name = db.exec_prepared('get_restaurant_name', [short_name]).first['restaurant_name']
    results = db.exec_prepared 'get_reviews', [short_name]
    haml :restaurant, :locals => {:reviews => results,
                             :name => name,
                             :short_name => short_name,
                             :cur_user => cur_user}
  end
end

def short_name(name)
  name.downcase().gsub(/[^a-zA-Z0-9]/, '-')
end

post '/add/restaurant' do
  db = connect_db()
  if check_user(db, session[:cur_user]).nil?
    redirect to("/")
  else
    db.prepare 'add_restaurant', 'INSERT INTO restaurants (restaurant_name, short_name) VALUES ($1,$2)'
    name = params[:restaurant_name]
    sn = short_name(name)
    db.exec_prepared 'add_restaurant', [name, sn]
    redirect to("/?#{sn}")
  end
end


post '/add/review' do
  db = connect_db()
  if check_user(db, session[:cur_user]).nil?
    redirect to("/?#{params[:short_name]}")
  else
    db.prepare 'put_review',  'SELECT * FROM put_review($1, $2, $3)'
    db.exec_prepared 'put_review', [ params[:short_name], session[:cur_user], params[:review] ]
    redirect to("/?#{params[:short_name]}")
  end
end