Fix websocket token (#1766)

server/build.gradle.kts

@@ -251,4 +251,8 @@ tasks {
     processResources {
         dependsOn(":server:server-config-generate:generateSettings")
     }
+
+    processTestResources {
+        dependsOn(":server:server-config-generate:generateSettings")
+    }
 }

server/src/main/kotlin/suwayomi/tachidesk/graphql/server/subscriptions/ApolloSubscriptionProtocolHandler.kt

@@ -155,7 +155,10 @@ class ApolloSubscriptionProtocolHandler(
     ): Flow<SubscriptionOperationMessage> {
         @Suppress("UNCHECKED_CAST")
         val user =
-            context.getAttributeOrSet(Attribute.TachideskUser) {
+            context.getAttributeOrSet(
+                Attribute.TachideskUser,
+                replaceIf = { it == UserType.Visitor },
+            ) {
                 val payload = operationMessage.payload as? Map<String, Any?>
                 val token = payload?.let { it[Header.AUTHORIZATION] as? String }
                 getUserFromToken(token)

server/src/main/kotlin/suwayomi/tachidesk/server/JavalinSetup.kt

@@ -319,8 +319,15 @@ object JavalinSetup {
 
     fun <T : Any> WsContext.getAttributeOrSet(
         attribute: Attribute<T>,
+        replaceIf: (T) -> Boolean = { false },
         set: () -> T,
-    ): T =
+    ): T {
-        attribute(attribute.name)
+        var item: T? = attribute(attribute.name)
-            ?: set().also { setAttribute(attribute, it) }
+
+        if (item != null && replaceIf(item)) {
+            item = null
+        }
+
+        return item ?: set().also { setAttribute(attribute, it) }
+    }
 }