com.didiglobal.booster.task.compression.pngquant.PngquantCompressFlatImages.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of booster-task-compression-pngquant Show documentation
booster-task-compression-pngquant
There is a newer version: 5.0.0-alpha3
Show newest version
package com.didiglobal.booster.task.compression.pngquant

import com.android.SdkConstants.DOT_PNG
import com.android.SdkConstants.FD_RES
import com.android.SdkConstants.FD_RES_DRAWABLE
import com.android.SdkConstants.FD_RES_MIPMAP
import com.android.builder.model.AndroidProject.FD_INTERMEDIATES
import com.android.sdklib.BuildToolInfo
import com.didiglobal.booster.aapt2.metadata
import com.didiglobal.booster.compression.CompressionResult
import com.didiglobal.booster.compression.task.Aapt2ActionData
import com.didiglobal.booster.gradle.buildTools
import com.didiglobal.booster.gradle.project
import com.didiglobal.booster.kotlinx.CSI_RED
import com.didiglobal.booster.kotlinx.CSI_RESET
import com.didiglobal.booster.kotlinx.file
import org.gradle.api.tasks.CacheableTask
import org.gradle.api.tasks.OutputDirectory
import java.io.File

/**
 * Represents a task for compiled image compression using pngquant
 *
 * @author johnsonlee
 */
@CacheableTask
internal abstract class PngquantCompressFlatImages : AbstractPngquantCompressImages() {

    private val intermediates: File
        get() = variant.project.buildDir.file(FD_INTERMEDIATES)

    @get:OutputDirectory
    val compressedRes: File
        get() = intermediates.file("compressed_${FD_RES}_pngquant", variant.dirName, this.name)

    @get:OutputDirectory
    val compiledRes: File
        get() = intermediates.file("compiled_${FD_RES}_pngquant", variant.dirName, this.name)

    override fun compress() {
        val pngquant = this.compressor.canonicalPath
        val aapt2 = variant.buildTools.getPath(BuildToolInfo.PathId.AAPT2)

        compiledRes.mkdirs()
        compressedRes.file(FD_RES_MIPMAP).mkdirs()
        compressedRes.file(FD_RES_DRAWABLE).mkdirs()

        images().parallelStream().map {
            it to it.metadata
        }.filter(this::includes).map {
            val output = compressedRes.file("${it.second.resourcePath.substringBeforeLast('.')}$DOT_PNG")
            Aapt2ActionData(it.first, it.second, output,
                    listOf(pngquant, "--strip", "--skip-if-larger", "-f", "-o", output.canonicalPath, "-s", "${options.speed}", "-Q", "${options.quality}-100", it.second.sourcePath),
                    listOf(aapt2, "compile", "-o", it.first.parent, output.canonicalPath))
        }.forEach {
            it.output.parentFile.mkdirs()
            val s0 = File(it.metadata.sourcePath).length()
            val rc = project.exec { spec ->
                spec.isIgnoreExitValue = true
                spec.commandLine = it.cmdline
            }
            when (rc.exitValue) {
                0 -> {
                    val s1 = it.output.length()
                    val rcAapt2 = project.exec { spec ->
                        spec.isIgnoreExitValue = true
                        spec.commandLine = it.aapt2
                    }

                    if (0 == rcAapt2.exitValue) {
                        results.add(CompressionResult(it.input, s0, s1, File(it.metadata.sourcePath)))
                    } else {
                        logger.error("${CSI_RED}Command `${it.aapt2.joinToString(" ")}` exited with non-zero value ${rc.exitValue}$CSI_RESET")
                        results.add(CompressionResult(it.input, s0, s0, File(it.metadata.sourcePath)))
                        rcAapt2.assertNormalExitValue()
                    }
                }
                else -> {
                    logger.error("${CSI_RED}Command `${it.cmdline.joinToString(" ")}` exited with non-zero value ${rc.exitValue}$CSI_RESET")
                    results.add(CompressionResult(it.input, s0, s0, File(it.metadata.sourcePath)))
                }
            }
        }
    }

}