#!/usr/bin/env ruby

require 'sinatra'
require 'erb'
require 'time'
require 'rinku'
require 'digest'

class TaskwarriorWeb::App < Sinatra::Base
  autoload :Helpers, 'taskwarrior-web/helpers'

  @@root = File.expand_path(File.join(File.dirname(__FILE__), '..', '..'))
  set :root,  @@root    
  set :app_file, __FILE__
  set :public_folder, File.dirname(__FILE__) + '/public'
  set :views, File.dirname(__FILE__) + '/views'
  
  def protected!
    response['WWW-Authenticate'] = %(Basic realm="Taskworrior Web") and throw(:halt, [401, "Not authorized\n"]) and return unless authorized?
  end

  def authorized?
    @auth ||=  Rack::Auth::Basic::Request.new(request.env)
    values = [TaskwarriorWeb::Config.property('task-web.user'), TaskwarriorWeb::Config.property('task-web.passwd')]
    @auth.provided? && @auth.basic? && @auth.credentials && @auth.credentials == values
  end

  # Before filter
  before do
    @current_page = request.path_info
    protected! if TaskwarriorWeb::Config.property('task-web.user')
  end

  # Helpers
  helpers Helpers

  # Redirects
  get '/' do
    redirect '/tasks/pending'
  end
  get '/tasks/?' do
    redirect '/tasks/pending'
  end

  # Task routes
  get '/tasks/:status/?' do
    pass unless ['pending', 'waiting', 'completed', 'deleted'].include?(params[:status])
    @title = "Tasks"
    @subnav = subnav('tasks')
    @tasks = TaskwarriorWeb::Task.find_by_status(params[:status]).sort_by! { |x| [x.priority.nil?.to_s, x.priority.to_s, x.due.nil?.to_s, x.due.to_s, x.project.to_s] }
    erb :listing
  end

  get '/tasks/new/?' do
    @title = 'New Task'
    @date_format = TaskwarriorWeb::Config.dateformat || 'm/d/yy'
    @date_format.gsub!('Y', 'yy')
    erb :task_form
  end

  post '/tasks/?' do
    results = passes_validation(params[:task], :task)
    if results.empty?
      task = TaskwarriorWeb::Task.new(params[:task])
      task.save!.to_s
      redirect '/tasks'
    else
      @task = params[:task]
      @title = 'New Task'
      @date_format = TaskwarriorWeb::Config.dateformat || 'm/d/yy'
      @date_format.gsub!('Y', 'yy')
      @messages = []
      results.each do |result|
        @messages << { :severity => 'alert-error', :message => result }
      end
      erb :task_form
    end
  end

  # Projects
  get '/projects' do
    redirect '/projects/overview'
  end

  get '/projects/overview/?' do
    @title = 'Projects'
    @subnav = subnav('projects')
    @tasks = TaskwarriorWeb::Task.query('status.not' => 'deleted', 'project.not' => '').sort_by! { |x| [x.priority.nil?.to_s, x.priority.to_s, x.due.nil?.to_s, x.due.to_s] }.group_by { |x| x.project.to_s }
    erb :projects
  end

  get '/projects/:name/?' do
    @subnav = subnav('projects')
    subbed = params[:name].gsub('--', '.') 
    @tasks = TaskwarriorWeb::Task.query('status.not' => 'deleted', :project => subbed).sort_by! { |x| [x.priority.nil?.to_s, x.priority.to_s, x.due.nil?.to_s, x.due.to_s] }
    @title = @tasks.select { |t| t.project.match(/^#{subbed}$/i) }.first.project
    erb :project
  end

  # AJAX callbacks
  get '/ajax/projects/?' do
    TaskwarriorWeb::Command.new(:projects).run.split("\n").to_json
  end

  get '/ajax/tags/?' do
    tags = TaskwarriorWeb::Command.new(:tags).run.split("\n")
    tags.keep_if { |tag| tag.include?(params[:query]) }

    json = []
    tags.each do |tag|
      json << { :name => tag, :id => tag }
    end

    json.to_json
  end

  get '/ajax/count/?' do
    TaskwarriorWeb::Task.count(:status => :pending).to_s
  end

  post '/ajax/task-complete/:id/?' do
    # Bummer that we have to directly use Command here, but apparently tasks
    # cannot be filtered by UUID.
    TaskwarriorWeb::Command.new(:complete, params[:id]).run
  end

  # Error handling
  not_found do
    @title = 'Page Not Found'
    @referrer = request.referrer
    erb :'404'
  end

  def passes_validation(item, method)
    results = [] 
    case method
      when :task
        if item['description'].empty?
          results << 'You must provide a description'
        end
    end
    results
  end
end