#ifndef BAWKWebView_WebP_h
#define BAWKWebView_WebP_h

//#import "BAURLSessionProtocol.h"
#import "NSURLProtocol+BAWebView.h"

/*!
 *********************************************************************************
 ************************************ 更新说明 ************************************
 *********************************************************************************
 
 欢迎使用 BAHome 系列开源代码 ！
 如有更多需求，请前往：https://github.com/BAHome
 
 项目源码地址：
 OC 版 ：https://github.com/BAHome/BAWKWebView_WebP
 
 最新更新时间：2017-08-02 【倒叙】<br>
 最新Version：【Version：1.0.0】<br>
 更新内容：<br>
 1.0.0.1、用分类封装 WKWebView，代码无任何侵入更改<br>
 1.0.0.2、WKWebView 目前可以兼容 GIF 动图显示，和 webp 的静态图片显示（webp 的动态图片显示需要等待后期版本更新）<br>
 */

#endif /* BAWKWebView_WebP_h */

/****
 NSURLProtocol, UIWebView 直接就可以支持,但是 WKWebView 是不支持的,如何让 WKWebView 也支持 NSURLProtocol
 **/
#import <Foundation/Foundation.h>

@interface NSURLProtocol (BAWebView)

/**
 NSURLProtocol：registerScheme

 @param scheme 【http/https】
 */
+ (void)ba_web_registerScheme:(NSString*)scheme;

/**
 NSURLProtocol：webView销毁的时候注销Scheme

 @param scheme 【http/https】
 */
+ (void)ba_web_unregisterScheme:(NSString*)scheme;

@end

// 示例1：
- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    
    [self.view setBackgroundColor:[UIColor whiteColor]];
    [self ba_registerURLProtocol];
    [self.view addSubview:self.wkWebview];
    [self.wkWebview loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:self.urlString]]];
}

- (void)viewDidLayoutSubviews
{
    [super viewDidLayoutSubviews];
    self.wkWebview.frame = CGRectMake(0, 0, BAKit_SCREEN_WIDTH, BAKit_SCREEN_HEIGHT);
}

#pragma mark - 注册自定义 NSURLProtocol
- (void)ba_registerURLProtocol
{
    [NSURLProtocol registerClass:NSClassFromString(@"BAURLSessionProtocol")];
    // 注册registerScheme使得WKWebView支持NSURLProtocol
    [NSURLProtocol ba_web_registerScheme:@"http"];
    [NSURLProtocol ba_web_registerScheme:@"https"];
}

- (void)dealloc
{
    [NSURLProtocol unregisterClass:NSClassFromString(@"BAURLSessionProtocol")];
    // 移除 registerScheme
    [NSURLProtocol ba_web_unregisterScheme:@"http"];
    [NSURLProtocol ba_web_unregisterScheme:@"https"];
}

- (WKWebView *)wkWebview
{
    if (!_wkWebview)
    {
        _wkWebview = [[WKWebView alloc]initWithFrame:CGRectZero];
    }
    return _wkWebview;
}

其他示例可下载 demo 查看源码！