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
|
// Copyright 2019 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/ash/policy/scheduled_task_handler/scoped_wake_lock.h"
#include "base/no_destructor.h"
#include "content/public/browser/device_service.h"
#include "services/device/public/mojom/wake_lock_provider.mojom.h"
namespace policy {
namespace {
ScopedWakeLock::WakeLockProviderBinder& GetBinderOverride() {
static base::NoDestructor<ScopedWakeLock::WakeLockProviderBinder> binder;
return *binder;
}
} // namespace
ScopedWakeLock::ScopedWakeLock(device::mojom::WakeLockType type,
const std::string& reason) {
mojo::Remote<device::mojom::WakeLockProvider> provider;
auto receiver = provider.BindNewPipeAndPassReceiver();
const auto& binder = GetBinderOverride();
if (binder)
binder.Run(std::move(receiver));
else
content::GetDeviceService().BindWakeLockProvider(std::move(receiver));
provider->GetWakeLockWithoutContext(
type, device::mojom::WakeLockReason::kOther, reason,
wake_lock_.BindNewPipeAndPassReceiver());
// This would violate |GetWakeLockWithoutContext|'s API contract.
DCHECK(wake_lock_);
wake_lock_->RequestWakeLock();
}
ScopedWakeLock::~ScopedWakeLock() {
// |wake_lock_| could have been moved.
if (wake_lock_)
wake_lock_->CancelWakeLock();
}
ScopedWakeLock::ScopedWakeLock(ScopedWakeLock&& other) = default;
ScopedWakeLock& ScopedWakeLock::operator=(ScopedWakeLock&& other) = default;
// static
void ScopedWakeLock::OverrideWakeLockProviderBinderForTesting(
WakeLockProviderBinder binder) {
GetBinderOverride() = std::move(binder);
}
} // namespace policy
|
