DownloadManagerController now uses WebState

ios/chrome/browser/ui/downloads/download_manager_controller.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/downloads/download_manager_controller.h"
 
 #include <stdint.h>
 #include <memory>
 
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/ios/weak_nsobject.h"
 #include "base/location.h"
 #include "base/mac/bind_objc_block.h"
 #include "base/mac/objc_property_releaser.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/memory/ref_counted.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/metrics/user_metrics.h"
 #include "base/metrics/user_metrics_action.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "components/strings/grit/components_strings.h"
 #import "ios/chrome/browser/installation_notifier.h"
 #include "ios/chrome/browser/native_app_launcher/ios_appstore_ids.h"
-#import "ios/chrome/browser/store_kit/store_kit_launcher.h"
+#import "ios/chrome/browser/store_kit/store_kit_tab_helper.h"
 #import "ios/chrome/browser/ui/alert_coordinator/alert_coordinator.h"
 #import "ios/chrome/browser/ui/colors/MDCPalette+CrAdditions.h"
 #import "ios/chrome/browser/ui/network_activity_indicator_manager.h"
 #include "ios/chrome/browser/ui/ui_util.h"
 #import "ios/chrome/browser/ui/uikit_ui_util.h"
 #include "ios/chrome/grit/ios_strings.h"
 #include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
 #import "ios/public/provider/chrome/browser/native_app_launcher/native_app_metadata.h"
 #import "ios/public/provider/chrome/browser/native_app_launcher/native_app_whitelist_manager.h"
 #import "ios/third_party/material_components_ios/src/components/ActivityIndicator/src/MaterialActivityIndicator.h"
 #import "ios/third_party/material_components_ios/src/components/Buttons/src/MaterialButtons.h"
 #import "ios/third_party/material_components_ios/src/components/Typography/src/MaterialTypography.h"
 #import "ios/third_party/material_roboto_font_loader_ios/src/src/MaterialRobotoFontLoader.h"
+#include "ios/web/public/browser_state.h"
+#include "ios/web/public/web_state/web_state.h"
 #include "ios/web/public/web_thread.h"
 #include "net/base/filename_util.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_util.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_status.h"
 #include "ui/base/l10n/l10n_util_mac.h"
 #import "ui/gfx/ios/uikit_util.h"
@@ skipping 317 matching lines @@
   // header field in the initial HEAD request. This is set to |kNoFileSizeGiven|
   // if the Content-Length header is not given.
   long long _totalFileSize;
 
   // YES if |_fileTypeLabel| is vertically centered in |_documentContainer|. NO
   // if |_fileTypeLabel| is lower to account for another view in
   // |_documentContainer|.
   BOOL _isFileTypeLabelCentered;
   BOOL _isDisplayingError;
   BOOL _didSuccessfullyFinishHeadFetch;
-  scoped_refptr<URLRequestContextGetter> _requestContextGetter;
+  // WebState provides access to the *TabHelper objects.
+  web::WebState* _webState;
   std::unique_ptr<URLFetcher> _fetcher;
   std::unique_ptr<DownloadHeadDelegate> _headFetcherDelegate;
   std::unique_ptr<DownloadContentDelegate> _contentFetcherDelegate;
-  base::WeakNSProtocol<id<StoreKitLauncher>> _storeKitLauncher;
   base::FilePath _downloadFilePath;
   base::scoped_nsobject<MDCActivityIndicator> _activityIndicator;
   // Set to YES when a download begins and is used to determine if the
   // DownloadFileResult histogram needs to be recorded on -dealloc.
   BOOL _recordDownloadResultHistogram;
   // Set to YES when a file is downloaded and is used to determine if the
   // DownloadedFileAction histogram needs to be recorded on -dealloc.
   BOOL _recordFileActionHistogram;
   base::mac::ObjCPropertyReleaser _propertyReleaser_DownloadManagerController;
 }
@@ skipping 119 matching lines @@
 @synthesize landscapeConstraintsArray = _landscapeConstraintsArray;
 @synthesize portraitActionBarOneButtonConstraintsArray =
     _portraitActionBarOneButtonConstraintsArray;
 @synthesize portraitActionBarTwoButtonConstraintsArray =
     _portraitActionBarTwoButtonConstraintsArray;
 @synthesize fileTypeLabelCentered = _fileTypeLabelCentered;
 @synthesize downloadStartedTime = _downloadStartedTime;
 @synthesize fractionDownloaded = _fractionDownloaded;
 @synthesize googleDriveMetadata = _googleDriveMetadata;
 
-- (id)initWithURL:(const GURL&)url
-    requestContextGetter:(URLRequestContextGetter*)requestContextGetter
-        storeKitLauncher:(id<StoreKitLauncher>)storeLauncher {
+- (instancetype)initWithWebState:(web::WebState*)webState
+                     downloadURL:(const GURL&)url {
   self = [super initWithNibName:@"DownloadManagerController" url:url];
   if (self) {
     _downloadManagerId = g_download_manager_id++;
     _propertyReleaser_DownloadManagerController.Init(
         self, [DownloadManagerController class]);
 
-    _requestContextGetter = requestContextGetter;
+    DCHECK(webState);
+    _webState = webState;
     _headFetcherDelegate.reset(new DownloadHeadDelegate(self));
     _contentFetcherDelegate.reset(new DownloadContentDelegate(self));
     _downloadFilePath = base::FilePath();
-    _storeKitLauncher.reset(storeLauncher);
 
     [_documentContainer
         setBackgroundColor:UIColorFromRGB(kUndownloadedDocumentColor)];
 
     _isFileTypeLabelCentered = YES;
     _isDisplayingError = NO;
     _didSuccessfullyFinishHeadFetch = NO;
 
     NSString* downloadText =
         l10n_util::GetNSString(IDS_IOS_DOWNLOAD_MANAGER_DOWNLOAD);
@@ skipping 402 matching lines @@
 // TODO(fulbright): Investigate moving this code into -wasShown instead
 // (-wasShown is currently not being called on this class even when it's
 // implemented).
 - (void)start {
   [self startHeadFetch];
 }
 
 - (void)startHeadFetch {
   _fetcher = URLFetcher::Create([self url], URLFetcher::HEAD,
                                 _headFetcherDelegate.get());
-  _fetcher->SetRequestContext(_requestContextGetter.get());
+  _fetcher->SetRequestContext(
+      _webState->GetBrowserState()->GetRequestContext());
   [[NetworkActivityIndicatorManager sharedInstance]
       startNetworkTaskForGroup:[self getNetworkActivityKey]];
   _fetcher->Start();
 }
 
 - (void)onHeadFetchComplete {
   [[NetworkActivityIndicatorManager sharedInstance]
       stopNetworkTaskForGroup:[self getNetworkActivityKey]];
   int responseCode = _fetcher->GetResponseCode();
   UMA_HISTOGRAM_SPARSE_SLOWLY(
@@ skipping 92 matching lines @@
       l10n_util::GetNSString(IDS_IOS_DOWNLOAD_MANAGER_UNABLE_TO_OPEN_FILE);
   NSString* message =
       l10n_util::GetNSString(IDS_IOS_DOWNLOAD_MANAGER_NO_APP_MESSAGE);
 
   _alertCoordinator.reset([[AlertCoordinator alloc]
       initWithBaseViewController:topViewController
                            title:title
                          message:message]);
 
   // |googleDriveMetadata| contains the information necessary to either launch
-  // |the Google Drive app or navigate to its StoreKit page.  If the metadata is
-  // |not present, do not show the upload button at all.
-  if (self.googleDriveMetadata) {
+  // the Google Drive app or navigate to its StoreKit page.  If the metadata is
+  // not present, do not show the upload button at all.
+  StoreKitTabHelper* tabHelper = StoreKitTabHelper::FromWebState(_webState);
+  if (self.googleDriveMetadata && tabHelper) {
     NSString* googleDriveButtonTitle =
         l10n_util::GetNSString(IDS_IOS_DOWNLOAD_MANAGER_UPLOAD_TO_GOOGLE_DRIVE);
     base::WeakNSObject<DownloadManagerController> weakSelf(self);
     [_alertCoordinator addItemWithTitle:googleDriveButtonTitle
                                  action:^{
                                    [weakSelf openGoogleDriveInAppStore];
                                  }
                                   style:UIAlertActionStyleDefault];
   }
   [_alertCoordinator addItemWithTitle:l10n_util::GetNSString(IDS_CANCEL)
@@ skipping 175 matching lines @@
   if (![DownloadManagerController
           fetchDownloadsDirectoryFilePath:&downloadsDirectoryPath]) {
     [self displayError];
     return;
   }
   _downloadFilePath = downloadsDirectoryPath.Append(
       base::SysNSStringToUTF8([_fileNameLabel text]));
 
   _fetcher = URLFetcher::Create([self url], URLFetcher::GET,
                                 _contentFetcherDelegate.get());
-  _fetcher->SetRequestContext(_requestContextGetter.get());
+  _fetcher->SetRequestContext(
+      _webState->GetBrowserState()->GetRequestContext());
   base::SequencedWorkerPool::SequenceToken sequenceToken =
       web::WebThread::GetBlockingPool()->GetSequenceToken();
   _fetcher->SaveResponseToFileAtPath(
       _downloadFilePath,
       web::WebThread::GetBlockingPool()->GetSequencedTaskRunner(sequenceToken));
   [[NetworkActivityIndicatorManager sharedInstance]
       startNetworkTaskForGroup:[self getNetworkActivityKey]];
   _fetcher->Start();
 
   self.downloadStartedTime = [NSDate date];
@@ skipping 290 matching lines @@
   if (!showedMenu) {
     [self displayUnableToOpenFileDialog];
   }
 }
 
 - (IBAction)googleDriveButtonTapped:(id)sender {
   [self openGoogleDriveInAppStore];
 }
 
 - (void)openGoogleDriveInAppStore {
-  [[InstallationNotifier sharedInstance]
-      registerForInstallationNotifications:self
-                              withSelector:@selector(hideGoogleDriveButton)
-                                 forScheme:[_googleDriveMetadata anyScheme]];
-
-  [_storeKitLauncher openAppStore:[_googleDriveMetadata appId]];
+  StoreKitTabHelper* helper = StoreKitTabHelper::FromWebState(_webState);
+  if (helper) {
+    [[InstallationNotifier sharedInstance]
+        registerForInstallationNotifications:self
+                                withSelector:@selector(hideGoogleDriveButton)
+                                   forScheme:[_googleDriveMetadata anyScheme]];
+    helper->OpenAppStore([_googleDriveMetadata appId]);
+  }
 }
 
 - (NSString*)getNetworkActivityKey {
   return
       [NSString stringWithFormat:
                     @"DownloadManagerController.NetworkActivityIndicatorKey.%d",
                     _downloadManagerId];
 }
 
 + (BOOL)fetchDownloadsDirectoryFilePath:(base::FilePath*)path {
@@ skipping 41 matching lines @@
 
 #pragma mark - CRWNativeContent
 
 - (void)close {
   // Makes sure that all outstanding network requests are shut down before
   // this controller is closed.
   _fetcher.reset();
 }
 
 @end