module Sinatra::AssetHelpers

  def stylesheet(s)
    s += '.css' unless s.include?('.css')
    s = asset_path(s) unless s =~ /^https?:\// or s =~ /^\/\/.+/
    "<link rel='stylesheet' href='#{s}' />"
  end

  def javascript(s, &block)
    s += '.js' unless s.include?('.js')
    s = asset_path(s) unless s =~ /^https?:\// or s =~ /^\/\/.+/
    html = "<script type='text/javascript' src='#{s}'>"
    if block.present?
      require 'haml'
      context = Object.new
      class << context
        include Haml::Helpers
      end
      context.init_haml_helpers
      html += context.capture_haml(&block)
    end
    html += '</script>'
  end

  def asset_path(source, options={})
    source = ('/assets/' + source).gsub('//', '/').gsub('/assets/assets/', '/assets/')
  end

  def image_path(source)
    asset_path(source)
  end

end