com.xml.guard.tasks.PackageChangeTask.kt Maven / Gradle / Ivy

Go to download
package com.xml.guard.tasks

import com.xml.guard.entensions.GuardExtension
import com.xml.guard.utils.*
import org.gradle.api.DefaultTask
import org.gradle.api.Project
import org.gradle.api.tasks.TaskAction
import java.io.File
import javax.inject.Inject

/**
 * User: ljx
 * Date: 2022/2/25
 * Time: 19:06
 */
open class PackageChangeTask @Inject constructor(
    private val guardExtension: GuardExtension,
    private val variantName: String,
) : DefaultTask() {

    init {
        group = "guard"
    }

    @TaskAction
    fun execute() {
        val packageExtension = guardExtension.packageChange
        if (packageExtension.isEmpty()) return

        packageExtension.forEach { (key, value) ->
            key.split(".").forEach { text ->
                if (text.inBlackList()) {
                    throw IllegalArgumentException("`${key.trim()}` is illegal, $text is a blacklist")
                }
            }

            value.split(".").forEach { text ->
                if (text.inBlackList()) {
                    throw IllegalArgumentException("`${value.trim()}` is illegal, $text is a blacklist")
                }
            }
        }

        val androidProjects = allDependencyAndroidProjects(project)
        androidProjects.forEach { it.changePackage(packageExtension) }
    }

    private fun Project.changePackage(map: Map) {
        val oldPackage = findPackage()
        val newPackage = map[oldPackage] ?: return
        val dirs = findXmlDirs(variantName, "layout")
        dirs.add(manifestFile())
        dirs.add(buildFile)
        // 1、修改layout文件、AndroidManifest文件、build.gradle文件
        files(dirs).asFileTree.forEach { file ->
            when {
                // 修改AndroidManifest.xml文件
                file.isAndroidManifest -> file.modifyManifestFile(oldPackage, newPackage)
                // 修改 build.gradle namespace
                file.name == buildFile.name -> file.modifyBuildGradleFile(oldPackage, newPackage)
                // 修改layout文件
                else -> file.modifyLayoutXml(oldPackage)
            }
        }

        val javaDirs = javaDirs(variantName)
        // 2.修改 kt/java文件
        files(javaDirs).asFileTree.forEach { javaFile ->
            var replaceText = javaFile.readText()
            if (replaceText.contains("${oldPackage}.*")) {
                replaceText = javaFile.insertImportXxxIfAbsent(oldPackage)
            }
            replaceText
                .replaceWords("$oldPackage.R", "$newPackage.R")
                .replaceWords("$oldPackage.BR", "$newPackage.BR")
                .replaceWords("$oldPackage.BuildConfig", "$newPackage.BuildConfig")
                .replaceWords("$oldPackage.databinding", "$newPackage.databinding")
                .let { javaFile.writeText(it) }
        }

        // 3.对旧包名下的直接子类，检测R类、BuildConfig类是否有用到，有的话，插入import语句
        val oldPackagePath = oldPackage.replace(".", File.separator)
        javaDirs.forEach {
            File(it, oldPackagePath).listFiles { f -> f.isFile }?.forEach { file ->
                file.insertImportXxxIfAbsent(newPackage)
            }
        }
    }

    // 修改AndroidManifest.xml文件，并返回新旧包名
    private fun File.modifyManifestFile(oldPackage: String, newPackage: String) {
        readText()
            .replaceWords("""package="$oldPackage"""", """package="$newPackage"""")
            .replaceWords("""android:name=".""", """android:name="$oldPackage.""")
            .let { writeText(it) }
    }

    // 修复build.gradle文件的 namespace 语句
    private fun File.modifyBuildGradleFile(oldPackage: String, newPackage: String) {
        readText()
            .replace("namespace\\s+['\"]${oldPackage}['\"]".toRegex(), "namespace '$newPackage'")
            .replace(
                "namespace\\s*=\\s*['\"]${oldPackage}['\"]".toRegex(),
                """namespace = "$newPackage""""
            )
            .let { writeText(it) }
    }

    // 修改layout文件
    private fun File.modifyLayoutXml(oldPackage: String) {
        readText()
            .replaceWords("""tools:context=".""", """tools:context="$oldPackage.""")
            .replaceWords("""app:layoutManager=".""", """app:layoutManager="$oldPackage.""")
            .let { writeText(it) }
    }
}