third_party/blink/renderer/modules/content_extraction/inner_text_agent.cc

// 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 "third_party/blink/renderer/modules/content_extraction/inner_text_agent.h"

#include "third_party/blink/renderer/core/frame/local_dom_window.h"
#include "third_party/blink/renderer/core/frame/local_frame.h"
#include "third_party/blink/renderer/modules/content_extraction/inner_text_builder.h"

namespace blink {

// static
InnerTextAgent* InnerTextAgent::From(Document& document) {
  return document.GetInnerTextAgent();
}

// static
void InnerTextAgent::BindReceiver(
    LocalFrame* frame,
    mojo::PendingReceiver<mojom::blink::InnerTextAgent> receiver) {
  DCHECK(frame && frame->GetDocument());
  auto& document = *frame->GetDocument();
  auto* agent = InnerTextAgent::From(document);
  if (!agent) {
    agent = MakeGarbageCollected<InnerTextAgent>(
        base::PassKey<InnerTextAgent>(), *frame);
    document.SetInnerTextAgent(agent);
  }
  agent->Bind(std::move(receiver));
}

InnerTextAgent::InnerTextAgent(base::PassKey<InnerTextAgent>, LocalFrame& frame)
    : document_(frame.GetDocument()), receiver_set_(this, frame.DomWindow()) {}

InnerTextAgent::~InnerTextAgent() = default;

void InnerTextAgent::Bind(
    mojo::PendingReceiver<mojom::blink::InnerTextAgent> receiver) {
  // Use `kInternalUserAction` as this task generally results in generating
  // a response to the user.
  receiver_set_.Add(
      std::move(receiver),
      document_->GetTaskRunner(TaskType::kInternalUserInteraction));
}

void InnerTextAgent::Trace(Visitor* visitor) const {
  visitor->Trace(document_);
  visitor->Trace(receiver_set_);
}

void InnerTextAgent::GetInnerText(mojom::blink::InnerTextParamsPtr params,
                                  GetInnerTextCallback callback) {
  LocalFrame* frame = document_->GetFrame();
  if (!frame) {
    std::move(callback).Run(nullptr);
    return;
  }
  if (params->max_words_per_aggregate_passage.has_value() ||
      params->greedily_aggregate_sibling_nodes.has_value()) {
    std::move(callback).Run(InnerTextPassagesBuilder::Build(*frame, *params));
  } else {
    std::move(callback).Run(InnerTextBuilder::Build(*frame, *params));
  }
}

}  // namespace blink