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_context_menu_delegate.h"
 #import "ios/web/public/web_state/ui/crw_native_content.h"
 #import "ios/web/public/web_view_creation_util.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;
@@ skipping 10 matching lines @@
   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 ()<CRWContextMenuDelegate,
+                                       WKNavigationDelegate> {
  @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 146 matching lines @@
                         (void (^)(WKNavigationActionPolicy))decisionHandler {
   decisionHandler(
       [self
           shouldStartLoadWithRequest:navigationAction.request
                        fromMainFrame:[navigationAction.sourceFrame isMainFrame]]
           ? WKNavigationActionPolicyAllow
           : WKNavigationActionPolicyCancel);
 }
 
 #pragma mark -
+#pragma mark CRWContextMenuDelegate implementation
+
+- (BOOL)webView:(WKWebView*)webView
+    handleContextMenu:(const web::ContextMenuParams&)params {
+  if ([delegate_
+          respondsToSelector:@selector(nativeContent:handleContextMenu:)]) {
+    return [delegate_ nativeContent:self handleContextMenu:params];
+  }
+  return NO;
+}
+
+#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 48 matching lines @@
   [components setScheme:@"file"];
   [components setHost:@""];
   [components setPath:path];
   resourceUrl_.reset([[components URL] retain]);
   resourcesRootDirectory_.reset([resourceUrl_ copy]);
   return resourceUrl_;
 }
 
 - (void)ensureWebViewCreated {
   if (!webView_) {
-    WKWebView* webView = web::BuildWKWebView(CGRectZero, browserState_);
+    WKWebView* webView = web::BuildWKWebViewWithCustomContextMenu(
+        CGRectZero, browserState_, self);
     [webView addObserver:self forKeyPath:@"title" options:0 context:nullptr];
     [webView setAutoresizingMask:UIViewAutoresizingFlexibleWidth |
                                  UIViewAutoresizingFlexibleHeight];
     [self loadWebViewContents:webView];
     [webView setNavigationDelegate:self];
     webView_.reset([webView retain]);
   }
 }
 
 - (void)loadWebViewContents:(WKWebView*)webView {
   if (!generator_) {
     [webView loadFileURL:[self resourceURL]
         allowingReadAccessToURL:resourcesRootDirectory_];
   } else {
     NSURL* resourceURL = [self resourceURL];
     [generator_ generateHtml:^(NSString* HTML) {
       [webView loadHTMLString:HTML baseURL:resourceURL];
     }];
   }
 }
 
 @end