Injecting block in Objective C Methods using RUNTIME

blockInjection

#import <BILib.h>
#import <objc/runtime.h>

// customClass.m
// under @implementation
//override
+(void)load
{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        
        // Iterate over the class and all superclasses
        Class currentClass = [self class];
        NSString *className = [NSString stringWithUTF8String:class_getName(currentClass)];
        // Iterate over all instance methods for this class
        unsigned int methodCount;
        Method *methodList = class_copyMethodList(currentClass, &methodCount);
        unsigned int i = 0;
        for (; i < methodCount; i++) {
            NSString *methodName = [NSString stringWithUTF8String:sel_getName(method_getName(methodList[i]))];
            [BILib injectToClassWithName:className methodName:methodName preprocess:^{
                assert(className);
                assert(methodName);
                NSLog(@"Running %@ '%@'", className,methodName);
                
            }];
        }
        free(methodList);
    });
}