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

[Eugene But (OOO till 7-30), 2017/01/13 16:19:41]

nit: Add a linebreak ?

[Olivier, 2017/01/13 18:08:58]

Done.

+- (BOOL)webView:(WKWebView*)webView
+    handleContextMenu:(const web::ContextMenuParams&)params {
+  if ([delegate_
+          respondsToSelector:@selector(nativeContent:handleContextMenu:)]) {
+    return [delegate_ nativeContent:self handleContextMenu:params];

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

I kept the current behavior of the delegate to pass the StaticHtmlViewController
instead of the nativeContent.
The nativeContent is not used anyway, but if you think it is better to forward
the call to the actual nativeContent, I can do it.

[Eugene But (OOO till 7-30), 2017/01/13 16:19:41]

Looks consistent with -[CRWWebController nativeContent:titleDidChange:] and
cleaner than my original suggestion, so your change lg :)

+  }
+  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