FLEX/Classes/GlobalStateExplorers/RuntimeBrowser/FLEXRuntimeKeyPath.m

//
//  FLEXRuntimeKeyPath.m
//  FLEX
//
//  Created by Tanner on 3/22/17.
//  Copyright © 2017 Tanner Bennett. All rights reserved.
//

#import "FLEXRuntimeKeyPath.h"
#include <dlfcn.h>

@interface FLEXRuntimeKeyPath () {
    NSString *flex_description;
}
@end

@implementation FLEXRuntimeKeyPath

+ (instancetype)empty {
    static FLEXRuntimeKeyPath *empty = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        FLEXSearchToken *any = FLEXSearchToken.any;

        empty = [self new];
        empty->_bundleKey = any;
        empty->flex_description = @"";
    });

    return empty;
}

+ (instancetype)bundle:(FLEXSearchToken *)bundle
                 class:(FLEXSearchToken *)cls
                method:(FLEXSearchToken *)method
            isInstance:(NSNumber *)instance
                string:(NSString *)keyPathString {
    FLEXRuntimeKeyPath *keyPath  = [self new];
    keyPath->_bundleKey = bundle;
    keyPath->_classKey  = cls;
    keyPath->_methodKey = method;

    keyPath->_instanceMethods = instance;

    // Remove irrelevant trailing '*' for equality purposes
    if ([keyPathString hasSuffix:@"*"]) {
        keyPathString = [keyPathString substringToIndex:keyPathString.length];
    }
    keyPath->flex_description = keyPathString;
    
    if (bundle.isAny && cls.isAny && method.isAny) {
        [self initializeWebKitLegacy];
    }

    return keyPath;
}

+ (void)initializeWebKitLegacy {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        void *handle = dlopen(
            "/System/Library/PrivateFrameworks/WebKitLegacy.framework/WebKitLegacy",
            RTLD_LAZY
        );
        void (*WebKitInitialize)() = dlsym(handle, "WebKitInitialize");
        if (WebKitInitialize) {
            NSAssert(NSThread.isMainThread,
                @"WebKitInitialize can only be called on the main thread"
            );
            WebKitInitialize();
        }
    });
}

- (NSString *)description {
    return flex_description;
}

- (NSUInteger)hash {
    return flex_description.hash;
}

- (BOOL)isEqual:(id)object {
    if ([object isKindOfClass:[FLEXRuntimeKeyPath class]]) {
        FLEXRuntimeKeyPath *kp = object;
        return [flex_description isEqualToString:kp->flex_description];
    }

    return NO;
}

@end