clean up

src/main/kotlin/space/luminic/budgerapp/services/TokenService.kt

@@ -1,6 +1,5 @@
 package space.luminic.budgerapp.services
 import org.springframework.cache.annotation.CacheEvict
-import org.springframework.cache.annotation.Cacheable
 import org.springframework.stereotype.Service
 import reactor.core.publisher.Mono
 import space.luminic.budgerapp.models.Token
@@ -12,29 +11,28 @@ import java.time.LocalDateTime
 class TokenService(private val tokenRepository: TokenRepo) {
 
     @CacheEvict("tokens", allEntries = true)
-    fun saveToken(token: String, username: String, expiresAt: LocalDateTime) {
+    fun saveToken(token: String, username: String, expiresAt: LocalDateTime): Mono<Token> {
         val newToken = Token(
             token = token,
             username = username,
             issuedAt = LocalDateTime.now(),
             expiresAt = expiresAt
         )
-        tokenRepository.save(newToken)
+        return tokenRepository.save(newToken)
     }
 
     @CacheEvict("tokens", allEntries = true)
     fun revokeToken(token: String): Mono<Void> {
         return tokenRepository.findByToken(token)
+            .switchIfEmpty(Mono.error(Exception("Token not found")))
             .flatMap { existingToken ->
                 val updatedToken = existingToken.copy(status = TokenStatus.REVOKED)
                 tokenRepository.save(updatedToken).then()
             }
-            .switchIfEmpty(Mono.error(Exception("Token not found")))
+
     }
 
 
-
-
     @CacheEvict("tokens", allEntries = true)
     fun deleteExpiredTokens() {
         tokenRepository.deleteByExpiresAtBefore(LocalDateTime.now())