mirror of
https://github.com/dalbodeule/chibot-chzzk-bot.git
synced 2025-06-08 14:58:21 +00:00
Merge branch 'develop'
# Conflicts: # chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/ChzzkHandler.kt # chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/utils/accessTokenRefresh.kt
This commit is contained in:
commit
5e3a350e15
@ -206,6 +206,7 @@ class UserHandler(
|
|||||||
var messageHandler: MessageHandler
|
var messageHandler: MessageHandler
|
||||||
var client: ChzzkClient
|
var client: ChzzkClient
|
||||||
var listener: ChzzkUserSession
|
var listener: ChzzkUserSession
|
||||||
|
var chatChannelId: String?
|
||||||
|
|
||||||
private val dispatcher: CoroutinesEventBus by inject(CoroutinesEventBus::class.java)
|
private val dispatcher: CoroutinesEventBus by inject(CoroutinesEventBus::class.java)
|
||||||
private var _isActive: Boolean
|
private var _isActive: Boolean
|
||||||
@ -221,18 +222,14 @@ class UserHandler(
|
|||||||
throw RuntimeException("AccessToken or RefreshToken is not valid.")
|
throw RuntimeException("AccessToken or RefreshToken is not valid.")
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
|
val tokens = Connector.client.refreshAccessToken(user.refreshToken!!)
|
||||||
client = Connector.getClient(user.accessToken!!, user.refreshToken!!)
|
client = Connector.getClient(tokens.first, tokens.second)
|
||||||
|
UserService.setRefreshToken(user, tokens.first, tokens.second)
|
||||||
|
chatChannelId = getChzzkChannelId(channel.channelId)
|
||||||
|
|
||||||
client.loginAsync().join()
|
client.loginAsync().join()
|
||||||
client.refreshTokenAsync().join()
|
|
||||||
|
|
||||||
UserService.setRefreshToken(user, client.loginResult.accessToken(), client.loginResult.refreshToken())
|
|
||||||
|
|
||||||
listener = ChzzkSessionBuilder(client).buildUserSession()
|
listener = ChzzkSessionBuilder(client).buildUserSession()
|
||||||
|
|
||||||
listener.createAndConnectAsync().join()
|
listener.createAndConnectAsync().join()
|
||||||
|
|
||||||
messageHandler = MessageHandler(this@UserHandler)
|
messageHandler = MessageHandler(this@UserHandler)
|
||||||
|
|
||||||
listener.on(SessionChatMessageEvent::class.java) {
|
listener.on(SessionChatMessageEvent::class.java) {
|
||||||
|
@ -353,15 +353,15 @@ class MessageHandler(
|
|||||||
// Replace followPattern
|
// Replace followPattern
|
||||||
result = followPattern.replace(result) { _ ->
|
result = followPattern.replace(result) { _ ->
|
||||||
try {
|
try {
|
||||||
val followingDate = getFollowDate(channel.channelId, msg.senderChannelId)
|
val followingDate = handler.chatChannelId?.let { getFollowDate(it, msg.senderChannelId) }
|
||||||
.content?.streamingProperty?.following?.followDate
|
?.content?.streamingProperty?.following?.followDate ?: LocalDateTime.now().minusDays(1).toString()
|
||||||
|
|
||||||
val period = followingDate?.let {
|
val period = followingDate.let {
|
||||||
val formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")
|
val formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")
|
||||||
val pastDate = LocalDateTime.parse(it, formatter)
|
val pastDate = LocalDateTime.parse(it, formatter)
|
||||||
val today = LocalDateTime.now()
|
val today = LocalDateTime.now()
|
||||||
ChronoUnit.DAYS.between(pastDate, today)
|
ChronoUnit.DAYS.between(pastDate, today)
|
||||||
} ?: 0
|
} + 1
|
||||||
|
|
||||||
period.toString()
|
period.toString()
|
||||||
} catch (e: Exception) {
|
} catch (e: Exception) {
|
||||||
|
@ -0,0 +1,55 @@
|
|||||||
|
package space.mori.chzzk_bot.chatbot.utils
|
||||||
|
|
||||||
|
import com.google.gson.Gson
|
||||||
|
import okhttp3.OkHttpClient
|
||||||
|
import okhttp3.Request
|
||||||
|
import okhttp3.RequestBody.Companion.toRequestBody
|
||||||
|
import space.mori.chzzk_bot.common.utils.client
|
||||||
|
import xyz.r2turntrue.chzzk4j.ChzzkClient
|
||||||
|
import java.io.IOException
|
||||||
|
|
||||||
|
val client = OkHttpClient.Builder()
|
||||||
|
.addNetworkInterceptor { chain ->
|
||||||
|
chain.proceed(
|
||||||
|
chain.request()
|
||||||
|
.newBuilder()
|
||||||
|
.header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
}
|
||||||
|
.build()
|
||||||
|
val gson = Gson()
|
||||||
|
|
||||||
|
data class RefreshTokenResponse(
|
||||||
|
val accessToken: String,
|
||||||
|
val refreshToken: String,
|
||||||
|
val expiresIn: Int,
|
||||||
|
val tokenType: String = "Bearer",
|
||||||
|
val scope: String
|
||||||
|
)
|
||||||
|
|
||||||
|
fun ChzzkClient.refreshAccessToken(refreshToken: String): Pair<String, String> {
|
||||||
|
val url = "https://chzzk.naver.com/auth/v1/token"
|
||||||
|
val request = Request.Builder()
|
||||||
|
.url(url)
|
||||||
|
.header("Content-Type", "application/json")
|
||||||
|
.post(gson.toJson(mapOf(
|
||||||
|
"grantType" to "refresh_token",
|
||||||
|
"refreshToken" to refreshToken,
|
||||||
|
"clientId" to this.apiClientId,
|
||||||
|
"clientSecret" to this.apiSecret
|
||||||
|
)).toRequestBody())
|
||||||
|
.build()
|
||||||
|
|
||||||
|
client.newCall(request).execute().use { response ->
|
||||||
|
try {
|
||||||
|
if(!response.isSuccessful) throw IOException("Unexpected code ${response.code}")
|
||||||
|
val body = response.body?.string()
|
||||||
|
val data = gson.fromJson(body, RefreshTokenResponse::class.java)
|
||||||
|
|
||||||
|
return Pair(data.accessToken, data.refreshToken)
|
||||||
|
} catch(e: Exception) {
|
||||||
|
throw e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -52,6 +52,20 @@ data class NicknameColor(
|
|||||||
val colorCode: String = ""
|
val colorCode: String = ""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
data class LiveStatus(
|
||||||
|
val liveTitle: String,
|
||||||
|
val status: String,
|
||||||
|
val concurrentUserCount: Int,
|
||||||
|
val accumulateCount: Int,
|
||||||
|
val paidPromotion: Boolean,
|
||||||
|
val adult: Boolean,
|
||||||
|
val krOnlyViewing: Boolean,
|
||||||
|
val openDate: String,
|
||||||
|
val closeDate: String?,
|
||||||
|
val clipActive: Boolean,
|
||||||
|
val chatChannelId: String
|
||||||
|
)
|
||||||
|
|
||||||
// OkHttpClient에 Interceptor 추가
|
// OkHttpClient에 Interceptor 추가
|
||||||
val client = OkHttpClient.Builder()
|
val client = OkHttpClient.Builder()
|
||||||
.addNetworkInterceptor { chain ->
|
.addNetworkInterceptor { chain ->
|
||||||
@ -84,3 +98,24 @@ fun getFollowDate(chatID: String, userId: String) : IData<IFollowContent?> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fun getChzzkChannelId(channelId: String): String? {
|
||||||
|
val url = "https://api.chzzk.naver.com/polling/v3/channels/$channelId/live-status?includePlayerRecommendContent=false"
|
||||||
|
val request = Request.Builder()
|
||||||
|
.url(url)
|
||||||
|
.header("Content-Type", "application/json")
|
||||||
|
.get()
|
||||||
|
.build()
|
||||||
|
|
||||||
|
client.newCall(request).execute().use { response ->
|
||||||
|
try {
|
||||||
|
if(!response.isSuccessful) throw IOException("Unexpected code ${response.code}")
|
||||||
|
val body = response.body?.string()
|
||||||
|
val data = gson.fromJson(body, object: TypeToken<IData<LiveStatus?>>() {})
|
||||||
|
|
||||||
|
return data.content?.chatChannelId
|
||||||
|
} catch(e: Exception) {
|
||||||
|
throw e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user