require 'fileutils'

module Nitro

# Adds support for caching.

module Caching

  # Output caching.

  module Output

    def self.append_features(base) # :nodoc: 
      super
      base.extend(ClassMethods)
      base.module_eval do
        cattr_accessor :output_cache_root, 'public'
      end
    end

    module ClassMethods

      def do_cache_output(path, content)
        filepath = output_cache_path(path)
        FileUtils.makedirs(File.dirname(filepath))
        File.open(filepath, 'w+') { |f| f.write(content) }
        Logger.debug "Cached page '#{filepath}'" if $DBG
      end

      # Enable output caching for the given actions.

      def cache_output(*actions)
        return unless caching_enabled?

        str = actions.collect { |a| ":#{a}" }.join(', ')

        module_eval %{
          after "do_cache_output", :on => [ #{str} ]
        }
      end

      private

      def output_cache_path(path)
        filename = ((path.empty? || path == '/') ? 'index.html' : path.dup)
#        filename.gsub!(/\/$/, '')
        filename << '/index.html' unless (filename.split('/').last || filename).include? '.'
        return output_cache_root + '/' + filename                     
      end
      
    end

    private

    def do_cache_output
      if caching_enabled? and caching_allowed?
        self.class.do_cache_output(@request.uri, @out)
      end
    end

    def expire_output(name)
      begin
        Logger.debug "Expirinig cache file '#{context.dispatcher.public_root}/#{name}'" if $DBG
        FileUtils.rm_rf("#{context.dispatcher.public_root}/#{name}")
      rescue Object
        # gmosx: is this the right thing to do?
      end
    end
    alias_method :delete_output, :expire_output
    
    def caching_allowed?
      not (@request.post? or @request.uri =~ /\?/)
    end

  end

end

end

# * George Moschovitis  <gm@navel.gr>