Refactor authentication

backend/build.gradle.kts

@@ -1,3 +1,5 @@
+import com.infendro.plugin.infendro
+
 group = "com.infendro.account"
 version = "0.0.1"
 
@@ -5,17 +7,18 @@ application {
     mainClass.set("com.infendro.account.ApplicationKt")
 }
 
-repositories {
-    maven("https://git.infendro.com/api/packages/Infendro/maven")
-    mavenCentral()
-}
-
 plugins {
+    alias(libs.plugins.infendro)
     alias(libs.plugins.kotlin)
     alias(libs.plugins.ktor)
     alias(libs.plugins.serialization)
 }
 
+repositories {
+    infendro()
+    mavenCentral()
+}
+
 dependencies {
     implementation(libs.ktor.server)
     implementation(libs.ktor.server.cio)
@@ -26,16 +29,20 @@ dependencies {
     implementation(libs.ktor.server.contentNegotiation)
     implementation(libs.ktor.server.serialization)
     implementation(libs.ktor.server.validation)
+
     implementation(libs.exposed)
     implementation(libs.exposed.dao)
     implementation(libs.exposed.jdbc)
     implementation(libs.migration)
     implementation(libs.postgres)
-    implementation(libs.random)
-    implementation(libs.otp)
-    implementation(libs.kdf)
-    implementation(libs.hash)
+
+    implementation(libs.bytes)
     implementation(libs.encoding)
+    implementation(libs.hash)
+    implementation(libs.kdf)
+    implementation(libs.otp)
+    implementation(libs.prng)
+
     implementation(libs.logback)
 }
 

backend/gradle/libs.versions.toml

@@ -1,17 +1,23 @@
 [versions]
-kotlin = "2.2.10"
-ktor = "3.2.3"
-exposed = "0.61.0"
-migration = "1.1.0"
-postgres = "42.7.5"
-random = "1.0.0"
-otp = "1.1.1"
-kdf = "1.0.3"
-hash = "1.2.2"
-encoding = "1.1.0"
-logback = "1.5.18"
+infendro = "1.1.0"
+kotlin = "2.3.0"
+ktor = "3.4.0"
+
+exposed = "1.0.0"
+migration = "1.2.0"
+postgres = "42.7.10"
+
+bytes = "1.4.0"
+encoding = "1.3.0"
+hash = "1.6.0"
+kdf = "1.2.0"
+otp = "1.3.0"
+prng = "1.3.0"
+
+logback = "1.5.32"
 
 [plugins]
+infendro = { id = "com.infendro.plugin", version.ref = "infendro" }
 kotlin = { id = "org.jetbrains.kotlin.jvm", version.ref = "kotlin" }
 ktor = { id = "io.ktor.plugin", version.ref = "ktor" }
 serialization = { id = "org.jetbrains.kotlin.plugin.serialization", version.ref = "kotlin" }
@@ -30,13 +36,14 @@ ktor-server-validation = { module = "io.ktor:ktor-server-request-validation", ve
 exposed = { module = "org.jetbrains.exposed:exposed-core", version.ref = "exposed" }
 exposed-dao = { module = "org.jetbrains.exposed:exposed-dao", version.ref = "exposed" }
 exposed-jdbc = { module = "org.jetbrains.exposed:exposed-jdbc", version.ref = "exposed" }
-migration = { module = "com.infendro:ktor-migration", version.ref = "migration" }
+migration = { module = "com.infendro:migration-ktor", version.ref = "migration" }
 postgres = { module = "org.postgresql:postgresql", version.ref = "postgres" }
 
-random = { module = "com.infendro:random", version.ref = "random" }
-otp = { module = "com.infendro:otp", version.ref = "otp" }
-kdf = { module = "com.infendro:kdf", version.ref = "kdf" }
-hash = { module = "com.infendro:hash", version.ref = "hash" }
+bytes = { module = "com.infendro:bytes", version.ref = "bytes" }
 encoding = { module = "com.infendro:encoding", version.ref = "encoding" }
+hash = { module = "com.infendro:hash", version.ref = "hash" }
+kdf = { module = "com.infendro:kdf", version.ref = "kdf" }
+otp = { module = "com.infendro:otp", version.ref = "otp" }
+prng = { module = "com.infendro:prng", version.ref = "prng" }
 
 logback = { module = "ch.qos.logback:logback-classic", version.ref = "logback" }

backend/settings.gradle.kts

@@ -1 +1,7 @@
 rootProject.name = "backend"
+
+pluginManagement.repositories {
+    maven("https://git.infendro.com/api/packages/Infendro/maven")
+    gradlePluginPortal()
+    mavenCentral()
+}

backend/src/main/kotlin/com/infendro/account/Application.kt

@@ -1,12 +1,11 @@
 package com.infendro.account
 
 import com.infendro.account.config.*
+import com.infendro.account.config.configureSession
 import io.ktor.server.application.*
 import io.ktor.server.cio.*
 
-fun main(
-    args: Array<String>,
-) = EngineMain.main(args)
+fun main(args: Array<String>) = EngineMain.main(args)
 
 fun Application.module() {
     configureDatabase()
@@ -15,6 +14,7 @@ fun Application.module() {
     configureValidation()
     configureException()
 
+    configureSession()
     configureSecurity()
     configureRouting()
 }

backend/src/main/kotlin/com/infendro/account/config/Database.kt

@@ -1,11 +1,9 @@
 package com.infendro.account.config
 
-import com.infendro.account.model.Role
-import com.infendro.account.model.repository.AccountRepository
-import com.infendro.account.util.SecureHasher
 import com.infendro.ktor.migration.Migration
+import initial
 import io.ktor.server.application.*
-import org.jetbrains.exposed.sql.Database
+import org.jetbrains.exposed.v1.jdbc.Database
 
 fun Application.configureDatabase() {
     val config = configuration
@@ -17,56 +15,6 @@ fun Application.configureDatabase() {
     )
 
     install(Migration) {
-        migration("initial") {
-            exec(
-                """
-                    CREATE TABLE access 
-                    ( 
-                        id BIGSERIAL
-                            PRIMARY KEY,
-                        token_hash TEXT
-                            NOT NULL
-                    );
-                    
-                    CREATE TABLE account
-                    (
-                        id BIGSERIAL
-                            PRIMARY KEY,
-                        username TEXT
-                            UNIQUE
-                            NOT NULL,
-                        password_hash TEXT
-                            NOT NULL,
-                        password_salt TEXT
-                            NOT NULL,
-                        secret TEXT
-                            NOT NULL,
-                        role TEXT
-                            NOT NULL
-                    );
-                    
-                    CREATE TABLE session
-                    (
-                        id BIGSERIAL
-                            PRIMARY KEY,
-                        token_hash TEXT
-                            NOT NULL,
-                        account_id BIGINT
-                            REFERENCES account (id) ON DELETE CASCADE
-                            NOT NULL
-                    );
-                """.trimIndent()
-            )
-
-            AccountRepository.insert {
-                val salt = SecureHasher.generateSalt()
-
-                this.username = "infendro"
-                this.passwordHash = SecureHasher.hash("password", salt)
-                this.passwordSalt = salt
-                this.secret = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA===="
-                this.role = Role.OWNER
-            }
-        }
+        initial()
     }
 }

backend/src/main/kotlin/com/infendro/account/config/Security.kt

@@ -1,25 +1,14 @@
 package com.infendro.account.config
 
-import com.infendro.account.exception.client.UnauthorizedException
 import com.infendro.account.model.Role
 import com.infendro.account.model.entity.AccountEntity
 import com.infendro.account.model.entity.SessionEntity
-import com.infendro.account.model.entity.SessionTable
 import com.infendro.account.model.repository.SessionRepository
-import com.infendro.account.util.Hasher
 import io.ktor.server.application.*
 import io.ktor.server.auth.*
 import io.ktor.server.sessions.*
-import kotlinx.serialization.Serializable
-import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
-import kotlin.time.Duration.Companion.days
 
-@Serializable
-data class AuthenticationSession(
-    val token: String,
-)
-
-data class AuthenticationPrincipal(
+data class Principal(
     val account: AccountEntity,
     val session: SessionEntity,
 ) {
@@ -28,34 +17,20 @@ data class AuthenticationPrincipal(
 }
 
 fun Application.configureSecurity() {
-    val config = configuration
-
-    install(Sessions) {
-        cookie<AuthenticationSession>("Authentication") {
-            this.cookie.domain = config.security.cookie.domain
-            this.cookie.path = config.security.cookie.path
-            this.cookie.httpOnly = true
-            this.cookie.secure = config.security.cookie.secure
-            this.cookie.extensions["SameSite"] = "Strict"
-            this.cookie.maxAge = 7.days
-        }
-    }
     install(Authentication) {
-        session<AuthenticationSession> {
-            validate { session ->
-                val session = SessionRepository
-                    .singleOrNull { SessionTable.tokenHash eq Hasher.hash(session.token) }
-                    ?: return@validate null
-
-                sessions.set(session)
-
-                AuthenticationPrincipal(session.account(), session)
+        bearer {
+            authenticate { credential ->
+                SessionRepository.by(credential.token)?.principal
             }
-            challenge {
-                call.sessions.clear<AuthenticationSession>()
-
-                throw UnauthorizedException()
+        }
+        session<Session> {
+            validate { session ->
+                SessionRepository.by(session.token)?.principal
+                    ?.also { sessions.set(session) }
             }
         }
     }
 }
+
+val SessionEntity.principal: Principal
+    get() = Principal(account(), this)

backend/src/main/kotlin/com/infendro/account/config/Session.kt

@@ -0,0 +1,26 @@
+package com.infendro.account.config
+
+import io.ktor.server.application.*
+import io.ktor.server.sessions.*
+import kotlinx.serialization.Serializable
+import kotlin.time.Duration.Companion.days
+
+@Serializable
+data class Session(
+    val token: String,
+)
+
+fun Application.configureSession() {
+    val config = configuration
+
+    install(Sessions) {
+        cookie<Session>("Authentication") {
+            this.cookie.domain = config.security.cookie.domain
+            this.cookie.path = config.security.cookie.path
+            this.cookie.httpOnly = true
+            this.cookie.secure = config.security.cookie.secure
+            this.cookie.extensions["SameSite"] = "Strict"
+            this.cookie.maxAge = 7.days
+        }
+    }
+}

backend/src/main/kotlin/com/infendro/account/dto/request/PostAccountRequest.kt

@@ -18,12 +18,10 @@ data class PostAccountRequest(
     )
 }
 
-fun RequestValidationConfig.validatePostAccountRequest() {
-    validate<PostAccountRequest> { request ->
-        when {
-            !USERNAME.matches(request.username) -> Invalid("")
-            !PASSWORD.matches(request.password) -> Invalid("")
-            else -> Valid
-        }
+fun RequestValidationConfig.validatePostAccountRequest() = validate<PostAccountRequest> { request ->
+    when {
+        !USERNAME.matches(request.username) -> Invalid("")
+        !PASSWORD.matches(request.password) -> Invalid("")
+        else -> Valid
     }
 }

backend/src/main/kotlin/com/infendro/account/dto/request/PutAccountCurrentPasswordRequest.kt

@@ -11,11 +11,9 @@ data class PutAccountCurrentPasswordRequest(
     val otp: String,
 )
 
-fun RequestValidationConfig.validatePutAccountCurrentPasswordRequest() {
-    validate<PutAccountCurrentPasswordRequest> { request ->
-        when {
-            !PASSWORD.matches(request.password) -> Invalid("")
-            else -> Valid
-        }
+fun RequestValidationConfig.validatePutAccountCurrentPasswordRequest() = validate<PutAccountCurrentPasswordRequest> { request ->
+    when {
+        !PASSWORD.matches(request.password) -> Invalid("")
+        else -> Valid
     }
 }

backend/src/main/kotlin/com/infendro/account/dto/request/PutAccountCurrentUsernameRequest.kt

@@ -11,11 +11,9 @@ data class PutAccountCurrentUsernameRequest(
     val otp: String,
 )
 
-fun RequestValidationConfig.validatePutAccountCurrentUsernameRequest() {
-    validate<PutAccountCurrentUsernameRequest> { request ->
-        when {
-            !USERNAME.matches(request.username) -> Invalid("")
-            else -> Valid
-        }
+fun RequestValidationConfig.validatePutAccountCurrentUsernameRequest() = validate<PutAccountCurrentUsernameRequest> { request ->
+    when {
+        !USERNAME.matches(request.username) -> Invalid("")
+        else -> Valid
     }
 }

backend/src/main/kotlin/com/infendro/account/dto/response/AccountResponse.kt

@@ -13,11 +13,9 @@ data class AccountResponse(
     val sessions: List<SessionResponse>,
 )
 
-fun AccountEntity.toResponse(): AccountResponse {
-    return AccountResponse(
-        id = id.value,
-        username = username,
-        role = role,
-        sessions = sessions().map(SessionEntity::toResponse),
-    )
-}
+fun AccountEntity.toResponse() = AccountResponse(
+    id = id.value,
+    username = username,
+    role = role,
+    sessions = sessions().map(SessionEntity::toResponse),
+)

backend/src/main/kotlin/com/infendro/account/dto/response/SessionResponse.kt

@@ -9,9 +9,7 @@ data class SessionResponse(
     val accountId: Long,
 )
 
-fun SessionEntity.toResponse(): SessionResponse {
-    return SessionResponse(
-        id = id.value,
-        accountId = account().id.value,
-    )
-}
+fun SessionEntity.toResponse() = SessionResponse(
+    id = id.value,
+    accountId = account().id.value,
+)

backend/src/main/kotlin/com/infendro/account/migration/Initial.kt

@@ -0,0 +1,56 @@
+import com.infendro.account.model.Role
+import com.infendro.account.model.repository.AccountRepository
+import com.infendro.account.util.PasswordHasher
+import com.infendro.ktor.migration.MigrationConfig
+
+fun MigrationConfig.initial() = migration("initial") {
+    exec(
+        """
+            CREATE TABLE access 
+            ( 
+                id BIGSERIAL
+                    PRIMARY KEY,
+                token_hash TEXT
+                    NOT NULL
+            );
+            
+            CREATE TABLE account
+            (
+                id BIGSERIAL
+                    PRIMARY KEY,
+                username TEXT
+                    UNIQUE
+                    NOT NULL,
+                password_hash TEXT
+                    NOT NULL,
+                password_salt TEXT
+                    NOT NULL,
+                secret TEXT
+                    NOT NULL,
+                role TEXT
+                    NOT NULL
+            );
+            
+            CREATE TABLE session
+            (
+                id BIGSERIAL
+                    PRIMARY KEY,
+                token_hash TEXT
+                    NOT NULL,
+                account_id BIGINT
+                    REFERENCES account (id) ON DELETE CASCADE
+                    NOT NULL
+            );
+        """.trimIndent(),
+    )
+
+    AccountRepository.insert {
+        val salt = PasswordHasher.generateSalt()
+
+        this.username = "infendro"
+        this.passwordHash = PasswordHasher.hash("password", salt)
+        this.passwordSalt = salt
+        this.secret = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA===="
+        this.role = Role.OWNER
+    }
+}

backend/src/main/kotlin/com/infendro/account/model/Role.kt

@@ -13,16 +13,5 @@ enum class Role(
     }
 }
 
-infix fun List<Role>.except(
-    roles: List<Role>,
-): List<Role> {
-    return filterNot { it in roles }
-        .toList()
-}
-
-infix fun List<Role>.except(
-    role: Role,
-): List<Role> {
-    return filterNot { it == role }
-        .toList()
-}
+infix fun List<Role>.except(roles: List<Role>) = filterNot { it in roles }
+infix fun List<Role>.except(role: Role) = filterNot { it == role }

backend/src/main/kotlin/com/infendro/account/model/entity/Access.kt

@@ -1,9 +1,9 @@
 package com.infendro.account.model.entity
 
-import org.jetbrains.exposed.dao.LongEntity
-import org.jetbrains.exposed.dao.LongEntityClass
-import org.jetbrains.exposed.dao.id.EntityID
-import org.jetbrains.exposed.dao.id.LongIdTable
+import org.jetbrains.exposed.v1.core.dao.id.EntityID
+import org.jetbrains.exposed.v1.core.dao.id.LongIdTable
+import org.jetbrains.exposed.v1.dao.LongEntity
+import org.jetbrains.exposed.v1.dao.LongEntityClass
 
 object AccessTable : LongIdTable("access", "id") {
     val tokenHash = text("token_hash")

backend/src/main/kotlin/com/infendro/account/model/entity/Account.kt

@@ -1,11 +1,11 @@
 package com.infendro.account.model.entity
 
 import com.infendro.account.model.Role
-import org.jetbrains.exposed.dao.LongEntity
-import org.jetbrains.exposed.dao.LongEntityClass
-import org.jetbrains.exposed.dao.id.EntityID
-import org.jetbrains.exposed.dao.id.LongIdTable
-import org.jetbrains.exposed.sql.transactions.transaction
+import org.jetbrains.exposed.v1.core.dao.id.EntityID
+import org.jetbrains.exposed.v1.core.dao.id.LongIdTable
+import org.jetbrains.exposed.v1.dao.LongEntity
+import org.jetbrains.exposed.v1.dao.LongEntityClass
+import org.jetbrains.exposed.v1.jdbc.transactions.transaction
 
 object AccountTable : LongIdTable("account", "id") {
     val username = text("username")
@@ -20,7 +20,7 @@ object AccountTable : LongIdTable("account", "id") {
                 else -> throw Error()
             }
         },
-        toDb = { it.name }
+        toDb = { it.name },
     )
 }
 

backend/src/main/kotlin/com/infendro/account/model/entity/Session.kt

@@ -1,10 +1,10 @@
 package com.infendro.account.model.entity
 
-import org.jetbrains.exposed.dao.LongEntity
-import org.jetbrains.exposed.dao.LongEntityClass
-import org.jetbrains.exposed.dao.id.EntityID
-import org.jetbrains.exposed.dao.id.LongIdTable
-import org.jetbrains.exposed.sql.transactions.transaction
+import org.jetbrains.exposed.v1.core.dao.id.EntityID
+import org.jetbrains.exposed.v1.core.dao.id.LongIdTable
+import org.jetbrains.exposed.v1.dao.LongEntity
+import org.jetbrains.exposed.v1.dao.LongEntityClass
+import org.jetbrains.exposed.v1.jdbc.transactions.transaction
 
 object SessionTable : LongIdTable("session", "id") {
     val tokenHash = text("token_hash")

backend/src/main/kotlin/com/infendro/account/model/repository/Repository.kt

@@ -1,9 +1,9 @@
 package com.infendro.account.model.repository
 
-import org.jetbrains.exposed.dao.LongEntity
-import org.jetbrains.exposed.dao.LongEntityClass
-import org.jetbrains.exposed.sql.Op
-import org.jetbrains.exposed.sql.transactions.transaction
+import org.jetbrains.exposed.v1.core.Op
+import org.jetbrains.exposed.v1.dao.LongEntity
+import org.jetbrains.exposed.v1.dao.LongEntityClass
+import org.jetbrains.exposed.v1.jdbc.transactions.transaction
 
 abstract class Repository<ENTITY : LongEntity>(
     private val entityClass: LongEntityClass<ENTITY>,
@@ -12,113 +12,75 @@ abstract class Repository<ENTITY : LongEntity>(
         entityClass.all().toList()
     }
 
-    fun all(
-        where: Op<Boolean>,
-    ): List<ENTITY> = transaction {
+    fun all(where: Op<Boolean>): List<ENTITY> = transaction {
         entityClass.find(where).toList()
     }
 
-    fun all(
-        where: () -> Op<Boolean>,
-    ): List<ENTITY> {
+    fun all(where: () -> Op<Boolean>): List<ENTITY> {
         return all(where())
     }
 
-    fun single(
-        where: Op<Boolean>,
-    ): ENTITY = transaction {
+    fun single(where: Op<Boolean>): ENTITY = transaction {
         entityClass.find(where).single()
     }
 
-    fun single(
-        where: () -> Op<Boolean>,
-    ): ENTITY {
+    fun single(where: () -> Op<Boolean>): ENTITY {
         return single(where())
     }
 
-    fun singleOrNull(
-        where: Op<Boolean>,
-    ): ENTITY? = transaction {
+    fun singleOrNull(where: Op<Boolean>): ENTITY? = transaction {
         entityClass.find(where).singleOrNull()
     }
 
-    fun singleOrNull(
-        where: () -> Op<Boolean>,
-    ): ENTITY? {
+    fun singleOrNull(where: () -> Op<Boolean>): ENTITY? {
         return singleOrNull(where())
     }
 
-    fun exists(
-        where: Op<Boolean>,
-    ): Boolean = transaction {
+    fun exists(where: Op<Boolean>): Boolean = transaction {
         entityClass.find(where).any()
     }
 
-    fun exists(
-        where: () -> Op<Boolean>,
-    ): Boolean {
+    fun exists(where: () -> Op<Boolean>): Boolean {
         return exists(where())
     }
 
-    fun insert(
-        block: ENTITY.() -> Unit,
-    ): Unit = transaction {
+    fun insert(block: ENTITY.() -> Unit): Unit = transaction {
         entityClass.new(block)
     }
 
-    fun update(
-        entity: ENTITY,
-        block: ENTITY.() -> Unit,
-    ): Unit = transaction {
+    fun update(entity: ENTITY, block: ENTITY.() -> Unit): Unit = transaction {
         entity.block()
     }
 
-    fun update(
-        entities: Iterable<ENTITY>,
-        block: ENTITY.() -> Unit,
-    ) {
+    fun update(entities: Iterable<ENTITY>, block: ENTITY.() -> Unit) {
         for (entity in entities) {
             update(entity, block)
         }
     }
 
-    fun update(
-        where: Op<Boolean>,
-        block: ENTITY.() -> Unit,
-    ) {
+    fun update(where: Op<Boolean>, block: ENTITY.() -> Unit) {
         update(all(where), block)
     }
 
-    fun update(
-        where: () -> Op<Boolean>,
-        block: ENTITY.() -> Unit,
-    ) {
+    fun update(where: () -> Op<Boolean>, block: ENTITY.() -> Unit) {
         update(where(), block)
     }
 
-    fun delete(
-        entity: ENTITY,
-    ): Unit = transaction {
+    fun delete(entity: ENTITY): Unit = transaction {
         entity.delete()
     }
 
-    fun delete(
-        entities: Iterable<ENTITY>,
-    ) {
+    fun delete(entities: Iterable<ENTITY>) {
         for (entity in entities) {
             delete(entity)
         }
     }
 
-    fun delete(
-        where: Op<Boolean>,
-    ) {
+    fun delete(where: Op<Boolean>) {
         delete(all(where))
     }
 
-    fun delete(
-        where: () -> Op<Boolean>,
-    ) {
+    fun delete(where: () -> Op<Boolean>) {
         delete(where())
     }
 }

backend/src/main/kotlin/com/infendro/account/model/repository/SessionRepository.kt

@@ -1,5 +1,10 @@
 package com.infendro.account.model.repository
 
 import com.infendro.account.model.entity.SessionEntity
+import com.infendro.account.model.entity.SessionTable
+import com.infendro.account.util.Hasher
+import org.jetbrains.exposed.v1.core.eq
 
-object SessionRepository : Repository<SessionEntity>(SessionEntity)
+object SessionRepository : Repository<SessionEntity>(SessionEntity) {
+    fun by(token: String) = singleOrNull { SessionTable.tokenHash eq Hasher.hash(token) }
+}

backend/src/main/kotlin/com/infendro/account/routing/Access.kt

@@ -1,6 +1,6 @@
 package com.infendro.account.routing
 
-import com.infendro.account.config.AuthenticationPrincipal
+import com.infendro.account.config.Principal
 import com.infendro.account.service.AccessService
 import io.ktor.http.HttpStatusCode.Companion.OK
 import io.ktor.resources.*
@@ -27,21 +27,21 @@ private class Access {
 fun Routing.access() {
     authenticate {
         post<Access> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccessService.post(principal)
                 .also { call.respond(it) }
         }
 
         get<Access.All> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccessService.getAll(principal)
                 .also { call.respond(it) }
         }
 
         delete<Access.Id> { resource ->
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val id = resource.id
 
             AccessService.deleteId(principal, id)

backend/src/main/kotlin/com/infendro/account/routing/Account.kt

@@ -1,6 +1,6 @@
 package com.infendro.account.routing
 
-import com.infendro.account.config.AuthenticationPrincipal
+import com.infendro.account.config.Principal
 import com.infendro.account.dto.request.*
 import com.infendro.account.service.AccountService
 import io.ktor.http.HttpStatusCode.Companion.OK
@@ -79,28 +79,28 @@ fun Routing.account() {
 
     authenticate {
         get<Account.All> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccountService.getAll(principal)
                 .also { call.respond(it) }
         }
 
         get<Account.Current> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccountService.getCurrent(principal)
                 .also { call.respond(it) }
         }
 
         delete<Account.Current> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccountService.deleteCurrent(principal)
             call.respond(OK)
         }
 
         put<Account.Current.Username> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val request = call.receive<PutAccountCurrentUsernameRequest>()
 
             AccountService.putCurrentUsername(principal, call.sessions, request)
@@ -108,7 +108,7 @@ fun Routing.account() {
         }
 
         put<Account.Current.Password> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val request = call.receive<PutAccountCurrentPasswordRequest>()
 
             AccountService.putCurrentPassword(principal, call.sessions, request)
@@ -116,7 +116,7 @@ fun Routing.account() {
         }
 
         post<Account.Current.Secret> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val request = call.receive<PostAccountCurrentSecretRequest>()
 
             AccountService.postCurrentSecret(principal, call.sessions, request)
@@ -124,14 +124,14 @@ fun Routing.account() {
         }
 
         get<Account.Current.Session.All> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             AccountService.getCurrentSessionAll(principal)
                 .also { call.respond(it) }
         }
 
         delete<Account.Current.Session.Id> { resource ->
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val id = resource.id
 
             AccountService.deleteCurrentSessionId(principal, call.sessions, id)
@@ -139,7 +139,7 @@ fun Routing.account() {
         }
 
         delete<Account.Id> { resource ->
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val id = resource.id
 
             AccountService.deleteId(principal, id)
@@ -147,7 +147,7 @@ fun Routing.account() {
         }
 
         put<Account.Id.Role> { resource ->
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val id = resource.parent.id
             val request = call.receive<PutAccountIdRoleRequest>()
 

backend/src/main/kotlin/com/infendro/account/routing/Session.kt

@@ -1,6 +1,6 @@
 package com.infendro.account.routing
 
-import com.infendro.account.config.AuthenticationPrincipal
+import com.infendro.account.config.Principal
 import com.infendro.account.dto.request.PostSessionRequest
 import com.infendro.account.service.SessionService
 import io.ktor.http.HttpStatusCode.Companion.OK
@@ -42,14 +42,14 @@ fun Routing.session() {
 
     authenticate {
         get<Session.All> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             SessionService.getAll(principal)
                 .also { call.respond(it) }
         }
 
         delete<Session.Id> { resource ->
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
             val id = resource.id
 
             SessionService.deleteId(principal, id)
@@ -57,14 +57,14 @@ fun Routing.session() {
         }
 
         get<Session.Current> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             SessionService.getCurrent(principal)
                 .also { call.respond(it) }
         }
 
         delete<Session.Current> {
-            val principal = call.principal<AuthenticationPrincipal>()!!
+            val principal = call.principal<Principal>()!!
 
             SessionService.deleteCurrent(principal, call.sessions)
             call.respond(OK)

backend/src/main/kotlin/com/infendro/account/service/AccessService.kt

@@ -1,6 +1,6 @@
 package com.infendro.account.service
 
-import com.infendro.account.config.AuthenticationPrincipal
+import com.infendro.account.config.Principal
 import com.infendro.account.dto.response.AccessResponse
 import com.infendro.account.dto.response.PostAccessResponse
 import com.infendro.account.dto.response.toResponse
@@ -11,12 +11,10 @@ import com.infendro.account.model.entity.AccessTable
 import com.infendro.account.model.repository.AccessRepository
 import com.infendro.account.util.Hasher
 import com.infendro.account.util.TokenGenerator
-import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
+import org.jetbrains.exposed.v1.core.eq
 
 object AccessService {
-    fun post(
-        principal: AuthenticationPrincipal,
-    ): PostAccessResponse {
+    fun post(principal: Principal): PostAccessResponse {
         if (principal.role != OWNER)
             throw ForbiddenException()
 
@@ -26,13 +24,11 @@ object AccessService {
         }
 
         return PostAccessResponse(
-            token = token
+            token = token,
         )
     }
 
-    fun getAll(
-        principal: AuthenticationPrincipal,
-    ): List<AccessResponse> {
+    fun getAll(principal: Principal): List<AccessResponse> {
         if (principal.role != OWNER)
             throw ForbiddenException()
 
@@ -41,10 +37,7 @@ object AccessService {
             .map { it.toResponse() }
     }
 
-    fun deleteId(
-        principal: AuthenticationPrincipal,
-        id: Long,
-    ) {
+    fun deleteId(principal: Principal, id: Long) {
         if (principal.role != OWNER)
             throw ForbiddenException()
 

backend/src/main/kotlin/com/infendro/account/service/AccountService.kt

@@ -1,7 +1,7 @@
 package com.infendro.account.service
 
-import com.infendro.account.config.AuthenticationPrincipal
-import com.infendro.account.config.AuthenticationSession
+import com.infendro.account.config.Principal
+import com.infendro.account.config.Session
 import com.infendro.account.dto.request.*
 import com.infendro.account.dto.response.*
 import com.infendro.account.exception.client.ConflictException
@@ -16,14 +16,12 @@ import com.infendro.account.model.repository.AccountRepository
 import com.infendro.account.model.repository.SessionRepository
 import com.infendro.account.util.Hasher
 import com.infendro.account.util.OTP
-import com.infendro.account.util.SecureHasher
+import com.infendro.account.util.PasswordHasher
 import io.ktor.server.sessions.*
-import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
+import org.jetbrains.exposed.v1.core.eq
 
 object AccountService {
-    fun post(
-        request: PostAccountRequest,
-    ): PostAccountResponse {
+    fun post(request: PostAccountRequest): PostAccountResponse {
         val access = AccessRepository
             .singleOrNull { AccessTable.tokenHash eq Hasher.hash(request.access.token) }
             ?: throw UnauthorizedException()
@@ -36,23 +34,21 @@ object AccountService {
 
         val secret = OTP.generateSecret()
         AccountRepository.insert {
-            val salt = SecureHasher.generateSalt()
+            val salt = PasswordHasher.generateSalt()
 
             this.username = request.username
-            this.passwordHash = SecureHasher.hash(request.password, salt)
+            this.passwordHash = PasswordHasher.hash(request.password, salt)
             this.passwordSalt = salt
             this.secret = secret
             this.role = USER
         }
 
         return PostAccountResponse(
-            secret = secret
+            secret = secret,
         )
     }
 
-    fun getAll(
-        principal: AuthenticationPrincipal,
-    ): List<AccountResponse> {
+    fun getAll(principal: Principal): List<AccountResponse> {
         if (principal.role != OWNER)
             throw ForbiddenException()
 
@@ -61,26 +57,18 @@ object AccountService {
             .map(AccountEntity::toResponse)
     }
 
-    fun getCurrent(
-        principal: AuthenticationPrincipal,
-    ): AccountResponse {
+    fun getCurrent(principal: Principal): AccountResponse {
         return principal.account.toResponse()
     }
 
-    fun deleteCurrent(
-        principal: AuthenticationPrincipal,
-    ) {
+    fun deleteCurrent(principal: Principal) {
         if (principal.role == OWNER)
             throw ForbiddenException()
 
         AccountRepository.delete(principal.account)
     }
 
-    fun putCurrentUsername(
-        principal: AuthenticationPrincipal,
-        sessions: CurrentSession,
-        request: PutAccountCurrentUsernameRequest,
-    ) {
+    fun putCurrentUsername(principal: Principal, sessions: CurrentSession, request: PutAccountCurrentUsernameRequest) {
         if (!OTP.verify(principal.account.secret, request.otp))
             throw UnauthorizedException()
 
@@ -89,34 +77,30 @@ object AccountService {
         }
         SessionRepository.delete(principal.account.sessions())
 
-        sessions.clear<AuthenticationSession>()
+        sessions.clear<Session>()
     }
 
-    fun putCurrentPassword(
-        principal: AuthenticationPrincipal,
-        sessions: CurrentSession,
-        request: PutAccountCurrentPasswordRequest,
-    ) {
+    fun putCurrentPassword(principal: Principal, sessions: CurrentSession, request: PutAccountCurrentPasswordRequest) {
         if (!OTP.verify(principal.account.secret, request.otp))
             throw UnauthorizedException()
 
         AccountRepository.update(principal.account) {
-            val salt = SecureHasher.generateSalt()
+            val salt = PasswordHasher.generateSalt()
 
-            this.passwordHash = SecureHasher.hash(request.password, salt)
+            this.passwordHash = PasswordHasher.hash(request.password, salt)
             this.passwordSalt = salt
         }
         SessionRepository.delete(principal.account.sessions())
 
-        sessions.clear<AuthenticationSession>()
+        sessions.clear<Session>()
     }
 
     fun postCurrentSecret(
-        principal: AuthenticationPrincipal,
+        principal: Principal,
         sessions: CurrentSession,
-        request: PostAccountCurrentSecretRequest,
+        request: PostAccountCurrentSecretRequest
     ): PostAccountCurrentSecretResponse {
-        if (principal.account.passwordHash != SecureHasher.hash(request.password, principal.account.passwordSalt))
+        if (principal.account.passwordHash != PasswordHasher.hash(request.password, principal.account.passwordSalt))
             throw UnauthorizedException()
 
         val secret = OTP.generateSecret()
@@ -125,25 +109,19 @@ object AccountService {
         }
         SessionRepository.delete(principal.account.sessions())
 
-        sessions.clear<AuthenticationSession>()
+        sessions.clear<Session>()
 
         return PostAccountCurrentSecretResponse(
-            secret = secret
+            secret = secret,
         )
     }
 
-    fun getCurrentSessionAll(
-        principal: AuthenticationPrincipal,
-    ): List<SessionResponse> {
+    fun getCurrentSessionAll(principal: Principal): List<SessionResponse> {
         return principal.account.sessions()
             .map(SessionEntity::toResponse)
     }
 
-    fun deleteCurrentSessionId(
-        principal: AuthenticationPrincipal,
-        sessions: CurrentSession,
-        id: Long,
-    ) {
+    fun deleteCurrentSessionId(principal: Principal, sessions: CurrentSession, id: Long) {
         val session = SessionRepository
             .singleOrNull { SessionTable.id eq id }
             ?: throw NotFoundException()
@@ -154,13 +132,10 @@ object AccountService {
         SessionRepository.delete(session)
 
         if (session.id == principal.session.id)
-            sessions.clear<AuthenticationSession>()
+            sessions.clear<Session>()
     }
 
-    fun deleteId(
-        principal: AuthenticationPrincipal,
-        id: Long,
-    ) {
+    fun deleteId(principal: Principal, id: Long) {
         val account = AccountRepository
             .singleOrNull { AccountTable.id eq id }
             ?: throw NotFoundException()
@@ -171,11 +146,7 @@ object AccountService {
         AccountRepository.delete(account)
     }
 
-    fun putIdRole(
-        principal: AuthenticationPrincipal,
-        id: Long,
-        request: PutAccountIdRoleRequest,
-    ) {
+    fun putIdRole(principal: Principal, id: Long, request: PutAccountIdRoleRequest) {
         val account = AccountRepository
             .singleOrNull { AccountTable.id eq id }
             ?: throw NotFoundException()

backend/src/main/kotlin/com/infendro/account/service/SessionService.kt

@@ -1,7 +1,7 @@
 package com.infendro.account.service
 
-import com.infendro.account.config.AuthenticationPrincipal
-import com.infendro.account.config.AuthenticationSession
+import com.infendro.account.config.Principal
+import com.infendro.account.config.Session
 import com.infendro.account.dto.request.PostSessionRequest
 import com.infendro.account.dto.response.SessionResponse
 import com.infendro.account.dto.response.toResponse
@@ -16,17 +16,14 @@ import com.infendro.account.model.repository.AccountRepository
 import com.infendro.account.model.repository.SessionRepository
 import com.infendro.account.util.Hasher
 import com.infendro.account.util.OTP
-import com.infendro.account.util.SecureHasher
+import com.infendro.account.util.PasswordHasher
 import com.infendro.account.util.TokenGenerator
 import io.ktor.server.sessions.*
-import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
-import org.jetbrains.exposed.sql.SqlExpressionBuilder.inList
+import org.jetbrains.exposed.v1.core.eq
+import org.jetbrains.exposed.v1.core.inList
 
 object SessionService {
-    fun post(
-        sessions: CurrentSession,
-        request: PostSessionRequest,
-    ) {
+    fun post(sessions: CurrentSession, request: PostSessionRequest) {
         val account = AccountRepository
             .singleOrNull { AccountTable.username eq request.username }
             ?: throw NotFoundException()
@@ -34,7 +31,7 @@ object SessionService {
         if (!OTP.verify(account.secret, request.otp))
             throw UnauthorizedException()
 
-        if (account.passwordHash != SecureHasher.hash(request.password, account.passwordSalt))
+        if (account.passwordHash != PasswordHasher.hash(request.password, account.passwordSalt))
             throw UnauthorizedException()
 
         val token = TokenGenerator.generate()
@@ -43,14 +40,10 @@ object SessionService {
             this.account = account
         }
 
-        sessions.set(
-            AuthenticationSession(token)
-        )
+        sessions.set(Session(token))
     }
 
-    fun getAll(
-        principal: AuthenticationPrincipal,
-    ): List<SessionResponse> {
+    fun getAll(principal: Principal): List<SessionResponse> {
         if (principal.role == USER)
             throw ForbiddenException()
 
@@ -60,10 +53,7 @@ object SessionService {
             .map(SessionEntity::toResponse)
     }
 
-    fun deleteId(
-        principal: AuthenticationPrincipal,
-        id: Long,
-    ) {
+    fun deleteId(principal: Principal, id: Long) {
         if (principal.role == USER)
             throw ForbiddenException()
 
@@ -77,18 +67,13 @@ object SessionService {
         SessionRepository.delete(session)
     }
 
-    fun getCurrent(
-        principal: AuthenticationPrincipal,
-    ): SessionResponse {
+    fun getCurrent(principal: Principal): SessionResponse {
         return principal.session.toResponse()
     }
 
-    fun deleteCurrent(
-        principal: AuthenticationPrincipal,
-        sessions: CurrentSession,
-    ) {
+    fun deleteCurrent(principal: Principal, sessions: CurrentSession) {
         SessionRepository.delete(principal.session)
 
-        sessions.clear<AuthenticationSession>()
+        sessions.clear<Session>()
     }
 }

backend/src/main/kotlin/com/infendro/account/util/Hasher.kt

@@ -1,13 +1,11 @@
 package com.infendro.account.util
 
-import com.infendro.encoding.Hex
-import com.infendro.hash.SHA256
+import com.infendro.encoding.Base16
+import com.infendro.hash.sha2.`SHA-256`
 
 object Hasher {
-    fun hash(
-        value: String,
-    ): String {
-        val bytes = SHA256.hash(value.toByteArray())
-        return Hex.encode(bytes).decodeToString()
+    fun hash(value: String): String {
+        val bytes = `SHA-256`.hash(value.encodeToByteArray())
+        return Base16.encode(bytes).decodeToString()
     }
 }

backend/src/main/kotlin/com/infendro/account/util/OTP.kt

@@ -1,29 +1,27 @@
 package com.infendro.account.util
 
 import com.infendro.encoding.Base32
-import com.infendro.hash.SHA256
-import com.infendro.otp.SecretGenerator
+import com.infendro.hash.sha2.`SHA-256`
 import com.infendro.otp.TOTP
+import com.infendro.random.csprng.`HMAC-DRBG`
 import kotlin.time.Clock
 import kotlin.time.Duration.Companion.seconds
 
 object OTP {
+    private val random = `HMAC-DRBG`(`SHA-256`)
     private val totp = TOTP(
-        function = SHA256,
+        function = `SHA-256`,
         length = 8,
         period = 30.seconds,
     )
 
-    fun verify(
-        secret: String,
-        otp: String,
-    ): Boolean {
-        val bytes = Base32.decode(secret.toByteArray())
+    fun verify(secret: String, otp: String): Boolean {
+        val bytes = Base32.decode(secret.encodeToByteArray())
         return totp.verify(bytes, Clock.System.now(), otp)
     }
 
     fun generateSecret(): String {
-        val bytes = SecretGenerator.generate(totp.hotp.function)
+        val bytes = random.nextBytes(32)
         return Base32.encode(bytes).decodeToString()
     }
 }

backend/src/main/kotlin/com/infendro/account/util/PasswordHasher.kt

@@ -0,0 +1,23 @@
+package com.infendro.account.util
+
+import com.infendro.encoding.Base16
+import com.infendro.hash.sha2.`SHA-256`
+import com.infendro.kdf.PBKDF2
+import com.infendro.random.csprng.`HMAC-DRBG`
+
+object PasswordHasher {
+    private val random = `HMAC-DRBG`(`SHA-256`)
+    private val kdf = PBKDF2(32, 100_000, `SHA-256`)
+
+    fun hash(value: String, salt: String): String {
+        val value = value.encodeToByteArray()
+        val salt = Base16.decode(salt.encodeToByteArray())
+        return kdf.hash(value, salt)
+            .let { Base16.encode(it).decodeToString() }
+    }
+
+    fun generateSalt(): String {
+        return random.nextBytes(16)
+            .let { Base16.encode(it).decodeToString() }
+    }
+}

backend/src/main/kotlin/com/infendro/account/util/SecureHasher.kt

@@ -1,26 +0,0 @@
-package com.infendro.account.util
-
-import com.infendro.encoding.Hex
-import com.infendro.hash.SHA256
-import com.infendro.kdf.PBKDF2
-import com.infendro.random.PRNG
-
-object SecureHasher {
-    private val random = PRNG.HMAC
-    private val kdf = PBKDF2(SHA256)
-
-    fun hash(
-        value: String,
-        salt: String,
-    ): String {
-        val value = value.encodeToByteArray()
-        val salt = Hex.decode(salt.encodeToByteArray())
-        return kdf.hash(value, salt, 100_000, 32)
-            .let { Hex.encode(it).decodeToString() }
-    }
-
-    fun generateSalt(): String {
-        return random.nextBytes(16)
-            .let { Hex.encode(it).decodeToString() }
-    }
-}

backend/src/main/kotlin/com/infendro/account/util/TokenGenerator.kt

@@ -1,13 +1,14 @@
 package com.infendro.account.util
 
-import com.infendro.encoding.Hex
-import com.infendro.random.PRNG
+import com.infendro.encoding.Base16
+import com.infendro.hash.sha2.`SHA-256`
+import com.infendro.random.csprng.`HMAC-DRBG`
 
 object TokenGenerator {
-    private val random = PRNG.HMAC
+    private val random = `HMAC-DRBG`(`SHA-256`)
 
     fun generate(): String {
         return random.nextBytes(32)
-            .let { Hex.encode(it).decodeToString() }
+            .let { Base16.encode(it).decodeToString() }
     }
 }