mlrpc.protocol.2.0.3.source-code.Util.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of protocol Show documentation

xml-rpc implementation written in Kotlin

The newest version!

/*
 * Copyright (c) 2022 Andreas Scheja. Use of this source code is governed by the Apache 2.0 license.
 */

package org.ascheja.xmlrpc.protocol

import org.w3c.dom.Document
import java.io.ByteArrayOutputStream
import java.io.OutputStream
import javax.xml.parsers.DocumentBuilder
import javax.xml.parsers.DocumentBuilderFactory
import javax.xml.transform.OutputKeys
import javax.xml.transform.Transformer
import javax.xml.transform.TransformerFactory
import javax.xml.transform.dom.DOMSource
import javax.xml.transform.stream.StreamResult

private val declarationBytes = "".toByteArray()

internal fun createDocumentBuilder(): DocumentBuilder {
    return DocumentBuilderFactory.newInstance().newDocumentBuilder()
}

internal fun createDefaultTransformer(prettyPrint: Boolean): Transformer {
    return TransformerFactory.newInstance().newTransformer().apply {
        if (prettyPrint) {
            setOutputProperty(OutputKeys.INDENT, "yes")
        }
        setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes")
    }
}

public fun Document.writeToByteArray(prettyPrint: Boolean = true): ByteArray {
    return ByteArrayOutputStream().use { out ->
        writeTo(out, prettyPrint)
        out.toByteArray()
    }
}

public fun Document.writeTo(out: OutputStream, prettyPrint: Boolean = true) {
    out.write(declarationBytes)
    if (prettyPrint) out.write(System.lineSeparator().toByteArray())
    createDefaultTransformer(prettyPrint).transform(DOMSource(this), StreamResult(out))
}

public fun Document.writeToString(prettyPrint: Boolean = true): String {
    return writeToByteArray(prettyPrint).toString(Charsets.UTF_8)
}