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
|
// 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 CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_MANAGER_H_
#define CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_MANAGER_H_
#include <memory>
#include "base/component_export.h"
#include "base/functional/callback_forward.h"
#include "base/sequence_checker.h"
#include "chromeos/ash/components/mantis/media_app/mantis_untrusted_service.h"
#include "chromeos/ash/components/mantis/mojom/mantis_service.mojom.h"
#include "chromeos/ash/components/mojo_service_manager/mojom/mojo_service_manager.mojom.h"
#include "chromeos/ash/components/specialized_features/feature_access_checker.h"
#include "chromeos/services/machine_learning/public/mojom/machine_learning_service.mojom.h"
#include "components/prefs/pref_service.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "mojo/public/cpp/bindings/unique_receiver_set.h"
namespace ash {
// A component that manages the creation and lifetime of a
// `MantisUntrustedService`.
class COMPONENT_EXPORT(CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP)
MantisUntrustedServiceManager {
public:
using CreateResult = media_app_ui::mojom::MantisUntrustedServiceResult;
using CreateCallback =
base::OnceCallback<void(mojo::StructPtr<CreateResult>)>;
explicit MantisUntrustedServiceManager(
std::unique_ptr<specialized_features::FeatureAccessChecker>
access_checker);
MantisUntrustedServiceManager(const MantisUntrustedServiceManager&) = delete;
MantisUntrustedServiceManager& operator=(
const MantisUntrustedServiceManager&) = delete;
~MantisUntrustedServiceManager();
static specialized_features::FeatureAccessConfig GetFeatureAccessConfig();
void IsAvailable(PrefService* pref_service,
base::OnceCallback<void(bool)> callback);
void Create(
mojo::PendingRemote<media_app_ui::mojom::MantisUntrustedPage> page,
const std::optional<base::Uuid>& dlc_uuid,
CreateCallback callback);
private:
mojo::PendingRemote<mantis::mojom::PlatformModelProgressObserver>
CreateProgressObserver(
mojo::PendingRemote<media_app_ui::mojom::MantisUntrustedPage> page);
void ResetService();
void OnQueryDone(
base::OnceCallback<void(bool)> callback,
chromeos::mojo_service_manager::mojom::ErrorOrServiceStatePtr result);
mojo::PendingRemote<chromeos::machine_learning::mojom::TextClassifier>
GetTextClassifier();
void OnInitializeDone(
CreateCallback callback,
mojo::PendingRemote<mantis::mojom::MantisProcessor> processor,
mantis::mojom::InitializeResult result);
mojo::Remote<mantis::mojom::MantisService> cros_service_;
mojo::UniqueReceiverSet<mantis::mojom::PlatformModelProgressObserver>
progress_observers_;
mojo::Remote<chromeos::machine_learning::mojom::MachineLearningService>
ml_service_;
std::unique_ptr<specialized_features::FeatureAccessChecker> access_checker_;
std::unique_ptr<MantisUntrustedService> mantis_untrusted_service_;
SEQUENCE_CHECKER(sequence_checker_);
base::WeakPtrFactory<MantisUntrustedServiceManager> weak_ptr_factory_{this};
};
} // namespace ash
#endif // CHROMEOS_ASH_COMPONENTS_MANTIS_MEDIA_APP_MANTIS_UNTRUSTED_SERVICE_MANAGER_H_
|
