1 /* <lambda>null2 * Copyright (C) 2023 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.intentresolver.widget 18 19 import android.util.Log 20 import android.view.View 21 import androidx.core.view.OneShotPreDrawListener 22 import kotlinx.coroutines.suspendCancellableCoroutine 23 import java.util.concurrent.atomic.AtomicBoolean 24 25 internal suspend fun View.waitForPreDraw(): Unit = suspendCancellableCoroutine { continuation -> 26 val isResumed = AtomicBoolean(false) 27 val callback = OneShotPreDrawListener.add( 28 this, 29 Runnable { 30 if (isResumed.compareAndSet(false, true)) { 31 continuation.resumeWith(Result.success(Unit)) 32 } else { 33 // it's not really expected but in some unknown corner-case let's not crash 34 Log.e("waitForPreDraw", "An attempt to resume a completed coroutine", Exception()) 35 } 36 } 37 ) 38 continuation.invokeOnCancellation { callback.removeListener() } 39 } 40