Reuse context menu in StaticHTMLViewController

ios/chrome/browser/ui/static_content/static_html_view_controller.mm

 // Copyright 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "ios/chrome/browser/ui/static_content/static_html_view_controller.h"
 
 #import <WebKit/WebKit.h>
 
 #include <stdlib.h>
 
 #include "base/logging.h"
 #include "base/mac/bundle_locations.h"
 #include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsobject.h"
 #include "ios/web/public/referrer.h"
 #import "ios/web/public/web_state/ui/crw_native_content.h"
 #import "ios/web/public/web_view_creation_util.h"
+#import "ios/web/public/web_state/ui/crw_context_menu_controller.h"
 #import "net/base/mac/url_conversions.h"
 #include "ui/base/page_transition_types.h"
 #include "ui/base/resource/resource_bundle.h"
 
 @implementation IdrHtmlGenerator
 - (id)initWithResourceId:(int)resourceId encoding:(NSStringEncoding)encoding {
   if ((self = [super init])) {
     resourceId_ = resourceId;
     encoding_ = encoding;
   }
   return self;
 }
 
 - (id)initWithResourceId:(int)resourceId {
   return [self initWithResourceId:resourceId encoding:NSUTF8StringEncoding];
 }
 
 - (void)generateHtml:(HtmlCallback)callback {
   base::StringPiece html(
       ResourceBundle::GetSharedInstance().GetRawDataResource(resourceId_));
   base::scoped_nsobject<NSString> result([[NSString alloc]
       initWithBytes:html.data()
              length:html.size()
            encoding:encoding_]);
   callback(result);
 }
 @end
 
-@interface StaticHtmlViewController ()<WKNavigationDelegate> {
+@interface StaticHtmlViewController ()<CRWContextMenuControllerWebView,
+                                       WKNavigationDelegate,
+                                       UIGestureRecognizerDelegate> {

[jif, 2017/01/12 17:13:12]

remove UIGestureRecognizerDelegate?

[Olivier, 2017/01/13 10:20:25]

Done.

  @private
   // The referrer that will be passed when navigating from this page.
   web::Referrer referrer_;
 
   // The URL that will be passed to the web page when loading.
   // If the page is displaying a local HTML file, it contains the file URL to
   // the file.
   // If the page is a generated HTML, it contains a random resource URL.
   base::scoped_nsobject<NSURL> resourceUrl_;
 
@@ skipping 12 matching lines @@
   // Browser state associated with the view controller, used to create the
   // WKWebView.
   web::BrowserState* browserState_;  // Weak.
 
   // The web view that is used to display the content.
   base::scoped_nsobject<WKWebView> webView_;
 
   // The delegate of the native content.
   id<CRWNativeContentDelegate> delegate_;  // weak
 
+  // The controller that will recognize the long press and trigger the context
+  // menu.
+  base::scoped_nsobject<CRWContextMenuController> contextMenuController_;
+
   // The loader to navigate from the page.
   id<UrlLoader> loader_;  // weak
 }
 
 // Returns the URL of the static page to display.
 - (NSURL*)resourceURL;
 // Ensures that webView_ has been created, creating it if necessary.
 - (void)ensureWebViewCreated;
 // Determines if the page load should begin based on the current |resourceURL|
 // and if the request is issued by the main frame (|fromMainFrame|).
@@ skipping 20 matching lines @@
   DCHECK(browserState);
   if ((self = [super init])) {
     generator_.reset([generator retain]);
     browserState_ = browserState;
   }
   return self;
 }
 
 - (instancetype)initWithFileURL:(const GURL&)URL
         allowingReadAccessToURL:(const GURL&)resourcesRoot
-                   browserState:(web::BrowserState*)browserState {
+                   browserState:(web::BrowserState*)browserState
+            contextMenuDelegate:
+                (id<CRWContextMenuControllerDelegate>)contextMenuDelegate {
   DCHECK(URL.is_valid());
   DCHECK(URL.SchemeIsFile());
   DCHECK(browserState);
   if ((self = [super init])) {
     resourceUrl_.reset([net::NSURLWithGURL(URL) retain]);
     resourcesRootDirectory_.reset([net::NSURLWithGURL(resourcesRoot) retain]);
     browserState_ = browserState;
+    if (contextMenuDelegate) {
+      contextMenuController_.reset([[CRWContextMenuController alloc]
+          initWithWebView:self
+                 delegate:contextMenuDelegate]);
+    }
   }
   return self;
 }
 
 - (void)dealloc {
   [self removeWebViewObservers];
   [super dealloc];
 }
 
 - (void)removeWebViewObservers {
@@ skipping 66 matching lines @@
                         (void (^)(WKNavigationActionPolicy))decisionHandler {
   decisionHandler(
       [self
           shouldStartLoadWithRequest:navigationAction.request
                        fromMainFrame:[navigationAction.sourceFrame isMainFrame]]
           ? WKNavigationActionPolicyAllow
           : WKNavigationActionPolicyCancel);
 }
 
 #pragma mark -
+#pragma mark CRWWKContextMenuWebView implementation
+- (UIView*)webScrollView {
+  return self.scrollView;
+}
+
+#pragma mark -
 #pragma mark KVO callback
 
 - (void)observeValueForKeyPath:(NSString*)keyPath
                       ofObject:(id)object
                         change:(NSDictionary*)change
                        context:(void*)context {
   DCHECK([keyPath isEqualToString:@"title"]);
   if ([delegate_ respondsToSelector:@selector(nativeContent:titleDidChange:)]) {
     // WKWebView's |title| changes to nil when its web process crashes.
     if ([webView_ title])
@@ skipping 71 matching lines @@
         allowingReadAccessToURL:resourcesRootDirectory_];
   } else {
     NSURL* resourceURL = [self resourceURL];
     [generator_ generateHtml:^(NSString* HTML) {
       [webView loadHTMLString:HTML baseURL:resourceURL];
     }];
   }
 }
 
 @end