From 8ceb458445ada93ca43600eb63f338622ac072d6 Mon Sep 17 00:00:00 2001
From: Gabriel Horner <gabriel.horner@gmail.com>
Date: Sat, 16 Apr 2011 02:27:26 -0400
Subject: [PATCH] exchange wasteful module creation for a hash

---
 lib/watchdog.rb | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/lib/watchdog.rb b/lib/watchdog.rb
index a44218b..949a045 100644
--- a/lib/watchdog.rb
+++ b/lib/watchdog.rb
@@ -1,4 +1,8 @@
 module Watchdog
+  # Maps objects or modules to their extension modules
+  class <<self; attr_accessor :extensions; end
+  self.extensions = {}
+
   class Error < StandardError
     def initialize(meth, from, to)
       super "#{from} can't add method '#{meth}' to #{to}"
@@ -7,17 +11,20 @@ def initialize(meth, from, to)
   class ExtendError < Error; end
   class IncludeError < Error; end
 
-  def self.guard(mod, guarded)
-    guard_mod = Module.new { class << self; attr_accessor :existing; end }
-    guard_mod.existing = mod
-    guard_meth = guarded.is_a?(Module) ? :method_added : :singleton_method_added
-    guard_mod.send(:define_method, guard_meth) do |meth|
-      if guard_mod.existing.instance_methods.include?(meth)
-        raise Watchdog::Error.new(meth, self, mod)
+  module GermanShepard
+    [:singleton_method_added, :method_added].each do |m|
+      define_method(m) do |meth|
+        if Watchdog.extensions[self].instance_methods.include?(meth)
+          raise Watchdog::Error.new(meth, self, Watchdog.extensions[self])
+        end
+        super
       end
-      super
     end
-    guarded.extend guard_mod
+  end
+
+  def self.guard(mod, guarded)
+    extensions[guarded] = mod
+    guarded.extend GermanShepard
   end
 
   def append_features(mod)