package cc.unitmesh.idea

object MvcUtil {
    fun isController(fileName: String, lang: String): Boolean {
        return fileName.endsWith("Controller.${lang.lowercase()}")
    }

    fun isService(fileName: String, lang: String) =
        fileName.endsWith("Service.$${lang.lowercase()}") || fileName.endsWith("ServiceImpl.$${lang.lowercase()}")

    fun isRepository(fileName: String, lang: String) =
        fileName.endsWith("Repository.$${lang.lowercase()}") || fileName.endsWith("Repo.$${lang.lowercase()}")
}