com.apollographql.apollo3.gradle.api.androidProject.kt Maven / Gradle / Ivy

Go to download
package com.apollographql.apollo3.gradle.api

import com.android.build.gradle.AppExtension
import com.android.build.gradle.BaseExtension
import com.android.build.gradle.LibraryExtension
import com.android.build.gradle.TestedExtension
import com.android.build.gradle.api.ApplicationVariant
import com.android.build.gradle.api.BaseVariant
import com.android.build.gradle.api.LibraryVariant
import com.android.build.gradle.api.TestVariant
import com.android.build.gradle.api.UnitTestVariant
import org.gradle.api.DomainObjectSet
import org.gradle.api.Project

object AndroidProject {
  fun onEachVariant(project: Project, withTestVariants: Boolean = false, block: (BaseVariant) -> Unit) {
    project.applicationVariants?.all {
      block(it)
    }
    project.libraryVariants?.all {
      block(it)
    }

    if (withTestVariants) {
      project.testVariants?.all {
        block(it)
      }
      project.unitTestVariants?.all {
        block(it)
      }
    }
  }
}

val Project.androidExtension
  get() = extensions.findByName("android") as? BaseExtension

val Project.androidExtensionOrThrow
  get() = androidExtension ?: throw IllegalStateException("Apollo: no 'android' extension found. Did you apply the Android plugin?")

val Project.libraryVariants: DomainObjectSet?
  get() {
    return (androidExtensionOrThrow as? LibraryExtension)
        ?.libraryVariants
  }

val Project.applicationVariants: DomainObjectSet?
  get() {
    return (androidExtensionOrThrow as? AppExtension)
        ?.applicationVariants
  }

val Project.unitTestVariants: DomainObjectSet?
  get() {
    return (androidExtensionOrThrow as? TestedExtension)
        ?.unitTestVariants
  }

val Project.testVariants: DomainObjectSet?
  get() {
    return (androidExtensionOrThrow as? TestedExtension)
        ?.testVariants
  }