1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
|
// 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/companion/text_finder/text_finder_manager.h"
#include <memory>
#include "base/barrier_callback.h"
#include "base/functional/bind.h"
#include "base/unguessable_token.h"
#include "chrome/browser/companion/text_finder/text_finder.h"
#include "content/public/browser/page.h"
#include "content/public/browser/page_user_data.h"
#include "content/public/browser/render_frame_host.h"
#include "services/service_manager/public/cpp/interface_provider.h"
namespace companion {
TextFinderManager::TextFinderManager(content::Page& page)
: PageUserData<TextFinderManager>(page) {
content::RenderFrameHost& frame = page.GetMainDocument();
frame.GetRemoteInterfaces()->GetInterface(
agent_container_.BindNewPipeAndPassReceiver());
}
TextFinderManager::~TextFinderManager() = default;
std::optional<base::UnguessableToken> TextFinderManager::CreateTextFinder(
const std::string& text_directive,
TextFinder::FinishedCallback callback) {
if (text_directive.empty()) {
std::move(callback).Run(std::make_pair(text_directive, false));
return std::nullopt;
}
// Generate a unique random id.
const base::UnguessableToken id = base::UnguessableToken::Create();
finder_map_.emplace(
id, std::make_unique<TextFinder>(
text_directive, agent_container_, std::move(callback),
// Remove this text finder from manager upon agent disconnection.
base::BindOnce(&TextFinderManager::RemoveTextFinder,
weak_ptr_factory_.GetWeakPtr(), id)));
return id;
}
void TextFinderManager::CreateTextFinders(
const std::vector<std::string>& text_directives,
AllDoneCallback all_done_callback) {
const auto finished_callback =
base::BarrierCallback<std::pair<std::string, bool>>(
text_directives.size(), std::move(all_done_callback));
for (const auto& text_directive : text_directives) {
CreateTextFinder(text_directive, finished_callback);
}
}
void TextFinderManager::RemoveTextFinder(base::UnguessableToken id) {
finder_map_.erase(id);
}
size_t TextFinderManager::Size() const {
return finder_map_.size();
}
PAGE_USER_DATA_KEY_IMPL(TextFinderManager);
} // namespace companion
|