[ios] WebStateList owns all WebState it manages.

ios/chrome/browser/ui/browser_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.
 
 #import "ios/chrome/browser/ui/browser_view_controller.h"
 
 #import <AssetsLibrary/AssetsLibrary.h>
 #import <MobileCoreServices/MobileCoreServices.h>
 #import <PassKit/PassKit.h>
 #import <Photos/Photos.h>
@@ skipping 153 matching lines @@
 #include "ios/chrome/grit/ios_strings.h"
 #import "ios/net/request_tracker.h"
 #include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
 #include "ios/public/provider/chrome/browser/ui/app_rating_prompt.h"
 #include "ios/public/provider/chrome/browser/ui/default_ios_web_view_factory.h"
 #import "ios/public/provider/chrome/browser/voice/voice_search_bar.h"
 #import "ios/public/provider/chrome/browser/voice/voice_search_bar_owner.h"
 #include "ios/public/provider/chrome/browser/voice/voice_search_controller.h"
 #include "ios/public/provider/chrome/browser/voice/voice_search_controller_delegate.h"
 #include "ios/public/provider/chrome/browser/voice/voice_search_provider.h"
+#import "ios/shared/chrome/browser/tabs/web_state_list.h"
+#import "ios/shared/chrome/browser/tabs/web_state_opener.h"
 #include "ios/web/navigation/navigation_manager_impl.h"
 #include "ios/web/public/active_state_manager.h"
 #include "ios/web/public/navigation_item.h"
 #import "ios/web/public/navigation_manager.h"
 #include "ios/web/public/referrer_util.h"
 #include "ios/web/public/ssl_status.h"
 #include "ios/web/public/url_scheme_util.h"
 #include "ios/web/public/user_agent.h"
 #include "ios/web/public/web_client.h"
 #import "ios/web/public/web_state/context_menu_params.h"
@@ skipping 3463 matching lines @@
   // ending up in the history causign the app to crash at every subsequent
   // restart.
   if (url.host() == kChromeUIBrowserCrashHost) {
     [self induceBrowserCrash];
     // In debug the app can continue working even after the CHECK. Adding a
     // return avoids the crash url to be added to the history.
     return;
   }
 
   if (url == [_preloadController prerenderedURL]) {
+    std::unique_ptr<web::WebState> newWebState =
+        [_preloadController releasePrerenderContents];
+    DCHECK(newWebState);
+
     Tab* oldTab = [_model currentTab];
-    Tab* newTab = [_preloadController releasePrerenderContents];
+    Tab* newTab = LegacyTabHelper::GetTabForWebState(newWebState.get());
     DCHECK(oldTab);
     DCHECK(newTab);
+
     bool canPruneItems =
         [newTab navigationManager]->CanPruneAllButLastCommittedItem();
+
     if (oldTab && newTab && canPruneItems) {
       [oldTab recordStateInHistory];
       [newTab navigationManager]->CopyStateFromAndPrune(
           [oldTab navigationManager]);
       [[newTab nativeAppNavigationController]
           copyStateFrom:[oldTab nativeAppNavigationController]];
-      [_model replaceTab:oldTab withTab:newTab];
+
+      [_model webStateList]->ReplaceWebStateAt([_model indexOfTab:oldTab],
+                                               std::move(newWebState));
 
       // Set isPrerenderTab to NO after replacing the tab. This will allow the
       // BrowserViewController to detect that a pre-rendered tab is switched in,
       // and show the prerendering animation.
       newTab.isPrerenderTab = NO;
 
       [self tabLoadComplete:newTab withSuccess:newTab.loadFinished];
-
       return;
     }
   }
 
   GURL urlToLoad = url;
   if ([_preloadController hasPrefetchedURL:url]) {
     // Prefetched URLs have modified URLs, so load the prefetched version of
     // |url| instead of the original |url|.
     urlToLoad = [_preloadController prefetchedURL];
   }
@@ skipping 963 matching lines @@
     [self updateToolbarControlsAlpha:1.0];
     [self updateToolbarBackgroundAlpha:1.0];
     [_tabStripController view].alpha = 1.0;
   };
 
   // After the transition animation completes, add the tab to the tab model
   // (on iPad this triggers the tab strip animation too), then fade out the
   // transitioning panel and remove it.
   void (^completion)(BOOL) = ^(BOOL finished) {
     _contextualSearchMask.alpha = 0;
-    Tab* newTab = [tabProvider releaseTab];
-    DCHECK(newTab);
-    DCHECK([newTab navigationManager]);
-    // Add the new tab to the tab model.
-    [newTab setParentTabModel:_model];
-    // Insert the new tab one after the current tab.
-    Tab* currentTab = [_model currentTab];
-    NSUInteger index = [_model indexOfTab:currentTab];
-    [_model insertTab:newTab atIndex:index + 1];
+    std::unique_ptr<web::WebState> webState = [tabProvider releaseWebState];
+    DCHECK(webState);
+    DCHECK(webState->GetNavigationManager());
+
+    Tab* newTab = LegacyTabHelper::GetTabForWebState(webState.get());
+    WebStateList* webStateList = [_model webStateList];
+
+    // Insert the new tab after the current tab.
+    DCHECK_NE(webStateList->active_index(), WebStateList::kInvalidIndex);
+    DCHECK_NE(webStateList->active_index(), INT_MAX);
+    int insertion_index = webStateList->active_index() + 1;
+    webStateList->InsertWebState(insertion_index, std::move(webState));
+    webStateList->SetOpenerOfWebStateAt(insertion_index,
+                                        [tabProvider webStateOpener]);
 
     // Set isPrerenderTab to NO after inserting the tab. This will allow the
     // BrowserViewController to detect that a pre-rendered tab is switched in,
     // and show the prerendering animation. This needs to happen before the
     // tab is made the current tab.
     // This also enables contextual search (if otherwise applicable) on
     // |newTab|.
+
     newTab.isPrerenderTab = NO;
     [_model setCurrentTab:newTab];
 
     if (newTab.loadFinished)
       [self tabLoadComplete:newTab withSuccess:YES];
 
     if (focusInput) {
       [_toolbarController focusOmnibox];
     }
     _infoBarContainer->RestoreInfobars();
@@ skipping 430 matching lines @@
 
 - (UIView*)voiceSearchButton {
   return _voiceSearchButton;
 }
 
 - (id<LogoAnimationControllerOwner>)logoAnimationControllerOwner {
   return [self currentLogoAnimationControllerOwner];
 }
 
 @end