Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 92f65330 authored by Christian Göllner's avatar Christian Göllner Committed by Chris Göllner
Browse files

Partial Screen Sharing: Task Switcher Data Layer partial implementation 

Partially implements the data layer:
- Defines data model and interfaces
- Implements task related repository
- For now uses a no-op repository for media projection

Bug: 286201261
Test: TasksRepositoryTest.kt

Change-Id: I58a7f648682f581645c6f283ffd8f0fa920f707f
parent 4321cd2b

packages/SystemUI/src/com/android/systemui/mediaprojection/taskswitcher/data/model/MediaProjectionState.kt

0 → 100644
+26 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.mediaprojection.taskswitcher.data.model



import android.app.TaskInfo



/** Represents the state of media projection. */

sealed interface MediaProjectionState {

    object NotProjecting : MediaProjectionState

    object EntireScreen : MediaProjectionState

    data class SingleTask(val task: TaskInfo) : MediaProjectionState

}

packages/SystemUI/src/com/android/systemui/mediaprojection/taskswitcher/data/repository/ActivityTaskManagerTasksRepository.kt

0 → 100644
+75 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.mediaprojection.taskswitcher.data.repository



import android.app.ActivityManager.RunningTaskInfo

import android.app.ActivityTaskManager

import android.app.TaskStackListener

import android.os.IBinder

import android.util.Log

import com.android.systemui.common.coroutine.ChannelExt.trySendWithFailureLogging

import com.android.systemui.common.coroutine.ConflatedCallbackFlow.conflatedCallbackFlow

import com.android.systemui.dagger.SysUISingleton

import com.android.systemui.dagger.qualifiers.Application

import com.android.systemui.dagger.qualifiers.Background

import javax.inject.Inject

import kotlinx.coroutines.CoroutineDispatcher

import kotlinx.coroutines.CoroutineScope

import kotlinx.coroutines.channels.awaitClose

import kotlinx.coroutines.flow.Flow

import kotlinx.coroutines.flow.SharingStarted

import kotlinx.coroutines.flow.shareIn

import kotlinx.coroutines.withContext



/** Implementation of [TasksRepository] that uses [ActivityTaskManager] as the data source. */

@SysUISingleton

class ActivityTaskManagerTasksRepository

@Inject

constructor(

    private val activityTaskManager: ActivityTaskManager,

    @Application private val applicationScope: CoroutineScope,

    @Background private val backgroundDispatcher: CoroutineDispatcher,

) : TasksRepository {



    override suspend fun findRunningTaskFromWindowContainerToken(

        windowContainerToken: IBinder

    ): RunningTaskInfo? =

        getRunningTasks().firstOrNull { taskInfo ->

            taskInfo.token.asBinder() == windowContainerToken

        }



    private suspend fun getRunningTasks(): List<RunningTaskInfo> =

        withContext(backgroundDispatcher) { activityTaskManager.getTasks(Integer.MAX_VALUE) }



    override val foregroundTask: Flow<RunningTaskInfo> =

        conflatedCallbackFlow {

                val listener =

                    object : TaskStackListener() {

                        override fun onTaskMovedToFront(taskInfo: RunningTaskInfo) {

                            Log.d(TAG, "onTaskMovedToFront: $taskInfo")

                            trySendWithFailureLogging(taskInfo, TAG)

                        }

                    }

                activityTaskManager.registerTaskStackListener(listener)

                awaitClose { activityTaskManager.unregisterTaskStackListener(listener) }

            }

            .shareIn(applicationScope, SharingStarted.Lazily, replay = 1)



    companion object {

        private const val TAG = "TasksRepository"

    }

}

packages/SystemUI/src/com/android/systemui/mediaprojection/taskswitcher/data/repository/MediaProjectionRepository.kt

0 → 100644
+27 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.mediaprojection.taskswitcher.data.repository



import com.android.systemui.mediaprojection.taskswitcher.data.model.MediaProjectionState

import kotlinx.coroutines.flow.Flow



/** Represents a repository to retrieve and change data related to media projection. */

interface MediaProjectionRepository {



    /** Represents the current [MediaProjectionState]. */

    val mediaProjectionState: Flow<MediaProjectionState>

}

packages/SystemUI/src/com/android/systemui/mediaprojection/taskswitcher/data/repository/NoOpMediaProjectionRepository.kt

0 → 100644
+32 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.mediaprojection.taskswitcher.data.repository



import com.android.systemui.dagger.SysUISingleton

import com.android.systemui.mediaprojection.taskswitcher.data.model.MediaProjectionState

import kotlinx.coroutines.flow.Flow

import kotlinx.coroutines.flow.emptyFlow



/**

 * No-op implementation of [MediaProjectionRepository] that does nothing. Currently used as a

 * placeholder, while the real implementation is not completed.

 */

@SysUISingleton

class NoOpMediaProjectionRepository : MediaProjectionRepository {



    override val mediaProjectionState: Flow<MediaProjectionState> = emptyFlow()

}

packages/SystemUI/src/com/android/systemui/mediaprojection/taskswitcher/data/repository/TasksRepository.kt

0 → 100644
+41 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.mediaprojection.taskswitcher.data.repository



import android.app.ActivityManager.RunningTaskInfo

import android.os.IBinder

import kotlinx.coroutines.flow.Flow



/** Repository responsible for retrieving data related to running tasks. */

interface TasksRepository {



    /**

     * Tries to find a [RunningTaskInfo] with a matching window container token. Returns `null` when

     * no matching task was found.

     */

    suspend fun findRunningTaskFromWindowContainerToken(

        windowContainerToken: IBinder

    ): RunningTaskInfo?



    /**

     * Emits a stream of [RunningTaskInfo] that have been moved to the foreground.

     *

     * Note: when subscribing for the first time, it will not immediately emit the current

     * foreground task. Only after a change in foreground task has occurred.

     */

    val foregroundTask: Flow<RunningTaskInfo>

}