iOS实战：实现网页任意位置的截屏

最近，在做网页截图需求时，发现了一个坑。

产品的要求是客户端可以随时截取网页置顶时的图片，即使当前的网页已经滑到底部了。然而，在实现的时候，发现UIWevView是可以实现的，而WKWebView则会出现截屏为空白或者一部分为空白的问题。

不着急，在说解决方案之前，我先把网页截图相关的需求都记录一边，需要的可以自取哈。

WebView的当前位置截图实现

UIWebView

虽然现在UIWebView已经被苹果废弃，但是考虑到很多app还有在用，因此也一起记录一下。

- (UIImage *)shotCurrentUIWebView:(UIWebView *)webView {
    UIGraphicsBeginImageContextWithOptions(webView.frame.size, 1, 2);
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGContextSaveGState(ctx);
    
    [webView.layer renderInContext:ctx];
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    CGContextRestoreGState(ctx);
    UIGraphicsEndImageContext();
    return image;
}

WKWebView

WKWebView是苹果在iOS 8之后推出的框架WebKit中的浏览器控件, 其加载速度比UIWebView快了许多, 但内存占用率却下降很多, 也解决了加载网页时的内存泄露问题。然而，事情并没有这么简单。。。

不过，当前位置的网页截屏还是可以用UIWebView那一套的，因此不重复写。

WebView的置顶位置截图实现

首先再明确一下需求，即使当前网页已经滚动到最底部，都要截取置顶时的网页，同时让用户无感知。

UIWebView

首先还是UIWebView的实现，UIWebView的实现比较简单，你第一时间想到的方案就是实现的方案。

具体思路是，当需要截图时，先保存当前网页的偏移量contentOffset，然后将网页置顶，使用上面的方法截图，然后设置网页刚保存的偏移量。

根据思路，代码如下：

- (UIImage *)shotTopUIWebView:(UIWebView *)webView {
    UIGraphicsBeginImageContextWithOptions(webView.frame.size, 1, 2);
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGContextSaveGState(ctx);
    
    CGPoint currentOffset = webView.scrollView.contentOffset;
    webView.scrollView.contentOffset = CGPointZero;
    [webView.layer renderInContext:ctx];
    webView.scrollView.contentOffset = currentOffset;
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    CGContextRestoreGState(ctx);
    UIGraphicsEndImageContext();
    return image;
}

效果如下：


UIWebView置顶截图效果.gif

demo中点击截图按钮后，会调用-(UIImage *)shotTopUIWebView:(UIWebView *)webView方法截取顶部图片，同时push到下一个控制器中，使用UIImageView展示出来。

WKWebView

UIWebView的实现比较简单，都是在主线程中同步完成。

下面尝试用同样的方法实现WKWebView的置顶截图。然而，发现效果和想象的不太一样：（我在第一个界面出现的时候延迟两秒进行了loadRequest处理）

WKWebView置顶截图效果1.gif

发现如果网页只是往下滑了一点，会截取一部分网页，如果偏移量比较大，截图全是空白。

为了解决这个问题，我上网查了下，网上推荐使用-(BOOL)drawViewHierarchyInRect:(CGRect)rect afterScreenUpdates:(BOOL)afterUpdates NS_AVAILABLE_IOS(7_0);方法，看方法名貌似可以实现，但是还是没有解决。

然后我开始猜测，是不是在网页还没滚动到顶部的时候，就已经开始截图，或者说在截图的时候，网页还没绘制完毕。

我在截屏的那行代码处打了断点，然后预览滚动到顶部后的网页，发现：

WKWebView断点截屏.png

果然是网页没有绘制完毕。

那么要解决这个问题，就得做个延时处理，比方说，设置网页置顶后，延时0.1秒截图。但是一旦这个做，用户就能感知到网页置顶了。有什么办法能让用户无感知的置顶然后延时截图呢？

方法就是在置顶之前，先在网页上放一个和当前网页一模一样的界面欺骗用户，然后置顶网页，延时截图，截完图后回归位置，然后移除顶部的视图。

按照逻辑，代码如下：

- (void)shotTopWKWebView:(WKWebView *)webView compete:(void (^)(UIImage *shotImage))compete {
    UIGraphicsBeginImageContextWithOptions(webView.frame.size, 1, 2);
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGContextSaveGState(ctx);
    
    UIView *tmpView = [webView snapshotViewAfterScreenUpdates:YES];
    tmpView.frame = webView.frame;
    [webView.superview addSubview:tmpView];
    
    CGPoint currentOffset = webView.scrollView.contentOffset;
    webView.scrollView.contentOffset = CGPointZero;
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        [webView drawViewHierarchyInRect:webView.bounds afterScreenUpdates:YES];
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        CGContextRestoreGState(ctx);
        UIGraphicsEndImageContext();
        webView.scrollView.contentOffset = currentOffset;
        [tmpView removeFromSuperview];
        compete(image);
    });
}

效果如下：（为了更好的展现问题，我在截取到图片后，延迟3秒再push下一界面）

WKWebView置顶截图效果2.gif

很明显，截图后，界面闪了一下，原因是[tmpView removeFromSuperview];。

解决方法：延迟0.1秒后移除。

- (void)shotTopWKWebView:(WKWebView *)webView compete:(void (^)(UIImage *shotImage))compete {
    UIGraphicsBeginImageContextWithOptions(webView.frame.size, 1, 2);
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGContextSaveGState(ctx);
    
    UIView *tmpView = [webView snapshotViewAfterScreenUpdates:YES];
    tmpView.frame = webView.frame;
    [webView.superview addSubview:tmpView];
    
    CGPoint currentOffset = webView.scrollView.contentOffset;
    webView.scrollView.contentOffset = CGPointZero;
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        [webView drawViewHierarchyInRect:webView.bounds afterScreenUpdates:YES];
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        CGContextRestoreGState(ctx);
        UIGraphicsEndImageContext();
        webView.scrollView.contentOffset = currentOffset;
        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
            [tmpView removeFromSuperview];
        });
        compete(image);
    });
}

一劳永逸

typedef NS_ENUM(NSInteger, WebViewShotType) {
    WebViewShotTypeTop = 0,
    WebViewShotTypeCurrent,
    WebViewShotTypeCustom,
};
- (void)shotWebView:(id)webView shotType:(WebViewShotType)shotType Offset:(CGPoint)offset compete:(void (^)(UIImage *shotImage))compete {
    if (!compete) {
        return;
    }
    
    if (![webView isKindOfClass:[UIWebView class]] && ![webView isKindOfClass:[WKWebView class]]) {
        compete(nil);
    }
    
    UIGraphicsBeginImageContextWithOptions(((UIView *)webView).frame.size, 1, 2);
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGContextSaveGState(ctx);
    if ([webView isKindOfClass:[UIWebView class]]) {
        UIWebView *wv = (UIWebView *)webView;
        if (shotType == WebViewShotTypeCurrent) {
            [wv.layer renderInContext:ctx];
        } else {
            CGPoint currentOffset = wv.scrollView.contentOffset;
            wv.scrollView.contentOffset = shotType == WebViewShotTypeTop ? CGPointZero : offset;
            [wv.layer renderInContext:ctx];
            wv.scrollView.contentOffset = currentOffset;
        }
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        CGContextRestoreGState(ctx);
        UIGraphicsEndImageContext();
        compete(image);
    } else {
        WKWebView *wv = (WKWebView *)webView;
        if (shotType == WebViewShotTypeCurrent) {
            [wv drawViewHierarchyInRect:wv.bounds afterScreenUpdates:YES];
            UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
            CGContextRestoreGState(ctx);
            UIGraphicsEndImageContext();
            compete(image);
        } else {
            UIView *tmpView = [wv snapshotViewAfterScreenUpdates:YES];
            tmpView.frame = wv.frame;
            [wv.superview addSubview:tmpView];
            
            CGPoint currentOffset = wv.scrollView.contentOffset;
            wv.scrollView.contentOffset = shotType == WebViewShotTypeTop ? CGPointZero : offset;
            dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
                [webView drawViewHierarchyInRect:wv.bounds afterScreenUpdates:YES];
                UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
                CGContextRestoreGState(ctx);
                UIGraphicsEndImageContext();
                wv.scrollView.contentOffset = currentOffset;
                dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
                    [tmpView removeFromSuperview];
                });
                compete(image);
            });
        }
    }
}

最后，我的博客地址：http://jabberyq.top/

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 227,022评论 6赞 528
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 97,760评论 3赞 412
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 174,687评论 0赞 373
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 62,318评论 1赞 307
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 71,112评论 6赞 405
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 54,645评论 1赞 320
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 42,724评论 3赞 435
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 41,885评论 0赞 285
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 48,387评论 1赞 330
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 40,368评论 3赞 354
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 42,506评论 1赞 365
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 38,063评论 5赞 355
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 43,759评论 3赞 343
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 34,150评论 0赞 25
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 35,414评论 1赞 281
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 51,080评论 3赞 386
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 47,494评论 2赞 370