Maintain CompatUIInfo in CompatUIController

CompatUIInfo (esp. TaskListener) is needed for
DisplayCompatShowRestartDialog, but WindowDecoration, which invokes
the functionality, doens't have access to TaskListener.

Initially we proposed the approach to expose
ShellTaskOrg#getTaskListener() but as TaskListener is already
exposed with onCompatInfoChanged(), it turns out they asked us to
reuse it.

This change adds mTaskIdToCompatUIInfoMap in CompatUIInfo for this
purpose.

Flag: com.android.window.flags.enable_restart_menu_for_connected_displays
Bug: 32086679
Test: CompatUIControllerTest
Change-Id: I222ef8680907e77f752448f902f98f3b4c646c39