More session token function

MixinNetwork · Jun 16, 2022 · 57051de · 57051de
1 parent c0747aa
commit 57051de
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/library/src/main/kotlin/one/mixin/bot/SessionToken.kt b/library/src/main/kotlin/one/mixin/bot/SessionToken.kt
@@ -1,6 +1,10 @@
 package one.mixin.bot
 
+import net.i2p.crypto.eddsa.EdDSAPrivateKey
+import one.mixin.bot.extension.base64Encode
+import one.mixin.bot.vo.User
 import java.security.Key
+import java.security.KeyPair
 
 sealed class SessionToken(open val userId: String, open val sessionId: String) {
     data class RSA(
@@ -15,3 +19,14 @@ sealed class SessionToken(open val userId: String, open val sessionId: String) {
         val seed: String
     ) : SessionToken(userId, sessionId)
 }
+
+fun getUserSessionToken(user: User, sessionKey: KeyPair, isRsa: Boolean = false): SessionToken {
+    return if (isRsa) {
+        SessionToken.RSA(user.userId, user.sessionId, sessionKey.private)
+    } else {
+        SessionToken.EdDSA(
+            user.userId, user.sessionId,
+            (sessionKey.private as EdDSAPrivateKey).seed.base64Encode()
+        )
+    }
+}