fr.speekha.httpmocker.sax.ScenarioHandler.kt Maven / Gradle / Ivy

/*
 * Copyright 2019-2021 David Blanc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package fr.speekha.httpmocker.sax

import fr.speekha.httpmocker.getLogger
import fr.speekha.httpmocker.model.Matcher
import fr.speekha.httpmocker.sax.builders.BodyBuilder
import fr.speekha.httpmocker.sax.builders.CaseBuilder
import fr.speekha.httpmocker.sax.builders.ErrorBuilder
import fr.speekha.httpmocker.sax.builders.HeaderBuilder
import fr.speekha.httpmocker.sax.builders.NodeBuilder
import fr.speekha.httpmocker.sax.builders.NodeWithBody
import fr.speekha.httpmocker.sax.builders.NodeWithHeaders
import fr.speekha.httpmocker.sax.builders.ParamBuilder
import fr.speekha.httpmocker.sax.builders.RequestBuilder
import fr.speekha.httpmocker.sax.builders.ResponseBuilder
import fr.speekha.httpmocker.sax.builders.ScenariosBuilder
import fr.speekha.httpmocker.sax.builders.UrlBuilder
import org.xml.sax.Attributes
import org.xml.sax.helpers.DefaultHandler
import java.util.*

class ScenarioHandler : DefaultHandler() {

    private val logger = getLogger()

    private val buildingStack = LinkedList()

    private lateinit var scenarios: ScenariosBuilder

    override fun startDocument() {
        scenarios = ScenariosBuilder()
    }

    @Suppress("TooGenericExceptionCaught")
    override fun startElement(
        uri: String?,
        localName: String?,
        qName: String?,
        attributes: Attributes?
    ) = try {
        val builder = getNodeBuilder(qName, buildingStack.peek(), attributes)
        buildingStack.push(builder)
    } catch (e: Throwable) {
        logger.error("Invalid XML", e)
        throw IllegalStateException("Invalid XML input", e)
    }

    @Suppress("UnsafeCast", "ComplexMethod")
    private fun getNodeBuilder(
        qName: String?,
        parent: NodeBuilder?,
        attributes: Attributes?
    ): NodeBuilder = when (qName) {
        "scenarios" -> scenarios
        "case" -> CaseBuilder(scenarios)
        "request" -> RequestBuilder(parent as CaseBuilder, attributes)
        "response" -> ResponseBuilder(parent as CaseBuilder, attributes)
        "error" -> ErrorBuilder(parent as CaseBuilder, attributes)
        "url" -> UrlBuilder(parent as RequestBuilder, attributes)
        "header" -> HeaderBuilder(parent as NodeWithHeaders, attributes)
        "param" -> ParamBuilder(parent as UrlBuilder, attributes)
        "body" -> BodyBuilder(parent as NodeWithBody, attributes)
        else -> error("Unkown tag $qName")
    }

    override fun endElement(uri: String?, localName: String?, qName: String?) {
        if (qName?.isNotEmpty() == true) {
            buildingStack.pop().build()
        }
    }

    override fun characters(ch: CharArray, start: Int, length: Int) {
        buildingStack.peek().addTextContent(String(ch, start, length))
    }

    fun getScenarios(): List = scenarios.build()
}