package expo.modules.webbrowser

import android.app.Activity
import android.content.Intent
import android.content.pm.PackageManager
import android.content.pm.ResolveInfo
import androidx.browser.customtabs.CustomTabsClient
import androidx.browser.customtabs.CustomTabsIntent
import androidx.browser.customtabs.CustomTabsService
import androidx.core.net.toUri
import expo.modules.kotlin.AppContext

private const val DUMMY_URL = "https://expo.dev"

internal class CustomTabsActivitiesHelper(
  private val appContext: AppContext
) {

  // region Actual custom tabs activities helper methods

  /**
   * @throws PackageManagerNotFoundException
   */
  fun canResolveIntent(customTabsIntent: CustomTabsIntent): Boolean =
    getResolvingActivities(customTabsIntent).isNotEmpty()

  /**
   * @throws PackageManagerNotFoundException
   */
  val customTabsResolvingActivities: ArrayList<String>
    get() = getResolvingActivities(createDefaultCustomTabsIntent())
      .mapToDistinctArrayList { resolveInfo: ResolveInfo ->
        resolveInfo.activityInfo.packageName
      }

  /**
   * @throws PackageManagerNotFoundException
   */
  val customTabsResolvingServices: ArrayList<String>
    get() = packageManager.queryIntentServices(createDefaultCustomTabsServiceIntent(), 0)
      .mapToDistinctArrayList { resolveInfo: ResolveInfo ->
        resolveInfo.serviceInfo.packageName
      }

  /**
   * @throws PackageManagerNotFoundException
   */
  fun getPreferredCustomTabsResolvingActivity(packages: List<String?>?): String? {
    val resolvedPackages = packages ?: customTabsResolvingActivities
    return CustomTabsClient.getPackageName(currentActivity, resolvedPackages)
  }

  /**
   * @throws PackageManagerNotFoundException
   */
  val defaultCustomTabsResolvingActivity: String?
    get() {
      val info = packageManager.resolveActivity(createDefaultCustomTabsIntent().intent, 0)
      return info?.activityInfo?.packageName
    }

  fun startCustomTabs(tabsIntent: CustomTabsIntent, options: OpenBrowserOptions) {
    val url = tabsIntent.intent.data ?: throw NoUrlProvidedException()

    if (!options.shouldCreateTask) {
      tabsIntent.launchUrl(currentActivity, url)
      return
    }

    // Use the proxy activity only when creating a new task AND useProxyActivity is enabled
    // This prevents the Custom Tab from being destroyed when the main activity
    // has singleTask launch mode
    if (options.useProxyActivity) {
      val proxyIntent = Intent(currentActivity, BrowserProxyActivity::class.java).apply {
        putExtra(BrowserProxyActivity.EXTRA_URL, url.toString())
        putExtra(BrowserProxyActivity.EXTRA_CUSTOM_TABS_INTENT_DATA, tabsIntent.intent)

        // The proxy activity will be in a different task due to its taskAffinity
        addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
      }

      currentActivity.startActivity(proxyIntent)
      return
    }

    // Launcher Custom Tab without proxy activity
    tabsIntent.intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
    if (!options.showInRecents) {
      tabsIntent.intent.addFlags(Intent.FLAG_ACTIVITY_EXCLUDE_FROM_RECENTS)
      tabsIntent.intent.addFlags(Intent.FLAG_ACTIVITY_NO_HISTORY)
    }

    tabsIntent.launchUrl(currentActivity, url)
  }

// endregion

// region Private helpers

  /**
   * @throws PackageManagerNotFoundException
   */
  private fun getResolvingActivities(customTabsIntent: CustomTabsIntent): List<ResolveInfo> {
    return packageManager.queryIntentActivities(customTabsIntent.intent, 0)
  }

  /**
   * @throws PackageManagerNotFoundException
   */
  private val packageManager: PackageManager
    get() = currentActivity.packageManager ?: throw PackageManagerNotFoundException()

  private val currentActivity: Activity
    get() = appContext.throwingActivity

// endregion
}

private inline fun <T, R> Collection<T>.mapToDistinctArrayList(mapper: (T) -> R): ArrayList<R> {
  val resultSet = LinkedHashSet<R>()
  for (element in this) {
    resultSet.add(mapper.invoke(element))
  }
  return ArrayList(resultSet)
}

private fun createDefaultCustomTabsIntent(): CustomTabsIntent {
  val customTabsIntent = CustomTabsIntent.Builder().build()
  return customTabsIntent.apply {
    intent.data = DUMMY_URL.toUri()
  }
}

private fun createDefaultCustomTabsServiceIntent() = Intent().apply {
  action = CustomTabsService.ACTION_CUSTOM_TABS_CONNECTION
}