stage.rb

=begin
# Staging Module
Allows for the configuration of execution stages, and provides code to be executed in
that stage. Node, code is executed in the current context. 

This is particularly useful for tasks that need to be performed at specific spots in 
the loading process. May also be useful for cleanup code.

## Usage
binding.stage :name do
  ... code ...
end

# Example
$result = []
class Ex
  val = 1
  $result.push(val)
  binding.stage(:after) {$result.push(val)}
  val = 2
end

Sereth::Stage.run_stage :after 

$result == [1, 2]
=end
require 'sourcify'

module Sereth
  module StageBindingExtender
    def run_stage(name)
      Sereth::Stage.run_stage(:sereth_util_loaded)
    end
  end

  module StageBinding
    # Convert the requested block to source code
    def stage(name, &block)
      Sereth::Stage.add_stage(name, self, block.to_source(:strip_enclosure => true))
    end

    # Extend target with the proper singleton methods
    def self.included(target)
      target.send(:extend, StageBindingExtender)
    end
  end

  class Stage
    @db = {}
    class << self
      def add_stage(name, target_binding, code)
        @db[name] ||= Stage.new(name)
        @db[name].add(target_binding, code)
      end

      def run_stage(name)
        return if [email protected]_key?(name)
        @db.delete(name).run
      end
    end

    def initialize(name)
      @name = name
      @targets = []
    end

    def add(target_binding, code)
      @targets.push([target_binding, code])
    end

    def run
      @targets.each {|target_binding, code| target_binding.eval(code)}
    end
  end
end


class Binding
  include Sereth::StageBinding
end