// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/supervised_user/supervised_user_interstitial_tab_closer.h"

#include "build/build_config.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/web_contents.h"

#if !BUILDFLAG(IS_ANDROID)
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#endif

TabCloser::~TabCloser() = default;

// static
void TabCloser::CheckIfInBrowserThenCloseTab(
    content::WebContents* web_contents) {
  DCHECK(web_contents);
  // Close the tab only if there is a browser for it (which is not the case
  // for example in a <webview>).
#if !BUILDFLAG(IS_ANDROID)
  if (!GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          web_contents)) {
    return;
  }
#endif
  TabCloser::CreateForWebContents(web_contents);
}

TabCloser::TabCloser(content::WebContents* web_contents)
    : content::WebContentsUserData<TabCloser>(*web_contents) {
  content::GetUIThreadTaskRunner({})->PostTask(
      FROM_HERE,
      base::BindOnce(&TabCloser::CloseTabImpl, weak_ptr_factory_.GetWeakPtr()));
}

void TabCloser::CloseTabImpl() {
  // On Android, FindBrowserWithTab doesn't exist.
#if !BUILDFLAG(IS_ANDROID)
  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          &GetWebContents());
  DCHECK(browser);
  if (browser->GetAllTabInterfaces().size() <= 1) {
    // Don't close the last tab in the window.
    GetWebContents().RemoveUserData(UserDataKey());
    return;
  }
#endif
  GetWebContents().Close();
}

WEB_CONTENTS_USER_DATA_KEY_IMPL(TabCloser);