layout.rb

require 'pathname'

module Rokko
  class Layout < Mustache
    self.template_file = File.join(File.dirname(__FILE__), 'layout.mustache')

    def initialize(doc)
      @doc = doc
      @options = @doc.options
    end

    def title
      File.basename(@doc.file)
    end

    def styles
      if @options[:local]
        docco = File.read(File.join(File.dirname(__FILE__), 'assets', 'docco.css'))
        highlight = File.read(File.join(File.dirname(__FILE__), 'assets', 'github.min.css'))

        %(<style type="text/css" media="screen, projection">#{docco}\n#{highlight}</style>)
      else
        stylesheet_link = @options[:stylesheet] || "http://vast.github.io/rokko/assets/v#{::Rokko::VERSION}/docco.css"
        %(<link rel="stylesheet" href="#{stylesheet_link}" />
         <link rel="stylesheet" href="http://yandex.st/highlightjs/7.5/styles/github.min.css" />)
      end
    end

    def highlight_js
      js = if @options[:local]
        "<script>#{File.read(File.join(File.dirname(__FILE__), 'assets', 'highlight.min.js'))}</script>"
      else
        %(<script src="http://yandex.st/highlightjs/7.5/highlight.min.js"></script>)
      end

      [js, '<script>hljs.initHighlightingOnLoad();</script>'].join("\n")
    end

    def sections
      num = 0
      @doc.sections.map do |docs,code|
        {
          docs: docs,
          code: code,
          num: (num += 1)
        }
      end
    end

    def sources?
      @doc.sources.length > 1
    end

    def sources
      sources = @doc.sources.sort.map do |source|
        {
          path: source,
          basename: File.basename(source),
          url: relative_url(source)
        }
      end

      if @doc.options[:index] || @doc.options[:generate_index]
        sources.unshift({ path: 'index.html', basename: 'index', url: relative_url('index.html') })
      end

      sources
    end

    private

    def relative_url(source)
      if(@doc.options[:preserve_urls])
        source.sub(Regexp.new("#{File.extname(source)}$"), ".html")
      else
        relative_path = Pathname.new(File.dirname(source)).relative_path_from(Pathname(File.dirname(@doc.file))).to_s
        "#{relative_path}/#{File.basename(source).sub(Regexp.new("#{File.extname(source)}$"), ".html")}"
      end
    end

  end
end