Create TabWindowManager#getTabAndWindowIdByTabId.

Bug: None
Change-Id: Id9ab6eb3e8b732d4c4fae8941646681d31773c5c
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/7049688
Commit-Queue: Sky Malice <skym@chromium.org>
Reviewed-by: Aishwarya Rajesh <aishwaryarj@google.com>
Cr-Commit-Position: refs/heads/main@{#1532364}

Author: Sky Malice

chrome/browser/tabwindow/BUILD.gn

@@ -8,6 +8,7 @@ import("//chrome/android/features/android_library_factory_tmpl.gni")
 android_library("java") {
   sources = [
     "android/java/src/org/chromium/chrome/browser/tabwindow/TabModelSelectorFactory.java",
+    "android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowInfo.java",
     "android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowManager.java",
     "android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowManagerUtils.java",
     "android/java/src/org/chromium/chrome/browser/tabwindow/WindowId.java",

chrome/browser/tabwindow/android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowInfo.java

@@ -0,0 +1,27 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.chrome.browser.tabwindow;
+
+import org.chromium.build.annotations.NullMarked;
+import org.chromium.chrome.browser.tab.Tab;
+import org.chromium.chrome.browser.tabmodel.TabModel;
+import org.chromium.chrome.browser.tabmodel.TabModelSelector;
+
+/** Holds objects related to a tab and where it lives. */
+@NullMarked
+public class TabWindowInfo {
+    public final @WindowId int windowId;
+    public final TabModelSelector tabModelSelector;
+    public final TabModel tabModel;
+    public final Tab tab;
+
+    public TabWindowInfo(
+            @WindowId int windowId, TabModelSelector tabModelSelector, TabModel tabModel, Tab tab) {
+        this.tab = tab;
+        this.tabModel = tabModel;
+        this.tabModelSelector = tabModelSelector;
+        this.windowId = windowId;
+    }
+}

chrome/browser/tabwindow/android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowManager.java

@@ -156,6 +156,15 @@ interface Observer {
      */
     @Nullable Tab getTabById(@TabId int tabId, @WindowId int windowId);
 
+    /**
+     * Similar to {@link #getTabById(int)} but returns a {@link TabWindowInfo}. Does not check the
+     * non-window sources for the tab.
+     *
+     * @param tabId The id of the tab to look for.
+     * @return The tab and related parents, null if it cannot be found.
+     */
+    @Nullable TabWindowInfo getTabWindowInfoById(@TabId int tabId);
+
     /**
      * @param windowId The ID of the window that holds the tab group.
      * @param tabGroupId The tab group ID of the tab group.

chrome/browser/tabwindow/internal/android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowManagerImpl.java

@@ -518,6 +518,21 @@ public int getIncognitoTabCount() {
         return getTabFromOtherSource(tabId);
     }
 
+    @Override
+    public @Nullable TabWindowInfo getTabWindowInfoById(@TabId int tabId) {
+        for (Map.Entry<@WindowId Integer, TabModelSelector> entry :
+                mWindowIdToSelectors.entrySet()) {
+            TabModelSelector selector = entry.getValue();
+            for (TabModel tabModel : selector.getModels()) {
+                @Nullable final Tab tab = tabModel.getTabById(tabId);
+                if (tab != null) {
+                    return new TabWindowInfo(entry.getKey(), selector, tabModel, tab);
+                }
+            }
+        }
+        return null;
+    }
+
     @Override
     public @Nullable List<Tab> getGroupedTabsByWindow(
             @WindowId int windowId, Token tabGroupId, boolean isIncognito) {

chrome/browser/tabwindow/internal/android/java/src/org/chromium/chrome/browser/tabwindow/TabWindowManagerImplUnitTest.java

@@ -1246,4 +1246,56 @@ public void testFindWindowIdForTabGroup_found() {
     public void testFindWindowIdForTabGroup_notFound() {
         assertEquals(INVALID_WINDOW_ID, mSubject.findWindowIdForTabGroup(GROUP_ID));
     }
+
+    @Test
+    public void testGetTabWindowInfoById() {
+        ActivityController<Activity> activityController0 = createActivity();
+        Activity activity0 = activityController0.get();
+        ActivityController<Activity> activityController1 = createActivity();
+        Activity activity1 = activityController1.get();
+        @WindowId int window0 = 0;
+        @WindowId int window1 = 1;
+        Pair<@WindowId Integer, TabModelSelector> assignment0 =
+                mSubject.requestSelector(
+                        activity0,
+                        mModalDialogManager,
+                        mProfileProviderSupplier,
+                        mTabCreatorManager,
+                        mNextTabPolicySupplier,
+                        mMultiInstanceManager,
+                        mMismatchedIndicesHandler0,
+                        window0);
+        Pair<@WindowId Integer, TabModelSelector> assignment1 =
+                mSubject.requestSelector(
+                        activity1,
+                        mModalDialogManager,
+                        mProfileProviderSupplier,
+                        mTabCreatorManager,
+                        mNextTabPolicySupplier,
+                        mMultiInstanceManager,
+                        mMismatchedIndicesHandler1,
+                        window1);
+        MockTabModelSelector selector0 = (MockTabModelSelector) assignment0.second;
+        MockTabModelSelector selector1 = (MockTabModelSelector) assignment1.second;
+        Tab tab1 = selector0.addMockTab();
+        Tab tab2 = selector1.addMockIncognitoTab();
+
+        TabWindowInfo info = mSubject.getTabWindowInfoById(tab1.getId() - 1);
+        assertNull(info);
+
+        info = mSubject.getTabWindowInfoById(tab1.getId());
+        assertNotNull(info);
+        assertEquals(tab1, info.tab);
+        assertEquals(selector0, info.tabModelSelector);
+        assertEquals(window0, info.windowId);
+
+        info = mSubject.getTabWindowInfoById(tab2.getId());
+        assertNotNull(info);
+        assertEquals(tab2, info.tab);
+        assertEquals(selector1, info.tabModelSelector);
+        assertEquals(window1, info.windowId);
+
+        info = mSubject.getTabWindowInfoById(tab2.getId() + 1);
+        assertNull(info);
+    }
 }