=begin Copyright 2010-2012 Tasos Laskos Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. =end module Arachni require Options.instance.dir['lib'] + 'module/utilities' module Mixins # # Provides a flexible way to make any Class observable via callbacks/hooks # using simple dynamic programming with the help of "method_missing()". # # The observable classes (those which include this module) use: # * call_( *args ) # to call specific hooks. # # The observers set hooks using: # * observer_instance.add_( &block ) # * observer_instance.on_( &block ) # # # @author Tasos "Zapotek" Laskos # module Observable include Arachni::Utilities def method_missing( sym, *args, &block ) # grab the action (add/call) and the hook name action, hook = sym.to_s.split( '_', 2 ) @__hooks ||= {} @__hooks[hook] ||= [] if action && hook case action when 'add', 'on' add_block( hook, &block ) return when 'call' call_blocks( hook, args ) return end end fail NoMethodError.new( "Undefined method '#{sym.to_s}'.", sym, args ) end private def add_block( hook, &block ) @__hooks[hook] << block end def call_blocks( hook, *args ) @__hooks[hook].each do |block| exception_jail { if args.first.size == 1 block.call( args.flatten[0] ) else block.call( *args ) end } end end end end end