sterrym/smithycms

module Smithy
  module Liquid
    module Tags
      class Nav < ::Liquid::Tag
        Syntax = /(#{::Liquid::Expression}+)?/

        # {% nav %} is equivalent to
        # {% nav site, depth: 1, id: 'nav', class: '', wrapper: true, active_class: 'on', include_root: 'true'}
        # {% nav site|site-section|page|section %}
        def initialize(tag_name, markup, tokens)
          @options = { :id => 'nav', :depth => 1, :class => '', :active_class => 'on', :include_root => false }
          if markup =~ Syntax
            @source = ($1 || 'site').gsub(/"|'/, '')
            markup.scan(::Liquid::TagAttributes) do |key, value|
              @options[key.to_sym] = value.gsub(/"|'/, '')
            end
            @options[:active_nested_class] = 'in'
            @options[:depth] = @options[:depth].to_i
            @options[:depth] = 100 if @options[:depth] == 0
            @options[:wrapper] = @options[:wrapper] == "false" ? false : true
            @options[:include_root] = @options[:include_root] == "true" ? true : false
            @options[:root] = @options[:root_id].present? ? Smithy::Page.find(@options[:root_id]) : Smithy::Page.root
          else
            raise ::Liquid::SyntaxError.new("Syntax Error in 'nav' - Valid syntax: nav <site|page|section> <options>")
          end
          super
        end

        def parse(tokens)
          @tokens = tokens
        end

        def render(context)
          @site = context.registers[:site]
          @page = context.registers[:page]
          @controller = context.registers[:controller]
          list_items = render_list_items(root_node)
          @options[:wrapper] ? render_wrapper(list_items, @options[:id]) : list_items
        end

        def render_children(parent, depth)
          list_items = render_list_items(parent, depth)
          return unless list_items.present?
          render_wrapper(list_items)
        end

        def render_list_item(item, depth)
          item_id = "#{@options[:id]}-#{item.permalink}"
          href = item.url
          label = item.title
          css_class = " class=\"#{@options[:active_class]}\"" if (@page && @page.id == item.id) || (@controller && [item.path, item.external_link].include?(@controller.request.path))
          css_class ||= " class=\"#{@options[:active_nested_class]}\"" if @page && @page.ancestors.include?(item)
          %Q{#{"  " * depth}<li id="#{item_id}"#{css_class}><a href="#{href}" id="#{item_id}-link">#{label}</a>#{render_children(item, depth.succ)}</li>}
        end

        def render_list_items(parent, depth=1)
          return unless write_child_list_items?(parent, depth)
          items = []
          items << render_list_item(parent, depth) if depth == 1 && @options[:include_root]
          parent.children.included_in_navigation.inject(items) do |items, item|
            items << render_list_item(item, depth)
          end.join("\n")
        end

        def render_wrapper(list_items, id = nil)
          list_id = id.present? ? " id=\"#{id}\"" : ''
          list_class = @options[:class].present? ? " class=\"#{@options[:class]}\"" : ''
          %Q{<ul#{list_id}#{list_class}>\n#{list_items}\n</ul>}
        end

        def root_node
          case @source
          when 'site', 'site-section'
            @options[:root]
          when 'page'
            @page
          when 'section'
            @page == @options[:root] ? @page : section_page
          end
        end

        private
          def section_page
            ancestors = @page.self_and_ancestors
            idx = ancestors.index(@options[:root])
            ancestors[idx+1].present? ? ancestors[idx+1] : ancestors.second
          end

          def write_child_list_items?(parent, depth)
            return false unless parent.present? && !parent.leaf?
            return true if @source == 'site-section' && @page.self_and_ancestors.include?(parent)
            depth > @options[:depth] ? false : true
          end

      end
      ::Liquid::Template.register_tag('nav', Nav)
    end
  end
end