refactor(youtube/resource-mapping): move to appropriate package

2022-11-21 00:25:30 +01:00
parent aaa820430b
commit 67e0db5ac9
17 changed files with 17 additions and 16 deletions
@@ -0,0 +1,73 @@
+package app.revanced.patches.shared.mapping.patch
+
+import app.revanced.patcher.annotation.Description
+import app.revanced.patcher.annotation.Name
+import app.revanced.patcher.annotation.Version
+import app.revanced.patcher.data.ResourceContext
+import app.revanced.patcher.patch.PatchResult
+import app.revanced.patcher.patch.PatchResultSuccess
+import app.revanced.patcher.patch.ResourcePatch
+import org.w3c.dom.Element
+import java.util.*
+import java.util.concurrent.Executors
+import java.util.concurrent.TimeUnit
+
+
+@Name("resource-mapping")
+@Description("Creates a map of public resources.")
+@Version("0.0.1")
+class ResourceMappingPatch : ResourcePatch {
+    companion object {
+        internal lateinit var resourceMappings: List<ResourceElement>
+            private set
+
+        private val THREAD_COUNT = Runtime.getRuntime().availableProcessors()
+        private val threadPoolExecutor = Executors.newFixedThreadPool(THREAD_COUNT)
+    }
+
+    override fun execute(context: ResourceContext): PatchResult {
+        // save the file in memory to concurrently read from
+        val resourceXmlFile = context["res/values/public.xml"].readBytes()
+
+        // create a synchronized list to store the resource mappings
+        val mappings = Collections.synchronizedList(mutableListOf<ResourceElement>())
+
+        for (threadIndex in 0 until THREAD_COUNT) {
+            threadPoolExecutor.execute thread@{
+                context.xmlEditor[resourceXmlFile.inputStream()].use { editor ->
+                    val resources = editor.file.documentElement.childNodes
+                    val resourcesLength = resources.length
+                    val jobSize = resourcesLength / THREAD_COUNT
+
+                    val batchStart = jobSize * threadIndex
+                    val batchEnd = jobSize * (threadIndex + 1)
+                    element@ for (i in batchStart until batchEnd) {
+                        // make sure to not to go out of bounds when rounding errors occur at calculating the jobSize
+                        if (i >= resourcesLength) return@thread
+
+                        val node = resources.item(i)
+                        if (node !is Element) continue
+
+                        val nameAttribute = node.getAttribute("name")
+                        val typeAttribute = node.getAttribute("type")
+
+                        if (node.nodeName != "public" || nameAttribute.startsWith("APKTOOL")) continue
+
+                        val id = node.getAttribute("id").substring(2).toLong(16)
+                        mappings.add(ResourceElement(typeAttribute, nameAttribute, id))
+                    }
+                }
+            }
+        }
+
+        threadPoolExecutor
+            .also { it.shutdown() }
+            .awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS)
+
+        resourceMappings = mappings
+
+        return PatchResultSuccess()
+    }
+}
+
+data class ResourceElement(val type: String, val name: String, val id: Long)