components/segmentation_platform/content/page_load_trigger_context.cc

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

#include "components/segmentation_platform/content/page_load_trigger_context.h"

#include "content/public/browser/web_contents.h"

#if BUILDFLAG(IS_ANDROID)
#include "base/android/jni_android.h"
#include "components/segmentation_platform/content/jni_headers/PageLoadTriggerContext_jni.h"
#endif  // BUILDFLAG(IS_ANDROID)

namespace segmentation_platform {

namespace {
// TODO(ssid): Figure out a good place to keep the param names for InputContext.
constexpr char kUrlKey[] = "url";
}  // namespace

PageLoadTriggerContext::PageLoadTriggerContext(
    content::WebContents* web_contents)
    : TriggerContext(TriggerType::kPageLoad),
      web_contents_(web_contents->GetWeakPtr()) {}

PageLoadTriggerContext::~PageLoadTriggerContext() = default;

base::flat_map<std::string, processing::ProcessedValue>
PageLoadTriggerContext::GetSelectionInputArgs() const {
  base::flat_map<std::string, processing::ProcessedValue> args;
  if (web_contents_) {
    args.emplace(kUrlKey, processing::ProcessedValue(
                              web_contents_->GetLastCommittedURL()));
  }
  return args;
}

#if BUILDFLAG(IS_ANDROID)
base::android::ScopedJavaLocalRef<jobject>
PageLoadTriggerContext::CreateJavaObject() const {
  JNIEnv* env = base::android::AttachCurrentThread();
  base::android::ScopedJavaLocalRef<jobject> j_web_contents;
  if (web_contents_ && !web_contents_->IsBeingDestroyed())
    j_web_contents = web_contents_->GetJavaWebContents();
  return Java_PageLoadTriggerContext_createPageLoadTriggerContext(
      env, j_web_contents);
}
#endif  // BUILDFLAG(IS_ANDROID)

}  // namespace segmentation_platform