Add CWVUIDelegate with method to customize context menu.

ios/web_view/shell/shell_view_controller.m

 // Copyright 2014 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.
 
 #import "ios/web_view/shell/shell_view_controller.h"
 
+#import <MobileCoreServices/UTCoreTypes.h>

[Eugene But (OOO till 7-30), 2017/02/27 16:34:07]

<MobileCoreServices/MobileCoreServices.h>

[michaeldo, 2017/02/27 19:02:06]

Done.

+
 #import "ios/web_view/public/cwv.h"
+#import "ios/web_view/public/cwv_html_element.h"
 #import "ios/web_view/public/cwv_web_view.h"
 #import "ios/web_view/shell/translate_controller.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
 @interface ShellViewController ()
 // Container for |webView|.
 @property(nonatomic, strong) UIView* containerView;
@@ skipping 97 matching lines @@
                 action:@selector(stopLoading)
       forControlEvents:UIControlEventTouchUpInside];
 
   [_toolbar addSubview:back];
   [_toolbar addSubview:forward];
   [_toolbar addSubview:stop];
   [_toolbar addSubview:_field];
 
   self.webView = [CWV webViewWithFrame:[_containerView bounds]];
   [_webView setDelegate:self];
+  [_webView setUIDelegate:self];
   [_webView setAutoresizingMask:UIViewAutoresizingFlexibleWidth |
                                 UIViewAutoresizingFlexibleHeight];
   [_containerView addSubview:_webView];
 
   NSURLRequest* request = [NSURLRequest
       requestWithURL:[NSURL URLWithString:@"https://www.google.com/"]];
   [_webView loadRequest:request];
 }
 
 - (UIBarPosition)positionForBar:(id<UIBarPositioning>)bar {
@@ skipping 30 matching lines @@
 
 - (void)updateToolbar {
   // Do not update the URL if the text field is currently being edited.
   if ([_field isFirstResponder]) {
     return;
   }
 
   [_field setText:[[_webView visibleURL] absoluteString]];
 }
 
+#pragma mark CWVUIDelegate methods
+
+- (void)webView:(CWVWebView*)webView
+    runContextMenuWithTitle:(NSString*)menuTitle
+             forHTMLElement:(CWVHTMLElement*)element
+                     inView:(UIView*)view
+        userGestureLocation:(CGPoint)location {
+  if (element.hyperlink == nil) {
+    return;
+  }
+
+  UIAlertController* alert = [UIAlertController
+      alertControllerWithTitle:menuTitle
+                       message:nil
+                preferredStyle:UIAlertControllerStyleActionSheet];
+  alert.popoverPresentationController.sourceView = view;
+  alert.popoverPresentationController.sourceRect =
+      CGRectMake(location.x, location.y, 1.0, 1.0);
+
+  void (^handler)(UIAlertAction*) = ^(UIAlertAction* action) {

[Eugene But (OOO till 7-30), 2017/02/27 16:34:07]

nit: s/handler/copyHandler ?

[michaeldo, 2017/02/27 19:02:06]

Done.

+    NSDictionary* item = @{
+      (NSString*)(kUTTypeURL) : element.hyperlink,
+      (NSString*)(kUTTypeUTF8PlainText) : [[element.hyperlink absoluteString]
+          dataUsingEncoding:NSUTF8StringEncoding],
+    };
+    [[UIPasteboard generalPasteboard] setItems:@[ item ]];
+  };
+  [alert addAction:[UIAlertAction actionWithTitle:@"Copy Link"
+                                            style:UIAlertActionStyleDefault
+                                          handler:handler]];
+
+  [alert addAction:[UIAlertAction actionWithTitle:@"Cancel"
+                                            style:UIAlertActionStyleCancel
+                                          handler:nil]];
+
+  [self presentViewController:alert animated:YES completion:nil];
+}
+
 #pragma mark CWVWebViewDelegate methods
 
 - (void)webView:(CWVWebView*)webView
     didFinishLoadingWithURL:(NSURL*)url
                 loadSuccess:(BOOL)loadSuccess {
   // TODO(crbug.com/679895): Add some visual indication that the page load has
   // finished.
   [self updateToolbar];
 }
 
@@ skipping 12 matching lines @@
   }
 }
 
 - (id<CWVTranslateDelegate>)translateDelegate {
   if (!_translateController)
     self.translateController = [[TranslateController alloc] init];
   return _translateController;
 }
 
 @end