Pastie/NSArray+Map.m at master · NSCake/Pastie · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
//
//  NSArray+Map.m
//  Pastie
//
//  Created by Tanner Bennett on 2021-05-11
//  Copyright © 2021 Tanner Bennett. All rights reserved.
//

#import "NSArray+Map.h"

@implementation NSArray (Functional)

- (__kindof NSMutableArray *)pastie_mapped:(id (^)(id, NSUInteger))mapFunc {
    NSMutableArray *map = [NSMutableArray new];
    [self enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
        id ret = mapFunc(obj, idx);
        if (ret) {
            [map addObject:ret];
        }
    }];

    return map;
}

+ (__kindof NSArray *)pastie_forEachUpTo:(NSUInteger)bound map:(id(^)(NSUInteger))block {
    NSMutableArray *array = [NSMutableArray new];
    for (NSUInteger i = 0; i < bound; i++) {
        id obj = block(i);
        if (obj) {
            [array addObject:obj];
        }
    }

    return array;
}

- (NSMutableArray *)pastie_filtered:(BOOL (^)(id, NSUInteger))filterFunc {
    return [self pastie_mapped:^id(id obj, NSUInteger idx) {
        return filterFunc(obj, idx) ? obj : nil;
    }];
}

- (id)pastie_firstWhere:(BOOL (^)(id, NSUInteger))predicate {
    return [self pastie_filtered:predicate].firstObject;
}

@end