chrome/browser/ash/mahi/web_contents/mahi_tab_helper.cc

// Copyright 2024 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/ash/mahi/web_contents/mahi_tab_helper.h"

#include <string_view>

#include "chromeos/components/mahi/public/cpp/mahi_web_contents_manager.h"
#include "chromeos/constants/chromeos_features.h"

namespace mahi {

// static
void MahiTabHelper::MaybeCreateForWebContents(
    content::WebContents* web_contents) {
  if (!chromeos::features::IsMahiEnabled()) {
    return;
  }
  MahiTabHelper::CreateForWebContents(web_contents);
}

MahiTabHelper::MahiTabHelper(content::WebContents* web_contents)
    : content::WebContentsUserData<MahiTabHelper>(*web_contents),
      content::WebContentsObserver(web_contents) {}

void MahiTabHelper::OnWebContentsFocused(
    content::RenderWidgetHost* render_widget_host) {
  focused_ = true;
  // Fires an event if the web content has finished document loading.
  // Otherwise, it would be handled by
  // `DocumentOnLoadCompletedInPrimaryMainFrame`.
  if (web_contents()->IsDocumentOnLoadCompletedInPrimaryMainFrame()) {
    chromeos::MahiWebContentsManager::Get()->OnFocusedPageLoadComplete(
        web_contents());
  } else {
    // Clears the previous focused page state so that it won't be shown before
    // the new page finishes loading.
    chromeos::MahiWebContentsManager::Get()->ClearFocusedWebContentState();
  }
}

void MahiTabHelper::OnWebContentsLostFocus(
    content::RenderWidgetHost* render_widget_host) {
  focused_ = false;
}

void MahiTabHelper::DocumentOnLoadCompletedInPrimaryMainFrame() {
  // Ignores the events from unfocused pages.
  if (!focused_) {
    return;
  }
  chromeos::MahiWebContentsManager::Get()->OnFocusedPageLoadComplete(
      web_contents());
}

void MahiTabHelper::WebContentsDestroyed() {
  chromeos::MahiWebContentsManager::Get()->WebContentsDestroyed(web_contents());
}

WEB_CONTENTS_USER_DATA_KEY_IMPL(MahiTabHelper);

}  // namespace mahi