diff --git a/lib/tracks/source_view.rb b/lib/tracks/source_view.rb index 9505282b..8cb17ca8 100644 --- a/lib/tracks/source_view.rb +++ b/lib/tracks/source_view.rb @@ -12,6 +12,10 @@ module Tracks def nil? yield if @source_view.nil? && block_given? end + + def context + yield if :context == @source_view && block_given? + end def method_missing(check_source_view,*args) yield if check_source_view == @source_view && block_given? @@ -61,4 +65,4 @@ module Tracks end -ActionController::Base.send(:include, Tracks::SourceViewSwitching::Controller) \ No newline at end of file +ActionController::Base.send(:include, Tracks::SourceViewSwitching::Controller)