add: share

cssxsh · cssxsh · commit fe769ad82656 · 2022-08-28T08:01:37.000+08:00
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDisk.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDisk.kt
@@ -8,7 +8,6 @@ import io.ktor.client.plugins.compression.*
 import io.ktor.client.request.*
 import io.ktor.client.statement.*
 import io.ktor.http.*
-import io.ktor.utils.io.charsets.*
 import io.ktor.utils.io.core.*
 import io.ktor.utils.io.errors.*
 import kotlinx.coroutines.*
@@ -51,6 +50,12 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
         }
     }
 
+    @JvmStatic
+    public val SHORT_URL_REGEX: Regex = """(?:surl=|s/1)([A-z0-9=_-]+)\W+([A-z0-9]{4})?""".toRegex()
+
+    @JvmStatic
+    public val STAND_CODE_REGEX: Regex = """[A-z0-9]{32}#[A-z0-9]{32}#\d+#\S+""".toRegex()
+
     private val logger: MiraiLogger by lazy {
         try {
             NetDiskFileSyncPlugin.logger
@@ -71,9 +76,23 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
         }
     }
 
+    private val downloader: HttpClient = HttpClient(OkHttp) {
+        ContentEncoding()
+        BrowserUserAgent()
+        // FIXME: MalformedInputException
+        expectSuccess = NetdiskUploadConfig.https.not()
+        HttpResponseValidator {
+            validateResponse { response ->
+                if (response.headers[HttpHeaders.ContentType] == "text/octet") {
+                    val bytes = response.readBytes()
+                    throw ClientRequestException(response, bytes.joinToString("") { "\\x%02x".format(it) })
+                }
+            }
+        }
+    }
+
     override val apiIgnore: suspend (Throwable) -> Boolean = { throwable ->
         when (throwable) {
-            is MalformedInputException -> false
             is IOException -> {
                 val count = ++throwable::class.count
                 if (count > 10) {
@@ -84,7 +103,6 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
                     true
                 }
             }
-
             else -> false
         }
     }
@@ -101,15 +119,15 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
     }
 
     @EventHandler
-    public fun MessageEvent.handle() {
+    public fun MessageEvent.upload() {
         val contact = subject as? Group ?: return
         val content = message.findIsInstance<FileMessage>() ?: return
         if (permission.testPermission(contact.permitteeId).not() &&
             permission.testPermission(sender.permitteeId).not()
         ) return
 
         launch {
-            logger.info { "发现文件消息 ${content}，开始上传" }
+            logger.info { "发现文件消息 $content 开始上传" }
             val file = withTimeout(10_000) {
                 content.toAbsoluteFile(contact) ?: throw NoSuchElementException("${content.name} in $contact")
             }
@@ -134,15 +152,81 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
         }
     }
 
+    @EventHandler
+    public fun MessageEvent.save() {
+        val plain = message.findIsInstance<PlainText>() ?: return
+        if (permission.testPermission(sender.permitteeId).not()) return
+
+        SHORT_URL_REGEX.findAll(plain.content).forEach { match ->
+            launch {
+                logger.info { "发现分享链接 ${match.value} 开始转存" }
+                val (surl, password) = match.destructured
+                val key = if (password.isNotEmpty()) {
+                    val verify = rest.verify(surl = surl, password = password)
+                    require(verify.errorNo == 0) { verify.errorMessage.ifEmpty { surl } }
+                    verify.key
+                } else {
+                    ""
+                }
+
+                val root = rest.view(surl = surl, key = key)
+
+                val dir = rest.mkdir(path = "${subject.id}/${root.uk}-${root.shareId}", ondup = OnDupType.NEW_COPY)
+
+                val info = TransferFileInfo(
+                    shareId = root.shareId,
+                    from = root.uk,
+                    key = key,
+                    files = emptyList()
+                )
+
+                val limit = user().vip.transferLimit
+
+                for (list in root.list.chunked(limit)) {
+                    val part = info.copy(files = list.map { it.id })
+
+                    rest.transfer(info = part, path = dir.path, ondup = OnDupType.NEW_COPY)
+
+                    delay(30_000)
+                }
+
+                launch {
+                    NetDiskFileSyncRecorder.record(source = source, surl = surl, password = password)
+                }
+                if (NetdiskUploadConfig.reply) {
+                    subject.sendMessage(message.quote() + "分享 $surl 转存成功")
+                }
+            }
+        }
+        STAND_CODE_REGEX.findAll(plain.content).forEach { match ->
+            launch {
+                logger.info { "发现秒传码 ${match.value} 开始保存" }
+                mkdir(path = "${subject.id}")
+                val upload = RapidUploadInfo.parse(code = match.value)
+                val path = "${subject.id}/${upload.path.substringAfterLast('/')}"
+
+                rapid(upload = upload.copy(path = path), ondup = OnDupType.NEW_COPY)
+
+                launch {
+                    NetDiskFileSyncRecorder.record(source = source, rapid = upload)
+                }
+                if (NetdiskUploadConfig.reply) {
+                    subject.sendMessage(message.quote() + "保存成功")
+                }
+            }
+        }
+    }
+
     private suspend fun uploadAbsoluteFile(file: AbsoluteFile): RapidUploadInfo {
         val url = requireNotNull(file.getUrl()) { "远程文件 URL 获取失败" }
+        @Suppress("INVISIBLE_MEMBER")
         val rapid = with(file) {
-            val content = file.md5.toUHexString("").lowercase()
+            val content = file.md5.toHexString()
             val slice = if (size <= SLICE_SIZE) {
                 content
             } else {
                 val slice = download(urlString = url, range = 0L until SLICE_SIZE.toLong()).body<ByteArray>()
-                slice.md5().toUHexString("").lowercase()
+                slice.md5().toHexString()
             }
             RapidUploadInfo(
                 content = content,
@@ -166,9 +250,9 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
             logger.warning({ "文件 ${file.name} 秒传失败, 进入文件上传" }, exception)
         }
 
-        val user = rest.user()
-        check(file.size <= user.vip.updateLimit) { "${file.contact}-${file.name} 超过了文件上传极限" }
+        val user = user()
         val limit = user.vip.superLimit.toLong()
+        check(file.size <= limit) { "${file.contact}-${file.name} 超过了文件上传极限" }
 
         if (file.size < limit) {
             try {
@@ -180,15 +264,13 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
             } catch (throwable: ClientRequestException) {
                 logger.info { "文件 ${file.name} 单文件上传失败, 进入文件上传, ${throwable.message}" }
             } catch (exception: Throwable) {
-                logger.info({ "文件 ${file.name} 快速存入失败, 进入文件上传" }, exception)
+                logger.info({ "文件 ${file.name} 单文件上传失败, 进入文件上传" }, exception)
             }
         }
 
 
         val prepare = rest.prepare(upload = rapid, blocks = LAZY_BLOCKS, ondup = OnDupType.NEW_COPY)
-        if (prepare.type == PrepareReturnType.EXIST) {
-            return rapid
-        }
+        if (prepare.type == PrepareReturnType.EXIST) return rapid
         val uploadId = requireNotNull(prepare.uploadId) { prepare }
 
         val blocks = download(urlString = url).execute { response ->
@@ -211,7 +293,7 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
         }.awaitAll()
 
         val merge = MergeFileInfo(
-            blocks = blocks.toMutableList(),
+            blocks = blocks,
             uploadId = uploadId,
             size = file.size,
             path = rapid.path
@@ -222,21 +304,6 @@ public object NetDisk : BaiduNetDiskClient(config = NetdiskOauthConfig), Listene
         return rapid
     }
 
-    private val downloader: HttpClient = HttpClient(OkHttp) {
-        ContentEncoding()
-        BrowserUserAgent()
-        // FIXME: MalformedInputException
-        expectSuccess = NetdiskUploadConfig.https.not()
-        HttpResponseValidator {
-            validateResponse { response ->
-                if (response.headers[HttpHeaders.ContentType] == "text/octet") {
-                    val bytes = response.readBytes()
-                    throw ClientRequestException(response, bytes.joinToString("") { "\\x%02x".format(it) })
-                }
-            }
-        }
-    }
-
     private suspend fun download(urlString: String, range: LongRange? = null): HttpStatement {
         val fragment = range?.run { "bytes=${start}-${endInclusive}" }
         logger.verbose { "download $urlString#$fragment" }
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDiskFileSyncRecorder.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDiskFileSyncRecorder.kt
@@ -3,6 +3,7 @@ package io.github.gnuf0rce.mirai.netdisk
 import io.github.gnuf0rce.mirai.netdisk.data.*
 import io.github.gnuf0rce.mirai.netdisk.entry.*
 import net.mamoe.mirai.contact.file.*
+import net.mamoe.mirai.message.data.*
 import org.hibernate.*
 import xyz.cssxsh.baidu.disk.*
 import xyz.cssxsh.hibernate.*
@@ -47,7 +48,50 @@ public object NetDiskFileSyncRecorder {
 
             factory.fromTransaction { session -> session.persist(record) }
         } else {
-            NetdiskSyncHistory.records.add(record)
+            NetdiskSyncHistory.sync.add(record)
+        }
+    }
+
+    public fun record(source: MessageSource, rapid: RapidUploadInfo) {
+
+        val record = CodeSaveRecord(
+            targetId = source.targetId,
+            fromId = source.fromId,
+            time = source.time,
+            ids = source.ids.joinToString(","),
+            code = rapid.format()
+        )
+
+        val factory = factory
+
+        if (factory != null) {
+            factory as SessionFactory
+
+            factory.fromTransaction { session -> session.persist(record) }
+        } else {
+            NetdiskSyncHistory.code.add(record)
+        }
+    }
+
+    public fun record(source: MessageSource, surl: String, password: String) {
+
+        val record = ShareSaveRecord(
+            targetId = source.targetId,
+            fromId = source.fromId,
+            time = source.time,
+            ids = source.ids.joinToString(","),
+            surl = surl,
+            password = password
+        )
+
+        val factory = factory
+
+        if (factory != null) {
+            factory as SessionFactory
+
+            factory.fromTransaction { session -> session.persist(record) }
+        } else {
+            NetdiskSyncHistory.share.add(record)
         }
     }
 
@@ -70,7 +114,7 @@ public object NetDiskFileSyncRecorder {
                 }.list()
             }
         } else {
-            NetdiskSyncHistory.records
+            NetdiskSyncHistory.sync
                 .asSequence()
                 .filter { it.uploaderId == uploaderId && it.uploadTime in start..end }
                 .toMutableList()
@@ -94,7 +138,7 @@ public object NetDiskFileSyncRecorder {
                 }.list()
             }
         } else {
-            NetdiskSyncHistory.records
+            NetdiskSyncHistory.sync
                 .asSequence()
                 .filter { it.contactId == contactId && it.uploadTime in start..end }
                 .toMutableList()
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/command/BaiduOAuthCommand.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/command/BaiduOAuthCommand.kt
@@ -30,27 +30,53 @@ internal object BaiduOAuthCommand : CompositeCommand(
             authorize { url ->
                 sendMessage("请打开连接，然后在十分钟内输入获得的认证码, $url")
                 read()
-            } to rest.user()
+            } to user()
         }.onSuccess { (token, user) ->
             logger.info { "百度云用户认证成功, ${user.baiduName} by $token" }
             sendMessage("百度云用户认证成功, ${user.baiduName} by $token")
-        }.onFailure {
-            logger.warning({ "认证失败" }, it)
-            sendMessage("百度云用户认证失败, ${it.message}")
+        }.onFailure { cause ->
+            logger.warning({ "认证失败" }, cause)
+            sendMessage("百度云用户认证失败, ${cause.message}")
         }
     }
 
     @SubCommand
     suspend fun CommandSender.refresh(token: String) {
         NetdiskUserData.refreshTokenValue = token
         NetDisk.runCatching {
-            refresh() to rest.user()
+            refresh() to user()
         }.onSuccess { (token, user) ->
             logger.info { "百度云用户认证成功, ${user.baiduName} by $token" }
             sendMessage("百度云用户认证成功, ${user.baiduName} by $token")
-        }.onFailure {
-            logger.warning({ "认证失败" }, it)
-            sendMessage("百度云用户认证失败, ${it.message}")
+        }.onFailure { cause ->
+            logger.warning({ "认证失败" }, cause)
+            sendMessage("百度云用户认证失败, ${cause.message}")
+        }
+    }
+
+    @SubCommand
+    suspend fun CommandSender.host() {
+        NetDisk.runCatching {
+            host()
+        }.onSuccess { host ->
+            logger.info { "host 刷新成功 $host" }
+            sendMessage("host 刷新成功，共 ${host.server.size} 个服务器")
+        }.onFailure { cause ->
+            logger.warning({ "刷新失败" }, cause)
+            sendMessage("host 刷新失败, ${cause.message}")
+        }
+    }
+
+    @SubCommand
+    suspend fun CommandSender.user() {
+        NetDisk.runCatching {
+            user()
+        }.onSuccess { user ->
+            logger.info { "百度云用户刷新成功, ${user.baiduName} - ${user.vip}" }
+            sendMessage("百度云用户刷新成功, ${user.baiduName} - ${user.vip}")
+        }.onFailure { cause ->
+            logger.warning({ "刷新失败" }, cause)
+            sendMessage("百度云用户刷新失败, ${cause.message}")
         }
     }
 }
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/data/NetdiskSyncHistory.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/data/NetdiskSyncHistory.kt
@@ -5,5 +5,11 @@ import net.mamoe.mirai.console.data.*
 
 internal object NetdiskSyncHistory : AutoSavePluginData("history") {
     @ValueName("sync_upload_records")
-    val records: MutableList<SyncUploadRecord> by value()
+    val sync: MutableList<SyncUploadRecord> by value()
+
+    @ValueName("code_save_records")
+    val code: MutableList<CodeSaveRecord> by value()
+
+    @ValueName("share_save_records")
+    val share: MutableList<ShareSaveRecord> by value()
 }
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/entry/CodeSaveRecord.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/entry/CodeSaveRecord.kt
@@ -0,0 +1,23 @@
+package io.github.gnuf0rce.mirai.netdisk.entry
+
+import jakarta.persistence.*
+
+@Entity
+@Table(name = "code_save_record")
+@kotlinx.serialization.Serializable
+public data class CodeSaveRecord(
+    @Id
+    @Column(name = "id", nullable = false, updatable = false)
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    val id: Long = 0,
+    @Column(name = "target_id", nullable = false, updatable = false)
+    val targetId: Long = 0,
+    @Column(name = "from_id", nullable = false, updatable = false)
+    val fromId: Long = 0,
+    @Column(name = "time", nullable = false, updatable = false)
+    val time: Int = 0,
+    @Column(name = "ids", nullable = true, updatable = false)
+    val ids: String = "",
+    @Column(name = "code", nullable = false, updatable = false)
+    val code: String = "",
+) : java.io.Serializable
diff --git a/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/entry/ShareSaveRecord.kt b/src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/entry/ShareSaveRecord.kt
