erik/kobalt
erik/kobalt
1
0
Fork 0
mirror of https://github.com/ethauvin/kobalt.git synced 2025-04-26 08:27:12 -07:00
Code Releases Activity

Refactor parameter extraction.

Browse source
This commit is contained in:
Cedric Beust 2017-04-03 10:00:18 -07:00
parent bb2eb19876
commit 0f7bb021c3
1 changed files with 25 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

47
src/main/kotlin/com/beust/kobalt/app/remote/GetDependencyGraphHandler.kt
View file

@ -22,12 +22,6 @@ class GetDependencyGraphHandler : WebSocketListener {
// so I have to do dependency injections manually :-(
val projectFinder = Kobalt.INJECTOR.getInstance(ProjectFinder::class.java)
// URL parameters sent by the client
val PARAMETER_PROJECT_ROOT = "projectRoot"
val PARAMETER_BUILD_FILE = "buildFile" // Deprecated
val PARAMETER_PROFILES = "profiles"
val PARAMETER_DOWNLOAD_SOURCES = "downloadSources"
var session: Session? = null
override fun onWebSocketClose(code: Int, reason: String?) {
@ -46,30 +40,39 @@ class GetDependencyGraphHandler : WebSocketListener {
endpoint.sendString(json)
}
private fun findProfiles(map: Map<String, List<String>>) = map[PARAMETER_PROFILES]?.getOrNull(0)
private fun findDownloadSources(map: Map<String, List<String>>) = map[PARAMETER_DOWNLOAD_SOURCES]
?.getOrNull(0)?.toBoolean() ?: false
/**
* Convenience class to extract the parameters from the WebSocket connection.
*/
class ParameterExtractor(val map: Map<String, List<String>>) {
// URL parameters sent by the client
private val PARAMETER_PROJECT_ROOT = "projectRoot"
private val PARAMETER_BUILD_FILE = "buildFile" // Deprecated
private val PARAMETER_PROFILES = "profiles"
private val PARAMETER_DOWNLOAD_SOURCES = "downloadSources"
private fun findBuildFile(map: Map<String, List<String>>) : BuildSources? {
val profiles = map[PARAMETER_PROFILES]?.getOrNull(0)
val downloadSources = map[PARAMETER_DOWNLOAD_SOURCES]?.getOrNull(0)?.toBoolean() ?: false
val projectRoot = map[PARAMETER_PROJECT_ROOT]
val buildFile = map[PARAMETER_BUILD_FILE]
val result =
if (projectRoot != null) {
BuildSources(File(projectRoot[0]))
} else if (buildFile != null) {
BuildSources(File(buildFile[0]))
} else {
null
val buildFile: BuildSources?
get() {
val bf = map[PARAMETER_BUILD_FILE]
return if (projectRoot != null) {
BuildSources(File(projectRoot[0]))
} else if (bf != null) {
BuildSources(File(bf[0]))
} else {
null
}
}
return result
}
override fun onWebSocketConnect(s: Session) {
session = s
val parameterMap = s.upgradeRequest.parameterMap
val buildSources = findBuildFile(parameterMap)
val profiles = findProfiles(parameterMap)
val downloadSources = findDownloadSources(s.upgradeRequest.parameterMap)
val parameters = ParameterExtractor(parameterMap)
val buildSources = parameters.buildFile
val profiles = parameters.profiles
val downloadSources = parameters.downloadSources
fun <T> getInstance(cls: Class<T>) : T = Kobalt.INJECTOR.getInstance(cls)