chrome/browser/android/webapps/twa_launch_queue_tab_helper.cc

// 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.

#include "chrome/browser/android/webapps/twa_launch_queue_tab_helper.h"

#include "chrome/browser/android/webapps/twa_launch_queue_delegate.h"
#include "components/webapps/browser/launch_queue/launch_queue.h"

namespace webapps {

// static
void TwaLaunchQueueTabHelper::Create(content::WebContents* contents) {
  auto helper = std::make_unique<TwaLaunchQueueTabHelper>(contents);
  contents->SetUserData(UserDataKey(), std::move(helper));
}

TwaLaunchQueueTabHelper* TwaLaunchQueueTabHelper::GetOrCreateForWebContents(
    content::WebContents* contents) {
  if (!webapps::TwaLaunchQueueTabHelper::FromWebContents(contents)) {
    webapps::TwaLaunchQueueTabHelper::Create(contents);
  }

  return webapps::TwaLaunchQueueTabHelper::FromWebContents(contents);
}

TwaLaunchQueueTabHelper::~TwaLaunchQueueTabHelper() = default;

LaunchQueue& TwaLaunchQueueTabHelper::EnsureLaunchQueue() {
  if (!launch_queue_) {
    std::unique_ptr<LaunchQueueDelegate> delegate =
        std::make_unique<TwaLaunchQueueDelegate>();
    launch_queue_ =
        std::make_unique<LaunchQueue>(&GetWebContents(), std::move(delegate));
  }
  return *launch_queue_;
}

void TwaLaunchQueueTabHelper::FlushLaunchQueueForTesting() const {
  if (!launch_queue_) {
    return;
  }
  launch_queue_->FlushForTesting();  // IN-TEST
}

TwaLaunchQueueTabHelper::TwaLaunchQueueTabHelper(content::WebContents* contents)
    : content::WebContentsUserData<TwaLaunchQueueTabHelper>(*contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(TwaLaunchQueueTabHelper);

}  // namespace webapps