autodimension.gradle.kts

import org.gradle.api.DefaultTask
import org.gradle.api.tasks.Input
import org.gradle.api.tasks.TaskAction
import org.gradle.kotlin.dsl.register
import org.w3c.dom.Element
import org.w3c.dom.Node
import java.io.File
import java.io.PrintWriter
import java.util.*
import javax.xml.parsers.DocumentBuilderFactory

enum class DimenType {
    /**
     * Best for auto generate the dimension by the default values/dimens.xml file.
     */
    FROM_DIMEN_FILE,

    /**
     * Best for auto generate the dimension as default.
     */
    AUTO_CREATE
}

open class DimenFactory : DefaultTask() {
    @Input
    private val dimens = intArrayOf(
        320,
        360,
        384,
        411,
        480,
        540,
        600,
        720,
        800,
        960,
        1024,
        1080,
        1280,
        1440,
        2560,
        3840
    )

    @Input
    private val fromDimen = 360f

    @Input
    private val positiveMaxDP = 400

    @Input
    private val positiveMaxSP = 60

    @Input
    private val negativeMaxDP = 60

    @Input
    private val type = DimenType.AUTO_CREATE

    @Input
    private val dimenFileName = "values/dimens.xml"

    private val resFolder = project.projectDir.path + "/src/main/res/"

    @TaskAction
    fun create() {
        when (type) {
            DimenType.AUTO_CREATE -> autoCreateDimen()
            DimenType.FROM_DIMEN_FILE -> createDimenFromDimenFile()
        }
    }

    private fun createDimenFromDimenFile() {
        println("Start convert dimen from value file to other screen size")
        val path = resFolder + dimenFileName
        val pairs = HashMap<String, String>()
        try {
            val fXmlFile = File(path)
            val dbFactory = DocumentBuilderFactory.newInstance()
            val dBuilder = dbFactory.newDocumentBuilder()
            val doc = dBuilder.parse(fXmlFile)
            doc.documentElement.normalize()

            val nList = doc.getElementsByTagName("dimen")
            for (i in 0 until nList.length) {
                val nNode = nList.item(i)
                if (nNode.nodeType == Node.ELEMENT_NODE) {
                    val eElement = nNode as Element
                    if (eElement.textContent.contains("@dimen")) continue
                    pairs[eElement.getAttribute("name")] = eElement.textContent
                }
            }
            val dps = HashMap<String, Double>()
            val sps = HashMap<String, Double>()

            for (entry in pairs) {
                val value = entry.value.replace("dp", "").replace("sp", "").toDouble()
                if (entry.value.contains("dp")) {
                    dps[entry.key] = value
                } else {
                    sps[entry.key] = value
                }
            }
//Sort
            val sortedDps = TreeMap(dps)
            val sortedSps = TreeMap(sps)
            for (dimen in dimens) {
                val folder = resFolder + "values-sw" + dimen + "dp"
                val fileName = "$folder/auto_dimens.xml"
                File(folder).mkdir()
                File(fileName).createNewFile()
                val printWriter = PrintWriter(fileName)
                printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>")
                printWriter.println("<resources>")
                for (entry in sortedDps) {
                    val ratio = dimen / fromDimen
                    val dp = ratio * entry.value
                    printWriter.printf(
                        "\t<dimen name=\"%s\">%.2fdp</dimen>\r\n",
                        entry.key,
                        dp
                    )
                }
                printWriter.println()
                for (entry in sortedSps) {
                    val ratio = dimen / fromDimen
                    val sp = ratio * entry.value
                    printWriter.printf(
                        "\t<dimen name=\"%s\">%.2fsp</dimen>\r\n",
                        entry.key,
                        sp
                    )
                }
                printWriter.println("</resources>")
                printWriter.close()
            }
        } catch (e: Exception) {
            e.printStackTrace()
        }
    }

    private fun autoCreateDimen() {
// write dimen.xml
        val defaultFolder = resFolder + "values"
        val defaultDimensFile = "$defaultFolder/dimens.xml"
        File(defaultFolder).mkdir()
        File(defaultDimensFile).createNewFile()
        writeAutoDimen(defaultDimensFile, 360)

// write other auto_dimens.xml
        for (dimen in dimens) {
            val folder = resFolder + "values-sw" + dimen + "dp"
            val fileName = "$folder/auto_dimens.xml"
            File(folder).mkdir()
            File(fileName).createNewFile()
            writeAutoDimen(fileName, dimen)
        }
    }

    private fun writeAutoDimen(fileName: String, dimen: Int) {
        println("Auto create dimension file and values $fileName")
        val printWriter = PrintWriter(fileName)
        printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>")
        printWriter.println("<resources>")
        for (i in 1..positiveMaxDP) {
            val ratio = dimen / fromDimen
            val dp = ratio * i
            printWriter.printf("\t<dimen name=\"dp_%d\">%.2fdp</dimen>\r\n", i, dp)
        }
        printWriter.println()
        for (i in 1..negativeMaxDP) {
            val ratio = dimen / fromDimen
            val dp = ratio * i
            printWriter.printf("\t<dimen name=\"dp_minus%d\">%.2fdp</dimen>\r\n", i, -dp)
        }
        printWriter.println()
        for (i in 1..positiveMaxSP) {
            val ratio = dimen / dimens[0].toFloat()
            val sp = ratio * i
            printWriter.printf("\t<dimen name=\"sp_%d\">%.2fsp</dimen>\r\n", i, sp)
        }
        printWriter.println("</resources>")
        printWriter.close()
    }
}

tasks.register("createDimen", DimenFactory::class) {
    create()
}