pl.allegro.mobile.logic.operators.OperatorFactory.kt Maven / Gradle / Ivy

Go to download
package pl.allegro.mobile.logic.operators

import pl.allegro.mobile.logic.ClientLogicElement
import pl.allegro.mobile.logic.ClientLogicOperator
import pl.allegro.mobile.logic.ListOfClientElements

internal abstract class OperatorFactory(protected val name: String) {
    fun create(vararg elements: ClientLogicElement): ClientLogicElement =
        ClientLogicOperator.Builder(name).addAll(elements).build()

    fun create(elements: ListOfClientElements) =
        ClientLogicOperator.Builder(name).addAll(elements.values).build()
}

// allows to flatten {"+" [1, {"+":[3, 4]}]} to {"+": [1, 3, 4]}
internal abstract class FlattenableOperatorFactory(protected val name: String) {
    fun create(vararg elements: ClientLogicElement): ClientLogicElement =
        ClientLogicOperator.Builder(name)
            .addAll(elements.map { flattenArgumentsIfTheSameOperator(it) }.flatten())
            .build()

    fun create(elements: List): ClientLogicElement =
        ClientLogicOperator.Builder(name)
            .addAll(elements.map { flattenArgumentsIfTheSameOperator(it) }.flatten())
            .build()

    private fun flattenArgumentsIfTheSameOperator(element: ClientLogicElement): List {
        return if (element is ClientLogicOperator && element.name == name) {
            element.arguments
        } else {
            listOf(element)
        }
    }
}