/*
 * Copyright (c) 2024 New Vector Ltd
 *
 * 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 io.element.android.features.call.impl.utils

import com.squareup.anvil.annotations.ContributesBinding
import io.element.android.features.call.impl.wellknown.CallWellknownAPI
import io.element.android.libraries.core.coroutine.CoroutineDispatchers
import io.element.android.libraries.di.AppScope
import io.element.android.libraries.di.SingleIn
import io.element.android.libraries.matrix.api.core.SessionId
import io.element.android.libraries.network.RetrofitFactory
import kotlinx.coroutines.withContext
import retrofit2.HttpException
import timber.log.Timber
import java.net.HttpURLConnection
import javax.inject.Inject

interface ElementCallBaseUrlProvider {
    suspend fun provides(sessionId: SessionId): String?
}

@SingleIn(AppScope::class)
@ContributesBinding(AppScope::class)
class DefaultElementCallBaseUrlProvider @Inject constructor(
    private val retrofitFactory: RetrofitFactory,
    private val coroutineDispatchers: CoroutineDispatchers,
) : ElementCallBaseUrlProvider {
    private val apiCache = mutableMapOf<SessionId, CallWellknownAPI>()

    override suspend fun provides(sessionId: SessionId): String? = withContext(coroutineDispatchers.io) {
        val domain = sessionId.value.substringAfter(":")
        val callWellknownAPI = apiCache.getOrPut(sessionId) {
            retrofitFactory.create("https://$domain")
                .create(CallWellknownAPI::class.java)
        }
        try {
            callWellknownAPI.getCallWellKnown().widgetUrl
        } catch (e: HttpException) {
            // Ignore Http 404, but re-throws any other exceptions
            if (e.code() != HttpURLConnection.HTTP_NOT_FOUND) {
                throw e
            }
            Timber.w(e, "Failed to fetch wellknown data")
            null
        }
    }
}
