1

Pods/LookinServer/Src/Main/Server/Others/LKS_HierarchyDisplayItemsMaker.m

@@ -0,0 +1,162 @@
+#ifdef SHOULD_COMPILE_LOOKIN_SERVER 
+
+//
+//  LKS_HierarchyDisplayItemsMaker.m
+//  LookinServer
+//
+//  Created by Li Kai on 2019/2/19.
+//  https://lookin.work
+//
+
+#import "LKS_HierarchyDisplayItemsMaker.h"
+#import "LookinDisplayItem.h"
+#import "LKS_TraceManager.h"
+#import "LKS_AttrGroupsMaker.h"
+#import "LKS_EventHandlerMaker.h"
+#import "LookinServerDefines.h"
+#import "UIColor+LookinServer.h"
+#import "LKSConfigManager.h"
+#import "LKS_CustomAttrGroupsMaker.h"
+#import "LKS_CustomDisplayItemsMaker.h"
+#import "LKS_CustomAttrSetterManager.h"
+#import "LKS_MultiplatformAdapter.h"
+
+@implementation LKS_HierarchyDisplayItemsMaker
+
++ (NSArray<LookinDisplayItem *> *)itemsWithScreenshots:(BOOL)hasScreenshots attrList:(BOOL)hasAttrList lowImageQuality:(BOOL)lowQuality readCustomInfo:(BOOL)readCustomInfo saveCustomSetter:(BOOL)saveCustomSetter {
+    
+    [[LKS_TraceManager sharedInstance] reload];
+    
+    NSArray<UIWindow *> *windows = [LKS_MultiplatformAdapter allWindows];
+    NSMutableArray *resultArray = [NSMutableArray arrayWithCapacity:windows.count];
+    [windows enumerateObjectsUsingBlock:^(__kindof UIWindow * _Nonnull window, NSUInteger idx, BOOL * _Nonnull stop) {
+        LookinDisplayItem *item = [self _displayItemWithLayer:window.layer screenshots:hasScreenshots attrList:hasAttrList lowImageQuality:lowQuality readCustomInfo:readCustomInfo saveCustomSetter:saveCustomSetter];
+        item.representedAsKeyWindow = window.isKeyWindow;
+        if (item) {
+            [resultArray addObject:item];
+        }
+    }];
+    
+    return [resultArray copy];
+}
+
++ (LookinDisplayItem *)_displayItemWithLayer:(CALayer *)layer screenshots:(BOOL)hasScreenshots attrList:(BOOL)hasAttrList lowImageQuality:(BOOL)lowQuality readCustomInfo:(BOOL)readCustomInfo saveCustomSetter:(BOOL)saveCustomSetter {
+    if (!layer) {
+        return nil;
+    }
+    
+    LookinDisplayItem *item = [LookinDisplayItem new];
+    CGRect layerFrame = layer.frame;
+    UIView *hostView = layer.lks_hostView;
+    if (hostView && hostView.superview) {
+        layerFrame = [hostView.superview convertRect:layerFrame toView:nil];
+    }
+    if ([self validateFrame:layerFrame]) {
+        item.frame = layer.frame;
+    } else {
+        NSLog(@"LookinServer - The layer frame(%@) seems really weird. Lookin will ignore it to avoid potential render error in Lookin.", NSStringFromCGRect(layer.frame));
+        item.frame = CGRectZero;
+    }
+    item.bounds = layer.bounds;
+    if (hasScreenshots) {
+        item.soloScreenshot = [layer lks_soloScreenshotWithLowQuality:lowQuality];
+        item.groupScreenshot = [layer lks_groupScreenshotWithLowQuality:lowQuality];
+        item.screenshotEncodeType = LookinDisplayItemImageEncodeTypeNSData;
+    }
+    
+    if (hasAttrList) {
+        item.attributesGroupList = [LKS_AttrGroupsMaker attrGroupsForLayer:layer];
+        LKS_CustomAttrGroupsMaker *maker = [[LKS_CustomAttrGroupsMaker alloc] initWithLayer:layer];
+        [maker execute];
+        item.customAttrGroupList = [maker getGroups];
+        item.customDisplayTitle = [maker getCustomDisplayTitle];
+        item.danceuiSource = [maker getDanceUISource];
+    }
+    
+    item.isHidden = layer.isHidden;
+    item.alpha = layer.opacity;
+    item.layerObject = [LookinObject instanceWithObject:layer];
+    item.shouldCaptureImage = [LKSConfigManager shouldCaptureScreenshotOfLayer:layer];
+    
+    if (layer.lks_hostView) {
+        UIView *view = layer.lks_hostView;
+        item.viewObject = [LookinObject instanceWithObject:view];
+        item.eventHandlers = [LKS_EventHandlerMaker makeForView:view];
+        item.backgroundColor = view.backgroundColor;
+        
+        UIViewController* vc = [view lks_findHostViewController];
+        if (vc) {
+            item.hostViewControllerObject = [LookinObject instanceWithObject:vc];
+        }
+    } else {
+        item.backgroundColor = [UIColor lks_colorWithCGColor:layer.backgroundColor];
+    }
+    
+    if (layer.sublayers.count) {
+        NSArray<CALayer *> *sublayers = [layer.sublayers copy];
+        NSMutableArray<LookinDisplayItem *> *allSubitems = [NSMutableArray arrayWithCapacity:sublayers.count];
+        [sublayers enumerateObjectsUsingBlock:^(__kindof CALayer * _Nonnull sublayer, NSUInteger idx, BOOL * _Nonnull stop) {
+            LookinDisplayItem *sublayer_item = [self _displayItemWithLayer:sublayer screenshots:hasScreenshots attrList:hasAttrList lowImageQuality:lowQuality readCustomInfo:readCustomInfo saveCustomSetter:saveCustomSetter];
+            if (sublayer_item) {
+                [allSubitems addObject:sublayer_item];
+            }
+        }];
+        item.subitems = [allSubitems copy];
+    }
+    if (readCustomInfo) {
+        NSArray<LookinDisplayItem *> *customSubitems = [[[LKS_CustomDisplayItemsMaker alloc] initWithLayer:layer saveAttrSetter:saveCustomSetter] make];
+        if (customSubitems.count > 0) {
+            if (item.subitems) {
+                item.subitems = [item.subitems arrayByAddingObjectsFromArray:customSubitems];
+            } else {
+                item.subitems = customSubitems;
+            }
+        }        
+    }
+    
+    return item;
+}
+
++ (NSArray<LookinDisplayItem *> *)subitemsOfLayer:(CALayer *)layer {
+    if (!layer || layer.sublayers.count == 0) {
+        return @[];
+    }
+    [[LKS_TraceManager sharedInstance] reload];
+    
+    NSMutableArray<LookinDisplayItem *> *resultSubitems = [NSMutableArray array];
+
+    NSArray<CALayer *> *sublayers = [layer.sublayers copy];
+    [sublayers enumerateObjectsUsingBlock:^(__kindof CALayer * _Nonnull sublayer, NSUInteger idx, BOOL * _Nonnull stop) {
+        LookinDisplayItem *sublayer_item = [self _displayItemWithLayer:sublayer screenshots:NO attrList:NO lowImageQuality:NO readCustomInfo:YES saveCustomSetter:YES];
+        if (sublayer_item) {
+            [resultSubitems addObject:sublayer_item];
+        }
+    }];
+
+    NSArray<LookinDisplayItem *> *customSubitems = [[[LKS_CustomDisplayItemsMaker alloc] initWithLayer:layer saveAttrSetter:YES] make];
+    if (customSubitems.count > 0) {
+        [resultSubitems addObjectsFromArray:customSubitems];
+    }
+    
+    return resultSubitems;
+}
+
++ (BOOL)validateFrame:(CGRect)frame {
+    return !CGRectIsNull(frame) && !CGRectIsInfinite(frame) && ![self cgRectIsNaN:frame] && ![self cgRectIsInf:frame] && ![self cgRectIsUnreasonable:frame];
+}
+
++ (BOOL)cgRectIsNaN:(CGRect)rect {
+    return isnan(rect.origin.x) || isnan(rect.origin.y) || isnan(rect.size.width) || isnan(rect.size.height);
+}
+
++ (BOOL)cgRectIsInf:(CGRect)rect {
+    return isinf(rect.origin.x) || isinf(rect.origin.y) || isinf(rect.size.width) || isinf(rect.size.height);
+}
+
++ (BOOL)cgRectIsUnreasonable:(CGRect)rect {
+    return ABS(rect.origin.x) > 100000 || ABS(rect.origin.y) > 100000 || rect.size.width < 0 || rect.size.height < 0 || rect.size.width > 100000 || rect.size.height > 100000;
+}
+
+@end
+
+#endif /* SHOULD_COMPILE_LOOKIN_SERVER */