Update documentation for NSSet+without

mxcl · Apr 25, 2016 · 1b16b3c · 1b16b3c
1 parent 35d4708
commit 1b16b3c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/NSSet+without.h b/NSSet+without.h
@@ -3,8 +3,8 @@
 @interface NSSet (YOLOWithout)
 
 /**
- Returns a new array where objects in the given array are removed from
- the receiver.
+ Returns a new set where objects in the given set/array/object
+ are removed from the receiver.
 
     id rv = @[@1, @2, @3, @4, @5, @6].without(@2);
     // rv => @[@1, @3, @4, @5, @6]

diff --git a/NSSet+without.m b/NSSet+without.m
@@ -3,15 +3,15 @@
 @implementation NSSet (YOLOWithout)
 
 - (NSSet *(^)(id))without {
-    return ^NSSet *(id arrayOrSet) {
-        if (!arrayOrSet)
+    return ^NSSet *(id arrayOrSetOrObject) {
+        if (!arrayOrSetOrObject)
             return self;
 
-        id objs = [arrayOrSet isKindOfClass:[NSSet class]]
-                ? [arrayOrSet allObjects]
-                : [arrayOrSet isKindOfClass:[NSArray class]]
-                ? arrayOrSet
-                : @[arrayOrSet];
+        id objs = [arrayOrSetOrObject isKindOfClass:[NSSet class]]
+                ? [arrayOrSetOrObject allObjects]
+                : [arrayOrSetOrObject isKindOfClass:[NSArray class]]
+                ? arrayOrSetOrObject
+                : @[arrayOrSetOrObject];
         id rv = self.mutableCopy;
         [rv removeObjectsInArray:objs];
         return rv;