Function registerBundleLlamaAssetsTask

app/build.gradle.kts:614–705

}

fun registerBundleLlamaAssetsTask(
	flavor: String,
	arch: String,
): TaskProvider<Task> {
	val capitalized =
		flavor.replaceFirstChar { if (it.isLowerCase()) it.titlecase(Locale.ROOT) else it.toString() }
	return tasks.register<Task>("bundle${capitalized}LlamaAssets") {
		dependsOn("assemble${capitalized}Assets")

		val assetsZipFile = project.layout.buildDirectory.file("outputs/assets/assets-$arch.zip")
		val outputDir =
			rootProject.layout.projectDirectory.dir("assets/release/$flavor/dynamic_libs")
		inputs.file(assetsZipFile)
		outputs.dir(outputDir)

		doLast {
			val assetsZip =
				project.layout.buildDirectory
					.file("outputs/assets/assets-$arch.zip")
					.get()
					.asFile
			if (!assetsZip.exists()) {
				throw GradleException("Assets zip not found: ${assetsZip.absolutePath}. Run assemble${capitalized}Assets first.")
			}

			val tempAar = Files.createTempFile("llama-$flavor", ".aar").toFile()
			var found = false
			ZipInputStream(assetsZip.inputStream()).use { zis ->
				var entry = zis.nextEntry
				while (entry != null) {
					if (entry.name == "dynamic_libs/llama.aar") {
						tempAar.outputStream().use { zis.copyTo(it) }
						found = true
						break
					}
					entry = zis.nextEntry
				}
			}

			if (!found) {
				tempAar.delete()
				throw GradleException("dynamic_libs/llama.aar not found inside ${assetsZip.name}")
			}

			val targetDir = project.rootProject.file("assets/release/$flavor/dynamic_libs")
			targetDir.mkdirs()
			val destBr = File(targetDir, "llama-$flavor.aar.br")
			val destAar = File(targetDir, "llama-$flavor.aar")

			destBr.delete()
			destAar.delete()

			val brotliAvailable =
				try {
					val result =
						project.exec {
							commandLine("brotli", "--version")
							isIgnoreExitValue = true