From c37e8d365b9470a593668476fc9be0af6da2e812 Mon Sep 17 00:00:00 2001 From: David Heinemeier Hansson Date: Tue, 25 Jan 2005 17:55:55 +0000 Subject: Added methods for removing subclasses -- couldnt make it work with the regular nested-module approach (ObjectSpace was being difficult), so this is a straight inclusion git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@506 5ecf4fe2-1ee6-0310-87b1-e25e094e27de --- activesupport/lib/core_ext/object_and_class.rb | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 activesupport/lib/core_ext/object_and_class.rb (limited to 'activesupport/lib/core_ext') diff --git a/activesupport/lib/core_ext/object_and_class.rb b/activesupport/lib/core_ext/object_and_class.rb new file mode 100644 index 0000000000..59a463ae29 --- /dev/null +++ b/activesupport/lib/core_ext/object_and_class.rb @@ -0,0 +1,24 @@ +class Object #:nodoc: + def remove_subclasses_of(superclass) + subclasses_of(superclass).each { |subclass| Object.send(:remove_const, subclass) rescue nil } + end + + def subclasses_of(superclass) + subclasses = [] + ObjectSpace.each_object(Class) do |k| + next if !k.ancestors.include?(superclass) || superclass == k || k.to_s.include?("::") || subclasses.include?(k.to_s) + subclasses << k.to_s + end + subclasses + end +end + +class Class #:nodoc: + def remove_subclasses + Object.remove_subclasses_of(self) + end + + def subclasses + Object.subclasses_of(self) + end +end -- cgit v1.2.3