xyz.cssxsh.arknights.weibo.MicroBlogDataHolder.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arknights-helper Show documentation
arknights-helper
There is a newer version: 2.3.1
package xyz.cssxsh.arknights.weibo

import io.ktor.client.request.*
import io.ktor.http.*
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.*
import kotlinx.serialization.*
import xyz.cssxsh.arknights.*
import java.io.*
import java.util.*
import kotlin.collections.*

public class MicroBlogDataHolder(override val folder: File, override val ignore: suspend (Throwable) -> Boolean) :
    CacheDataHolder() {

    override val cache: MutableMap> = EnumMap(BlogUser::class.java)

    private fun timestamp(id: Long): Long = (id shr 22) + 515483463L

    override suspend fun load(key: BlogUser) {
        val blogs: MutableMap = HashMap()

        try {
            val text = http.prepareGet(BLOG_API) { parameter("containerid", "107803${key.uid}") }
                .body()
            val temp = CustomJson.decodeFromString>(text)

            for (blog in temp.data().blogs()) {
                blogs.compute(blog.id) { _, old ->
                    old?.copy(pictures = old.pictures + blog.pictures) ?: blog
                }
            }

            blogs.replaceAll { _, blog ->
                blog.copy(
                    created = TimestampSerializer.timestamp(second = timestamp(id = blog.id)),
                    user = blog.user.copy(
                        id = key.uid,
                        name = "此微博被锁定为热门，机器人无法获取详情，请打开链接自行查看"
                    )
                )
            }
        } catch (_: Exception) {
            //
        }

        try {
            val text = http.prepareGet(BLOG_API) { parameter("containerid", "107603${key.uid}") }
                .body()
            val temp = CustomJson.decodeFromString>(text)

            for (blog in temp.data().blogs()) {
                blogs[blog.id] = blog
            }
        } catch (_: Exception) {
            //
        }

        key.write(blogs.values.toList())

        cache[key] = blogs.values.toList()
    }

    override suspend fun raw(key: BlogUser): List {
        return cache[key] ?: try {
            key.read()
        } catch (_: FileNotFoundException) {
            emptyList()
        }
    }

    override suspend fun clear(): Unit = mutex.withLock {
        runInterruptible(context = Dispatchers.IO) {
            for (item in folder.listFiles() ?: return@runInterruptible) {
                if (!item.isDirectory) continue
                item.deleteRecursively()
            }
        }
    }

    private fun MicroBlog.folder(): File = folder.resolve(created.toLocalDate().toString()).apply { mkdirs() }

    public suspend fun images(blog: MicroBlog): List = mutex.withLock {
        val cache = ArrayList(blog.pictures.size)
        val folder = blog.folder()
        for (pid in blog.pictures) {
            val url = image(pid = pid)
            val file = folder.resolve(url.substringAfterLast('/'))
            if (file.exists().not()) {
                http.prepareGet(url) {
                    header(HttpHeaders.Referrer, blog.url)
                }.copyTo(file)
            }
            cache.add(file)
        }
        cache
    }

    public suspend fun content(blog: MicroBlog): String = mutex.withLock {
        val file = blog.folder().resolve("${blog.id}.content.json")
        val json = when {
            !blog.isLongText -> return blog.raw ?: blog.text
            file.exists() -> file.readText()
            else -> {
                val statement = http.prepareGet(CONTENT_API) {
                    parameter("id", blog.id)
                }
                statement.copyTo(file)
                val json = file.readText()
                if ("请求超时" in json) {
                    file.delete()
                    throw IllegalStateException("请求超时")
                }
                if ("登录注册更精彩" in json) {
                    file.delete()
                    throw IllegalStateException("登陆锁定")
                }
                if ("打开微博客户端，查看全文" in json) {
                    file.delete()
                    throw IllegalStateException("微博客户端锁定")
                }
                json
            }
        }

        CustomJson.decodeFromString>(json).data().content
    }
}