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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// Copyright 2025 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_OMNIBOX_BROWSER_AIM_ELIGIBILITY_SERVICE_H_
#define COMPONENTS_OMNIBOX_BROWSER_AIM_ELIGIBILITY_SERVICE_H_
#include <memory>
#include <string>
#include "base/memory/raw_ref.h"
#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "base/observer_list.h"
#include "components/keyed_service/core/keyed_service.h"
#include "third_party/omnibox_proto/aim_eligibility_response.pb.h"
class AimEligibilityServiceObserver;
class PrefRegistrySimple;
class PrefService;
class TemplateURLService;
namespace network {
class SimpleURLLoader;
class SharedURLLoaderFactory;
} // namespace network
// Utility service to check if the profile is eligible for AI mode features.
class AimEligibilityService : public KeyedService {
public:
// See comment for `WriteToPref()`.
static void RegisterProfilePrefs(PrefRegistrySimple* registry);
AimEligibilityService(
PrefService& pref_service,
TemplateURLService& template_url_service,
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory);
~AimEligibilityService() override;
// Checks if the application country matches the given country.
bool IsCountry(const std::string& country) const;
// Checks if the application language matches the given language.
bool IsLanguage(const std::string& language) const;
// Register observers to notify when eligibility may have changed. Eligibility
// is re-checked periodically.
void AddObserver(AimEligibilityServiceObserver* observer);
void RemoveObserver(AimEligibilityServiceObserver* observer);
// Checks if server eligibility checking is enabled.
// Virtual for testing purposes.
virtual bool IsServerEligibilityEnabled() const;
// Checks if user is locally eligible for AI mode (excludes server checks).
// Virtual for testing purposes.
virtual bool IsAimLocallyEligible() const;
// Checks if user is eligible for AI mode (includes server checks).
// Virtual for testing purposes.
virtual bool IsAimEligible() const;
protected:
// Virtual methods for platform-specific country and locale access.
virtual std::string GetCountryCode() const = 0;
virtual std::string GetLocale() const = 0;
private:
// Notify `observers_` ineligibles may have changed.
void NotifyObservers() const;
// Parses `response_string` into `most_recent_response_`. Does not modify
// `most_recent_response_` if parsing fails.
bool ParseResponseString(const std::string& response_string);
// Write and read `most_recent_response_` to pref so it can be used when the
// server is unavailable.
void WriteToPref(const std::string& response_string) const;
void ReadFromPref();
// Fetch eligibility from the server.
void StartServerEligibilityRequest();
void OnServerEligibilityResponse(
std::unique_ptr<network::SimpleURLLoader> loader,
std::unique_ptr<std::string> response_string);
static constexpr char kResponsePrefName[] =
"aim_eligibility_service.aim_eligibility_response";
base::ObserverList<AimEligibilityServiceObserver> observers_;
const raw_ref<PrefService> pref_service_;
// Guaranteed not-null because this server `DependsOn` `TemplateURLService`.
const raw_ref<TemplateURLService> template_url_service_;
const scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory_;
// Update on each successfully server response.
omnibox::AimEligibilityResponse most_recent_response_;
// For binding the `OnServerEligibilityResponse()` callback.
base::WeakPtrFactory<AimEligibilityService> weak_factory_{this};
};
#endif // COMPONENTS_OMNIBOX_BROWSER_AIM_ELIGIBILITY_SERVICE_H_
|
