commonMain.io.ktor.http.parsing.GrammarBuilder.kt Maven / Gradle / Ivy

Go to download
/*
* Copyright 2014-2021 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
*/

package io.ktor.http.parsing

internal class GrammarBuilder {
    private val grammars = mutableListOf()

    infix fun then(grammar: Grammar): GrammarBuilder {
        grammars += grammar
        return this
    }

    infix fun then(value: String): GrammarBuilder {
        grammars += StringGrammar(value)
        return this
    }

    operator fun (() -> Grammar).unaryPlus() {
        grammars += this()
    }

    operator fun Grammar.unaryPlus() {
        grammars += this
    }

    operator fun String.unaryPlus() {
        grammars += StringGrammar(this)
    }

    public fun build(): Grammar = if (grammars.size == 1) grammars.first() else SequenceGrammar(grammars)
}

internal fun grammar(block: GrammarBuilder.() -> Unit): Grammar = GrammarBuilder().apply(block).build()