dalbodeule/chibot-chzzk-bot
1
0
Fork 0
mirror of https://github.com/dalbodeule/chibot-chzzk-bot.git synced 2025-06-08 23:08:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

debug chzzk login 4

Browse Source
This commit is contained in:
dalbodeule 2025-02-08 04:45:01 +09:00
parent 61a5f985c1
commit 3c3b9a79a2
No known key found for this signature in database
GPG Key ID: EFA860D069C9FA65
3 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/src/main/kotlin/space/mori/chzzk_bot/common/models/User.kt
View File

@ -15,6 +15,8 @@ object Users: IntIdTable("users") {
val liveAlertMessage = text("live_alert_message").nullable()
val isDisableStartupMsg = bool("is_disable_startup_msg").default(false)
val isDisabled = bool("is_disabled").default(false)
val accessToken = varchar("access_token", 255).nullable()
val refreshToken = varchar("refresh_token", 255).nullable()
}
class User(id: EntityID<Int>) : IntEntity(id) {
@ -29,6 +31,9 @@ class User(id: EntityID<Int>) : IntEntity(id) {
var isDisableStartupMsg by Users.isDisableStartupMsg
var isDisabled by Users.isDisabled
var accessToken by Users.accessToken
var refreshToken by Users.refreshToken
// 유저가 가진 매니저들
var managers by User.via(UserManagers.user, UserManagers.manager)

15
common/src/main/kotlin/space/mori/chzzk_bot/common/services/UserService.kt
View File

@ -97,4 +97,19 @@ object UserService {
user
}
}
fun setAccessToken(user: User, accessToken: String): User {
return transaction {
user.accessToken = accessToken
user
}
}
fun setRefreshToken(user: User, accessToken: String, refreshToken: String): User {
return transaction {
user.accessToken = accessToken
user.refreshToken = refreshToken
user
}
}
}

2
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/Main.kt
View File

@ -191,6 +191,7 @@ val server = embeddedServer(Netty, port = 8080, ) {
val userInfo = getChzzkUser(tokenResponse.content.accessToken)
if(userInfo.content != null) {
val user = UserService.getUser(userInfo.content.channelId)
call.sessions.set(
UserSession(
session.state,
@ -198,6 +199,7 @@ val server = embeddedServer(Netty, port = 8080, ) {
listOf()
)
)
user?.let { UserService.setRefreshToken(it, tokenResponse.content.accessToken, tokenResponse.content.refreshToken ?: "") }
call.respondRedirect(getFrontendURL(""))
}
} catch (e: Exception) {