Added hook.before / hook.after for convenient function wrapping.

brianreavis · brianreavis · commit 70dcbcc4a8c0 · 2013-06-22T23:00:41.000-07:00
diff --git a/src/utils.js b/src/utils.js
@@ -10,6 +10,41 @@ var quoteRegExp = function(str) {
 	return (str + '').replace(/([.?*+^$[\]\\(){}|-])/g, '\\$1');
 };
 
+var hook = {};
+
+/**
+ * Wraps `method` on `self` so that `fn`
+ * is invoked before the original method.
+ *
+ * @param {object} self
+ * @param {string} method
+ * @param {function} fn
+ */
+hook.before = function(self, method, fn) {
+	var original = self[method];
+	self[method] = function() {
+		fn.apply(self, arguments);
+		return original.apply(self, arguments);
+	};
+};
+
+/**
+ * Wraps `method` on `self` so that `fn`
+ * is invoked after the original method.
+ *
+ * @param {object} self
+ * @param {string} method
+ * @param {function} fn
+ */
+hook.after = function(self, method, fn) {
+	var original = self[method];
+	self[method] = function() {
+		var result = original.apply(self, arguments);
+		fn.apply(self, arguments);
+		return result;
+	};
+};
+
 var once = function(fn) {
 	var called = false;
 	return function() {
