79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
69 lines
2.2 KiB
Plaintext
69 lines
2.2 KiB
Plaintext
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#if FB_SONARKIT_ENABLED
|
|
|
|
#import "SKDescriptorMapper.h"
|
|
|
|
#import <FlipperKitLayoutIOSDescriptors/SKApplicationDescriptor.h>
|
|
#import <FlipperKitLayoutIOSDescriptors/SKButtonDescriptor.h>
|
|
#import <FlipperKitLayoutIOSDescriptors/SKScrollViewDescriptor.h>
|
|
#import <FlipperKitLayoutIOSDescriptors/SKViewControllerDescriptor.h>
|
|
#import <FlipperKitLayoutIOSDescriptors/SKViewDescriptor.h>
|
|
|
|
@implementation SKDescriptorMapper {
|
|
NSMutableDictionary<NSString*, SKNodeDescriptor*>* _descriptors;
|
|
}
|
|
|
|
- (instancetype)initWithDefaults {
|
|
if (self = [super init]) {
|
|
_descriptors = [NSMutableDictionary new];
|
|
|
|
[self registerDescriptor:[[SKApplicationDescriptor alloc]
|
|
initWithDescriptorMapper:self]
|
|
forClass:[UIApplication class]];
|
|
[self registerDescriptor:[[SKViewControllerDescriptor alloc]
|
|
initWithDescriptorMapper:self]
|
|
forClass:[UIViewController class]];
|
|
[self registerDescriptor:[[SKScrollViewDescriptor alloc]
|
|
initWithDescriptorMapper:self]
|
|
forClass:[UIScrollView class]];
|
|
[self registerDescriptor:[[SKButtonDescriptor alloc]
|
|
initWithDescriptorMapper:self]
|
|
forClass:[UIButton class]];
|
|
[self registerDescriptor:[[SKViewDescriptor alloc]
|
|
initWithDescriptorMapper:self]
|
|
forClass:[UIView class]];
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (SKNodeDescriptor*)descriptorForClass:(Class)cls {
|
|
SKNodeDescriptor* classDescriptor = nil;
|
|
|
|
while (classDescriptor == nil && cls != nil) {
|
|
classDescriptor = [_descriptors objectForKey:NSStringFromClass(cls)];
|
|
cls = [cls superclass];
|
|
}
|
|
|
|
return classDescriptor;
|
|
}
|
|
|
|
- (void)registerDescriptor:(SKNodeDescriptor*)descriptor forClass:(Class)cls {
|
|
NSString* className = NSStringFromClass(cls);
|
|
_descriptors[className] = descriptor;
|
|
}
|
|
|
|
- (NSArray<SKNodeDescriptor*>*)allDescriptors {
|
|
return [_descriptors allValues];
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|