From 6f72c4c4c051e48c8d03d2a7b2cfc1c53028ed86 Mon Sep 17 00:00:00 2001
From: oSumAtrIX <johan.melkonyan1@web.de>
Date: Sat, 2 Jul 2022 17:44:08 +0200
Subject: [PATCH] feat: streams overload for `XmlFileHolder`

---
 .../revanced/patcher/data/impl/ResourceData.kt | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/main/kotlin/app/revanced/patcher/data/impl/ResourceData.kt b/src/main/kotlin/app/revanced/patcher/data/impl/ResourceData.kt
index b4f26202..fddabde6 100644
--- a/src/main/kotlin/app/revanced/patcher/data/impl/ResourceData.kt
+++ b/src/main/kotlin/app/revanced/patcher/data/impl/ResourceData.kt
@@ -4,6 +4,8 @@ import app.revanced.patcher.data.Data
 import org.w3c.dom.Document
 import java.io.Closeable
 import java.io.File
+import java.io.InputStream
+import java.io.OutputStream
 import javax.xml.parsers.DocumentBuilderFactory
 import javax.xml.transform.TransformerFactory
 import javax.xml.transform.dom.DOMSource
@@ -17,14 +19,20 @@ class ResourceData(private val resourceCacheDirectory: File) : Data, Iterable<Fi
     override fun iterator() = resourceCacheDirectory.walkTopDown().iterator()
 
     inner class XmlFileHolder {
+        operator fun get(inputStream: InputStream, outputStream: OutputStream) =
+            DomFileEditor(inputStream, outputStream)
+
         operator fun get(path: String) = DomFileEditor(this@ResourceData[path])
     }
 }
 
-class DomFileEditor internal constructor(private val domFile: File) : Closeable {
-    val file: Document = DocumentBuilderFactory.newInstance().newDocumentBuilder()
-        .parse(domFile).also(Document::normalize)
+class DomFileEditor internal constructor(inputStream: InputStream, private val outputStream: OutputStream) : Closeable {
+    constructor(file: File) : this(file.inputStream(), file.outputStream())
+
+    val file: Document =
+        DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(inputStream).also(Document::normalize)
+
+    override fun close() =
+        TransformerFactory.newInstance().newTransformer().transform(DOMSource(file), StreamResult(outputStream))
 
-    override fun close() = TransformerFactory.newInstance().newTransformer()
-        .transform(DOMSource(file), StreamResult(domFile.outputStream()))
 }