Contents

require 'alki/support'
require 'alki/dsl/registry'

module Alki
  module Reload
    class Unloader
      def initialize(handlers,whitelist=[])
        @handlers = handlers
        @whitelist = whitelist
      end

      def find_unloadable
        files = []
        consts = []
        $LOADED_FEATURES.each do |path|
          if path.end_with?('.rb')
            result = @handlers.lazy.map{|h| h.handle_path path }.find{|r| r != nil}
            if result
              add_const consts, *result
              files << path
            end
          end
        end
        unless files.empty?
          {files: files, consts: consts}
        end
      end

      def unload(files:, consts:)
        $LOADED_FEATURES.delete_if {|f| files.include? f}
        consts.each {|(parent,const)| parent.send :remove_const, const}
      end

      private

      def add_const(consts,parent,name)
        unless whitelisted? parent, name
          parent = parent ? Alki::Support.load_class(parent) : Object
          name = Alki::Support.classify(name).to_sym
          if parent && parent.is_a?(Module) && parent.const_defined?(name,false)
            consts << [parent,name]
            true
          end
        end
      end

      def whitelisted?(parent,name)
        @whitelist.include?(parent ? File.join(parent,name) : name)
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
alki-reload-0.1.0	lib/alki/reload/unloader.rb