module Middleman::Features::Slickmap

def self.build_sitemap(&block)    
  @@utility = []
  [recurse_sitemap(Middleman::Server.views, &block), @@utility]
end

def self.recurse_sitemap(path, &block)
  bad_ext = path.split('.html')[1]
  path = path.gsub(bad_ext, '') if bad_ext
  entry = Entry.new(path, [])
  #no "." or ".." dirs
  Dir[File.join(path, "*")].each do |e|
    next if !File.directory?(e) && !e.include?(".html")
    if File.directory?(e)
      entry.children << recurse_sitemap(e, &block)
    elsif block_given?
      how_to_handle = block.call(e)
      if how_to_handle == :valid
        entry.children << recurse_sitemap(e, &block)
      elsif how_to_handle == :utility
        bad_ext = e.split('.html')[1]
        e = e.gsub(bad_ext, '') if bad_ext
        @@utility << e.gsub(Middleman::Server.views + "/", '')
      end
    end
  end
  entry
end

def registered(app)
  require 'slickmap'
  @sitemap_url = config[:url] || "sitemap.html"
  if Middleman::Server.environment == :build
    Middleman::Builder.template :slickmap, @sitemap_url, @sitemap_url
  end
  Middleman::Server.helpers do
    def sitemap_node(n, first=false)
      if n.children.length < 1
        if !first && File.extname(n.dir).length > 0
          haml_tag :li do
            path = n.dir.gsub(self.class.views, '')
            haml_concat link_to(File.basename(path), path)
          end
        end
      else  
        haml_tag(:li, :id => first ? "home" : nil) do
          if first
            haml_concat link_to("Homepage", "/" + self.class.index_file)
          else
            # we are a dir
            index = n.children.find { |c| c.dir.include?(self.class.index_file) }
            haml_concat link_to(index.dir.gsub(self.class.views + "/", '').gsub("/" + File.basename(index.dir), '').capitalize, index.dir.gsub(self.class.views, ''))
          end
          other_children = n.children.select { |c| !c.dir.include?(self.class.index_file) }
          if other_children.length > 0
            if first
              other_children.each { |i| sitemap_node(i) }
            else
              haml_tag :ul do
                other_children.each { |i| sitemap_node(i) }
              end
            end
          end
        end  
      end
    end
  end
  Middleman::Server.get "/#{@sitemap_url}" do
    # Return :utility to put it util top menu. False to ignore
    @tree, @utility = Middleman::Features::Slickmap.build_sitemap do |file_name|
      :valid
    end
    haml "template.html".to_sym, :layout => false, :views => File.expand_path(File.join(File.dirname(__FILE__), "slickmap"))
  end
end

def sitemap_node(n, first=false)
  if n.children.length < 1
    if !first && File.extname(n.dir).length > 0
      haml_tag :li do
        path = n.dir.gsub(self.class.views, '')
        haml_concat link_to(File.basename(path), path)
      end
    end
  else  
    haml_tag(:li, :id => first ? "home" : nil) do
      if first
        haml_concat link_to("Homepage", "/" + self.class.index_file)
      else
        # we are a dir
        index = n.children.find { |c| c.dir.include?(self.class.index_file) }
        haml_concat link_to(index.dir.gsub(self.class.views + "/", '').gsub("/" + File.basename(index.dir), '').capitalize, index.dir.gsub(self.class.views, ''))
      end
      other_children = n.children.select { |c| !c.dir.include?(self.class.index_file) }
      if other_children.length > 0
        if first
          other_children.each { |i| sitemap_node(i) }
        else
          haml_tag :ul do
            other_children.each { |i| sitemap_node(i) }
          end
        end
      end
    end  
  end
end