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
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// 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.
#ifndef COMPONENTS_LANGUAGE_DETECTION_CONTENT_RENDERER_LANGUAGE_DETECTION_MODEL_MANAGER_H_
#define COMPONENTS_LANGUAGE_DETECTION_CONTENT_RENDERER_LANGUAGE_DETECTION_MODEL_MANAGER_H_
#include "base/functional/callback_forward.h"
#include "base/memory/weak_ptr.h"
#include "components/language_detection/content/common/language_detection.mojom.h"
#include "components/language_detection/core/language_detection_model.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "third_party/blink/public/platform/browser_interface_broker_proxy.h"
namespace language_detection {
// Provides a language detection model. It tries to ensure that it has a valid
// model file loaded by requesting one from the browser.
class LanguageDetectionModelManager {
public:
// `model` is not owned by and must outlive `this`.
explicit LanguageDetectionModelManager(LanguageDetectionModel& model);
LanguageDetectionModelManager(const LanguageDetectionModelManager&) = delete;
LanguageDetectionModelManager& operator=(
const LanguageDetectionModelManager&) = delete;
~LanguageDetectionModelManager();
// `nullptr` indicates that no valid model could be loaded.
using GetLanuageDetectionModelCallback =
base::OnceCallback<void(LanguageDetectionModel* model)>;
using GetLanguageDetectionModelStatusCallback = language_detection::mojom::
ContentLanguageDetectionDriver::GetLanguageDetectionModelStatusCallback;
// Passes a model to `callback`. If no valid model can possibly be loaded, an
// invalid model will be passed. If the current model is valid, this will
// succeed immediately. If it's not valid, it will request a model via the
// `interface_broker` and call `callback` when that response arrives. This may
// still result in no valid model.
void GetLanguageDetectionModel(
const blink::BrowserInterfaceBrokerProxy& interface_broker,
GetLanuageDetectionModelCallback callback);
// Checks if the model has been downloaded in the browser process.
void GetLanguageDetectionModelStatus(
const blink::BrowserInterfaceBrokerProxy& interface_broker,
GetLanguageDetectionModelStatusCallback callback);
private:
// Ensures that the driver is connected before returning it.
mojo::Remote<mojom::ContentLanguageDetectionDriver>&
GetLanguageDetectionDriver(
const blink::BrowserInterfaceBrokerProxy& interface_broker);
// The callback to receive the language detection model file.
void UpdateLanguageDetectionModel(base::File model_file,
base::OnceClosure callback);
// Callback for when the browser returns a model file.
static void OnModelFileAvailable(
base::WeakPtr<LanguageDetectionModelManager> manager,
LanguageDetectionModelManager::GetLanuageDetectionModelCallback callback,
base::File model_file);
// Callback for when the returned model file has been loaded.
void OnModelFileLoaded(
LanguageDetectionModelManager::GetLanuageDetectionModelCallback callback);
mojo::Remote<mojom::ContentLanguageDetectionDriver>
language_detection_driver_;
const raw_ref<language_detection::LanguageDetectionModel>
language_detection_model_;
base::WeakPtrFactory<LanguageDetectionModelManager> weak_pointer_factory_{
this};
};
} // namespace language_detection
#endif // COMPONENTS_LANGUAGE_DETECTION_CONTENT_RENDERER_LANGUAGE_DETECTION_MODEL_MANAGER_H_
|
