# -*- ruby -*-
#
# Copyright (C) 2010  Kouhei Sutou <kou@clear-code.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

Encoding.default_external = "utf-8"

require 'pathname'

base_dir = Pathname.new(__FILE__).dirname.cleanpath.realpath
lib_dir = base_dir + "lib"
home_dir = Pathname.new('/home/okkez')

bitclust_dir = home_dir + 'ruby' + "bitclust"
bitclust_lib_dir = bitclust_dir + "lib"
# rroonga_dir = home_dir + 'ruby' + "rroonga"
# rroonga_lib_dir = rroonga_dir + "lib"
# rroonga_ext_dir = rroonga_dir + "ext" + "groonga"
# racknga_dir = home_dir + 'ruby' + 'racknga'
# racknga_lib_dir = racknga_dir + 'lib'

$LOAD_PATH.unshift(bitclust_lib_dir.to_s)
# $LOAD_PATH.unshift(rroonga_ext_dir.to_s)
# $LOAD_PATH.unshift(rroonga_lib_dir.to_s)
# $LOAD_PATH.unshift(racknga_lib_dir.to_s)
$LOAD_PATH.unshift(lib_dir.to_s)

require 'racknga'
require 'racknga/middleware/log'
require 'racknga/middleware/cache'

require 'rurema_search'
require 'rurema_search/groonga_searcher'

database = RuremaSearch::GroongaDatabase.new
database.open((base_dir + "groonga-database").to_s, "utf-8")

suggest_database = RuremaSearch::GroongaSuggestDatabase.new
suggest_database.open((base_dir + "var" + "lib" + "suggest").to_s)

environment = ENV["RACK_ENV"] || "development"

searcher_options = {}

load_yaml = Proc.new do |file_name|
  configuration_file = base_dir + file_name
  if configuration_file.exist?
    require 'yaml'
    YAML.load(configuration_file.read)
  else
    nil
  end
end

load_searcher_option = Proc.new do |key, file_name|
  configuration = load_yaml.call(file_name)
  if configuration
    require 'yaml'
    searcher_options[key] = configuration
  end
end

load_searcher_option.call(:document, "document.yaml")

configuration = load_yaml.call("#{environment}.yaml") || {}

searcher = RuremaSearch::GroongaSearcher.new(database,
                                             suggest_database,
                                             base_dir.to_s,
                                             searcher_options)

case environment
when "development"
  use Rack::ShowExceptions
when "production"
  show_error_page = Class.new do
    def initialize(app, options={})
      @app = app
      @searcher = options[:searcher]
    end

    def call(env)
      @app.call(env)
    rescue Exception => exception
      @searcher.error_page(env, exception)
    end
  end
  use show_error_page, :searcher => searcher

  load_searcher_option.call(:smtp, "smtp.yaml")
  notifiers = [Racknga::ExceptionMailNotifier.new(searcher_options[:smtp])]
  use Racknga::Middleware::ExceptionNotifier, :notifiers => notifiers
end

if configuration["use_log"]
  log_database_path = base_dir + "var" + "log" + "db"
  use Racknga::Middleware::Log, :database_path => log_database_path.to_s
end

case environment
when "development"
  use Rack::CommonLogger
end

use Rack::Runtime
use Rack::ContentLength

urls = ["/favicon.", "/css/", "/images/", "/javascripts/", "/1.8.", "/1.9."]

case environment
when "development"
  class DirectoryIndex
    def initialize(app, options={})
      @app = app
      @urls = options[:urls]
    end

    def call(env)
      path = env["PATH_INFO"]
      can_serve = @urls.any? { |url| path.index(url) == 0 }
      env["PATH_INFO"] += "index.html" if can_serve and /\/\z/ =~ path
      @app.call(env)
    end
  end

  use DirectoryIndex, :urls => urls
end

use Rack::Static, :urls => urls, :root => (base_dir + "public").to_s

use Racknga::Middleware::Deflater
use Rack::ConditionalGet

use Racknga::Middleware::JSONP

if configuration["use_cache"]
  cache_database_path = base_dir + "var" + "cache" + "db"
  use Racknga::Middleware::Cache, :database_path => cache_database_path.to_s
end

use Rack::Lint
use Rack::Head
map "/ja/search" do
  use Rack::Chunked
  # This is workaround!
  use Rack::Static, :urls => ["/favicon.ico", "/css", "/javascripts", "/images", "apple-touch-icon.png", "apple-touch-icon.svg", "favicon.png", "favicon.svg", "robots.txt"], :root => "public"
  run searcher
end