wulkanowy/sdk
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Always fetch student site cookies, fetch studentplus only if eduOne enabled

Browse source
This commit is contained in:
Mikołaj Pich 2024-01-18 18:00:41 +01:00
parent 06a2e33822
commit 48af6bfc14
1 changed files with 24 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
sdk-scrapper/src/main/kotlin/io/github/wulkanowy/sdk/scrapper/interceptor/AutoLoginInterceptor.kt
View file

@ -40,6 +40,7 @@ import java.util.concurrent.locks.ReentrantLock
private val lock = ReentrantLock(true) private val lock = ReentrantLock(true)
private var studentModuleHeaders: ModuleHeaders? = null private var studentModuleHeaders: ModuleHeaders? = null
private var studentPlusModuleHeaders: ModuleHeaders? = null
private var messagesModuleHeaders: ModuleHeaders? = null private var messagesModuleHeaders: ModuleHeaders? = null
internal class AutoLoginInterceptor( internal class AutoLoginInterceptor(
@ -88,17 +89,19 @@ internal class AutoLoginInterceptor(
val studentModuleUrls = homePageResponse.studentSchools.map { it.attr("href") } val studentModuleUrls = homePageResponse.studentSchools.map { it.attr("href") }
val isEduOne = isCurrentLoginHasEduOne(studentModuleUrls) val isEduOne = isCurrentLoginHasEduOne(studentModuleUrls)
isEduOneStudent(isEduOne) isEduOneStudent(isEduOne)
studentModuleHeaders = null
messagesModuleHeaders = null messagesModuleHeaders = null
studentPlusModuleHeaders = null
studentModuleHeaders = null
val messages = getModuleCookies(UrlGenerator.Site.MESSAGES) val messages = getModuleCookies(UrlGenerator.Site.MESSAGES)
val student = getModuleCookies( val student = getModuleCookies(UrlGenerator.Site.STUDENT)
site = if (isEduOne) { val studentPlus = if (isEduOne) {
UrlGenerator.Site.STUDENT_PLUS getModuleCookies(UrlGenerator.Site.STUDENT_PLUS)
} else UrlGenerator.Site.STUDENT, } else null
)
when { when {
"wiadomosciplus" in uri.host -> messages.getOrThrow() "wiadomosciplus" in uri.host -> messages.getOrThrow()
"uczenplus" in uri.host -> studentPlus?.getOrThrow()
"uczen" in uri.host -> student.getOrThrow() "uczen" in uri.host -> student.getOrThrow()
else -> logger.info("Resource don't need further login") else -> logger.info("Resource don't need further login")
} }
@ -139,18 +142,27 @@ internal class AutoLoginInterceptor(
private fun saveModuleHeaders(doc: Document, url: HttpUrl) { private fun saveModuleHeaders(doc: Document, url: HttpUrl) {
when { when {
"uonetplus-uczen" in url.host -> { "uonetplus-wiadomosciplus" in url.host -> {
val htmlContent = doc.select("script").html() val htmlContent = doc.select("script").html()
studentModuleHeaders = ModuleHeaders( messagesModuleHeaders = ModuleHeaders(
token = getScriptParam("antiForgeryToken", htmlContent), token = getScriptParam("antiForgeryToken", htmlContent),
appGuid = getScriptParam("appGuid", htmlContent), appGuid = getScriptParam("appGuid", htmlContent),
appVersion = getScriptParam("version", htmlContent), appVersion = getScriptParam("version", htmlContent),
) )
} }
"uonetplus-wiadomosciplus" in url.host -> { "uonetplus-uczenplus" in url.host -> {
val htmlContent = doc.select("script").html() val htmlContent = doc.select("script").html()
messagesModuleHeaders = ModuleHeaders( studentPlusModuleHeaders = ModuleHeaders(
token = getScriptParam("antiForgeryToken", htmlContent),
appGuid = getScriptParam("appGuid", htmlContent),
appVersion = getScriptParam("version", htmlContent),
)
}
"uonetplus-uczen" in url.host -> {
val htmlContent = doc.select("script").html()
studentModuleHeaders = ModuleHeaders(
token = getScriptParam("antiForgeryToken", htmlContent), token = getScriptParam("antiForgeryToken", htmlContent),
appGuid = getScriptParam("appGuid", htmlContent), appGuid = getScriptParam("appGuid", htmlContent),
appVersion = getScriptParam("version", htmlContent), appVersion = getScriptParam("version", htmlContent),
@ -170,8 +182,9 @@ internal class AutoLoginInterceptor(
private fun Request.attachModuleHeaders(): Request { private fun Request.attachModuleHeaders(): Request {
val headers = when { val headers = when {
"uonetplus-uczen" in url.host -> studentModuleHeaders
"uonetplus-wiadomosciplus" in url.host -> messagesModuleHeaders "uonetplus-wiadomosciplus" in url.host -> messagesModuleHeaders
"uonetplus-uczenplus" in url.host -> studentPlusModuleHeaders
"uonetplus-uczen" in url.host -> studentModuleHeaders
else -> return this else -> return this
} }
return newBuilder() return newBuilder()