Introduced StoreKitTabHelper class

ios/chrome/browser/native_app_launcher/native_app_navigation_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.
 
 #import "ios/chrome/browser/native_app_launcher/native_app_navigation_controller.h"
 
 #import <StoreKit/StoreKit.h>
 
 #include "base/memory/ptr_util.h"
 #include "base/metrics/user_metrics.h"
 #include "base/metrics/user_metrics_action.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "components/image_fetcher/ios/ios_image_data_fetcher_wrapper.h"
 #include "components/infobars/core/infobar_manager.h"
 #include "ios/chrome/browser/infobars/infobar_manager_impl.h"
 #import "ios/chrome/browser/installation_notifier.h"
 #include "ios/chrome/browser/native_app_launcher/native_app_infobar_delegate.h"
 #import "ios/chrome/browser/native_app_launcher/native_app_navigation_controller_protocol.h"
 #include "ios/chrome/browser/native_app_launcher/native_app_navigation_util.h"
 #import "ios/chrome/browser/open_url_util.h"
+#import "ios/chrome/browser/store_kit/store_kit_tab_helper.h"
+#import "ios/chrome/browser/tabs/legacy_tab_helper.h"
 #import "ios/chrome/browser/tabs/tab.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_types.h"
 #import "ios/public/provider/chrome/browser/native_app_launcher/native_app_whitelist_manager.h"
+#include "ios/web/public/browser_state.h"
 #include "ios/web/public/web_state/web_state.h"
 #import "ios/web/public/web_state/web_state_observer_bridge.h"
 #include "ios/web/public/web_thread.h"
 #import "net/base/mac/url_conversions.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
 using base::UserMetricsAction;
 
 @interface NativeAppNavigationController ()<
     CRWWebStateObserver,
     NativeAppNavigationControllerProtocol>
 // Shows a native app infobar by looking at the page's URL and by checking
 // wheter that infobar should be bypassed or not.
 - (void)showInfoBarIfNecessary;
 
 // Returns a pointer to the NSMutableSet of |_appsPossiblyBeingInstalled|
 - (NSMutableSet*)appsPossiblyBeingInstalled;
 
 // Records what type of infobar was opened.
 - (void)recordInfobarDisplayedOfType:(NativeAppControllerType)type
                     onLinkNavigation:(BOOL)isLinkNavigation;
 
 @end
 
 @implementation NativeAppNavigationController {
-  // WebState provides access to the *TabHelper objects. This will eventually
-  // replace the need to have |_tab| in this object.
+  // WebState provides access to the *TabHelper objects.
   web::WebState* _webState;
   // ImageFetcher needed to fetch the icons.
   std::unique_ptr<image_fetcher::IOSImageDataFetcherWrapper> _imageFetcher;
-  // DEPRECATED: Tab hosting the infobar and is also used for accessing Tab
-  // states such as navigation manager and whether it is a pre-rendered tab.
-  // Use |webState| whenever possible.
-  __weak Tab* _tab;
   id<NativeAppMetadata> _metadata;
   // A set of appIds encoded as NSStrings.
   NSMutableSet* _appsPossiblyBeingInstalled;
   // Allows this class to subscribe for CRWWebStateObserver callbacks.
   std::unique_ptr<web::WebStateObserverBridge> _webStateObserver;
 }
 
 // Designated initializer. Use this instead of -init.
-- (instancetype)initWithWebState:(web::WebState*)webState
-            requestContextGetter:(net::URLRequestContextGetter*)context
-                             tab:(Tab*)tab {
+- (instancetype)initWithWebState:(web::WebState*)webState {
   self = [super init];
   if (self) {
-    DCHECK(context);
-    _imageFetcher = base::MakeUnique<image_fetcher::IOSImageDataFetcherWrapper>(
-        context, web::WebThread::GetBlockingPool());
     DCHECK(webState);
     _webState = webState;
-    // Allows |tab| to be nil for unit testing. If not nil, it should have the
-    // same webState.
-    DCHECK(!tab || [tab webState] == webState);
-    _tab = tab;
+    _imageFetcher = base::MakeUnique<image_fetcher::IOSImageDataFetcherWrapper>(
+        _webState->GetBrowserState()->GetRequestContext(),
+        web::WebThread::GetBlockingPool());
     _appsPossiblyBeingInstalled = [[NSMutableSet alloc] init];
     _webStateObserver =
         base::MakeUnique<web::WebStateObserverBridge>(webState, self);
   }
   return self;
 }
 
 - (void)copyStateFrom:(NativeAppNavigationController*)controller {
   DCHECK(controller);
   _appsPossiblyBeingInstalled = [[NSMutableSet alloc]
@@ skipping 92 matching lines @@
       registerForInstallationNotifications:self
                               withSelector:@selector(appDidInstall:)
                                  forScheme:[_metadata anyScheme]];
   NSString* appIdString = [self appId];
   // Defensively early return if native app metadata returns an nil string for
   // appId which can cause subsequent -addObject: to throw exceptions.
   if (![appIdString length])
     return;
   DCHECK(![_appsPossiblyBeingInstalled containsObject:appIdString]);
   [_appsPossiblyBeingInstalled addObject:appIdString];
-  // TODO(crbug.com/684063): Preferred method is to add a helper object to
-  // WebState and use the helper object to launch Store Kit.
-  [_tab openAppStore:appIdString];
+  StoreKitTabHelper* tabHelper = StoreKitTabHelper::FromWebState(_webState);
+  if (tabHelper)
+    tabHelper->OpenAppStore(appIdString);
 }
 
 - (void)launchApp:(const GURL&)URL {
   // TODO(crbug.com/353957): Pass the ChromeIdentity to
   // -launchURLWithURL:identity:
   GURL launchURL([_metadata launchURLWithURL:URL identity:nil]);
   if (launchURL.is_valid()) {
     OpenUrlWithCompletionHandler(net::NSURLWithGURL(launchURL), nil);
   }
 }
 
 - (void)updateMetadataWithUserAction:(NativeAppActionType)userAction {
   [_metadata updateWithUserAction:userAction];
 }
 
 #pragma mark - CRWWebStateObserver methods
 
 - (void)webState:(web::WebState*)webState didLoadPageWithSuccess:(BOOL)success {
-  if (success && ![_tab isPrerenderTab])
+  Tab* tab = LegacyTabHelper::GetTabForWebState(_webState);
+  if (success && ![tab isPrerenderTab])
     [self showInfoBarIfNecessary];
 }
 
 - (void)webStateDestroyed:(web::WebState*)webState {
   _webState = nullptr;
   _webStateObserver.reset();
 }
 
 #pragma mark - Private methods
 
@@ skipping 14 matching lines @@
         if ([[appURL scheme] isEqualToString:schemeOfInstalledApp]) {
           appIDToRemove = appID;
           *stop = YES;
         }
       }];
   DCHECK(appIDToRemove);
   [_appsPossiblyBeingInstalled removeObject:appIDToRemove];
 }
 
 @end