Hippy/ios/sdk/debug/devtools/inspector/model/HippyPageModel.m

/*!
* iOS SDK
*
* Tencent is pleased to support the open source community by making
* Hippy available.
*
* Copyright (C) 2019 THL A29 Limited, a Tencent company.
* All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*   http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#import "HippyPageModel.h"
#import "HippyLog.h"
#import "HippyUIManager.h"

NSString *const HippyPageKeyFormat = @"format";
NSString *const HippyPageKeyQuality = @"quality";
NSString *const HippyPageKeyMaxWidth = @"maxWidth";
NSString *const HippyPageKeyMaxHeight = @"maxHeight";
NSString *const HippyPageKeyOffsetTop = @"offsetTop";
NSString *const HippyPageKeyPageScaleFactor = @"pageScaleFactor";
NSString *const HippyPageKeyDeviceWidth = @"deviceWidth";
NSString *const HippyPageKeyDeviceHeight = @"deviceHeight";
NSString *const HippyPageKeyScrollOffsetX = @"scrollOffsetX";
NSString *const HippyPageKeyScrollOffsetY = @"scrollOffsetY";
NSString *const HippyPageKeyTimestamp = @"timestamp";
NSString *const HippyPageKeyData = @"data";
NSString *const HippyPageKeyMetaData = @"metadata";
NSString *const HippyPageKeySessionId = @"sessionId";

NSString *const HippyPageImageFormatPNG = @"png";
NSString *const HippyPageImageFormatJPEG = @"jpeg";

@interface HippyPageModel ()

@property (nonatomic, copy) NSString *format;
@property (nonatomic, assign) CGSize maxSize;
@property (nonatomic, assign) NSInteger quality;
@property (nonatomic, assign) NSTimeInterval lastTimestamp;
@property (atomic, assign) BOOL needScreenCast;

@end

@implementation HippyPageModel

- (instancetype)init {
    self = [super init];
    if (self) {
        self.maxSize = CGSizeMake(0, 0);
        self.format = HippyPageImageFormatJPEG;
        self.needScreenCast = NO;
    }
    return self;
}

- (BOOL)startScreenCastWithUIManager:(HippyUIManager *)manager
                              params:(NSDictionary *)params
                          completion:(void (^)(NSDictionary *rspObject))completion {
    self.needScreenCast = YES;
    if (params.count > 0) {
        self.format = [params objectForKey:HippyPageKeyFormat] ? params[HippyPageKeyFormat] : HippyPageImageFormatJPEG;
        self.quality = [params objectForKey:HippyPageKeyQuality] ? [params[HippyPageKeyQuality] integerValue] : 0;
        CGFloat maxWidth = [params objectForKey:HippyPageKeyMaxWidth] ? [params[HippyPageKeyMaxWidth] doubleValue] : 0;
        CGFloat maxHeight = [params objectForKey:HippyPageKeyMaxHeight] ? [params[HippyPageKeyMaxHeight] doubleValue] : 0;
        self.maxSize = CGSizeMake(maxWidth, maxHeight);
    }
    return [self screencastJSONWithManager:manager completion:completion];
}

- (void)stopScreenCastWithUIManager:(HippyUIManager *)manager {
    self.needScreenCast = NO;
}

- (BOOL)screencastFrameAckWithUIManager:(HippyUIManager *)manager
                                 params:(NSDictionary *)params
                             completion:(void (^)(NSDictionary *rspObject))completion {
    if (!completion || !manager) {
        return NO;
    }
    NSTimeInterval sessionId = [params[HippyPageKeySessionId] doubleValue];
    if (!self.needScreenCast || sessionId != self.lastTimestamp) {
        completion(@{});
        return NO;
    }
    return [self screencastJSONWithManager:manager completion:completion];
}

- (BOOL)screencastJSONWithManager:(HippyUIManager *)manager
                       completion:(void (^)(NSDictionary *rspObject))completion {
    if (!completion) {
        return NO;
    }
    if (!manager) {
        HippyLogWarn(@"PageModel, screen cast, manager is nil");
        return NO;
    }
    dispatch_async(dispatch_get_main_queue(), ^{
        UIView *rootView = [manager viewForHippyTag:[manager rootHippyTag]];
        if (!rootView) {
            HippyLogWarn(@"PageModel, screen cast, root view is nil");
            completion(@{});
            return;
        }
        CGFloat viewWidth = rootView.frame.size.width;
        CGFloat viewHeight = rootView.frame.size.height;
        CGFloat scale = 1.f;
        if (viewWidth != 0 && viewHeight != 0) {
            CGFloat scaleX = self.maxSize.width / viewWidth;
            CGFloat scaleY = self.maxSize.height / viewHeight;
            scale = MIN(scaleX, scaleY);
        } else {
            HippyLogWarn(@"PageModel, screen cast, root view size is 0");
            completion(@{});
            return;
        }
        // root view snapshot
        UIGraphicsImageRendererFormat *format = [UIGraphicsImageRendererFormat preferredFormat];
        format.scale = scale;
        UIGraphicsImageRenderer *renderer = [[UIGraphicsImageRenderer alloc] initWithSize:rootView.frame.size format:format];
        UIImage *resultImage = [renderer imageWithActions:^(UIGraphicsImageRendererContext * _Nonnull context) {
            [rootView drawViewHierarchyInRect:rootView.bounds afterScreenUpdates:YES];
        }];
        
        if (resultImage) {
            NSTimeInterval timestamp = [[NSDate date] timeIntervalSince1970];
            NSString *base64String = [self imageToBase64String:resultImage format:self.format];
            NSMutableDictionary *meta = [NSMutableDictionary dictionary];
            meta[HippyPageKeyOffsetTop] = @(0);
            meta[HippyPageKeyPageScaleFactor] = @(1);
            meta[HippyPageKeyDeviceWidth] = @(viewWidth);
            meta[HippyPageKeyDeviceHeight] = @(viewHeight);
            meta[HippyPageKeyScrollOffsetX] = @(0);
            meta[HippyPageKeyScrollOffsetY] = @(0);
            meta[HippyPageKeyTimestamp] = @(timestamp);
            NSMutableDictionary *resultJSON = [NSMutableDictionary dictionary];
            resultJSON[HippyPageKeyData] = base64String != nil ? base64String : @"";
            resultJSON[HippyPageKeyMetaData] = meta;
            resultJSON[HippyPageKeySessionId] = @(timestamp);
            self.lastTimestamp = timestamp;
            completion(resultJSON);
        } else {
            completion(@{});
        }
    });
    return YES;
}

- (NSString *)imageToBase64String:(UIImage *)image
                           format:(NSString *)format {
    if (!image) {
        return @"";
    }
    NSString *resultFormat = format;
    if (resultFormat.length <= 0) {
        resultFormat = HippyPageImageFormatJPEG;
    }
    NSData *imageData = nil;
    if ([resultFormat caseInsensitiveCompare:HippyPageImageFormatJPEG] == NSOrderedSame) {
        imageData = UIImageJPEGRepresentation(image, 0.8f);
    } else if ([resultFormat caseInsensitiveCompare:HippyPageImageFormatPNG] == NSOrderedSame) {
        imageData = UIImagePNGRepresentation(image);
    }
    return [imageData base64EncodedStringWithOptions:NSDataBase64Encoding64CharacterLineLength];
}

@end