Cleanup SysUIState callback interface

This gets rid of the old method without displayId parameter and refactors the few places that were using it

Bug: 362719719
Bug: 398011576
Test: SysUiStateTest
Flag: com.android.systemui.shade_window_goes_around
Change-Id: I79e72ffd3271e170ef264e73a0d70567650432f6