UwbControllerSessionScopeAospImpl.kt

/*
 * Copyright 2022 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 androidx.core.uwb.impl

import androidx.core.uwb.RangingCapabilities
import androidx.core.uwb.RangingParameters
import androidx.core.uwb.RangingResult
import androidx.core.uwb.UwbAddress
import androidx.core.uwb.UwbComplexChannel
import androidx.core.uwb.UwbControllerSessionScope
import androidx.core.uwb.backend.IUwbClient
import kotlinx.coroutines.flow.Flow

internal class UwbControllerSessionScopeAospImpl(
    private val uwbClient: IUwbClient,
    override val rangingCapabilities: RangingCapabilities,
    override val localAddress: UwbAddress,
    override val uwbComplexChannel: UwbComplexChannel
) : UwbControllerSessionScope {
    private val uwbClientSessionScope =
        UwbClientSessionScopeAospImpl(uwbClient, rangingCapabilities, localAddress)
    override suspend fun addControlee(address: UwbAddress) {
        val uwbAddress = androidx.core.uwb.backend.UwbAddress()
        uwbAddress.address = address.address
        try {
            uwbClient.addControlee(uwbAddress)
        } catch (e: Exception) {
            throw(e)
        }
    }

    override suspend fun removeControlee(address: UwbAddress) {
        val uwbAddress = androidx.core.uwb.backend.UwbAddress()
        uwbAddress.address = address.address
        try {
            uwbClient.removeControlee(uwbAddress)
        } catch (e: Exception) {
            throw(e)
        }
    }

    override fun prepareSession(parameters: RangingParameters): Flow<RangingResult> {
        return uwbClientSessionScope.prepareSession(parameters)
    }
}