class RDoc::RI::Servlet

# File lib/rdoc/ri/servlet.rb, line 71
def initialize(server, stores, cache, mount_path = nil, extra_doc_dirs = [])
  super server

  @cache      = cache
  @mount_path = mount_path
  @extra_doc_dirs = extra_doc_dirs
  @stores     = stores

  @options = Options.new
  @options.op_dir = '.'

  darkfish_dir = nil

  # HACK dup
  $LOAD_PATH.each do |path|
    darkfish_dir = File.join path, 'rdoc/generator/template/darkfish/'
    next unless File.directory? darkfish_dir
    @options.template_dir = darkfish_dir
    break
  end

  @asset_dirs = {
    :darkfish   => darkfish_dir,
    :json_index =>
      File.expand_path('../generator/template/json_index/', __FILE__),
  }
end

# File lib/rdoc/ri/servlet.rb, line 102
def asset(generator_name, req, res)
  asset_dir = @asset_dirs[generator_name]

  asset_path = File.join asset_dir, req.path

  if_modified_since req, res, asset_path

  res.body = File.read asset_path

  res.content_type = case req.path
                     when /\.css\z/ then 'text/css'
                     when /\.js\z/  then 'application/javascript'
                     else                'application/octet-stream'
                     end
end

# File lib/rdoc/ri/servlet.rb, line 121
def do_GET(req, res)
  req.path.sub!(/\A#{Regexp.escape @mount_path}/, '') if @mount_path

  case req.path
  when '/'
    root req, res
  when '/js/darkfish.js', '/js/jquery.js', '/js/search.js',
       %r%^/css/%, %r%^/images/%, %r%^/fonts/%
    asset :darkfish, req, res
  when '/js/navigation.js', '/js/searcher.js'
    asset :json_index, req, res
  when '/js/search_index.js'
    root_search req, res
  else
    show_documentation req, res
  end
rescue WEBrick::HTTPStatus::NotFound => e
  generator = generator_for ::RDoc::Store.new(@options)

  not_found generator, req, res, e.message
rescue WEBrick::HTTPStatus::Status
  raise
rescue => e
  error e, req, res
end

# File lib/rdoc/ri/servlet.rb, line 154
def documentation_page(store, generator, path, req, res)
  text_name = path.chomp '.html'
  name = text_name.gsub '/', '::'

  if klass = store.find_class_or_module(name)
    res.body = generator.generate_class klass
  elsif page = store.find_text_page(name.sub(/_([^_]*)\z/, '.\1'))
    res.body = generator.generate_page page
  elsif page = store.find_text_page(text_name.sub(/_([^_]*)\z/, '.\1'))
    res.body = generator.generate_page page
  else
    not_found generator, req, res
  end
end

# File lib/rdoc/ri/servlet.rb, line 173
def documentation_search(store, generator, req, res)
  json_index = @cache[store].fetch :json_index do
    @cache[store][:json_index] =
      JSON.dump generator.json_index.build_index
  end

  res.content_type = 'application/javascript'
  res.body = "var search_data = #{json_index}"
end

# File lib/rdoc/ri/servlet.rb, line 187
def documentation_source(path)
  _, source_name, path = path.split '/', 3

  store = @stores[source_name]
  return store, path if store

  store = store_for source_name

  store.load_all

  @stores[source_name] = store

  return store, path
end

# File lib/rdoc/ri/servlet.rb, line 205
      def error(exception, req, res)
        backtrace = exception.backtrace.join "\n"

        res.content_type = 'text/html'
        res.status = 500
        res.body = <<-BODY
<!DOCTYPE html>
<html>
<head>
<meta content="text/html; charset=UTF-8" http-equiv="Content-Type">

<title>Error - #{ERB::Util.html_escape exception.class}</title>

<link type="text/css" media="screen" href="#{@mount_path}/css/rdoc.css" rel="stylesheet">
</head>
<body>
<h1>Error</h1>

<p>While processing <code>#{ERB::Util.html_escape req.request_uri}</code> the
RDoc (#{ERB::Util.html_escape(VERSION)}) server has encountered a
<code>#{ERB::Util.html_escape exception.class}</code>
exception:

<pre>#{ERB::Util.html_escape exception.message}</pre>

<p>Please report this to the
<a href="https://github.com/ruby/rdoc/issues">RDoc issues tracker</a>.  Please
include the RDoc version, the URI above and exception class, message and
backtrace.  If you're viewing a gem's documentation, include the gem name and
version.  If you're viewing Ruby's documentation, include the version of ruby.

<p>Backtrace:

<pre>#{ERB::Util.html_escape backtrace}</pre>

</body>
</html>
    BODY
      end

# File lib/rdoc/ri/servlet.rb, line 248
def generator_for(store)
  generator = Generator::Darkfish.new store, @options
  generator.file_output = false
  generator.asset_rel_path = '..'
  generator.setup

  rdoc = RDoc.new
  rdoc.store     = store
  rdoc.generator = generator
  rdoc.options   = @options

  @options.main_page = store.main
  @options.title     = store.title

  generator
end

# File lib/rdoc/ri/servlet.rb, line 270
def if_modified_since(req, res, path = nil)
  last_modified = File.stat(path).mtime if path

  res['last-modified'] = last_modified.httpdate

  return unless ims = req['if-modified-since']

  ims = Time.parse ims

  unless ims < last_modified
    res.body = ''
    raise WEBrick::HTTPStatus::NotModified
  end
end

# File lib/rdoc/ri/servlet.rb, line 293
def installed_docs
  extra_counter = 0
  ri_paths.map do |path, type|
    store = ::RDoc::Store.new(@options, path: path, type: type)
    exists = File.exist? store.cache_path

    case type
    when :gem
      gem_path = path[%r%/([^/]*)/ri$%, 1]
      [gem_path, "#{gem_path}/", exists, type, path]
    when :system
      ['Ruby Documentation', 'ruby/', exists, type, path]
    when :site
      ['Site Documentation', 'site/', exists, type, path]
    when :home
      ['Home Documentation', 'home/', exists, type, path]
    when :extra
      extra_counter += 1
      store.load_cache if exists
      title = store.title || "Extra Documentation"
      [title, "extra-#{extra_counter}/", exists, type, path]
    end
  end
end

# File lib/rdoc/ri/servlet.rb, line 321
def not_found(generator, req, res, message = nil)
  message ||= "The page <kbd>#{ERB::Util.h req.path}</kbd> was not found"
  res.body = generator.generate_servlet_not_found message
  res.status = 404
end

# File lib/rdoc/ri/servlet.rb, line 330
def ri_paths(&block)
  RI::Paths.each true, true, true, :all, *@extra_doc_dirs, &block #TODO: pass extra_dirs
end

# File lib/rdoc/ri/servlet.rb, line 337
def root(req, res)
  generator = Generator::Darkfish.new nil, @options

  res.body = generator.generate_servlet_root installed_docs

  res.content_type = 'text/html'
end

# File lib/rdoc/ri/servlet.rb, line 348
def root_search(req, res)
  search_index = []
  info         = []

  installed_docs.map do |name, href, exists, type, path|
    next unless exists

    search_index << name

    case type
    when :gem
      gemspec = path.gsub(%r%/doc/([^/]*?)/ri$%,
                          '/specifications/\1.gemspec')

      spec = Gem::Specification.load gemspec

      path    = spec.full_name
      comment = spec.summary
    when :system
      path    = 'ruby'
      comment = 'Documentation for the Ruby standard library'
    when :site
      path    = 'site'
      comment = 'Documentation for non-gem libraries'
    when :home
      path    = 'home'
      comment = 'Documentation from your home directory'
    when :extra
      comment = name
    end

    info << [name, '', path, '', comment]
  end

  index = {
    :index => {
      :searchIndex     => search_index,
      :longSearchIndex => search_index,
      :info            => info,
    }
  }

  res.body = "var search_data = #{JSON.dump index};"
  res.content_type = 'application/javascript'
end

# File lib/rdoc/ri/servlet.rb, line 398
def show_documentation(req, res)
  store, path = documentation_source req.path

  if_modified_since req, res, store.cache_path

  generator = generator_for store

  case path
  when nil, '', 'index.html'
    res.body = generator.generate_index
  when 'table_of_contents.html'
    res.body = generator.generate_table_of_contents
  when 'js/search_index.js'
    documentation_search store, generator, req, res
  else
    documentation_page store, generator, path, req, res
  end
ensure
  res.content_type ||= 'text/html'
end

# File lib/rdoc/ri/servlet.rb, line 422
def store_for(source_name)
  case source_name
  when 'home'
    ::RDoc::Store.new(@options, path: RI::Paths.home_dir, type: :home)
  when 'ruby'
    ::RDoc::Store.new(@options, path: RI::Paths.system_dir, type: :system)
  when 'site'
    ::RDoc::Store.new(@options, path: RI::Paths.site_dir, type: :site)
  when /\Aextra-(\d+)\z/
    index = $1.to_i - 1
    ri_dir = installed_docs[index][4]
    ::RDoc::Store.new(@options, path: ri_dir, type: :extra)
  else
    ri_dir, type = ri_paths.find do |dir, dir_type|
      next unless dir_type == :gem

      source_name == dir[%r%/([^/]*)/ri$%, 1]
    end

    raise WEBrick::HTTPStatus::NotFound,
          "Could not find gem \"#{ERB::Util.html_escape(source_name)}\". Are you sure you installed it?" unless ri_dir

    store = ::RDoc::Store.new(@options, path: ri_dir, type: type)

    return store if File.exist? store.cache_path

    raise WEBrick::HTTPStatus::NotFound,
          "Could not find documentation for \"#{ERB::Util.html_escape(source_name)}\". Please run `gem rdoc --ri gem_name`"

  end
end