middleman--middleman/lib/middleman/core_extensions/sitemap.rb

require 'find'

module Middleman::CoreExtensions::Sitemap
  class << self
    def registered(app)
      app.extend ClassMethods
      app.helpers Helpers
      
      app.file_changed do |file|
        sitemap.touch_file(file)
      end
      
      app.file_deleted do |file|
        sitemap.remove_file(file)
      end
    end
    alias :included :registered
  end
  
  module Helpers
    def sitemap
      self.class.sitemap
    end
  end
  
  module ClassMethods
    def sitemap
      @sitemap ||= SitemapStore.new(self)
    end
    
    # Keep a path from building
    def ignore(path)
      sitemap.ignore_path(path)
    end
    
    def reroute(url, target)
      sitemap.set_path(url, target)
    end
  end
  
  class SitemapStore
    def initialize(app)
      @app = app
      @source = File.expand_path(@app.views, @app.root)
      @map = {}
      @source_map = {}
      @ignored_paths = false
      @generic_paths = false
      @proxied_paths = false
      
      build_static_map
    end
    
    # Check to see if we know about a specific path
    def path_exists?(path)
      path = path.sub(/^\//, "")
      @map.has_key?(path)
    end
    
    def path_is_proxy?(path)
      path = path.sub(/^\//, "")
      return false if !path_exists?(path)
      @map[path].is_a?(String)
    end
    
    def path_target(path)
      path = path.sub(/^\//, "")
      @map[path]
    end
    
    def set_path(path, target=true)
      path   = path.sub(/^\//, "")
      target = target.sub(/^\//, "") if target.is_a?(String)
      
      @map[path] = target
      
      @ignored_paths = false if target === false
      @generic_paths = false if target === true
      @proxied_paths = false if target.is_a?(String)
    end
    
    def ignore_path(path)
      set_path(path, false)
    end
    
    def each(&block)
      @map.each do |k, v|
        yield(k, v)
      end
    end
    
    def all_paths
      @map.keys
    end
    
    def all_values
      @map.values
    end
    
    def ignored_path?(path)
      path = path.sub(/^\//, "")
      ignored_paths.include?(path)
    end
    
    def ignored_paths
      @ignored_paths ||= begin
        ignored = []
        each do |k, v|
          ignored << k if v === false
        end
        ignored
      end
    end
    
    def generic_path?(path)
      path = path.sub(/^\//, "")
      generic_paths.include?(path)
    end
    
    def generic_paths
      @generic_paths ||= begin
        generic = []
        each do |k, v|
          generic << k if v === true
        end
        generic
      end
    end
    
    def proxied_path?(path)
      path = path.sub(/^\//, "")
      proxied_paths.include?(path)
    end
    
    def proxied_paths
      @proxied_paths ||= begin
        proxied = []
        each do |k, v|
          proxied << k if v.is_a?(String)
        end
        proxied
      end
    end
    
    def touch_file(file)
      add_file(file)
    end
    
    def remove_file(file)
      path = file_to_path(file)
      remove_path(path) if path
    end
    
    def remove_path(path)
      path = path.sub(/^\//, "")
      @map.delete(path) if path_exists?(path)
    end
    
    def source_map
      @source_map
    end
    
  protected
    def build_static_map
      Find.find(@source) do |file|
        add_file(file)
      end
    end
    
    def file_to_path(file)
      @source ||= File.expand_path(@app.views, @app.root)
      file = File.expand_path(file, @app.root)
      
      prefix = @source + "/"
      return false unless file.include?(prefix)
      
      path = file.sub(prefix, "")
      
      end_of_the_line = false
      while !end_of_the_line
        file_extension = File.extname(path)
      
        if ::Tilt.mappings.has_key?(file_extension.gsub(/^\./, ""))
          path = path.sub(file_extension, "")
        else
          end_of_the_line = true
        end
      end
      
      @source_map[path] = file
      path
    end
    
    def add_file(file)
      return false if file == @source ||
                      file.match(/\/\./) ||
                      (file.match(/\/_/) && !file.match(/\/__/)) ||
                      File.directory?(file)
                      
      path = file_to_path(file)
      
      add_path(path) if path && !path_exists?(path)
    end
    
    def add_path(path)
      return false if path == "layout" ||
                      path.match(/^layouts/)
    
      set_path(path)
      
      true
    end
  end
end