FancyBot/src/main/kotlin/cn/rtast/fancybot/commands/parse/BilibiliParseCommand.kt

/*
 * Copyright © 2024 RTAkland
 * Author: RTAkland
 * Date: 2024/9/6
 */


package cn.rtast.fancybot.commands.parse

import cn.rtast.fancybot.entity.bili.*
import cn.rtast.fancybot.util.Http
import cn.rtast.fancybot.util.misc.Resources
import cn.rtast.fancybot.util.misc.drawCustomImage
import cn.rtast.fancybot.util.misc.drawString
import cn.rtast.fancybot.util.misc.toBufferedImage
import cn.rtast.fancybot.util.misc.toByteArray
import cn.rtast.fancybot.util.misc.toURL
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.fancybot.util.str.formatNumber
import cn.rtast.fancybot.util.str.fromJson
import cn.rtast.fancybot.util.str.setTruncate
import cn.rtast.rob.entity.GroupMessage
import cn.rtast.rob.enums.ArrayMessageType
import cn.rtast.rob.onebot.MessageChain
import okhttp3.OkHttpClient
import okhttp3.Request
import java.awt.Color
import java.awt.Font
import java.awt.Image
import java.awt.RenderingHints
import java.awt.image.BufferedImage
import java.math.BigInteger
import java.net.URI
import javax.imageio.ImageIO


object BiliVideoParseCommand {
    private val avRegex = Regex("av(\\d+)", RegexOption.IGNORE_CASE)
    private val bvRegex = Regex("BV[\\dA-Za-z]{10}")
    private val shortUrlRegex = Regex("https://b23\\.tv/\\w{6,8}")
    private val backgroundColor = Color(43, 43, 43)
    private val upBarColor = Color(102, 102, 102)
    private val textColor = Color.WHITE
    private val titleFont = Font("Serif", Font.ITALIC, 40)
    private val numberFont = Font("Serif", Font.ITALIC, 25)
    private val twoTwoLogo = ImageIO.read(Resources.loadFromResources("bili/22-coin.png"))
    private val likeIcon = ImageIO.read(Resources.loadFromResources("bili/like.png"))
    private val coinIcon = ImageIO.read(Resources.loadFromResources("bili/coin.png"))
    private val viewIcon = ImageIO.read(Resources.loadFromResources("bili/view.png"))
    private val shareIcon = ImageIO.read(Resources.loadFromResources("bili/share.png"))
    private val favoriteIcon = ImageIO.read(Resources.loadFromResources("bili/favorite.png"))
    private val replyIcon = ImageIO.read(Resources.loadFromResources("bili/reply.png"))
    private const val VIDEO_STAT_URL = "https://api.bilibili.com/x/web-interface/view"
    private const val USER_STAT_URL = "https://api.bilibili.com/x/relation/stat"
    private const val SHORT_URL_API_URL = "https://api.bilibili.com/x/share/click"
    private const val VIEW_COUNT_URL = "https://api.bilibili.com/x/player/online/total"
    private const val CANVAS_WIDTH = 1000
    private const val CANVAS_HEIGHT = 600
    private const val BASE = 58
    private const val DATA = "FcwAPNKTMug3GV5Lj7EJnHpWsx4tb8haYeviqBz6rkCy12mUSDQX9RdoZf"
    private val XOR_CODE = BigInteger.valueOf(23442827791579L)
    private val MAX_AID = BigInteger.ONE.shiftLeft(51)
    private val tempOkHttpClient = OkHttpClient()

    private fun generateShortUrl(bvid: String, oid: Long): String {
        val result = Http.post<ShortUrl>(
            SHORT_URL_API_URL, mapOf(
                "buvid" to bvid,
                "build" to 6114514,
                "platform" to "unix",
                "share_channel" to "COPY",
                "share_mode" to 2,
                "share_id" to "main.ugc-video-detail.0.0.pv",
                "oid" to oid
            )
        )
        return result.data.content.split(" ").last()
    }

    private fun getViewCount(bvid: String, cid: Long): String {
        return Http.get<ViewCount>(
            VIEW_COUNT_URL, mapOf(
                "bvid" to bvid,
                "cid" to cid
            )
        ).data.total
    }

    private fun getVideoStat(bvid: String): VideoStat {
        return Http.get<VideoStat>(VIDEO_STAT_URL, mapOf("bvid" to bvid))
    }

    private fun createResponseImage(
        title: String,
        author: String,
        authorFace: String,
        picUrl: String,
        reply: String,
        view: String,
        coin: String,
        share: String,
        like: String,
        favorite: String,
        fans: Int,
        viewCount: String,
    ): String {
        val coverImage = ImageIO.read(URI(picUrl).toURL())
        val faceImage = ImageIO.read(URI(authorFace).toURL())
        val canvas = BufferedImage(CANVAS_WIDTH, CANVAS_HEIGHT, BufferedImage.TYPE_INT_RGB)
        val g2d = canvas.createGraphics()
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON)
        g2d.color = backgroundColor
        g2d.fillRect(0, 0, CANVAS_WIDTH, CANVAS_HEIGHT)
        // draw rect
        g2d.color = upBarColor
        g2d.fillRect(0, 0, CANVAS_WIDTH, 120)
        g2d.color = textColor
        // draw numbers
        g2d.font = numberFont
        g2d.drawString(like, 110, 200)
        g2d.drawString(coin, 110, 290)
        g2d.drawString(view, 110, 380)
        g2d.drawString(favorite, 300, 380)
        g2d.drawString(reply, 300, 290)
        g2d.drawString(share, 300, 200)
        // draw video title
        val truncatedText = setTruncate(title, g2d)
        g2d.font = titleFont
        g2d.drawString(truncatedText, 20, 70)
        // draw author name
        g2d.drawString(author, 40, 580)
        // draw icons
        g2d.drawCustomImage(favoriteIcon, 230, 340, 50.0, 50.0, false)
        g2d.drawCustomImage(replyIcon, 230, 250, 50.0, 50.0, false)
        g2d.drawCustomImage(shareIcon, 230, 160, 50.0, 50.0, false)
        g2d.drawCustomImage(viewIcon, 40, 340, 50.0, 50.0, false)
        g2d.drawCustomImage(coinIcon, 40, 250, 50.0, 50.0, false)
        g2d.drawCustomImage(likeIcon, 40, 160, 50.0, 50.0, false)
        // draw 22 logo
        g2d.drawCustomImage(twoTwoLogo, 800, 450, 90.0, 160.0, false)
        // draw fans count
        g2d.font = numberFont
        g2d.drawString(fans.formatNumber() + "粉丝", 120, 520)
        // draw viewing count
        g2d.drawString("${viewCount}人正在观看", 40, 460)
        // draw author face
        g2d.drawCustomImage(faceImage, 40, 480, 60.0, 60.0, true)
        // draw cover image
        g2d.drawCustomImage(coverImage, 430, 140, 600.0, 300.0, true)
        g2d.dispose()
        return canvas.toByteArray().encodeToBase64()
    }

    private fun getShortUrlBVID(shortUrl: String): String {
        val request = Request.Builder().url(shortUrl.split(" ").last()).build()
        val redirectedUrl = tempOkHttpClient.newCall(request).execute()
        redirectedUrl.use {
            return it.request.url.toString().split("/")[4].split("?").first()
        }
    }

    private fun String.extractAv(): Long = avRegex.find(this)!!.groupValues[1].toLong()
    private fun String.extractBv(): String = bvRegex.find(this)!!.value

    private fun swap(array: CharArray, i: Int, j: Int) {
        val temp = array[i]
        array[i] = array[j]
        array[j] = temp
    }

    private fun Long.avToBv(): String {
        val aid = BigInteger.valueOf(this)
        val bytes = charArrayOf('B', 'V', '1', '0', '0', '0', '0', '0', '0', '0', '0', '0')
        var bvIndex = bytes.size - 1
        var tmp = MAX_AID.or(aid).xor(XOR_CODE)
        while (tmp > BigInteger.ZERO) {
            bytes[bvIndex] = DATA[tmp.mod(BigInteger.valueOf(BASE.toLong())).toInt()]
            tmp = tmp.divide(BigInteger.valueOf(BASE.toLong()))
            bvIndex--
        }
        swap(bytes, 3, 9)
        swap(bytes, 4, 7)
        return String(bytes)
    }

    suspend fun parse(message: GroupMessage) {
        try {
            val bvid = if (bvRegex.containsMatchIn(message.rawMessage)) {
                message.rawMessage.extractBv()
            } else if (avRegex.containsMatchIn(message.rawMessage)) {
                message.rawMessage.extractAv().avToBv()
            } else if (shortUrlRegex.containsMatchIn(message.rawMessage)) {
                val shortUrl = shortUrlRegex.find(message.rawMessage)!!.value
                getShortUrlBVID(shortUrl)
            } else if (bvRegex.containsMatchIn(message.rawMessage)) {
                bvRegex.find(message.rawMessage)!!.value
            } else if (message.message.find { it.type == ArrayMessageType.json } != null) {
                val card = message.message.find { it.type == ArrayMessageType.json }!!
                    .data.data!!.toString().fromJson<CardShare>()
                val shortUrl = if (card.meta.detail == null) {
                    if (!card.meta.news?.tag!!.contains("哔哩哔哩")) return
                    card.meta.news.jumpUrl
                } else {
                    if (!card.meta.detail.title.contains("哔哩哔哩")) return
                    card.meta.detail.qqDocUrl
                }
                getShortUrlBVID(shortUrl)
            } else {
                return
            }
            val videoInfo = this.getVideoStat(bvid)
            val viewCount = this.getViewCount(bvid, videoInfo.data.cid)
            val shortUrl = this.generateShortUrl(bvid, videoInfo.data.aid)
            val fans = Http.get<UserStat>(USER_STAT_URL, mapOf("vmid" to videoInfo.data.owner.mid)).data.follower
            val authorFace = videoInfo.data.owner.face
            val title = videoInfo.data.title
            val author = videoInfo.data.owner.name
            val coverPicUrl = videoInfo.data.pic
            val view = videoInfo.data.stat.view.formatNumber()
            val share = videoInfo.data.stat.share.formatNumber()
            val like = videoInfo.data.stat.like.formatNumber()
            val favorite = videoInfo.data.stat.favorite.formatNumber()
            val coin = videoInfo.data.stat.coin.formatNumber()
            val reply = videoInfo.data.stat.reply.formatNumber()
            val image = this.createResponseImage(
                title, author, authorFace,
                coverPicUrl, reply, view, coin,
                share, like, favorite, fans, viewCount
            )
            val msg = MessageChain.Builder()
                .addImage(image, true)
                .addText(shortUrl)
                .build()
            message.reply(msg)
        } catch (_: Exception) {
        }
    }
}

object BiliUserParseCommand {

    private val idRegex = Regex("https://space\\.bilibili\\.com/(\\d+)")
    private const val USER_INFO_API_URL = "https://api.bilibili.com/x/web-interface/card"
    private val font = Font("Serif", Font.BOLD, 24)
    private val signFont = Font("Serif", Font.BOLD, 18)
    private const val IMAGE_WIDTH = 500
    private const val IMAGE_HEIGHT = 300


    private fun getUserInfo(userId: String): BiliUserInfo {
        val result = Http.get<UserInfo>(USER_INFO_API_URL, mapOf("mid" to userId)).data.card
        return BiliUserInfo(result.name, result.levelInfo.currentLevel, result.face, result.fans, result.sign)
    }

    private fun getUserFace(url: String) = url.toURL().readBytes().toBufferedImage()

    private fun createUserImage(userInfo: BiliUserInfo): String {
        val image = BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_ARGB)
        val g2d = image.createGraphics()
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON)
        g2d.color = Color.WHITE
        g2d.fillRect(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT)
        g2d.font = font
        g2d.color = Color.BLACK
        g2d.drawString("用户名: ${userInfo.name}", 150, 50)
        g2d.drawString("粉丝数: ${userInfo.fans}", 150, 100)
        g2d.drawString("等级: ${userInfo.level}", 150, 150)
        g2d.font = signFont
        g2d.drawString(setTruncate("签名: ${userInfo.sign}", g2d), 40, 180, 100)
        val faceImage = getUserFace(userInfo.avatar)
        val resizedFaceImage = faceImage.getScaledInstance(100, 100, Image.SCALE_SMOOTH)
        g2d.drawImage(resizedFaceImage, 20, 20, null)
        g2d.dispose()
        return image.toByteArray().encodeToBase64()
    }

    suspend fun parse(message: GroupMessage) {
        val match = idRegex.find(message.rawMessage)
        if (match != null) {
            val userId = match.groupValues[1]
            val userInfo = this.getUserInfo(userId)
            val userCardImage = this.createUserImage(userInfo)
            val msg = MessageChain.Builder()
                .addImage(userCardImage, true)
                .build()
            message.reply(msg)
        }
    }
}