Initial

1 files changed, 131 insertions, 0 deletions

diff --git a/Firebase/Database/FImmutableSortedDictionary/FImmutableSortedDictionary/FImmutableSortedSet.m b/Firebase/Database/FImmutableSortedDictionary/FImmutableSortedDictionary/FImmutableSortedSet.m
new file mode 100644
index 0000000..09c4164
--- /dev/null
+++ b/Firebase/Database/FImmutableSortedDictionary/FImmutableSortedDictionary/FImmutableSortedSet.m
@@ -0,0 +1,131 @@
+/*
+ * Copyright 2017 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#import "FImmutableSortedSet.h"
+#import "FImmutableSortedDictionary.h"
+
+@interface FImmutableSortedSet ()
+
+@property (nonatomic, strong) FImmutableSortedDictionary *dictionary;
+
+@end
+
+@implementation FImmutableSortedSet
+
++ (FImmutableSortedSet *)setWithKeysFromDictionary:(NSDictionary *)dictionary comparator:(NSComparator)comparator
+{
+    FImmutableSortedDictionary *setDict = [FImmutableSortedDictionary fromDictionary:dictionary withComparator:comparator];
+    return [[FImmutableSortedSet alloc] initWithDictionary:setDict];
+}
+
+- (id)initWithDictionary:(FImmutableSortedDictionary *)dictionary
+{
+    self = [super init];
+    if (self != nil) {
+        self->_dictionary = dictionary;
+    }
+    return self;
+}
+
+- (BOOL)contains:(id)object
+{
+    return [self.dictionary contains:object];
+}
+
+- (FImmutableSortedSet *)addObject:(id)object
+{
+    FImmutableSortedDictionary *newDictionary = [self.dictionary insertKey:object withValue:[NSNull null]];
+    if (newDictionary != self.dictionary) {
+        return [[FImmutableSortedSet alloc] initWithDictionary:newDictionary];
+    } else {
+        return self;
+    }
+}
+
+- (FImmutableSortedSet *)removeObject:(id)object
+{
+    FImmutableSortedDictionary *newDictionary = [self.dictionary removeObjectForKey:object];
+    if (newDictionary != self.dictionary) {
+        return [[FImmutableSortedSet alloc] initWithDictionary:newDictionary];
+    } else {
+        return self;
+    }
+}
+
+- (BOOL)containsObject:(id)object
+{
+    return [self.dictionary contains:object];
+}
+
+- (id)firstObject
+{
+    return [self.dictionary minKey];
+}
+
+- (id)lastObject
+{
+    return [self.dictionary maxKey];
+}
+
+- (id)predecessorEntry:(id)entry
+{
+    return [self.dictionary getPredecessorKey:entry];
+}
+
+- (NSUInteger)count
+{
+    return [self.dictionary count];
+}
+
+- (BOOL)isEmpty
+{
+    return [self.dictionary isEmpty];
+}
+
+- (void)enumerateObjectsUsingBlock:(void (^)(id, BOOL *))block
+{
+    [self enumerateObjectsReverse:NO usingBlock:block];
+}
+
+- (void)enumerateObjectsReverse:(BOOL)reverse usingBlock:(void (^)(id, BOOL *))block
+{
+    [self.dictionary enumerateKeysAndObjectsReverse:reverse usingBlock:^(id key, id value, BOOL *stop) {
+        block(key, stop);
+    }];
+}
+
+- (NSEnumerator *)objectEnumerator
+{
+    return [self.dictionary keyEnumerator];
+}
+
+- (NSString *)description
+{
+    NSMutableString *str = [[NSMutableString alloc] init];
+    __block BOOL first = YES;
+    [str appendString:@"FImmutableSortedSet ( "];
+    [self enumerateObjectsUsingBlock:^(id obj, BOOL *stop) {
+        if (!first) {
+            [str appendString:@", "];
+        }
+        first = NO;
+        [str appendString:[NSString stringWithFormat:@"%@", obj]];
+    }];
+    [str appendString:@" )"];
+    return str;
+}
+
+@end