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

Commit e1f6743e authored by Matt Casey's avatar Matt Casey
Browse files

Connect screenshot detection to actual API 

If the screenshot isn't from overview, notify listening apps. Return the
names of the apps that were listening.

Bug: 265937777
Test: atest ScreenshotDetectionControllerTest
Change-Id: I61ecf0c5de3e392243272e1f058ea85e31a36b25
parent 4ab33a94

packages/SystemUI/src/com/android/systemui/screenshot/ScreenshotDetectionController.kt

+14 −2
Original line number Diff line number Diff line
@@ -17,8 +17,10 @@ 
package com.android.systemui.screenshot



import android.content.pm.PackageManager

import android.view.Display

import android.view.IWindowManager

import android.view.ViewGroup

import android.view.WindowManager

import android.widget.TextView

import com.android.systemui.R

import javax.inject.Inject
@@ -34,8 +36,18 @@ constructor( 
     * notified.

     */

    fun maybeNotifyOfScreenshot(data: ScreenshotData): List<CharSequence> {

        // TODO: actually ask the window manager once API is available.

        return listOf()

        // No notification for screenshots from overview.

        if (data.source == WindowManager.ScreenshotSource.SCREENSHOT_OVERVIEW) return listOf()



        // Notify listeners, retrieve a list of listening components.

        val components = windowManager.notifyScreenshotListeners(Display.DEFAULT_DISPLAY)



        // Convert component names to app names.

        return components.map {

            packageManager

                .getActivityInfo(it, PackageManager.ComponentInfoFlags.of(0))

                .loadLabel(packageManager)

        }

    }



    fun populateView(view: ViewGroup, appNames: List<CharSequence>) {

packages/SystemUI/tests/src/com/android/systemui/screenshot/ScreenshotDetectionControllerTest.kt

0 → 100644
+161 −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.screenshot



import android.content.ComponentName

import android.content.pm.ActivityInfo

import android.content.pm.PackageManager

import android.testing.AndroidTestingRunner

import android.view.Display

import android.view.IWindowManager

import android.view.WindowManager

import androidx.test.filters.SmallTest

import com.android.systemui.util.mockito.any

import com.android.systemui.util.mockito.eq

import com.android.systemui.util.mockito.whenever

import junit.framework.Assert.assertEquals

import junit.framework.Assert.assertTrue

import org.junit.Before

import org.junit.Test

import org.junit.runner.RunWith

import org.mockito.Mock

import org.mockito.Mockito.mock

import org.mockito.Mockito.never

import org.mockito.Mockito.verify

import org.mockito.MockitoAnnotations



@SmallTest

@RunWith(AndroidTestingRunner::class)

class ScreenshotDetectionControllerTest {



    @Mock lateinit var windowManager: IWindowManager



    @Mock lateinit var packageManager: PackageManager



    lateinit var controller: ScreenshotDetectionController



    @Before

    fun setup() {

        MockitoAnnotations.initMocks(this)



        controller = ScreenshotDetectionController(windowManager, packageManager)

    }



    @Test

    fun testMaybeNotifyOfScreenshot_ignoresOverview() {

        val data = ScreenshotData.forTesting()

        data.source = WindowManager.ScreenshotSource.SCREENSHOT_OVERVIEW



        val list = controller.maybeNotifyOfScreenshot(data)



        assertTrue(list.isEmpty())

        verify(windowManager, never()).notifyScreenshotListeners(any())

    }



    @Test

    fun testMaybeNotifyOfScreenshot_emptySet() {

        val data = ScreenshotData.forTesting()

        data.source = WindowManager.ScreenshotSource.SCREENSHOT_KEY_CHORD



        whenever(windowManager.notifyScreenshotListeners(eq(Display.DEFAULT_DISPLAY)))

            .thenReturn(listOf())



        val list = controller.maybeNotifyOfScreenshot(data)



        assertTrue(list.isEmpty())

    }



    @Test

    fun testMaybeNotifyOfScreenshot_oneApp() {

        val data = ScreenshotData.forTesting()

        data.source = WindowManager.ScreenshotSource.SCREENSHOT_KEY_CHORD



        val component = ComponentName("package1", "class1")

        val appName = "app name"

        val activityInfo = mock(ActivityInfo::class.java)



        whenever(

                packageManager.getActivityInfo(

                    eq(component),

                    any(PackageManager.ComponentInfoFlags::class.java)

                )

            )

            .thenReturn(activityInfo)

        whenever(activityInfo.loadLabel(eq(packageManager))).thenReturn(appName)



        whenever(windowManager.notifyScreenshotListeners(eq(Display.DEFAULT_DISPLAY)))

            .thenReturn(listOf(component))



        val list = controller.maybeNotifyOfScreenshot(data)



        assertEquals(1, list.size)

        assertEquals(appName, list[0])

    }



    @Test

    fun testMaybeNotifyOfScreenshot_multipleApps() {

        val data = ScreenshotData.forTesting()

        data.source = WindowManager.ScreenshotSource.SCREENSHOT_KEY_CHORD



        val component1 = ComponentName("package1", "class1")

        val component2 = ComponentName("package2", "class2")

        val component3 = ComponentName("package3", "class3")

        val appName1 = "app name 1"

        val appName2 = "app name 2"

        val appName3 = "app name 3"



        val activityInfo1 = mock(ActivityInfo::class.java)

        val activityInfo2 = mock(ActivityInfo::class.java)

        val activityInfo3 = mock(ActivityInfo::class.java)



        whenever(

                packageManager.getActivityInfo(

                    eq(component1),

                    any(PackageManager.ComponentInfoFlags::class.java)

                )

            )

            .thenReturn(activityInfo1)

        whenever(

                packageManager.getActivityInfo(

                    eq(component2),

                    any(PackageManager.ComponentInfoFlags::class.java)

                )

            )

            .thenReturn(activityInfo2)

        whenever(

                packageManager.getActivityInfo(

                    eq(component3),

                    any(PackageManager.ComponentInfoFlags::class.java)

                )

            )

            .thenReturn(activityInfo3)



        whenever(activityInfo1.loadLabel(eq(packageManager))).thenReturn(appName1)

        whenever(activityInfo2.loadLabel(eq(packageManager))).thenReturn(appName2)

        whenever(activityInfo3.loadLabel(eq(packageManager))).thenReturn(appName3)



        whenever(windowManager.notifyScreenshotListeners(eq(Display.DEFAULT_DISPLAY)))

            .thenReturn(listOf(component1, component2, component3))



        val list = controller.maybeNotifyOfScreenshot(data)



        assertEquals(3, list.size)

        assertEquals(appName1, list[0])

        assertEquals(appName2, list[1])

        assertEquals(appName3, list[2])

    }

}