com.infobip.kafkistry.webapp.security.CurrentRequestUserResolver.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafkistry-security Show documentation

The newest version!

package com.infobip.kafkistry.webapp.security

import org.springframework.security.core.context.SecurityContextHolder
import org.springframework.stereotype.Component

@Component
class CurrentRequestUserResolver {

    private val unknown = User("Unknown", "Unknown", "Unknown", "Unknown", UserRole.USER)
    private val scopedUser: ThreadLocal = ThreadLocal()

    fun resolveUser(): User? {
        scopedUser.get()?.run { return this }
        val context = SecurityContextHolder.getContext()
        val authentication = context.authentication ?: return null
        val userObj = authentication.principal
        return if (userObj is User) userObj else null
    }

    fun resolveUserOrUnknown(): User = resolveUser() ?: unknown

    fun  withUser(user: User, block: () -> T): T {
        return try {
            scopedUser.set(user)
            block()
        } finally {
            scopedUser.remove()
        }
    }

}