1 /*
2  * Copyright (C) 2023 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.systemui.flags
18 
19 import com.android.systemui.common.coroutine.ConflatedCallbackFlow.conflatedCallbackFlow
20 import com.android.systemui.statusbar.policy.BatteryController
21 import dagger.Lazy
22 import javax.inject.Inject
23 import kotlinx.coroutines.channels.awaitClose
24 
25 /** Returns true when the device is plugged in. */
26 class PluggedInCondition
27 @Inject
28 constructor(
29     private val batteryControllerLazy: Lazy<BatteryController>,
30 ) : ConditionalRestarter.Condition {
31 
<lambda>null32     override val canRestartNow = conflatedCallbackFlow {
33         val batteryCallback =
34             object : BatteryController.BatteryStateChangeCallback {
35                 override fun onBatteryLevelChanged(
36                     level: Int,
37                     pluggedIn: Boolean,
38                     charging: Boolean
39                 ) {
40                     trySend(pluggedIn)
41                 }
42             }
43         batteryControllerLazy.get().addCallback(batteryCallback)
44 
45         trySend(batteryControllerLazy.get().isPluggedIn)
46 
47         awaitClose { batteryControllerLazy.get().removeCallback(batteryCallback) }
48     }
49 }
50