require 'cgi'
require 'action_view/helpers/url_helper'
require 'action_view/helpers/tag_helper'
module SweetAssets
def self.included(base)
base.class_eval do
include ClassMethods
extend SweetAssetsShortcuts
include AppendAssetsAfterRescue
before_filter :initialize_assets_accessor
before_filter :style_like_current_controller
before_filter :script_like_current_controller
after_filter :apply_sweet_assets
end
end
module SweetAssetsShortcuts
def style_like(*assets)
assets.each {|asset| before_filter "style_like_#{asset}".intern }
end
def script_like(*assets)
assets.each {|asset| before_filter "script_like_#{asset}".intern }
end
end
module ClassMethods
def method_missing_with_sweet_assets(method_name, *args, &block)
if match = method_name.to_s.match(/^style_like_(\w+)(\!)?$/)
@sweet_assets[:stylesheets][match[2] ? :bottom : :top] << match[1]
elsif match = method_name.to_s.match(/^script_like_(\w+)(\!)?$/)
@sweet_assets[:javascripts][match[2] ? :bottom : :top] << match[1]
else
method_missing_without_sweet_assets(method_name, *args, &block)
end
end
alias_method_chain :method_missing, :sweet_assets
def initialize_assets_accessor
@sweet_assets = { :javascripts => {:top => [], :bottom => []},
:stylesheets => {:top => [], :bottom => []} }
end
def style_like_current_controller
@sweet_assets[:stylesheets][:bottom] << controller_name
end
def script_like_current_controller
@sweet_assets[:javascripts][:bottom] << controller_name
end
def apply_sweet_assets
return true unless @sweet_assets.any? {|asset_type, assets| assets.any? {|placement, files| !files.blank? } }
generator = SweetAssetsGenerator.new(@sweet_assets, self)
response.body.gsub! '<head>', "<head>\n#{generator.tags(:top)}\n" if response.body.respond_to?(:gsub!)
response.body.gsub! '</head>', "\n#{generator.tags(:bottom)}\n</head>" if response.body.respond_to?(:gsub!)
@sweet_assets = {}
end
class SweetAssetsGenerator
include ActionView::Helpers::TagHelper
include ActionView::Helpers::AssetTagHelper
def initialize(assets, controller)
@assets = assets
@controller = controller
end
def tags(placement)
javascript_tags(placement) + "\n" + stylesheet_tags(placement)
end
def stylesheet_tags(placement)
files = @assets[:stylesheets][placement].dup
files = files.select {|file| File.exists?("#{STYLESHEETS_DIR}/#{file}.css") } unless RAILS_ENV.eql?('test')
return '' if files.blank?
files << {:cache => "sweet_stylesheets_#{files.join(',')}" } if ActionController::Base.perform_caching
stylesheet_link_tag *files
end
def javascript_tags(placement)
files = @assets[:javascripts][placement].dup
files = files.select {|file| File.exists?("#{JAVASCRIPTS_DIR}/#{file}.js") } unless RAILS_ENV.eql?('test')
return '' if files.blank?
files << {:cache => "sweet_javascripts_#{files.join(',')}" } if ActionController::Base.perform_caching
javascript_include_tag *files
end
end
end
# we're going to wrap our rescue_action around the top-level rescue_action method chain
# this ensures that we can apply the stylesheets even if the after filter was aborted due to an exception
module AppendAssetsAfterRescue
def self.included(base)
base.class_eval do
def initialize_with_append_sweet_assets_rescue_action(*args)
initialize_without_append_sweet_assets_rescue_action(*args)
self.class.send :alias_method_chain, :rescue_action, :append_sweet_assets unless respond_to?(:rescue_action_without_append_sweet_assets)
end
alias_method_chain :initialize, :append_sweet_assets_rescue_action
end
end
def rescue_action_with_append_sweet_assets(*args)
rescue_action_without_append_sweet_assets(*args)
apply_sweet_assets
end
end
end
) is private.
