config.ru

# -*- ruby -*-
#
# Copyright (C) 2010-2016  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"

bitclust_dir = base_dir.parent + "bitclust"
bitclust_lib_dir = bitclust_dir + "lib"
rroonga_dir = base_dir.parent + "rroonga"
rroonga_lib_dir = rroonga_dir + "lib"
rroonga_ext_dir = rroonga_dir + "ext" + "groonga"
racknga_dir = base_dir.parent + "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 "groonga"

keep_n_latest_logs = 10
log_dir = base_dir + "var" + "log"
log_dir.mkpath
logs = Pathname.glob(log_dir + "groonga.log.*")
logs.sort[0..(-keep_n_latest_logs + -1)].each do |old_log|
  old_log.delete
end
Groonga::Logger.path = (log_dir + "groonga.log").to_s
Groonga::Logger.rotate_threshold_size = 1 * 1024 * 1024

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)

if defined?(PhusionPassenger)
  PhusionPassenger.on_event(:starting_worker_process) do |forked|
    if forked
      database.reopen
      suggest_database.reopen
    end
  end
end

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 "production"
  show_error_page = Class.new do
    def initialize(app, options={})
      @app = app
      @searcher = options[:searcher]
      @target_exception = options[:target_exception] || Exception
    end

    def call(env)
      @app.call(env)
    rescue @target_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

  options = {
    :searcher => searcher,
    :target_exception => RuremaSearch::GroongaSearcher::ClientError,
  }
  use show_error_page, options
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

use Rack::Runtime
use Rack::ContentLength

urls = [
  "/favicon.",
  "/css/",
  "/images/",
  "/javascripts/",
  "/1.",
  "/2.",
  "/3.",
]

case environment
when "development"
  use Rack::Static,
      :urls => urls,
      :root => (base_dir + "public").to_s,
      :index => "index.html"
end

use Racknga::Middleware::Deflater
use Rack::Lint
use Rack::Head
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 Racknga::Middleware::InstanceName, :application_name => "Rurema Search"

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

require 'rack/protection'
use Rack::Protection::FrameOptions

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