Add public key retrieval

Author: satiracode

src/main/kotlin/org/exploit/keeper/controller/CentralController.kt

@@ -2,23 +2,28 @@ package org.exploit.keeper.controller
 
 import io.smallrye.mutiny.Uni
 import jakarta.inject.Singleton
+import jakarta.ws.rs.GET
 import jakarta.ws.rs.POST
 import jakarta.ws.rs.Path
+import jakarta.ws.rs.QueryParam
 import org.exploit.keeper.constant.KeeperCurve
 import org.exploit.keeper.constant.SessionType.FROST
 import org.exploit.keeper.constant.SessionType.GG20
 import org.exploit.keeper.extension.decodeBase64
+import org.exploit.keeper.model.PublicKeyDto
 import org.exploit.keeper.model.TSSResult
 import org.exploit.keeper.model.request.Sign
 import org.exploit.keeper.service.frost.initiator.FrostSignatureService
 import org.exploit.keeper.service.gg20.initiator.GG20SignatureService
+import org.exploit.keeper.service.pub.PublicKeyService
 import org.jboss.resteasy.reactive.ResponseStatus
 
 @Singleton
 @Path("/v1/keeper")
 class CentralController(
     private val gg20: GG20SignatureService,
     private val frost: FrostSignatureService,
+    private val pub: PublicKeyService
 ) {
     @Path("/ping")
     @ResponseStatus(200)
@@ -41,4 +46,9 @@ class CentralController(
             curve = KeeperCurve.fromName(body.curve)
         )
     }
+
+    @GET
+    @Path("/publicKey")
+    fun publicKey(@QueryParam("keyId") keyId: String): PublicKeyDto =
+        pub.getPublicKey(keyId)
 }

src/main/kotlin/org/exploit/keeper/model/PublicKeyDto.kt

@@ -0,0 +1,3 @@
+package org.exploit.keeper.model
+
+data class PublicKeyDto(val data64: String = "")

src/main/kotlin/org/exploit/keeper/service/pub/NaivePublicKeyService.kt

@@ -0,0 +1,14 @@
+package org.exploit.keeper.service.pub
+
+import jakarta.inject.Singleton
+import jakarta.ws.rs.NotFoundException
+import org.exploit.keeper.db.LMKDB
+import org.exploit.keeper.extension.toBase64
+import org.exploit.keeper.model.PublicKeyDto
+
+@Singleton
+class NaivePublicKeyService(private val db: LMKDB): PublicKeyService {
+    override fun getPublicKey(keyId: String): PublicKeyDto {
+        return db.public.get(keyId)?.let { PublicKeyDto(it.toBase64()) } ?: throw NotFoundException()
+    }
+}

src/main/kotlin/org/exploit/keeper/service/pub/PublicKeyService.kt

@@ -0,0 +1,7 @@
+package org.exploit.keeper.service.pub
+
+import org.exploit.keeper.model.PublicKeyDto
+
+interface PublicKeyService {
+    fun getPublicKey(keyId: String): PublicKeyDto
+}