├── .gitignore ├── .idea ├── .gitignore ├── .name ├── codeStyles │ ├── Project.xml │ └── codeStyleConfig.xml ├── compiler.xml ├── gradle.xml ├── jarRepositories.xml ├── libraries-with-intellij-classes.xml ├── misc.xml ├── modules │ └── ktor-chat.iml └── uiDesigner.xml ├── build.gradle.kts ├── gradle.properties ├── gradle └── wrapper │ ├── gradle-wrapper.jar │ └── gradle-wrapper.properties ├── gradlew ├── gradlew.bat ├── settings.gradle.kts └── src ├── main ├── kotlin │ └── com │ │ └── plcoding │ │ ├── Application.kt │ │ ├── data │ │ ├── MessageDataSource.kt │ │ ├── MessageDataSourceImpl.kt │ │ └── model │ │ │ └── Message.kt │ │ ├── di │ │ └── MainModule.kt │ │ ├── plugins │ │ ├── Monitoring.kt │ │ ├── Routing.kt │ │ ├── Security.kt │ │ ├── Serialization.kt │ │ └── Sockets.kt │ │ ├── room │ │ ├── Member.kt │ │ ├── MemberAlreadyExistsException.kt │ │ └── RoomController.kt │ │ ├── routes │ │ └── ChatRoutes.kt │ │ └── session │ │ └── ChatSession.kt └── resources │ ├── application.conf │ └── logback.xml └── test └── kotlin └── com └── plcoding └── ApplicationTest.kt /.gitignore: -------------------------------------------------------------------------------- 1 | # Project exclude paths 2 | /.gradle/ 3 | /build/ -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/.name: -------------------------------------------------------------------------------- 1 | com.plcoding.ktor-chat -------------------------------------------------------------------------------- /.idea/codeStyles/Project.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 6 | 7 | 17 | 18 | -------------------------------------------------------------------------------- /.idea/codeStyles/codeStyleConfig.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 5 | -------------------------------------------------------------------------------- /.idea/compiler.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | -------------------------------------------------------------------------------- /.idea/gradle.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 15 | 16 | -------------------------------------------------------------------------------- /.idea/jarRepositories.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 9 | 10 | 14 | 15 | 19 | 20 | -------------------------------------------------------------------------------- /.idea/libraries-with-intellij-classes.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 64 | 65 | -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /.idea/modules/ktor-chat.iml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | -------------------------------------------------------------------------------- /.idea/uiDesigner.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | 21 | 22 | 23 | 24 | 25 | 26 | 27 | 28 | 29 | 30 | 31 | 32 | 33 | 34 | 35 | 36 | 37 | 38 | 39 | 40 | 41 | 42 | 43 | 44 | 45 | 46 | 47 | 48 | 49 | 50 | 51 | 52 | 53 | 54 | 55 | 56 | 57 | 58 | 59 | 60 | 61 | 62 | 63 | 64 | 65 | 66 | 67 | 68 | 69 | 70 | 71 | 72 | 73 | 74 | 75 | 76 | 77 | 78 | 79 | 80 | 81 | 82 | 83 | 84 | 85 | 86 | 87 | 88 | 89 | 90 | 91 | 92 | 93 | 94 | 95 | 96 | 97 | 98 | 99 | 100 | 101 | 102 | 103 | 104 | 105 | 106 | 107 | 108 | 109 | 110 | 111 | 112 | 113 | 114 | 115 | 116 | 117 | 118 | 119 | 120 | 121 | 122 | 123 | 124 | -------------------------------------------------------------------------------- /build.gradle.kts: -------------------------------------------------------------------------------- 1 | val ktor_version: String by project 2 | val kotlin_version: String by project 3 | val logback_version: String by project 4 | val kmongo_version: String by project 5 | val koin_version: String by project 6 | 7 | plugins { 8 | application 9 | kotlin("jvm") version "1.5.31" 10 | id("org.jetbrains.kotlin.plugin.serialization") version "1.6.0" 11 | } 12 | 13 | group = "com.plcoding" 14 | version = "0.0.1" 15 | application { 16 | mainClass.set("io.ktor.server.netty.EngineMain") 17 | } 18 | 19 | repositories { 20 | mavenCentral() 21 | } 22 | 23 | dependencies { 24 | implementation("io.ktor:ktor-server-core:$ktor_version") 25 | implementation("io.ktor:ktor-websockets:$ktor_version") 26 | implementation("io.ktor:ktor-serialization:$ktor_version") 27 | implementation("io.ktor:ktor-server-sessions:$ktor_version") 28 | implementation("io.ktor:ktor-server-netty:$ktor_version") 29 | implementation("ch.qos.logback:logback-classic:$logback_version") 30 | testImplementation("io.ktor:ktor-server-tests:$ktor_version") 31 | testImplementation("org.jetbrains.kotlin:kotlin-test-junit:$kotlin_version") 32 | 33 | // KMongo 34 | implementation("org.litote.kmongo:kmongo:$kmongo_version") 35 | implementation("org.litote.kmongo:kmongo-coroutine:$kmongo_version") 36 | 37 | // Koin core features 38 | implementation("io.insert-koin:koin-core:$koin_version") 39 | implementation("io.insert-koin:koin-ktor:$koin_version") 40 | implementation("io.insert-koin:koin-logger-slf4j:$koin_version") 41 | } -------------------------------------------------------------------------------- /gradle.properties: -------------------------------------------------------------------------------- 1 | ktor_version=1.6.5 2 | kotlin_version=1.5.31 3 | logback_version=1.2.3 4 | koin_version=3.1.2 5 | kmongo_version=4.3.0 6 | kotlin.code.style=official 7 | -------------------------------------------------------------------------------- /gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/philipplackner/ChatApp-Server/498e656dec1510e974e5be3d97bc9b921e35ceaf/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- 1 | distributionBase=GRADLE_USER_HOME 2 | distributionPath=wrapper/dists 3 | distributionUrl=https\://services.gradle.org/distributions/gradle-6.8-bin.zip 4 | zipStoreBase=GRADLE_USER_HOME 5 | zipStorePath=wrapper/dists 6 | -------------------------------------------------------------------------------- /gradlew: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env sh 2 | 3 | # 4 | # Copyright 2015 the original author or authors. 5 | # 6 | # Licensed under the Apache License, Version 2.0 (the "License"); 7 | # you may not use this file except in compliance with the License. 8 | # You may obtain a copy of the License at 9 | # 10 | # https://www.apache.org/licenses/LICENSE-2.0 11 | # 12 | # Unless required by applicable law or agreed to in writing, software 13 | # distributed under the License is distributed on an "AS IS" BASIS, 14 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 | # See the License for the specific language governing permissions and 16 | # limitations under the License. 17 | # 18 | 19 | ############################################################################## 20 | ## 21 | ## Gradle start up script for UN*X 22 | ## 23 | ############################################################################## 24 | 25 | # Attempt to set APP_HOME 26 | # Resolve links: $0 may be a link 27 | PRG="$0" 28 | # Need this for relative symlinks. 29 | while [ -h "$PRG" ] ; do 30 | ls=`ls -ld "$PRG"` 31 | link=`expr "$ls" : '.*-> \(.*\)$'` 32 | if expr "$link" : '/.*' > /dev/null; then 33 | PRG="$link" 34 | else 35 | PRG=`dirname "$PRG"`"/$link" 36 | fi 37 | done 38 | SAVED="`pwd`" 39 | cd "`dirname \"$PRG\"`/" >/dev/null 40 | APP_HOME="`pwd -P`" 41 | cd "$SAVED" >/dev/null 42 | 43 | APP_NAME="Gradle" 44 | APP_BASE_NAME=`basename "$0"` 45 | 46 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 47 | DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"' 48 | 49 | # Use the maximum available, or set MAX_FD != -1 to use that value. 50 | MAX_FD="maximum" 51 | 52 | warn () { 53 | echo "$*" 54 | } 55 | 56 | die () { 57 | echo 58 | echo "$*" 59 | echo 60 | exit 1 61 | } 62 | 63 | # OS specific support (must be 'true' or 'false'). 64 | cygwin=false 65 | msys=false 66 | darwin=false 67 | nonstop=false 68 | case "`uname`" in 69 | CYGWIN* ) 70 | cygwin=true 71 | ;; 72 | Darwin* ) 73 | darwin=true 74 | ;; 75 | MINGW* ) 76 | msys=true 77 | ;; 78 | NONSTOP* ) 79 | nonstop=true 80 | ;; 81 | esac 82 | 83 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar 84 | 85 | 86 | # Determine the Java command to use to start the JVM. 87 | if [ -n "$JAVA_HOME" ] ; then 88 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then 89 | # IBM's JDK on AIX uses strange locations for the executables 90 | JAVACMD="$JAVA_HOME/jre/sh/java" 91 | else 92 | JAVACMD="$JAVA_HOME/bin/java" 93 | fi 94 | if [ ! -x "$JAVACMD" ] ; then 95 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME 96 | 97 | Please set the JAVA_HOME variable in your environment to match the 98 | location of your Java installation." 99 | fi 100 | else 101 | JAVACMD="java" 102 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 103 | 104 | Please set the JAVA_HOME variable in your environment to match the 105 | location of your Java installation." 106 | fi 107 | 108 | # Increase the maximum file descriptors if we can. 109 | if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then 110 | MAX_FD_LIMIT=`ulimit -H -n` 111 | if [ $? -eq 0 ] ; then 112 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then 113 | MAX_FD="$MAX_FD_LIMIT" 114 | fi 115 | ulimit -n $MAX_FD 116 | if [ $? -ne 0 ] ; then 117 | warn "Could not set maximum file descriptor limit: $MAX_FD" 118 | fi 119 | else 120 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT" 121 | fi 122 | fi 123 | 124 | # For Darwin, add options to specify how the application appears in the dock 125 | if $darwin; then 126 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\"" 127 | fi 128 | 129 | # For Cygwin or MSYS, switch paths to Windows format before running java 130 | if [ "$cygwin" = "true" -o "$msys" = "true" ] ; then 131 | APP_HOME=`cygpath --path --mixed "$APP_HOME"` 132 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"` 133 | 134 | JAVACMD=`cygpath --unix "$JAVACMD"` 135 | 136 | # We build the pattern for arguments to be converted via cygpath 137 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null` 138 | SEP="" 139 | for dir in $ROOTDIRSRAW ; do 140 | ROOTDIRS="$ROOTDIRS$SEP$dir" 141 | SEP="|" 142 | done 143 | OURCYGPATTERN="(^($ROOTDIRS))" 144 | # Add a user-defined pattern to the cygpath arguments 145 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then 146 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)" 147 | fi 148 | # Now convert the arguments - kludge to limit ourselves to /bin/sh 149 | i=0 150 | for arg in "$@" ; do 151 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -` 152 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option 153 | 154 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition 155 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"` 156 | else 157 | eval `echo args$i`="\"$arg\"" 158 | fi 159 | i=`expr $i + 1` 160 | done 161 | case $i in 162 | 0) set -- ;; 163 | 1) set -- "$args0" ;; 164 | 2) set -- "$args0" "$args1" ;; 165 | 3) set -- "$args0" "$args1" "$args2" ;; 166 | 4) set -- "$args0" "$args1" "$args2" "$args3" ;; 167 | 5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;; 168 | 6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;; 169 | 7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;; 170 | 8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;; 171 | 9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;; 172 | esac 173 | fi 174 | 175 | # Escape application args 176 | save () { 177 | for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done 178 | echo " " 179 | } 180 | APP_ARGS=`save "$@"` 181 | 182 | # Collect all arguments for the java command, following the shell quoting and substitution rules 183 | eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS" 184 | 185 | exec "$JAVACMD" "$@" 186 | -------------------------------------------------------------------------------- /gradlew.bat: -------------------------------------------------------------------------------- 1 | @rem 2 | @rem Copyright 2015 the original author or authors. 3 | @rem 4 | @rem Licensed under the Apache License, Version 2.0 (the "License"); 5 | @rem you may not use this file except in compliance with the License. 6 | @rem You may obtain a copy of the License at 7 | @rem 8 | @rem https://www.apache.org/licenses/LICENSE-2.0 9 | @rem 10 | @rem Unless required by applicable law or agreed to in writing, software 11 | @rem distributed under the License is distributed on an "AS IS" BASIS, 12 | @rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 | @rem See the License for the specific language governing permissions and 14 | @rem limitations under the License. 15 | @rem 16 | 17 | @if "%DEBUG%" == "" @echo off 18 | @rem ########################################################################## 19 | @rem 20 | @rem Gradle startup script for Windows 21 | @rem 22 | @rem ########################################################################## 23 | 24 | @rem Set local scope for the variables with windows NT shell 25 | if "%OS%"=="Windows_NT" setlocal 26 | 27 | set DIRNAME=%~dp0 28 | if "%DIRNAME%" == "" set DIRNAME=. 29 | set APP_BASE_NAME=%~n0 30 | set APP_HOME=%DIRNAME% 31 | 32 | @rem Resolve any "." and ".." in APP_HOME to make it shorter. 33 | for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi 34 | 35 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 36 | set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m" 37 | 38 | @rem Find java.exe 39 | if defined JAVA_HOME goto findJavaFromJavaHome 40 | 41 | set JAVA_EXE=java.exe 42 | %JAVA_EXE% -version >NUL 2>&1 43 | if "%ERRORLEVEL%" == "0" goto execute 44 | 45 | echo. 46 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 47 | echo. 48 | echo Please set the JAVA_HOME variable in your environment to match the 49 | echo location of your Java installation. 50 | 51 | goto fail 52 | 53 | :findJavaFromJavaHome 54 | set JAVA_HOME=%JAVA_HOME:"=% 55 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe 56 | 57 | if exist "%JAVA_EXE%" goto execute 58 | 59 | echo. 60 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% 61 | echo. 62 | echo Please set the JAVA_HOME variable in your environment to match the 63 | echo location of your Java installation. 64 | 65 | goto fail 66 | 67 | :execute 68 | @rem Setup the command line 69 | 70 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar 71 | 72 | 73 | @rem Execute Gradle 74 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %* 75 | 76 | :end 77 | @rem End local scope for the variables with windows NT shell 78 | if "%ERRORLEVEL%"=="0" goto mainEnd 79 | 80 | :fail 81 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of 82 | rem the _cmd.exe /c_ return code! 83 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1 84 | exit /b 1 85 | 86 | :mainEnd 87 | if "%OS%"=="Windows_NT" endlocal 88 | 89 | :omega 90 | -------------------------------------------------------------------------------- /settings.gradle.kts: -------------------------------------------------------------------------------- 1 | rootProject.name = "com.plcoding.ktor-chat" -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/Application.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding 2 | 3 | import com.plcoding.di.mainModule 4 | import io.ktor.application.* 5 | import com.plcoding.plugins.* 6 | import org.koin.ktor.ext.Koin 7 | import org.koin.ktor.ext.modules 8 | 9 | fun main(args: Array): Unit = 10 | io.ktor.server.netty.EngineMain.main(args) 11 | 12 | @Suppress("unused") // application.conf references the main function. This annotation prevents the IDE from marking it as unused. 13 | fun Application.module() { 14 | install(Koin) { 15 | modules(mainModule) 16 | } 17 | configureSockets() 18 | configureRouting() 19 | configureSerialization() 20 | configureMonitoring() 21 | configureSecurity() 22 | } 23 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/data/MessageDataSource.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.data 2 | 3 | import com.plcoding.data.model.Message 4 | 5 | interface MessageDataSource { 6 | 7 | suspend fun getAllMessages(): List 8 | 9 | suspend fun insertMessage(message: Message) 10 | } -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/data/MessageDataSourceImpl.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.data 2 | 3 | import com.plcoding.data.model.Message 4 | import org.litote.kmongo.coroutine.CoroutineDatabase 5 | 6 | class MessageDataSourceImpl( 7 | private val db: CoroutineDatabase 8 | ): MessageDataSource { 9 | 10 | private val messages = db.getCollection() 11 | 12 | override suspend fun getAllMessages(): List { 13 | return messages.find() 14 | .descendingSort(Message::timestamp) 15 | .toList() 16 | } 17 | 18 | override suspend fun insertMessage(message: Message) { 19 | messages.insertOne(message) 20 | } 21 | } -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/data/model/Message.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.data.model 2 | 3 | import kotlinx.serialization.Serializable 4 | import org.bson.codecs.pojo.annotations.BsonId 5 | import org.bson.types.ObjectId 6 | 7 | @Serializable 8 | data class Message( 9 | val text: String, 10 | val username: String, 11 | val timestamp: Long, 12 | @BsonId 13 | val id: String = ObjectId().toString() 14 | ) 15 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/di/MainModule.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.di 2 | 3 | import com.plcoding.data.MessageDataSource 4 | import com.plcoding.data.MessageDataSourceImpl 5 | import com.plcoding.room.RoomController 6 | import org.koin.dsl.module 7 | import org.litote.kmongo.coroutine.coroutine 8 | import org.litote.kmongo.reactivestreams.KMongo 9 | 10 | val mainModule = module { 11 | single { 12 | KMongo.createClient() 13 | .coroutine 14 | .getDatabase("message_db_yt") 15 | } 16 | single { 17 | MessageDataSourceImpl(get()) 18 | } 19 | single { 20 | RoomController(get()) 21 | } 22 | } -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/plugins/Monitoring.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.plugins 2 | 3 | import io.ktor.features.* 4 | import org.slf4j.event.* 5 | import io.ktor.application.* 6 | import io.ktor.response.* 7 | import io.ktor.request.* 8 | 9 | fun Application.configureMonitoring() { 10 | install(CallLogging) { 11 | level = Level.INFO 12 | filter { call -> call.request.path().startsWith("/") } 13 | } 14 | 15 | } 16 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/plugins/Routing.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.plugins 2 | 3 | import com.plcoding.room.RoomController 4 | import com.plcoding.routes.chatSocket 5 | import com.plcoding.routes.getAllMessages 6 | import io.ktor.routing.* 7 | import io.ktor.http.* 8 | import io.ktor.application.* 9 | import io.ktor.response.* 10 | import io.ktor.request.* 11 | import org.koin.ktor.ext.inject 12 | 13 | fun Application.configureRouting() { 14 | val roomController by inject() 15 | install(Routing) { 16 | chatSocket(roomController) 17 | getAllMessages(roomController) 18 | } 19 | } 20 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/plugins/Security.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.plugins 2 | 3 | import com.plcoding.session.ChatSession 4 | import io.ktor.sessions.* 5 | import io.ktor.application.* 6 | import io.ktor.response.* 7 | import io.ktor.request.* 8 | import io.ktor.routing.* 9 | import io.ktor.util.* 10 | 11 | fun Application.configureSecurity() { 12 | install(Sessions) { 13 | cookie("SESSION") 14 | } 15 | 16 | intercept(ApplicationCallPipeline.Features) { 17 | if(call.sessions.get() == null) { 18 | val username = call.parameters["username"] ?: "Guest" 19 | call.sessions.set(ChatSession(username, generateNonce())) 20 | } 21 | } 22 | } 23 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/plugins/Serialization.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.plugins 2 | 3 | import io.ktor.serialization.* 4 | import io.ktor.features.* 5 | import io.ktor.application.* 6 | import io.ktor.response.* 7 | import io.ktor.request.* 8 | import io.ktor.routing.* 9 | 10 | fun Application.configureSerialization() { 11 | install(ContentNegotiation) { 12 | json() 13 | } 14 | } 15 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/plugins/Sockets.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.plugins 2 | 3 | import io.ktor.http.cio.websocket.* 4 | import io.ktor.websocket.* 5 | import java.time.* 6 | import io.ktor.application.* 7 | import io.ktor.response.* 8 | import io.ktor.request.* 9 | import io.ktor.routing.* 10 | 11 | fun Application.configureSockets() { 12 | install(WebSockets) { 13 | pingPeriod = Duration.ofSeconds(15) 14 | timeout = Duration.ofSeconds(15) 15 | maxFrameSize = Long.MAX_VALUE 16 | masking = false 17 | } 18 | } 19 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/room/Member.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.room 2 | 3 | import io.ktor.http.cio.websocket.* 4 | 5 | data class Member( 6 | val username: String, 7 | val sessionId: String, 8 | val socket: WebSocketSession 9 | ) 10 | -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/room/MemberAlreadyExistsException.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.room 2 | 3 | class MemberAlreadyExistsException: Exception( 4 | "There is already a member with that username in the room." 5 | ) -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/room/RoomController.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.room 2 | 3 | import com.plcoding.data.MessageDataSource 4 | import com.plcoding.data.model.Message 5 | import io.ktor.http.cio.websocket.* 6 | import kotlinx.serialization.encodeToString 7 | import kotlinx.serialization.json.Json 8 | import java.util.concurrent.ConcurrentHashMap 9 | 10 | class RoomController( 11 | private val messageDataSource: MessageDataSource 12 | ) { 13 | private val members = ConcurrentHashMap() 14 | 15 | fun onJoin( 16 | username: String, 17 | sessionId: String, 18 | socket: WebSocketSession 19 | ) { 20 | if(members.containsKey(username)) { 21 | throw MemberAlreadyExistsException() 22 | } 23 | members[username] = Member( 24 | username = username, 25 | sessionId = sessionId, 26 | socket = socket 27 | ) 28 | } 29 | 30 | suspend fun sendMessage(senderUsername: String, message: String) { 31 | members.values.forEach { member -> 32 | val messageEntity = Message( 33 | text = message, 34 | username = senderUsername, 35 | timestamp = System.currentTimeMillis() 36 | ) 37 | messageDataSource.insertMessage(messageEntity) 38 | 39 | val parsedMessage = Json.encodeToString(messageEntity) 40 | member.socket.send(Frame.Text(parsedMessage)) 41 | } 42 | } 43 | 44 | suspend fun getAllMessages(): List { 45 | return messageDataSource.getAllMessages() 46 | } 47 | 48 | suspend fun tryDisconnect(username: String) { 49 | members[username]?.socket?.close() 50 | if(members.containsKey(username)) { 51 | members.remove(username) 52 | } 53 | } 54 | } -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/routes/ChatRoutes.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.routes 2 | 3 | import com.plcoding.room.MemberAlreadyExistsException 4 | import com.plcoding.room.RoomController 5 | import com.plcoding.session.ChatSession 6 | import io.ktor.application.* 7 | import io.ktor.http.* 8 | import io.ktor.http.cio.websocket.* 9 | import io.ktor.response.* 10 | import io.ktor.routing.* 11 | import io.ktor.sessions.* 12 | import io.ktor.websocket.* 13 | import kotlinx.coroutines.channels.consumeEach 14 | 15 | fun Route.chatSocket(roomController: RoomController) { 16 | webSocket("/chat-socket") { 17 | val session = call.sessions.get() 18 | if(session == null) { 19 | close(CloseReason(CloseReason.Codes.VIOLATED_POLICY, "No session.")) 20 | return@webSocket 21 | } 22 | try { 23 | roomController.onJoin( 24 | username = session.username, 25 | sessionId = session.sessionId, 26 | socket = this 27 | ) 28 | incoming.consumeEach { frame -> 29 | if(frame is Frame.Text) { 30 | roomController.sendMessage( 31 | senderUsername = session.username, 32 | message = frame.readText() 33 | ) 34 | } 35 | } 36 | } catch(e: MemberAlreadyExistsException) { 37 | call.respond(HttpStatusCode.Conflict) 38 | } catch (e: Exception) { 39 | e.printStackTrace() 40 | } finally { 41 | roomController.tryDisconnect(session.username) 42 | } 43 | } 44 | } 45 | 46 | fun Route.getAllMessages(roomController: RoomController) { 47 | get("/messages") { 48 | call.respond( 49 | HttpStatusCode.OK, 50 | roomController.getAllMessages() 51 | ) 52 | } 53 | } -------------------------------------------------------------------------------- /src/main/kotlin/com/plcoding/session/ChatSession.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding.session 2 | 3 | data class ChatSession( 4 | val username: String, 5 | val sessionId: String 6 | ) 7 | -------------------------------------------------------------------------------- /src/main/resources/application.conf: -------------------------------------------------------------------------------- 1 | ktor { 2 | deployment { 3 | port = 8082 4 | port = ${?PORT} 5 | } 6 | application { 7 | modules = [ com.plcoding.ApplicationKt.module ] 8 | } 9 | } 10 | -------------------------------------------------------------------------------- /src/main/resources/logback.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | %d{YYYY-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | -------------------------------------------------------------------------------- /src/test/kotlin/com/plcoding/ApplicationTest.kt: -------------------------------------------------------------------------------- 1 | package com.plcoding 2 | 3 | import io.ktor.routing.* 4 | import io.ktor.http.* 5 | import io.ktor.http.cio.websocket.* 6 | import io.ktor.websocket.* 7 | import java.time.* 8 | import io.ktor.serialization.* 9 | import io.ktor.features.* 10 | import org.slf4j.event.* 11 | import io.ktor.sessions.* 12 | import io.ktor.application.* 13 | import io.ktor.response.* 14 | import io.ktor.request.* 15 | import kotlin.test.* 16 | import io.ktor.server.testing.* 17 | import com.plcoding.plugins.* 18 | 19 | class ApplicationTest { 20 | @Test 21 | fun testRoot() { 22 | withTestApplication({ configureRouting() }) { 23 | handleRequest(HttpMethod.Get, "/").apply { 24 | assertEquals(HttpStatusCode.OK, response.status()) 25 | assertEquals("Hello World!", response.content) 26 | } 27 | } 28 | } 29 | } --------------------------------------------------------------------------------