module SinatraMore
  module OutputHelpers
    # Captures the html from a block of template code for erb or haml
    # capture_html(&block) => "...html..."
    def capture_html(*args, &block)
      if self.respond_to?(:is_haml?) && is_haml?
         block_is_haml?(block) ? capture_haml(*args, &block) : block.call
      else
        result_text = capture_erb(*args, &block)
        result_text.present? ? result_text : block.call
      end
    end
    
    # Outputs the given text to the templates buffer directly
    # concat_content("This will be output to the template buffer in erb or haml")
    def concat_content(text="")
      if self.respond_to?(:is_haml?) && is_haml?
        haml_concat(text)
      else
        @_out_buf << text
      end
    end
    
    # Returns true if the block is from an ERB or HAML template; false otherwise.
    # Used to determine if html should be returned or concatted to view
    # block_is_template?(block)
    def block_is_template?(block)
       block && (block_is_erb?(block) || (self.respond_to?(:block_is_haml?) && block_is_haml?(block)))
    end
    
    protected

    # Used to capture the html from a block of erb code
    # capture_erb(&block) => '...html...'
    def capture_erb(*args, &block)
      erb_with_output_buffer { block.call(*args) }
    end
    
    # Used to determine if a block is called from ERB.
    # NOTE: This doesn't actually work yet because the variable __in_erb_template 
    # hasn't been defined in ERB. We need to find a way to fix this.
    def block_is_erb?(block)
      block && eval('defined? __in_erb_template', block)
    end

    # Used to direct the buffer for the erb capture
    def erb_with_output_buffer(buf = '') #:nodoc:
      @_out_buf, old_buffer = buf, @_out_buf
      yield
      @_out_buf
    ensure
      @_out_buf = old_buffer
    end
  end
end