Handle fold when bubble bar is expanded

When folding the device with an expanded bubble bar we now keep
the bubble expanded if the device is configured to stay awake
on fold. Otherwise the bubble is collapsed.

Bug: 380105874
Flag: com.android.wm.shell.enable_bubble_bar
Test: atest BubbleStackViewTest
Test: atest BubbleControllerTest
Test: atest ExpandedAnimationControllerTest
Test: manual
       - create a bubble and unfold
       - expand bubble
       - fold
       - observe bubble is expanded
Change-Id: Iaf0a04f6dd98362388f0a421b018d3d71697be64