Contents

require 'hamster'
require 'middleman-core/contracts'

# Immutable Callback Management, complete with Contracts validation.
module Middleman
  class CallbackManager
    include Contracts

    Contract Any
    def initialize
      @callbacks = ::Hamster.hash
      @subscribers = ::Hamster.vector
    end

    Contract RespondTo[:define_singleton_method], ArrayOf[Symbol] => Any
    def install_methods!(install_target, names)
      manager = self

      names.each do |method_name|
        install_target.define_singleton_method(method_name) do |*keys, &b|
          key_set = keys.unshift(method_name)
          manager.add(key_set.length > 1 ? key_set : key_set[0], &b)
        end
      end

      install_target.define_singleton_method(:execute_callbacks) do |*args|
        keys = args.shift
        manager.execute(keys, args[0], self)
      end

      install_target.define_singleton_method(:callbacks_for, &method(:callbacks_for))
      install_target.define_singleton_method(:subscribe_to_callbacks, &method(:subscribe))
    end

    Contract Or[Symbol, ArrayOf[Symbol]], Proc => Any
    def add(keys, &block)
      immutable_keys = keys.is_a?(Symbol) ? keys : ::Hamster::Vector.new(keys)

      @callbacks = @callbacks.put(immutable_keys) do |v|
        v.nil? ? ::Hamster::Vector.new([block]) : v.push(block)
      end
    end

    Contract Proc => Any
    def subscribe(&block)
      @subscribers = @subscribers.push(block)
    end

    Contract Or[Symbol, ArrayOf[Symbol]], Maybe[ArrayOf[Any]], Maybe[RespondTo[:instance_exec]] => Any
    def execute(keys, args=[], scope=self)
      callbacks_for(keys).each { |b| scope.instance_exec(*args, &b) }
      @subscribers.each { |b| scope.instance_exec(keys, args, &b) }
    end

    Contract Or[Symbol, ArrayOf[Symbol]] => ::Hamster::Vector
    def callbacks_for(keys)
      immutable_keys = keys.is_a?(Symbol) ? keys : ::Hamster::Vector.new(keys)
      @callbacks.get(immutable_keys) || ::Hamster.vector
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
middleman-core-4.0.0.rc.1	lib/middleman-core/callback_manager.rb
middleman-core-4.0.0.beta.2	lib/middleman-core/callback_manager.rb