io.javalin.routing.PathMatcher.kt Maven / Gradle / Ivy

Go to download

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

There is a newer version: 6.3.0

/*
 * Javalin - https://javalin.io
 * Copyright 2017 David Åse
 * Licensed under Apache 2.0: https://github.com/tipsy/javalin/blob/master/LICENSE
 */

package io.javalin.routing

import io.javalin.http.HandlerType
import java.util.*

class PathMatcher {

    private val handlerEntries: Map> =
        HandlerType.values().associateWithTo(EnumMap(HandlerType::class.java)) { arrayListOf() }

    fun add(entry: HandlerEntry) {
        if (entry.type.isHttpMethod() && handlerEntries[entry.type]!!.find { it.type == entry.type && it.path == entry.path } != null) {
            throw IllegalArgumentException("Handler with type='${entry.type}' and path='${entry.path}' already exists.")
        }
        handlerEntries[entry.type]!!.add(entry)
    }

    fun findEntries(handlerType: HandlerType, requestUri: String) =
        handlerEntries[handlerType]!!.filter { he -> match(he, requestUri) }

    internal fun hasEntries(handlerType: HandlerType, requestUri: String): Boolean =
        handlerEntries[handlerType]!!.any { entry -> match(entry, requestUri) }

    internal fun getAllEntriesOfType(handlerType: HandlerType) =
        handlerEntries[handlerType]!!

    private fun match(entry: HandlerEntry, requestPath: String): Boolean = when (entry.path) {
        "*" -> true
        requestPath -> true
        else -> entry.matches(requestPath)
    }

}