mirror of
https://github.com/middleman/middleman.git
synced 2022-11-09 12:20:27 -05:00
90 lines
2.9 KiB
Ruby
Executable file
90 lines
2.9 KiB
Ruby
Executable file
Entry = Struct.new(:dir, :children)
|
|
|
|
module Middleman::Features::Slickmap
|
|
class << self
|
|
def registered(app)
|
|
require 'slickmap'
|
|
|
|
@sitemap_url = "sitemap.html"
|
|
@sitemap_url = app.sitemap_url if app.respond_to?(:slickmap_url)
|
|
|
|
if app.environment == :build
|
|
Middleman::Builder.template :slickmap, @sitemap_url, @sitemap_url
|
|
end
|
|
|
|
app.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
|
|
|
|
app.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
|
|
alias :included :registered
|
|
end
|
|
|
|
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
|
|
end
|