Expose thumbnails of pages to iOS share extensions.

ios/chrome/browser/ui/activity_services/chrome_activity_item_source.mm

 // 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/chrome/browser/ui/activity_services/chrome_activity_item_source.h"
 
 #import <MobileCoreServices/MobileCoreServices.h>
 
 #include "base/logging.h"
 #import "ios/chrome/browser/ui/activity_services/activity_type_util.h"
 #import "ios/chrome/browser/ui/activity_services/appex_constants.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
-#pragma mark - UIActivityURLSource
-
-@interface UIActivityURLSource () {
-  // The shared subject.
-  NSString* _subject;
-  // The shared url.
-  NSURL* _url;
-}
-
-@end
-
-@implementation UIActivityURLSource
-
-- (instancetype)init {
-  NOTREACHED();
-  return nil;
-}
-
-- (instancetype)initWithURL:(NSURL*)url subject:(NSString*)subject {
-  DCHECK(subject);
-  DCHECK(url);
-  self = [super init];
-  if (self) {
-    _subject = [subject copy];
-    _url = url;
-  }
-  return self;
-}
-
-#pragma mark - UIActivityItemSource
-
-- (id)activityViewController:(UIActivityViewController*)activityViewController
-         itemForActivityType:(NSString*)activityType {
-  return _url;
-}
-
-- (id)activityViewControllerPlaceholderItem:
-    (UIActivityViewController*)activityViewController {
-  return _url;
-}
-
-- (NSString*)activityViewController:
-                 (UIActivityViewController*)activityViewController
-             subjectForActivityType:(NSString*)activityType {
-  return _subject;
-}
-
-@end
-
 #pragma mark - UIActivityTextSource
 
 @interface UIActivityTextSource () {
   // The shared text.
   NSString* _text;
 }
 
 @end
 
 @implementation UIActivityTextSource
@@ skipping 74 matching lines @@
   return _image;
 }
 
 @end
 
 #pragma mark - UIActivityFindLoginActionSource
 
 @interface UIActivityFindLoginActionSource () {
   NSString* _subject;
   NSURL* _url;
+  ThumbnailGenerator _thumbnailGenerator;
 }
 @end
 
 @implementation UIActivityFindLoginActionSource
 
 - (instancetype)init {
   NOTREACHED();
   return nil;
 }
 
-- (instancetype)initWithURL:(NSURL*)url subject:(NSString*)subject {
+- (instancetype)initWithURL:(NSURL*)url
+                    subject:(NSString*)subject
+         thumbnailGenerator:(ThumbnailGenerator)thumbnailGenerator {
   DCHECK(url);
   DCHECK(subject);
+  DCHECK(thumbnailGenerator);
   self = [super init];
   if (self) {
     _url = url;
     _subject = [subject copy];
+    _thumbnailGenerator = thumbnailGenerator;
   }
   return self;
 }
 
 #pragma mark - UIActivityItemSource
 
 - (id)activityViewControllerPlaceholderItem:
     (UIActivityViewController*)activityViewController {
   // Return the current URL as a placeholder
   return _url;
@@ skipping 45 matching lines @@
   // after user made a choice of which AppEx to run, this method may be called
   // with |activityType| equals to the bundle ID of the AppEx selected.
   // Default action is to return @"public.url" UTType.
   if (!activityType ||
       activity_type_util::PasswordAppExActivityVersion(activityType)) {
     return findLoginType;
   }
   return (NSString*)kUTTypeURL;
 }
 
+- (UIImage*)activityViewController:
+                (UIActivityViewController*)activityViewController
+     thumbnailImageForActivityType:(UIActivityType)activityType
+                     suggestedSize:(CGSize)size {
+  return _thumbnailGenerator(size);

[Olivier, 2017/01/20 20:41:59]

What happens if the block is nil?
(it may work... I don't really know).

[jif, 2017/01/24 10:29:50]

_thumbnailGenerator can't be nil (see DCHECK in initializer, but don't see
comment on initializer...).

+}
+
 @end