Use Opensearch as the tool for searching posts

.env.example

@@ -2,6 +2,10 @@
 API_URL=http://localhost:8080
 UI_URL=http://localhost:3000
 
+SE_PROTOCOL="http"
+SE_HOST=localhost
+SE_PORT=9200
+
 # Discord application secrets - make sure these are never committed!
 DISCORD_GUILD_ID=GUILD_ID
 DISCORD_CLIENT_ID=CLIENT_ID

api/build.gradle.kts

@@ -32,6 +32,11 @@ dependencies {
     // DB
     implementation("org.litote.kmongo:kmongo:4.11.0")
 
+    // SE
+    implementation("org.apache.httpcomponents.core5:httpcore5:5.2.4")
+    implementation("org.apache.httpcomponents.core5:httpcore5-h2:5.2.4")
+    implementation("org.opensearch.client:opensearch-java:2.10.0")
+
     // Discord bot
     implementation("org.javacord:javacord:3.8.0")
 

api/src/main/kotlin/com/gmtkgamejam/Application.kt

@@ -2,13 +2,13 @@ package com.gmtkgamejam
 
 import com.gmtkgamejam.koin.DatabaseModule
 import com.gmtkgamejam.koin.DiscordBotModule
+import com.gmtkgamejam.koin.SearchEngineModule
 import com.gmtkgamejam.routing.*
 import io.ktor.http.*
 import io.ktor.serialization.kotlinx.json.*
 import io.ktor.server.application.*
 import io.ktor.server.plugins.contentnegotiation.*
-import io.ktor.server.plugins.cors.*
-import io.ktor.server.plugins.cors.routing.CORS
+import io.ktor.server.plugins.cors.routing.*
 import kotlinx.serialization.json.Json
 import org.koin.core.context.startKoin
 import org.koin.environmentProperties
@@ -19,16 +19,21 @@ fun main(args: Array<String>): Unit = io.ktor.server.netty.EngineMain.main(args)
 fun Application.module() {
     startKoin {
         environmentProperties()
-        modules(DatabaseModule, DiscordBotModule)
+        modules(
+            DatabaseModule,
+            DiscordBotModule,
+            SearchEngineModule
+        )
     }
 
-    configureRequestHandling()
-    configureUserInfoRouting()
-    configureAuthRouting()
     configureAdminRouting()
-    configurePostRouting()
-    configureFavouritesRouting()
+    configureAuthRouting()
     configureDiscordBotRouting()
+    configureFavouritesRouting()
+    configureInfraRouting()
+    configurePostRouting()
+    configureRequestHandling()
+    configureUserInfoRouting()
 }
 
 fun Application.configureRequestHandling() {

api/src/main/kotlin/com/gmtkgamejam/ApplicationCallExtensions.kt

@@ -0,0 +1,21 @@
+package com.gmtkgamejam
+
+import com.auth0.jwt.JWT
+import com.gmtkgamejam.models.auth.AuthTokenSet
+import com.gmtkgamejam.services.AuthService
+import io.ktor.http.*
+import io.ktor.server.application.*
+import io.ktor.server.request.*
+import io.ktor.server.response.*
+
+fun ApplicationCall.getAuthTokenSet(authService: AuthService): AuthTokenSet? {
+    return this.request.header("Authorization")
+        ?.substringAfter("Bearer ")
+        ?.let { JWT.decode(it) }?.getClaim("id")?.asString()
+        ?.let { authService.getTokenSet(it) }
+}
+
+suspend fun ApplicationCall.respondJSON(text: String, status: HttpStatusCode? = null) {
+    status?.let { response.status(it) }
+    respond(mapOf("message" to text))
+}

api/src/main/kotlin/com/gmtkgamejam/Config.kt

@@ -16,6 +16,11 @@ object Config {
      */
     fun getString(key: String): String = config.property(key).getString()
 
+    /**
+     * Get property int value
+     */
+    fun getInt(key: String): Int = getString(key).toInt()
+
     /**
      * Get property list value
      */

api/src/main/kotlin/com/gmtkgamejam/EnumExtensions.kt

@@ -1,10 +1,5 @@
 package com.gmtkgamejam
 
-/**
- * Floating function to cast a String to an Enum without throwing an exception
- *
- * Suggest using with mapNotNull{} where possible
- */
-inline fun <reified A : Enum<A>> enumFromStringSafe(value: String) : A? {
-    return enumValues<A>().find { s -> s.name == value.uppercase() }
+inline fun <reified A : Enum<A>> enumSetFromInput(commaSeparatedString: String) : Set<A> {
+    return commaSeparatedString.split(',').filter(String::isNotBlank).map { enumValueOf<A>(it) }.toSet()
 }

api/src/main/kotlin/com/gmtkgamejam/koin/SearchEngineModule.kt

@@ -0,0 +1,28 @@
+package com.gmtkgamejam.koin
+
+import com.gmtkgamejam.Config
+import org.apache.hc.core5.http.HttpHost
+import org.koin.dsl.module
+import org.opensearch.client.json.jackson.JacksonJsonpMapper
+import org.opensearch.client.opensearch.OpenSearchClient
+import org.opensearch.client.transport.OpenSearchTransport
+import org.opensearch.client.transport.httpclient5.ApacheHttpClient5TransportBuilder
+
+val SearchEngineModule = module(createdAtStart = true) {
+    single<OpenSearchClient> {
+        // Depending on hosting requirements, this is likely `http` while inside a VPC
+        val httpHost = HttpHost(
+            // TODO: Consider if there's a more flexible connection option w/o going into cert management hell
+            Config.getString("se.protocol"),
+            Config.getString("se.host"),
+            Config.getInt("se.port")
+        )
+
+        val transport: OpenSearchTransport = ApacheHttpClient5TransportBuilder
+            .builder(httpHost)
+            .setMapper(JacksonJsonpMapper())
+            .build()
+
+        OpenSearchClient(transport)
+    }
+}

api/src/main/kotlin/com/gmtkgamejam/models/posts/SearchItem.kt

@@ -0,0 +1,34 @@
+package com.gmtkgamejam.models.posts
+
+/**
+ * Data model for an entry in the search engine
+ *
+ * This is a stripped-down PostItem with just the fields we use for searching
+ *
+ * Any index mappings (i.e. the `description_shingle` defined in OpensearchClusterConfigurer) will be applied and
+ * computed by the SE at index-time
+ */
+data class SearchItem(
+    val id: String,
+    var description_shingle: String,
+    var size: Int,
+    var skillsPossessed: Set<Skills>?,
+    var skillsSought: Set<Skills>?,
+    var preferredTools: Set<Tools>?,
+    var availability: Availability,
+    var timezoneOffsets: Set<Int>,
+    var languages: Set<String>,
+) {
+    constructor(postItem: PostItem): this(
+        id = postItem.id,
+        // TODO: what work do we want to do here that the SE won't already do?
+        description_shingle = postItem.description.lowercase().replace("\n", " "),
+        size = postItem.size,
+        skillsPossessed = postItem.skillsPossessed,
+        skillsSought = postItem.skillsSought,
+        preferredTools = postItem.preferredTools,
+        availability = postItem.availability,
+        timezoneOffsets = postItem.timezoneOffsets,
+        languages = postItem.languages,
+    )
+}

api/src/main/kotlin/com/gmtkgamejam/routing/InfraRoutes.kt

@@ -0,0 +1,28 @@
+package com.gmtkgamejam.routing
+
+import com.gmtkgamejam.models.posts.PostItem
+import com.gmtkgamejam.respondJSON
+import com.gmtkgamejam.search.OpensearchClusterConfigurer
+import com.gmtkgamejam.services.PostService
+import io.ktor.http.*
+import io.ktor.server.application.*
+import io.ktor.server.routing.*
+import org.litote.kmongo.eq
+
+// TODO: Auth control
+fun Application.configureInfraRouting() {
+
+    val postService = PostService()
+
+    routing {
+        route("/infra") {
+            route("/se") {
+                get("/reset") {
+                    val posts = postService.getPosts(PostItem::deletedAt eq null)
+                    OpensearchClusterConfigurer().initCluster(posts)
+                    call.respondJSON("Search engine reset complete", HttpStatusCode.OK)
+                }
+            }
+        }
+    }
+}